mirror of https://github.com/zulip/zulip.git
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from typing import Any, Dict
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
from zerver.decorator import webhook_view
|
|
from zerver.lib.request import REQ, has_request_variables
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.models import UserProfile
|
|
|
|
GRAFANA_TOPIC_TEMPLATE = "{alert_title}"
|
|
|
|
GRAFANA_ALERT_STATUS_TEMPLATE = "{alert_icon} **{alert_state}**\n\n"
|
|
|
|
GRAFANA_MESSAGE_TEMPLATE = (
|
|
"{alert_status}[{rule_name}]({rule_url})\n\n{alert_message}{eval_matches}"
|
|
)
|
|
|
|
ALL_EVENT_TYPES = ["ok", "pending", "alerting", "paused"]
|
|
|
|
|
|
@webhook_view("Grafana", all_event_types=ALL_EVENT_TYPES)
|
|
@has_request_variables
|
|
def api_grafana_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
payload: Dict[str, Any] = REQ(argument_type="body"),
|
|
) -> HttpResponse:
|
|
|
|
topic = GRAFANA_TOPIC_TEMPLATE.format(alert_title=payload["title"])
|
|
|
|
eval_matches_text = ""
|
|
eval_matches = payload.get("evalMatches")
|
|
if eval_matches is not None:
|
|
for match in eval_matches:
|
|
eval_matches_text += "**{}:** {}\n".format(match["metric"], match["value"])
|
|
|
|
message_text = ""
|
|
if payload.get("message") is not None:
|
|
message_text = payload["message"] + "\n\n"
|
|
|
|
if payload.get("state") is not None:
|
|
if payload.get("state") == "alerting":
|
|
alert_status = GRAFANA_ALERT_STATUS_TEMPLATE.format(
|
|
alert_icon=":alert:", alert_state=payload["state"].upper()
|
|
)
|
|
elif payload.get("state") == "ok":
|
|
alert_status = GRAFANA_ALERT_STATUS_TEMPLATE.format(
|
|
alert_icon=":squared_ok:", alert_state=payload["state"].upper()
|
|
)
|
|
else:
|
|
alert_status = GRAFANA_ALERT_STATUS_TEMPLATE.format(
|
|
alert_icon=":info:", alert_state=payload["state"].upper()
|
|
)
|
|
|
|
body = GRAFANA_MESSAGE_TEMPLATE.format(
|
|
alert_message=message_text,
|
|
alert_status=alert_status,
|
|
rule_name=payload["ruleName"],
|
|
rule_url=payload["ruleUrl"],
|
|
eval_matches=eval_matches_text,
|
|
)
|
|
|
|
if payload.get("imageUrl") is not None:
|
|
body += "\n[Click to view visualization]({visualization})".format(
|
|
visualization=payload["imageUrl"]
|
|
)
|
|
|
|
body = body.strip()
|
|
|
|
# send the message
|
|
check_send_webhook_message(request, user_profile, topic, body, payload.get("state"))
|
|
|
|
return json_success(request)
|