Files
zulip/zerver/webhooks/uptimerobot/view.py
Prakhar Pratyush 3afc8ed7ae webhooks: Rename *topic local variables to *topic_name.
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.
2024-01-17 08:35:29 -08:00

93 lines
3.6 KiB
Python

from django.core.exceptions import ValidationError
from django.http import HttpRequest, HttpResponse
from django.utils.translation import gettext as _
from zerver.actions.message_send import send_rate_limited_pm_notification_to_bot_owner
from zerver.decorator import webhook_view
from zerver.lib.exceptions import JsonableError
from zerver.lib.response import json_success
from zerver.lib.send_email import FromAddress
from zerver.lib.typed_endpoint import 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
MISCONFIGURED_PAYLOAD_ERROR_MESSAGE = """
Hi there! Your bot {bot_name} just received a UptimeRobot payload that is missing
some data that Zulip requires. This usually indicates a configuration issue
in your UptimeRobot webhook settings. Please make sure that you set the required parameters
when configuring the UptimeRobot webhook. Contact {support_email} if you
need further help!
"""
UPTIMEROBOT_TOPIC_TEMPLATE = "{monitor_friendly_name}"
UPTIMEROBOT_MESSAGE_UP_TEMPLATE = """
{monitor_friendly_name} ({monitor_url}) is back UP ({alert_details}).
It was down for {alert_friendly_duration}.
""".strip()
UPTIMEROBOT_MESSAGE_DOWN_TEMPLATE = (
"{monitor_friendly_name} ({monitor_url}) is DOWN ({alert_details})."
)
ALL_EVENT_TYPES = ["up", "down"]
@webhook_view("UptimeRobot", all_event_types=ALL_EVENT_TYPES)
@typed_endpoint
def api_uptimerobot_webhook(
request: HttpRequest,
user_profile: UserProfile,
*,
payload: JsonBodyPayload[WildValue],
) -> HttpResponse:
event_type = payload["alert_type_friendly_name"].tame(check_string)
if event_type == "Up":
event = "up"
elif event_type == "Down":
event = "down"
try:
body = get_body_for_http_request(payload, event_type)
topic_name = get_topic_for_http_request(payload)
except ValidationError:
message = MISCONFIGURED_PAYLOAD_ERROR_MESSAGE.format(
bot_name=user_profile.full_name,
support_email=FromAddress.SUPPORT,
).strip()
send_rate_limited_pm_notification_to_bot_owner(user_profile, user_profile.realm, message)
raise JsonableError(_("Invalid payload"))
check_send_webhook_message(request, user_profile, topic_name, body, event)
return json_success(request)
def get_topic_for_http_request(payload: WildValue) -> str:
return UPTIMEROBOT_TOPIC_TEMPLATE.format(
monitor_friendly_name=payload["monitor_friendly_name"].tame(check_string)
)
def get_body_for_http_request(payload: WildValue, event_type: str) -> str:
if event_type == "Up":
monitor_friendly_name = payload["monitor_friendly_name"].tame(check_string)
monitor_url = payload["monitor_url"].tame(check_string)
alert_details = payload["alert_details"].tame(check_string)
alert_friendly_duration = payload["alert_friendly_duration"].tame(check_string)
body = UPTIMEROBOT_MESSAGE_UP_TEMPLATE.format(
monitor_friendly_name=monitor_friendly_name,
monitor_url=monitor_url,
alert_details=alert_details,
alert_friendly_duration=alert_friendly_duration,
)
elif event_type == "Down":
monitor_friendly_name = payload["monitor_friendly_name"].tame(check_string)
monitor_url = payload["monitor_url"].tame(check_string)
alert_details = payload["alert_details"].tame(check_string)
body = UPTIMEROBOT_MESSAGE_DOWN_TEMPLATE.format(
monitor_friendly_name=monitor_friendly_name,
monitor_url=monitor_url,
alert_details=alert_details,
)
return body