import yaml from flask import Flask, request, abort from matrix_client.client import MatrixClient application = Flask(__name__) """ config.yml Example: secret: "..." matrix: server: https://matrix.org username: ... password: "..." """ with open("config.yml", 'r') as ymlfile: cfg = yaml.safe_load(ymlfile) @application.route('/matrix', methods=['POST']) def notify(): channel = request.args.get('channel') if channel is None or len(channel) == 0: abort(401) gitlab_token = request.headers.get('X-Gitlab-Token') if gitlab_token is None or len(gitlab_token) == 0 or gitlab_token != cfg['secret']: abort(403) gitlab_event = request.headers.get("X-Gitlab-Event") if gitlab_event == "Push Hook": client = MatrixClient(cfg["matrix"]["server"]) client.login(username=cfg["matrix"]["username"], password=cfg["matrix"]["password"]) room = client.join_room(room_id_or_alias=channel) def sort_commits_by_time(commits): return sorted(commits, key=lambda commit: commit["timestamp"]) def extract_commit_message(commit): return next(iter(commit["message"].splitlines(keepends=False)), "$EMPTY_COMMIT_MESSAGE - impossibruh").strip() 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"
  • {msg}
  • " for msg in commit_messages)) text_commits = "\n".join((f"- {msg}" for msg in commit_messages)) room.send_html(f"{username} pushed {len(commit_messages)} commits to {project_name}
    \n" f"\n", body=f"{username} pushed {len(commit_messages)} commits to {project_name}\n{text_commits}\n", msgtype="m.notice") return ""