T3MP3ST public launch build

bench/cve-zero/samples/GHSA-fqvv-jvhr-g5jc/source/src/firefighter/raid/messages.py

@@ -0,0 +1,190 @@
+from __future__ import annotations
+
+import textwrap
+from typing import TYPE_CHECKING
+
+from django.conf import settings
+from slack_sdk.models.blocks.basic_components import MarkdownTextObject
+from slack_sdk.models.blocks.blocks import (
+    Block,
+    DividerBlock,
+    SectionBlock,
+)
+
+from firefighter.slack.messages.base import SlackMessageSurface
+from firefighter.slack.slack_templating import user_slack_handle_or_name
+
+RAID_JIRA_API_URL: str = settings.RAID_JIRA_API_URL
+
+if TYPE_CHECKING:
+    from firefighter.incidents.models.user import User
+    from firefighter.raid.models import JiraTicket
+
+
+class SlackMessageRaidCreatedIssue(SlackMessageSurface):
+    id = "raid_created_issue"
+
+    def __init__(self, ticket: JiraTicket, reporter_user: User | None = None) -> None:
+        self.ticket = ticket
+        self.reporter_user = reporter_user or ticket.reporter.user
+        super().__init__()
+
+    def get_text(self) -> str:
+        return f"New {self.ticket.issue_type}: {self.ticket.key} '{self.ticket.summary}' was created."
+
+    def get_blocks(self) -> list[Block]:
+        blocks: list[Block] = [
+            SectionBlock(
+                text=f"New {self.ticket.issue_type}: *{self.ticket.key}* _ {self.ticket.summary} _ was created."
+            ),
+            # XXX Add issue type emoji from RAID
+            DividerBlock(),
+            SectionBlock(
+                fields=[
+                    MarkdownTextObject(text=f":tag: *Title:*\n{self.ticket.summary}"),
+                    MarkdownTextObject(
+                        text=textwrap.shorten(
+                            f":spiral_note_pad: *Description:*\n{self.ticket.description}",
+                            width=1997,
+                            placeholder="...",
+                        )
+                    ),
+                    MarkdownTextObject(
+                        text=f":happydoge: *Reporter:*\n{user_slack_handle_or_name(self.reporter_user)}"
+                    ),
+                    MarkdownTextObject(text=f":jira_new: *Jira:*\n{self.ticket.url}"),
+                    MarkdownTextObject(
+                        text=f":warning: *Business impact*: {self.ticket.business_impact or 'N/A'}"
+                    ),
+                ]
+            ),
+        ]
+        return blocks
+
+
+class SlackMessageRaidModifiedIssue(SlackMessageSurface):
+    id = "raid_modify_issue"
+
+    def __init__(
+        self,
+        jira_ticket_key: str,
+        jira_author_name: str,
+        jira_field_modified: str,
+        jira_field_from: str,
+        jira_field_to: str,
+    ) -> None:
+        self.jira_ticket_key = jira_ticket_key
+        self.jira_author_name = jira_author_name
+        self.jira_field_modified = jira_field_modified
+        self.jira_field_from = jira_field_from
+        self.jira_field_to = jira_field_to
+        super().__init__()
+
+    def get_text(self) -> str:
+        return f"Modified Jira ticket {self.jira_ticket_key}"
+
+    def get_blocks(self) -> list[Block]:
+        return [
+            SectionBlock(
+                text=f":jira_new: Jira ticket *{self.jira_ticket_key}* was *modified* :hammer:"
+            ),
+            DividerBlock(),
+            SectionBlock(
+                fields=[
+                    MarkdownTextObject(
+                        text=f":face_with_monocle: *Author:*\n{self.jira_author_name}"
+                    ),
+                    MarkdownTextObject(
+                        text=f":tag: *Field:*\n{self.jira_field_modified}"
+                    ),
+                    MarkdownTextObject(text=f":x: *From:*\n{self.jira_field_from}"),
+                    MarkdownTextObject(
+                        text=f":light_check_mark: *To:*\n{self.jira_field_to}"
+                    ),
+                    MarkdownTextObject(
+                        text=f":jira_new: *URL:*\n{RAID_JIRA_API_URL}/browse/{self.jira_ticket_key}"
+                    ),
+                ]
+            ),
+        ]
+
+
+class SlackMessageRaidComment(SlackMessageSurface):
+    id = "raid_comment"
+
+    def __init__(
+        self,
+        jira_ticket_key: str,
+        author_jira_name: str,
+        comment: str,
+        webhook_event: str,
+    ) -> None:
+        self.jira_ticket_key = jira_ticket_key
+        self.author_jira_name = author_jira_name
+        self.comment = comment
+        self.webhook_event = webhook_event
+        match self.webhook_event:
+            case "comment_created":
+                self.comment_event_text = ":green_book: New comment added"
+            case "comment_updated":
+                self.comment_event_text = ":blue_book: Updated comment"
+            case "comment_deleted":
+                self.comment_event_text = ":closed_book: Deleted comment"
+        super().__init__()
+
+    def get_text(self) -> str:
+        return f"{self.comment_event_text} on {self.jira_ticket_key}"
+
+    def get_blocks(self) -> list[Block]:
+        return [
+            SectionBlock(
+                text=f"{self.comment_event_text} on the Jira ticket *{self.jira_ticket_key}*"
+            ),
+            DividerBlock(),
+            SectionBlock(
+                fields=[
+                    MarkdownTextObject(
+                        text=f":face_with_monocle: *Author:*\n{self.author_jira_name}"
+                    ),
+                    MarkdownTextObject(text=f":pencil2: *Comment:*\n{self.comment}"),
+                    MarkdownTextObject(
+                        text=f":jira_new: *Jira:*\n{RAID_JIRA_API_URL}/browse/{self.jira_ticket_key}"
+                    ),
+                ]
+            ),
+        ]
+
+
+class SlackMessageJiraClosedBlockedByCanBeClosed(SlackMessageSurface):
+    """Sent to the incident channel when a Jira 'Closed' webhook was received but the incident cannot be automatically closed (e.g. missing key events)."""
+
+    id = "raid_jira_closed_blocked"
+
+    def __init__(self, jira_ticket_key: str, reasons: list[tuple[str, str]]) -> None:
+        self.jira_ticket_key = jira_ticket_key
+        self.reasons = reasons
+        super().__init__()
+
+    def get_text(self) -> str:
+        return (
+            f"Jira ticket {self.jira_ticket_key} was closed, "
+            "but the incident could not be automatically closed."
+        )
+
+    def get_blocks(self) -> list[Block]:
+        reasons_text = "\n".join(f"• {reason}" for _, reason in self.reasons)
+        return [
+            SectionBlock(
+                text=(
+                    f":jira_new: Jira ticket *{self.jira_ticket_key}* was closed, "
+                    "but the incident was *not* automatically closed."
+                )
+            ),
+            DividerBlock(),
+            SectionBlock(
+                text=f":warning: *The following must be resolved before closing:*\n{reasons_text}"
+            ),
+            SectionBlock(
+                text="Once resolved, close the incident using the _Close incident_ action in this channel."
+            ),
+        ]