zulip/zerver/webhooks/newrelic/view.py

77 lines
2.6 KiB
Python

# Webhooks for external integrations.
from typing import Any, Dict
from django.http import HttpRequest, HttpResponse
from django.utils.translation import ugettext as _
from zerver.decorator import webhook_view
from zerver.lib.request import REQ, has_request_variables
from zerver.lib.response import json_error, json_success
from zerver.lib.webhooks.common import check_send_webhook_message, unix_milliseconds_to_timestamp
from zerver.models import UserProfile
OPEN_TEMPLATE = """
[Incident]({incident_url}) **opened** for condition: **{condition_name}** at <time:{iso_timestamp}>
``` quote
{details}
```
""".strip()
DEFAULT_TEMPLATE = (
"""[Incident]({incident_url}) **{status}** {owner}for condition: **{condition_name}**""".strip()
)
TOPIC_TEMPLATE = """{policy_name} ({incident_id})""".strip()
@webhook_view("NewRelic")
@has_request_variables
def api_newrelic_webhook(
request: HttpRequest,
user_profile: UserProfile,
payload: Dict[str, Any] = REQ(argument_type="body"),
) -> HttpResponse:
info = {
"condition_name": payload.get("condition_name", "Unknown condition"),
"details": payload.get("details", "No details."),
"incident_url": payload.get("incident_url", "https://alerts.newrelic.com"),
"incident_acknowledge_url": payload.get(
"incident_acknowledge_url", "https://alerts.newrelic.com"
),
"status": payload.get("current_state", "None"),
"iso_timestamp": "",
"owner": payload.get("owner", ""),
}
unix_time = payload.get("timestamp", None)
if unix_time is None:
return json_error(_("The newrelic webhook requires timestamp in milliseconds"))
info["iso_timestamp"] = unix_milliseconds_to_timestamp(unix_time, "newrelic")
# Add formatting to the owner field if owner is present
if info["owner"] != "":
info["owner"] = "by **{}** ".format(info["owner"])
# These are the three promised current_state values
if "open" in info["status"]:
content = OPEN_TEMPLATE.format(**info)
elif "acknowledged" in info["status"]:
content = DEFAULT_TEMPLATE.format(**info)
elif "closed" in info["status"]:
content = DEFAULT_TEMPLATE.format(**info)
else:
return json_error(
_("The newrelic webhook requires current_state be in [open|acknowledged|closed]")
)
topic_info = {
"policy_name": payload.get("policy_name", "Unknown Policy"),
"incident_id": payload.get("incident_id", "Unknown ID"),
}
topic = TOPIC_TEMPLATE.format(**topic_info)
check_send_webhook_message(request, user_profile, topic, content)
return json_success()