You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.9 KiB
53 lines
1.9 KiB
6 years ago
|
from .bitbucket import BitBucketRequestParser
|
||
|
|
from .github import GitHubRequestParser
|
||
|
|
from .gitlab import GitLabRequestParser
|
||
|
|
from .gitlabci import GitLabCIRequestParser
|
||
|
|
from .generic import GenericRequestParser
|
||
|
|
from .coding import CodingRequestParser
|
||
|
|
|
||
|
|
|
||
|
|
def get_service_handler(request_headers, request_body, action):
|
||
|
|
"""Parses the incoming request and attempts to determine whether
|
||
|
|
it originates from GitHub, GitLab or any other known service."""
|
||
|
|
import json
|
||
|
|
|
||
|
|
payload = json.loads(request_body)
|
||
|
|
|
||
|
|
if not isinstance(payload, dict):
|
||
|
|
raise ValueError("Invalid JSON object")
|
||
|
|
|
||
|
|
user_agent = 'user-agent' in request_headers and request_headers['user-agent']
|
||
|
|
content_type = 'content-type' in request_headers and request_headers['content-type']
|
||
|
|
|
||
|
|
# Assume Coding if the X-Coding-Event HTTP header is set
|
||
|
|
if 'x-coding-event' in request_headers:
|
||
|
|
return CodingRequestParser
|
||
|
|
|
||
|
|
# Assume GitLab if the X-Gitlab-Event HTTP header is set
|
||
|
|
elif 'x-gitlab-event' in request_headers:
|
||
|
|
|
||
|
|
# Special Case for Gitlab CI
|
||
|
|
if content_type == "application/json" and "build_status" in payload:
|
||
|
|
return GitLabCIRequestParser
|
||
|
|
else:
|
||
|
|
return GitLabRequestParser
|
||
|
|
|
||
|
|
# Assume GitHub if the X-GitHub-Event HTTP header is set
|
||
|
|
elif 'x-github-event' in request_headers:
|
||
|
|
|
||
|
|
return GitHubRequestParser
|
||
|
|
|
||
|
|
# Assume BitBucket if the User-Agent HTTP header is set to
|
||
|
|
# 'Bitbucket-Webhooks/2.0' (or something similar)
|
||
|
|
elif user_agent and user_agent.lower().find('bitbucket') != -1:
|
||
|
|
|
||
|
|
return BitBucketRequestParser
|
||
|
|
|
||
|
|
# This handles old GitLab requests and Gogs requests for example.
|
||
|
|
elif content_type == "application/json":
|
||
|
|
|
||
|
|
action.log_info("Received event from unknown origin.")
|
||
|
|
return GenericRequestParser
|
||
|
|
|
||
|
|
action.log_error("Unable to recognize request origin. Don't know how to handle the request.")
|
||
|
|
return
|