mirror of https://github.com/zulip/zulip.git
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
|
# Webhooks for external integrations.
|
||
|
|
from __future__ import absolute_import
|
||
|
|
from zerver.models import get_client
|
||
|
|
from zerver.lib.actions import check_send_message
|
||
|
|
from zerver.lib.response import json_success, json_error
|
||
|
|
from zerver.decorator import REQ, has_request_variables, api_key_only_webhook_view
|
||
|
|
|
||
|
|
import ujson
|
||
|
|
|
||
|
|
|
||
|
|
PINGDOM_SUBJECT_TEMPLATE = '{name} status.'
|
||
|
|
PINGDOM_MESSAGE_TEMPLATE = 'Service {service_url} changed its {type} status from {previous_state} to {current_state}.'
|
||
|
|
PINGDOM_MESSAGE_DESCRIPTION_TEMPLATE = 'Description: {description}.'
|
||
|
|
|
||
|
|
|
||
|
|
SUPPORTED_CHECK_TYPES = (
|
||
|
|
'HTTP',
|
||
|
|
'HTTP_CUSTOM'
|
||
|
|
'HTTPS',
|
||
|
|
'SMTP',
|
||
|
|
'POP3',
|
||
|
|
'IMAP',
|
||
|
|
'PING',
|
||
|
|
'DNS',
|
||
|
|
'UDP',
|
||
|
|
'PORT_TCP',
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@api_key_only_webhook_view
|
||
|
|
@has_request_variables
|
||
|
|
def api_pingdom_webhook(request, user_profile, stream=REQ(default='pingdom')):
|
||
|
|
payload = ujson.loads(request.body)
|
||
|
|
check_type = get_check_type(payload)
|
||
|
|
|
||
|
|
if check_type in SUPPORTED_CHECK_TYPES:
|
||
|
|
subject = get_subject_for_http_request(payload)
|
||
|
|
body = get_body_for_http_request(payload)
|
||
|
|
else:
|
||
|
|
return json_error('Unsupported check_type: {check_type}'.format(check_type=check_type))
|
||
|
|
|
||
|
|
check_send_message(user_profile, get_client('ZulipPingdomWebhook'), 'stream', [stream], subject, body)
|
||
|
|
return json_success()
|
||
|
|
|
||
|
|
|
||
|
|
def get_subject_for_http_request(payload):
|
||
|
|
return PINGDOM_SUBJECT_TEMPLATE.format(name=payload['check_name'])
|
||
|
|
|
||
|
|
|
||
|
|
def get_body_for_http_request(payload):
|
||
|
|
current_state = payload['current_state']
|
||
|
|
previous_state = payload['previous_state']
|
||
|
|
|
||
|
|
data = {
|
||
|
|
'service_url': payload['check_params']['hostname'],
|
||
|
|
'previous_state': previous_state,
|
||
|
|
'current_state': current_state,
|
||
|
|
'type': get_check_type(payload)
|
||
|
|
}
|
||
|
|
body = PINGDOM_MESSAGE_TEMPLATE.format(**data)
|
||
|
|
if current_state == 'DOWN' and previous_state == 'UP':
|
||
|
|
description = PINGDOM_MESSAGE_DESCRIPTION_TEMPLATE.format(description=payload['long_description'])
|
||
|
|
body += '\n{description}'.format(description=description)
|
||
|
|
return body
|
||
|
|
|
||
|
|
|
||
|
|
def get_check_type(payload):
|
||
|
|
return payload['check_type']
|