mirror of https://github.com/zulip/zulip.git
114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
# Webhooks for external integrations.
|
|
import base64
|
|
import re
|
|
from functools import wraps
|
|
from typing import Dict, List, Optional, Tuple, cast
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
from zerver.decorator import authenticated_rest_api_view
|
|
from zerver.lib.request import REQ, has_request_variables
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.types import ViewFuncT
|
|
from zerver.lib.validator import WildValue, check_int, check_string, to_wild_value
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
|
from zerver.lib.webhooks.git import TOPIC_WITH_BRANCH_TEMPLATE, get_push_commits_event_message
|
|
from zerver.models import UserProfile
|
|
|
|
|
|
def build_message_from_gitlog(
|
|
user_profile: UserProfile,
|
|
name: str,
|
|
ref: str,
|
|
commits: WildValue,
|
|
before: str,
|
|
after: str,
|
|
url: str,
|
|
pusher: str,
|
|
forced: Optional[str] = None,
|
|
created: Optional[str] = None,
|
|
deleted: bool = False,
|
|
) -> Tuple[str, str]:
|
|
short_ref = re.sub(r"^refs/heads/", "", ref)
|
|
subject = TOPIC_WITH_BRANCH_TEMPLATE.format(repo=name, branch=short_ref)
|
|
|
|
commits_data = _transform_commits_list_to_common_format(commits)
|
|
content = get_push_commits_event_message(pusher, url, short_ref, commits_data, deleted=deleted)
|
|
|
|
return subject, content
|
|
|
|
|
|
def _transform_commits_list_to_common_format(commits: WildValue) -> List[Dict[str, str]]:
|
|
new_commits_list = []
|
|
for commit in commits:
|
|
new_commits_list.append(
|
|
{
|
|
"name": commit["author"]["name"].tame(check_string),
|
|
"sha": commit["id"].tame(check_string),
|
|
"url": commit["url"].tame(check_string),
|
|
"message": commit["message"].tame(check_string),
|
|
}
|
|
)
|
|
return new_commits_list
|
|
|
|
|
|
# Beanstalk's web hook UI rejects URL with a @ in the username section
|
|
# So we ask the user to replace them with %40
|
|
# We manually fix the username here before passing it along to @authenticated_rest_api_view
|
|
def beanstalk_decoder(view_func: ViewFuncT) -> ViewFuncT:
|
|
@wraps(view_func)
|
|
def _wrapped_view_func(request: HttpRequest, *args: object, **kwargs: object) -> HttpResponse:
|
|
auth_type: str
|
|
encoded_value: str
|
|
auth_type, encoded_value = request.META["HTTP_AUTHORIZATION"].split()
|
|
if auth_type.lower() == "basic":
|
|
email, api_key = base64.b64decode(encoded_value).decode().split(":")
|
|
email = email.replace("%40", "@")
|
|
credentials = f"{email}:{api_key}"
|
|
encoded_credentials: str = base64.b64encode(credentials.encode()).decode()
|
|
request.META["HTTP_AUTHORIZATION"] = "Basic " + encoded_credentials
|
|
|
|
return view_func(request, *args, **kwargs)
|
|
|
|
return cast(ViewFuncT, _wrapped_view_func) # https://github.com/python/mypy/issues/1927
|
|
|
|
|
|
@beanstalk_decoder
|
|
@authenticated_rest_api_view(webhook_client_name="Beanstalk")
|
|
@has_request_variables
|
|
def api_beanstalk_webhook(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
payload: WildValue = REQ(converter=to_wild_value),
|
|
branches: Optional[str] = REQ(default=None),
|
|
) -> HttpResponse:
|
|
# Beanstalk supports both SVN and Git repositories
|
|
# We distinguish between the two by checking for a
|
|
# 'uri' key that is only present for Git repos
|
|
git_repo = "uri" in payload
|
|
if git_repo:
|
|
if branches is not None and branches.find(payload["branch"].tame(check_string)) == -1:
|
|
return json_success(request)
|
|
|
|
subject, content = build_message_from_gitlog(
|
|
user_profile,
|
|
payload["repository"]["name"].tame(check_string),
|
|
payload["ref"].tame(check_string),
|
|
payload["commits"],
|
|
payload["before"].tame(check_string),
|
|
payload["after"].tame(check_string),
|
|
payload["repository"]["url"].tame(check_string),
|
|
payload["pusher_name"].tame(check_string),
|
|
)
|
|
else:
|
|
author = payload["author_full_name"].tame(check_string)
|
|
url = payload["changeset_url"].tame(check_string)
|
|
revision = payload["revision"].tame(check_int)
|
|
(short_commit_msg, _, _) = payload["message"].tame(check_string).partition("\n")
|
|
|
|
subject = f"svn r{revision}"
|
|
content = f"{author} pushed [revision {revision}]({url}):\n\n> {short_commit_msg}"
|
|
|
|
check_send_webhook_message(request, user_profile, subject, content)
|
|
return json_success(request)
|