mirror of https://github.com/zulip/zulip.git
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# Webhooks for external integrations.
|
|
from typing import Dict, List
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from typing_extensions import Annotated
|
|
|
|
from zerver.decorator import webhook_view
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.typed_endpoint import ApiParamConfig, JsonBodyPayload, typed_endpoint
|
|
from zerver.lib.validator import WildValue, check_string
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.models import UserProfile
|
|
|
|
|
|
@webhook_view("Alertmanager")
|
|
@typed_endpoint
|
|
def api_alertmanager_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
*,
|
|
payload: JsonBodyPayload[WildValue],
|
|
name_field: Annotated[str, ApiParamConfig("name")] = "instance",
|
|
desc_field: Annotated[str, ApiParamConfig("desc")] = "alertname",
|
|
) -> HttpResponse:
|
|
topics: Dict[str, Dict[str, List[str]]] = {}
|
|
|
|
for alert in payload["alerts"]:
|
|
labels = alert.get("labels", {})
|
|
annotations = alert.get("annotations", {})
|
|
|
|
name = labels.get(name_field, annotations.get(name_field, "(unknown)")).tame(check_string)
|
|
desc = labels.get(
|
|
desc_field, annotations.get(desc_field, f"<missing field: {desc_field}>")
|
|
).tame(check_string)
|
|
|
|
url = alert["generatorURL"].tame(check_string).replace("tab=1", "tab=0")
|
|
|
|
body = f"{desc} ([graph]({url}))"
|
|
if name not in topics:
|
|
topics[name] = {"firing": [], "resolved": []}
|
|
topics[name][alert["status"].tame(check_string)].append(body)
|
|
|
|
for topic_name, statuses in topics.items():
|
|
for status, messages in statuses.items():
|
|
if len(messages) == 0:
|
|
continue
|
|
|
|
if status == "firing":
|
|
icon = ":alert:"
|
|
title = "FIRING"
|
|
else:
|
|
title = "Resolved"
|
|
icon = ":squared_ok:"
|
|
|
|
if len(messages) == 1:
|
|
body = f"{icon} **{title}** {messages[0]}"
|
|
else:
|
|
message_list = "\n".join(f"* {m}" for m in messages)
|
|
body = f"{icon} **{title}**\n{message_list}"
|
|
|
|
check_send_webhook_message(request, user_profile, topic_name, body)
|
|
|
|
return json_success(request)
|