webhook-matrix-notifier/wmn.py

98 lines
3.1 KiB
Python
Raw Normal View History

2019-07-19 14:45:58 +00:00
import re
import yaml
from flask import Flask, request, abort
from matrix_client.client import MatrixClient
2019-07-19 12:14:16 +00:00
from matrix_client.errors import MatrixRequestError
application = Flask(__name__)
2019-07-19 14:45:58 +00:00
# Not going to care for specifics like the underscore.
# Generally match !anything:example.com with unicode support.
room_pattern = re.compile(r'^!\w+:[\w\-.]+$')
2019-07-18 12:47:28 +00:00
"""
config.yml Example:
secret: "..."
matrix:
server: https://matrix.org
2019-07-18 12:47:28 +00:00
username: ...
password: "..."
"""
with open("config.yml", 'r') as ymlfile:
cfg = yaml.safe_load(ymlfile)
2019-07-19 14:45:58 +00:00
def check_token(header_field: str):
token = request.headers.get(header_field)
if token != cfg['secret']:
abort(401)
def get_a_room():
if 'channel' not in request.args:
2019-07-19 12:14:16 +00:00
abort(400)
2019-07-19 14:45:58 +00:00
room = request.args.get('channel')
# sanitize input
if room_pattern.fullmatch(room) is None:
abort(400)
return room
def process_gitlab_request():
check_token('X-Gitlab-Token')
room = get_a_room()
gitlab_event = request.headers.get("X-Gitlab-Event")
if gitlab_event == "Push Hook":
2019-07-19 12:14:16 +00:00
try:
client = MatrixClient(cfg["matrix"]["server"])
client.login(username=cfg["matrix"]["username"], password=cfg["matrix"]["password"])
2019-07-19 14:45:58 +00:00
room = client.join_room(room_id_or_alias=room)
2019-07-19 12:14:16 +00:00
except MatrixRequestError as e:
2019-07-19 14:45:58 +00:00
# see Flask.make_response, this is interpreted as (body, status)
2019-07-19 12:14:16 +00:00
return f"Error from Matrix: {e.content}", e.code
def sort_commits_by_time(commits):
return sorted(commits, key=lambda commit: commit["timestamp"])
def extract_commit_message(commit):
2019-07-19 12:14:16 +00:00
return next(iter(commit["message"].lstrip().splitlines(keepends=False)),
"$EMPTY_COMMIT_MESSAGE - impossibruh").rstrip()
username = request.json["user_name"]
commit_messages = list(map(extract_commit_message, sort_commits_by_time(request.json["commits"])))
project_name = request.json["project"]["name"]
html_commits = "\n".join((f" <li>{msg}</li>" for msg in commit_messages))
text_commits = "\n".join((f"- {msg}" for msg in commit_messages))
room.send_html(f"<strong>{username} pushed {len(commit_messages)} commits to {project_name}</strong><br>\n"
f"<ul>\n{html_commits}\n</ul>\n",
body=f"{username} pushed {len(commit_messages)} commits to {project_name}\n{text_commits}\n",
msgtype="m.notice")
2019-07-19 11:53:00 +00:00
# see Flask.make_response, this is interpreted as (body, status)
return "", 204
2019-07-19 12:14:16 +00:00
def process_jenkins_request():
2019-07-19 14:45:58 +00:00
check_token('X-Jenkins-Token')
# room = get_a_room()
2019-07-19 12:14:16 +00:00
from pprint import pprint
pprint(request.json)
# see Flask.make_response, this is interpreted as (body, status)
return "", 204
@application.route('/matrix', methods=("POST",))
def notify():
if 'X-Gitlab-Token' in request.headers:
return process_gitlab_request()
elif 'X-Jenkins-Token' in request.headers:
return process_jenkins_request()
else:
return "Cannot determine the request's webhook cause", 400