Add gitlab webhook support

ops_bot/gitlab/hook.py

@@ -0,0 +1,70 @@
+import re
+import attr
+import logging
+from typing import Any, Tuple
+from jinja2 import TemplateNotFound
+
+from mautrix.types import (EventType, RoomID, StateEvent, Membership, MessageType, JSON,
+                           TextMessageEventContent, Format, ReactionEventContent, RelationType)
+from mautrix.util.formatter import parse_html
+
+from ..util.template import TemplateManager, TemplateUtil
+
+from .types import EventParse, OTHER_ENUMS, Action
+
+from ..common import COLOR_ALARM, COLOR_OK, COLOR_UNKNOWN
+
+spaces = re.compile(" +")
+space = " "
+
+
+messages = TemplateManager("gitlab", "messages")
+templates = TemplateManager("gitlab", "mixins")
+
+async def parse_event(x_gitlab_event: str, payload: Any) -> Tuple[str, str]:
+    evt = EventParse[x_gitlab_event].deserialize(payload)
+    print("processing", evt)
+    try:
+        tpl = messages[evt.template_name]
+    except TemplateNotFound as e:
+        msg = f"Received unhandled gitlab event type {x_gitlab_event}"
+        logging.info(msg)
+        logging.info(payload)
+        return [(msg, msg)]
+
+    aborted = False
+
+    def abort() -> None:
+        nonlocal aborted
+        aborted = True
+    base_args = {
+        **{field.key: field for field in Action if field.key.isupper()},
+        **OTHER_ENUMS,
+        "util": TemplateUtil,
+    }
+
+    msgs = []
+    for subevt in evt.preprocess():
+        print("preprocessing", subevt)
+        args = {
+            **attr.asdict(subevt, recurse=False),
+            **{key: getattr(subevt, key) for key in subevt.event_properties},
+            "abort": abort,
+            **base_args,
+        }
+        args["templates"] = templates.proxy(args)
+
+        html = tpl.render(**args)
+        if not html or aborted:
+            aborted = False
+            continue
+        html = spaces.sub(space, html.strip())
+
+        content = TextMessageEventContent(msgtype=MessageType.TEXT, format=Format.HTML,
+                                            formatted_body=html, body=await parse_html(html))
+        content["xyz.maubot.gitlab.webhook"] = {
+            "event_type": x_gitlab_event,
+            **subevt.meta,
+        }
+        msgs.append((content.body, content.formatted_body))
+    return msgs