zulip/zerver/webhooks/netlify/view.py

from typing import Any, Dict, Iterable

from django.http import HttpRequest, HttpResponse

from zerver.decorator import api_key_only_webhook_view
from zerver.lib.request import REQ, has_request_variables
from zerver.lib.response import json_success
from zerver.lib.webhooks.common import UnexpectedWebhookEventType, \
    check_send_webhook_message, get_http_headers_from_filename, \
    validate_extract_webhook_http_header
from zerver.models import UserProfile

EVENTS = ['deploy_failed', 'deploy_locked', 'deploy_unlocked', 'deploy_building', 'deploy_created']

fixture_to_headers = get_http_headers_from_filename("HTTP_X_NETLIFY_EVENT")

@api_key_only_webhook_view('Netlify')
@has_request_variables
def api_netlify_webhook(
        request: HttpRequest, user_profile: UserProfile,
        payload: Dict[str, Iterable[Dict[str, Any]]]=REQ(argument_type='body')
) -> HttpResponse:

    message_template = get_template(request, payload)

    body = message_template.format(build_name=payload['name'],
                                   build_url=payload['url'],
                                   branch_name=payload['branch'],
                                   state=payload['state'])

    topic = "{topic}".format(topic=payload['branch'])

    check_send_webhook_message(request, user_profile, topic, body)

    return json_success()

def get_template(request: HttpRequest, payload: Dict[str, Any]) -> str:

    message_template = u'The build [{build_name}]({build_url}) on branch {branch_name} '
    event = validate_extract_webhook_http_header(request, 'X_NETLIFY_EVENT', 'Netlify')

    if event == 'deploy_failed':
        return message_template + payload['error_message']
    elif event == 'deploy_locked':
        return message_template + 'is now locked.'
    elif event == 'deploy_unlocked':
        return message_template + 'is now unlocked.'
    elif event in EVENTS:
        return message_template + 'is now {state}.'.format(state=payload['state'])
    else:
        raise UnexpectedWebhookEventType('Netlify', event)