zulip/zerver/webhooks/netlify/view.py

from typing import Tuple

from django.http import HttpRequest, HttpResponse

from zerver.decorator import webhook_view
from zerver.lib.exceptions import UnsupportedWebhookEventTypeError
from zerver.lib.request import REQ, has_request_variables
from zerver.lib.response import json_success
from zerver.lib.validator import WildValue, check_string, to_wild_value
from zerver.lib.webhooks.common import (
    check_send_webhook_message,
    get_http_headers_from_filename,
    validate_extract_webhook_http_header,
)
from zerver.models import UserProfile

ALL_EVENT_TYPES = [
    "deploy_failed",
    "deploy_locked",
    "deploy_unlocked",
    "deploy_building",
    "deploy_created",
]

fixture_to_headers = get_http_headers_from_filename("HTTP_X_NETLIFY_EVENT")


@webhook_view("Netlify", all_event_types=ALL_EVENT_TYPES)
@has_request_variables
def api_netlify_webhook(
    request: HttpRequest,
    user_profile: UserProfile,
    payload: WildValue = REQ(argument_type="body", converter=to_wild_value),
) -> HttpResponse:

    message_template, event = get_template(request, payload)

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

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

    check_send_webhook_message(request, user_profile, topic, body, event)

    return json_success(request)


def get_template(request: HttpRequest, payload: WildValue) -> Tuple[str, str]:

    message_template = "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":
        message_template += payload["error_message"].tame(check_string)
    elif event == "deploy_locked":
        message_template += "is now locked."
    elif event == "deploy_unlocked":
        message_template += "is now unlocked."
    elif event in ALL_EVENT_TYPES:
        message_template += "is now {state}.".format(state=payload["state"].tame(check_string))
    else:
        raise UnsupportedWebhookEventTypeError(event)

    return message_template, event