mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-04 05:53:43 +00:00 
			
		
		
		
	This kind of payload that's loaded from json in the body of the request is not only used for webhooks, but also in the push bouncer, and may get used elsewhere too - so a general name is better.
		
			
				
	
	
		
			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, 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, body)
 | 
						|
 | 
						|
    return json_success(request)
 |