70 lines
No EOL
2.1 KiB
Python
70 lines
No EOL
2.1 KiB
Python
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 |