mirror of https://github.com/zulip/zulip.git
92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
# Webhooks for external integrations.
|
|
from __future__ import absolute_import
|
|
import re
|
|
from datetime import datetime
|
|
from typing import Any, Dict, List
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.utils.translation import ugettext as _
|
|
from zerver.lib.actions import check_send_message
|
|
from zerver.lib.response import json_success, json_error
|
|
from zerver.lib.request import JsonableError
|
|
from zerver.decorator import REQ, has_request_variables, api_key_only_webhook_view
|
|
from zerver.models import UserProfile, Client
|
|
|
|
|
|
SUBJECT_TEMPLATE = "{service_url}"
|
|
|
|
|
|
class UnsupportedUpdownEventType(JsonableError):
|
|
pass
|
|
|
|
def send_message_for_event(event, user_profile, client, stream):
|
|
# type: (Dict[str, Any], UserProfile, Client, str) -> None
|
|
try:
|
|
event_type = get_event_type(event)
|
|
subject = SUBJECT_TEMPLATE.format(service_url=event['check']['url'])
|
|
body = EVENT_TYPE_BODY_MAPPER[event_type](event)
|
|
except KeyError as e:
|
|
return json_error(_("Missing key {} in JSON").format(str(e)))
|
|
check_send_message(user_profile, client, 'stream', [stream], subject, body)
|
|
|
|
def get_body_for_up_event(event):
|
|
# type: (Dict[str, Any]) -> str
|
|
body = "Service is `up`"
|
|
event_downtime = event['downtime']
|
|
if event_downtime['started_at']:
|
|
body = "{} again".format(body)
|
|
string_date = get_time_string_based_on_duration(event_downtime['duration'])
|
|
if string_date:
|
|
body = "{} after {}".format(body, string_date)
|
|
return "{}.".format(body)
|
|
|
|
def get_time_string_based_on_duration(duration):
|
|
# type: (int) -> str
|
|
days, reminder = divmod(duration, 86400)
|
|
hours, reminder = divmod(reminder, 3600)
|
|
minutes, seconds = divmod(reminder, 60)
|
|
|
|
string_date = ''
|
|
string_date += add_time_part_to_string_date_if_needed(days, 'day')
|
|
string_date += add_time_part_to_string_date_if_needed(hours, 'hour')
|
|
string_date += add_time_part_to_string_date_if_needed(minutes, 'minute')
|
|
string_date += add_time_part_to_string_date_if_needed(seconds, 'second')
|
|
return string_date.rstrip()
|
|
|
|
def add_time_part_to_string_date_if_needed(value, text_name):
|
|
# type: (int, str) -> str
|
|
if value == 1:
|
|
return "1 {} ".format(text_name)
|
|
if value > 1:
|
|
return "{} {}s ".format(value, text_name)
|
|
return ''
|
|
|
|
def get_body_for_down_event(event):
|
|
# type: (Dict[str, Any]) -> str
|
|
return "Service is `down`. It returned a {} error at {}.".format(
|
|
event['downtime']['error'],
|
|
event['downtime']['started_at'].replace('T', ' ').replace('Z', ' UTC'))
|
|
|
|
@api_key_only_webhook_view('Updown')
|
|
@has_request_variables
|
|
def api_updown_webhook(request, user_profile,
|
|
payload=REQ(argument_type='body'),
|
|
stream=REQ(default='updown')):
|
|
# type: (HttpRequest, UserProfile, List[Dict[str, Any]], str) -> HttpResponse
|
|
for event in payload:
|
|
send_message_for_event(event, user_profile, request.client, stream)
|
|
return json_success()
|
|
|
|
EVENT_TYPE_BODY_MAPPER = {
|
|
'up': get_body_for_up_event,
|
|
'down': get_body_for_down_event
|
|
}
|
|
|
|
def get_event_type(event):
|
|
# type: (Dict[str, Any]) -> str
|
|
event_type_match = re.match('check.(.*)', event['event'])
|
|
if event_type_match:
|
|
event_type = event_type_match.group(1)
|
|
if event_type in EVENT_TYPE_BODY_MAPPER:
|
|
return event_type
|
|
raise UnsupportedUpdownEventType(event['event'])
|