mirror of
https://github.com/zulip/zulip.git
synced 2025-11-06 15:03:34 +00:00
3afc8ed7ae
This is preparatory work towards adding a Topic model. We plan to use the local variable name as 'topic' for the Topic model objects. Currently, we use *topic as the local variable name for topic names. We rename local variables of the form *topic to *topic_name so that we don't need to think about type collisions in individual code paths where we might want to talk about both Topic objects and strings for the topic name.
283 lines
10 KiB
Python
283 lines
10 KiB
Python
from email.headerregistry import Address
|
|
from typing import Dict, Union
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from typing_extensions import TypeAlias
|
|
|
|
from zerver.decorator import webhook_view
|
|
from zerver.lib.exceptions import UnsupportedWebhookEventTypeError
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.typed_endpoint import JsonBodyPayload, typed_endpoint
|
|
from zerver.lib.validator import WildValue, check_int, check_none_or, check_string
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.models import UserProfile
|
|
|
|
FormatDictType: TypeAlias = Dict[str, Union[str, int]]
|
|
|
|
PAGER_DUTY_EVENT_NAMES = {
|
|
"incident.trigger": "triggered",
|
|
"incident.acknowledge": "acknowledged",
|
|
"incident.unacknowledge": "unacknowledged",
|
|
"incident.resolve": "resolved",
|
|
"incident.assign": "assigned",
|
|
"incident.escalate": "escalated",
|
|
"incident.delegate": "delineated",
|
|
}
|
|
|
|
PAGER_DUTY_EVENT_NAMES_V2 = {
|
|
"incident.trigger": "triggered",
|
|
"incident.acknowledge": "acknowledged",
|
|
"incident.resolve": "resolved",
|
|
"incident.assign": "assigned",
|
|
}
|
|
|
|
PAGER_DUTY_EVENT_NAMES_V3 = {
|
|
"incident.triggered": "triggered",
|
|
"incident.acknowledged": "acknowledged",
|
|
"incident.unacknowledged": "unacknowledged",
|
|
"incident.resolved": "resolved",
|
|
"incident.reassigned": "reassigned",
|
|
}
|
|
|
|
ALL_EVENT_TYPES = [
|
|
"resolved",
|
|
"assigned",
|
|
"unacknowledged",
|
|
"acknowledged",
|
|
"triggered",
|
|
"reassigned",
|
|
]
|
|
|
|
AGENT_TEMPLATE = "[{username}]({url})"
|
|
|
|
INCIDENT_WITH_SERVICE_AND_ASSIGNEE = (
|
|
"Incident [{incident_num_title}]({incident_url}) {action} by [{service_name}]"
|
|
"({service_url}) (assigned to {assignee_info}).\n\n{trigger_message}"
|
|
)
|
|
|
|
TRIGGER_MESSAGE = "``` quote\n{message}\n```"
|
|
|
|
NUM_TITLE = "{incident_title} (#{incident_num})"
|
|
|
|
INCIDENT_WITH_ASSIGNEE = """
|
|
Incident [{incident_num_title}]({incident_url}) {action} by {assignee_info}.
|
|
|
|
{trigger_message}
|
|
""".strip()
|
|
|
|
INCIDENT_ASSIGNED = """
|
|
Incident [{incident_num_title}]({incident_url}) {action} to {assignee_info}.
|
|
|
|
{trigger_message}
|
|
""".strip()
|
|
|
|
INCIDENT_RESOLVED_WITH_AGENT = """
|
|
Incident [{incident_num_title}]({incident_url}) resolved by {agent_info}.
|
|
|
|
{trigger_message}
|
|
""".strip()
|
|
|
|
INCIDENT_RESOLVED = """
|
|
Incident [{incident_num_title}]({incident_url}) resolved.
|
|
|
|
{trigger_message}
|
|
""".strip()
|
|
|
|
|
|
def build_pagerduty_formatdict(message: WildValue) -> FormatDictType:
|
|
format_dict: FormatDictType = {}
|
|
format_dict["action"] = PAGER_DUTY_EVENT_NAMES[message["type"].tame(check_string)]
|
|
|
|
format_dict["incident_id"] = message["data"]["incident"]["id"].tame(check_string)
|
|
format_dict["incident_num_title"] = message["data"]["incident"]["incident_number"].tame(
|
|
check_int
|
|
)
|
|
format_dict["incident_url"] = message["data"]["incident"]["html_url"].tame(check_string)
|
|
|
|
format_dict["service_name"] = message["data"]["incident"]["service"]["name"].tame(check_string)
|
|
format_dict["service_url"] = message["data"]["incident"]["service"]["html_url"].tame(
|
|
check_string
|
|
)
|
|
|
|
if message["data"]["incident"].get("assigned_to_user"):
|
|
assigned_to_user = message["data"]["incident"]["assigned_to_user"]
|
|
format_dict["assignee_info"] = AGENT_TEMPLATE.format(
|
|
username=Address(addr_spec=assigned_to_user["email"].tame(check_string)).username,
|
|
url=assigned_to_user["html_url"].tame(check_string),
|
|
)
|
|
else:
|
|
format_dict["assignee_info"] = "nobody"
|
|
|
|
if message["data"]["incident"].get("resolved_by_user"):
|
|
resolved_by_user = message["data"]["incident"]["resolved_by_user"]
|
|
format_dict["agent_info"] = AGENT_TEMPLATE.format(
|
|
username=Address(addr_spec=resolved_by_user["email"].tame(check_string)).username,
|
|
url=resolved_by_user["html_url"].tame(check_string),
|
|
)
|
|
|
|
trigger_message = []
|
|
trigger_summary_data = message["data"]["incident"].get("trigger_summary_data")
|
|
if trigger_summary_data:
|
|
trigger_subject = trigger_summary_data.get("subject", "").tame(check_string)
|
|
if trigger_subject:
|
|
trigger_message.append(trigger_subject)
|
|
|
|
trigger_description = trigger_summary_data.get("description", "").tame(check_string)
|
|
if trigger_description:
|
|
trigger_message.append(trigger_description)
|
|
|
|
format_dict["trigger_message"] = TRIGGER_MESSAGE.format(message="\n".join(trigger_message))
|
|
return format_dict
|
|
|
|
|
|
def build_pagerduty_formatdict_v2(message: WildValue) -> FormatDictType:
|
|
format_dict: FormatDictType = {}
|
|
format_dict["action"] = PAGER_DUTY_EVENT_NAMES_V2[message["event"].tame(check_string)]
|
|
|
|
format_dict["incident_id"] = message["incident"]["id"].tame(check_string)
|
|
format_dict["incident_num_title"] = message["incident"]["incident_number"].tame(check_int)
|
|
format_dict["incident_url"] = message["incident"]["html_url"].tame(check_string)
|
|
|
|
format_dict["service_name"] = message["incident"]["service"]["name"].tame(check_string)
|
|
format_dict["service_url"] = message["incident"]["service"]["html_url"].tame(check_string)
|
|
|
|
assignments = message["incident"]["assignments"]
|
|
if assignments:
|
|
assignee = assignments[0]["assignee"]
|
|
format_dict["assignee_info"] = AGENT_TEMPLATE.format(
|
|
username=assignee["summary"].tame(check_string),
|
|
url=assignee["html_url"].tame(check_string),
|
|
)
|
|
else:
|
|
format_dict["assignee_info"] = "nobody"
|
|
|
|
last_status_change_by = message["incident"].get("last_status_change_by")
|
|
if last_status_change_by is not None:
|
|
format_dict["agent_info"] = AGENT_TEMPLATE.format(
|
|
username=last_status_change_by["summary"].tame(check_string),
|
|
url=last_status_change_by["html_url"].tame(check_string),
|
|
)
|
|
|
|
trigger_description = message["incident"].get("description").tame(check_none_or(check_string))
|
|
if trigger_description is not None:
|
|
format_dict["trigger_message"] = TRIGGER_MESSAGE.format(message=trigger_description)
|
|
return format_dict
|
|
|
|
|
|
def build_pagerduty_formatdict_v3(event: WildValue) -> FormatDictType:
|
|
format_dict: FormatDictType = {}
|
|
format_dict["action"] = PAGER_DUTY_EVENT_NAMES_V3[event["event_type"].tame(check_string)]
|
|
|
|
format_dict["incident_id"] = event["data"]["id"].tame(check_string)
|
|
format_dict["incident_url"] = event["data"]["html_url"].tame(check_string)
|
|
format_dict["incident_num_title"] = NUM_TITLE.format(
|
|
incident_num=event["data"]["number"].tame(check_int),
|
|
incident_title=event["data"]["title"].tame(check_string),
|
|
)
|
|
|
|
format_dict["service_name"] = event["data"]["service"]["summary"].tame(check_string)
|
|
format_dict["service_url"] = event["data"]["service"]["html_url"].tame(check_string)
|
|
|
|
assignees = event["data"]["assignees"]
|
|
if assignees:
|
|
assignee = assignees[0]
|
|
format_dict["assignee_info"] = AGENT_TEMPLATE.format(
|
|
username=assignee["summary"].tame(check_string),
|
|
url=assignee["html_url"].tame(check_string),
|
|
)
|
|
else:
|
|
format_dict["assignee_info"] = "nobody"
|
|
|
|
agent = event.get("agent")
|
|
if agent is not None:
|
|
format_dict["agent_info"] = AGENT_TEMPLATE.format(
|
|
username=agent["summary"].tame(check_string),
|
|
url=agent["html_url"].tame(check_string),
|
|
)
|
|
|
|
# V3 doesn't have trigger_message
|
|
format_dict["trigger_message"] = ""
|
|
|
|
return format_dict
|
|
|
|
|
|
def send_formatted_pagerduty(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
message_type: str,
|
|
format_dict: FormatDictType,
|
|
) -> None:
|
|
if message_type in (
|
|
"incident.trigger",
|
|
"incident.triggered",
|
|
"incident.unacknowledge",
|
|
"incident.unacknowledged",
|
|
):
|
|
template = INCIDENT_WITH_SERVICE_AND_ASSIGNEE
|
|
elif message_type in ("incident.resolve", "incident.resolved"):
|
|
if "agent_info" in format_dict:
|
|
template = INCIDENT_RESOLVED_WITH_AGENT
|
|
else:
|
|
template = INCIDENT_RESOLVED
|
|
elif message_type in ("incident.assign", "incident.reassigned"):
|
|
template = INCIDENT_ASSIGNED
|
|
else:
|
|
template = INCIDENT_WITH_ASSIGNEE
|
|
|
|
topic_name = "Incident {incident_num_title}".format(**format_dict)
|
|
body = template.format(**format_dict)
|
|
assert isinstance(format_dict["action"], str)
|
|
check_send_webhook_message(request, user_profile, topic_name, body, format_dict["action"])
|
|
|
|
|
|
@webhook_view("PagerDuty", all_event_types=ALL_EVENT_TYPES)
|
|
@typed_endpoint
|
|
def api_pagerduty_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
*,
|
|
payload: JsonBodyPayload[WildValue],
|
|
) -> HttpResponse:
|
|
messages = payload.get("messages")
|
|
if messages:
|
|
for message in messages:
|
|
message_type = message.get("type").tame(check_none_or(check_string))
|
|
|
|
# If the message has no "type" key, then this payload came from a
|
|
# Pagerduty Webhook V2.
|
|
if message_type is None:
|
|
break
|
|
|
|
if message_type not in PAGER_DUTY_EVENT_NAMES:
|
|
raise UnsupportedWebhookEventTypeError(message_type)
|
|
|
|
format_dict = build_pagerduty_formatdict(message)
|
|
send_formatted_pagerduty(request, user_profile, message_type, format_dict)
|
|
|
|
for message in messages:
|
|
message_event = message.get("event").tame(check_none_or(check_string))
|
|
|
|
# If the message has no "event" key, then this payload came from a
|
|
# Pagerduty Webhook V1.
|
|
if message_event is None:
|
|
break
|
|
|
|
if message_event not in PAGER_DUTY_EVENT_NAMES_V2:
|
|
raise UnsupportedWebhookEventTypeError(message_event)
|
|
|
|
format_dict = build_pagerduty_formatdict_v2(message)
|
|
send_formatted_pagerduty(request, user_profile, message_event, format_dict)
|
|
else:
|
|
if "event" in payload:
|
|
# V3 has no "messages" field, and it has key "event" instead
|
|
event = payload["event"]
|
|
event_type = event.get("event_type").tame(check_none_or(check_string))
|
|
|
|
if event_type not in PAGER_DUTY_EVENT_NAMES_V3:
|
|
raise UnsupportedWebhookEventTypeError(event_type)
|
|
|
|
format_dict = build_pagerduty_formatdict_v3(event)
|
|
send_formatted_pagerduty(request, user_profile, event_type, format_dict)
|
|
|
|
return json_success(request)
|