from typing import Any, Dict, List import dateutil.parser from django.http import HttpRequest, HttpResponse from django.utils.translation import ugettext as _ from zerver.decorator import REQ, has_request_variables, webhook_view from zerver.lib.actions import send_rate_limited_pm_notification_to_bot_owner from zerver.lib.response import json_error, json_success from zerver.lib.send_email import FromAddress from zerver.lib.webhooks.common import check_send_webhook_message, get_setup_webhook_message from zerver.models import UserProfile MISCONFIGURED_PAYLOAD_ERROR_MESSAGE = """ Hi there! Your bot {bot_name} just received a Freshstatus payload that is missing some data that Zulip requires. This usually indicates a configuration issue in your Freshstatus webhook settings. Please make sure that you provide all the required parameters when configuring the Freshstatus webhook. Contact {support_email} if you need further help! """ FRESHSTATUS_TOPIC_TEMPLATE = "{title}".strip() FRESHSTATUS_TOPIC_TEMPLATE_TEST = "Freshstatus" FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_OPEN = """ The following incident has been opened: **{title}** **Description:** {description} **Start Time:** {start_time} **Affected Services:** {affected_services} """.strip() FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_CLOSED = """ The following incident has been closed: **{title}** **Note:** {message} """.strip() FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_NOTE_CREATED = """ The following note has been added to the incident: **{title}** **Note:** {message} """.strip() FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_PLANNED = """ The following scheduled maintenance has been opened: **{title}** **Description:** {description} **Scheduled Start Time:** {scheduled_start_time} **Scheduled End Time:** {scheduled_end_time} **Affected Services:** {affected_services} """.strip() FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_CLOSED = """ The following scheduled maintenance has been closed: **{title}** **Note:** {message} """.strip() FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_NOTE_CREATED = """ The following note has been added to the scheduled maintenance: **{title}** **Note:** {message} """.strip() FRESHSTATUS_MESSAGE_EVENT_NOT_SUPPORTED = "The event ({event_type}) is not supported yet." FRESHSTATUS_SERVICES_ROW_TEMPLATE = "* {service_name}\n" FRESHSTATUS_SERVICES_OTHERS_ROW_TEMPLATE = "[and {services_number} more service(s)]" FRESHSTATUS_SERVICES_LIMIT = 5 @webhook_view("Freshstatus") @has_request_variables def api_freshstatus_webhook( request: HttpRequest, user_profile: UserProfile, payload: Dict[str, Any] = REQ(argument_type="body"), ) -> HttpResponse: try: body = get_body_for_http_request(payload) subject = get_subject_for_http_request(payload) except KeyError: 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) return json_error(_("Invalid payload")) check_send_webhook_message(request, user_profile, subject, body) return json_success() def get_services_content(services_data: List[Dict[str, Any]]) -> str: services_content = "" for service in services_data[:FRESHSTATUS_SERVICES_LIMIT]: services_content += FRESHSTATUS_SERVICES_ROW_TEMPLATE.format( service_name=service.get("service_name") ) if len(services_data) > FRESHSTATUS_SERVICES_LIMIT: services_content += FRESHSTATUS_SERVICES_OTHERS_ROW_TEMPLATE.format( services_number=len(services_data) - FRESHSTATUS_SERVICES_LIMIT, ) return services_content.rstrip() def get_subject_for_http_request(payload: Dict[str, Any]) -> str: event_data = payload["event_data"] if event_data["event_type"] == "INCIDENT_OPEN" and payload["id"] == "1": return FRESHSTATUS_TOPIC_TEMPLATE_TEST else: return FRESHSTATUS_TOPIC_TEMPLATE.format(title=payload["title"]) def get_body_for_maintenance_planned_event(payload: Dict[str, Any]) -> str: services_data = [] for service in payload["affected_services"].split(","): services_data.append({"service_name": service}) data = { "title": payload["title"], "description": payload["description"], "scheduled_start_time": dateutil.parser.parse(payload["scheduled_start_time"]).strftime( "%Y-%m-%d %H:%M %Z" ), "scheduled_end_time": dateutil.parser.parse(payload["scheduled_end_time"]).strftime( "%Y-%m-%d %H:%M %Z" ), "affected_services": get_services_content(services_data), } return FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_PLANNED.format(**data) def get_body_for_incident_open_event(payload: Dict[str, Any]) -> str: services_data = [] for service in payload["affected_services"].split(","): services_data.append({"service_name": service}) data = { "title": payload["title"], "description": payload["description"], "start_time": dateutil.parser.parse(payload["start_time"]).strftime("%Y-%m-%d %H:%M %Z"), "affected_services": get_services_content(services_data), } return FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_OPEN.format(**data) def get_body_for_http_request(payload: Dict[str, Any]) -> str: event_data = payload["event_data"] if event_data["event_type"] == "INCIDENT_OPEN" and payload["id"] == "1": return get_setup_webhook_message("Freshstatus") elif event_data["event_type"] == "INCIDENT_OPEN": return get_body_for_incident_open_event(payload) elif event_data["event_type"] == "INCIDENT_NOTE_CREATE": if payload["incident_status"] == "Closed": return FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_CLOSED.format(**payload) elif payload["incident_status"] == "Open": return FRESHSTATUS_MESSAGE_TEMPLATE_INCIDENT_NOTE_CREATED.format(**payload) elif event_data["event_type"] == "MAINTENANCE_PLANNED": return get_body_for_maintenance_planned_event(payload) elif event_data["event_type"] == "MAINTENANCE_NOTE_CREATE": if payload["incident_status"] == "Closed": return FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_CLOSED.format(**payload) else: return FRESHSTATUS_MESSAGE_TEMPLATE_SCHEDULED_MAINTENANCE_NOTE_CREATED.format(**payload) return FRESHSTATUS_MESSAGE_EVENT_NOT_SUPPORTED.format(event_type=event_data["event_type"])