156 lines
4.3 KiB
Python
Executable file
156 lines
4.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import argparse
|
|
import asyncio
|
|
import dataclasses
|
|
import logging
|
|
|
|
import aiohttp.web
|
|
import jinja2
|
|
import markdown
|
|
import nio
|
|
import yaml
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class Config:
|
|
listen_addr: str
|
|
listen_port: int
|
|
matrix_homeserver: str
|
|
matrix_user: str
|
|
matrix_password: str
|
|
matrix_room: str
|
|
webhook_token: str
|
|
|
|
|
|
TEMPLATES = {
|
|
"task.create": (
|
|
"<b>[{{ event_data.task.project_name }}]</b> "
|
|
"<font color='{{ event_data.task.color_id }}'>"
|
|
"{{ event_data.task.title }}"
|
|
"</font> "
|
|
"(#{{ event_data.task.id }}): "
|
|
"{{ event_author }} created the task in "
|
|
"'{{ event_data.task.column_title }}'\n"
|
|
"<blockquote>"
|
|
"{{ event_data.task.description | markdown }}"
|
|
"</blockquote>"
|
|
),
|
|
"task.move.column": (
|
|
"<b>[{{ event_data.task.project_name }}]</b> "
|
|
"<font color='{{ event_data.task.color_id }}'>"
|
|
"{{ event_data.task.title }}"
|
|
"</font> "
|
|
"(#{{ event_data.task.id }}): "
|
|
"{{ event_author }} moved the task to "
|
|
"'{{ event_data.task.column_title }}'"
|
|
),
|
|
"task.close": (
|
|
"<b>[{{ event_data.task.project_name }}]</b> "
|
|
"<font color='{{ event_data.task.color_id }}'>"
|
|
"{{ event_data.task.title }}"
|
|
"</font>"
|
|
"(#{{ event_data.task.id }}): "
|
|
"{{ event_author }} closed the task"
|
|
),
|
|
"comment.create": (
|
|
"<b>[{{ event_data.task.project_name }}]</b> "
|
|
"<font color='{{ event_data.task.color_id }}'>"
|
|
"{{ event_data.task.title }}"
|
|
"</font> "
|
|
"(#{{ event_data.task.id }}): "
|
|
"{{ event_author }} added a comment:\n"
|
|
"<blockquote>"
|
|
"{{ event_data.comment.comment | markdown }}"
|
|
"</blockquote>"
|
|
),
|
|
}
|
|
|
|
|
|
async def format_events(config, src, dest):
|
|
|
|
env = jinja2.Environment()
|
|
md = markdown.Markdown(extensions=["fenced_code"])
|
|
env.filters["markdown"] = lambda src: jinja2.Markup(md.convert(src))
|
|
|
|
while True:
|
|
event = await src.get()
|
|
logging.debug("Formatting message: %s", event)
|
|
try:
|
|
template = env.from_string(TEMPLATES[event["event_name"]])
|
|
except KeyError:
|
|
logging.info("Unknown message type: %s", event["event_name"])
|
|
continue
|
|
|
|
rendered = template.render(**event)
|
|
await dest.put(rendered)
|
|
|
|
|
|
async def send_notices(config, src):
|
|
client = nio.AsyncClient(config.matrix_homeserver, config.matrix_user)
|
|
await client.login(config.matrix_password)
|
|
|
|
while True:
|
|
formatted = str(await src.get())
|
|
|
|
await client.room_send(
|
|
config.matrix_room,
|
|
message_type="m.room.message",
|
|
content={
|
|
"msgtype": "m.notice",
|
|
"format": "org.matrix.custom.html",
|
|
"body": formatted, # à corriger
|
|
"formatted_body": formatted,
|
|
},
|
|
)
|
|
|
|
|
|
async def run_app(config, dest):
|
|
async def webhook(request):
|
|
tokens = request.query.getall("token", [])
|
|
if config.webhook_token not in tokens:
|
|
logging.warning("Invalid tokens: %s", tokens)
|
|
raise aiohttp.web.HTTPForbidden("Invalid tokens")
|
|
event = await request.json()
|
|
await dest.put(event)
|
|
|
|
app = aiohttp.web.Application()
|
|
app.add_routes([aiohttp.web.post("/webhook", webhook)])
|
|
|
|
runner = aiohttp.web.AppRunner(app)
|
|
await runner.setup()
|
|
await aiohttp.web.TCPSite(
|
|
runner, config.listen_addr, config.listen_port
|
|
).start()
|
|
|
|
|
|
async def main():
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-c", "--config-file", default="config.yaml")
|
|
|
|
args = parser.parse_args()
|
|
|
|
with open(args.config_file) as f:
|
|
config = yaml.safe_load(f)
|
|
|
|
config = Config(**config)
|
|
|
|
events = asyncio.Queue()
|
|
formatted = asyncio.Queue()
|
|
|
|
logging.info("Started Kanbot")
|
|
|
|
tasks = (
|
|
asyncio.create_task(run_app(config, events)),
|
|
asyncio.create_task(format_events(config, events, formatted)),
|
|
asyncio.create_task(send_notices(config, formatted)),
|
|
)
|
|
|
|
# on propage les exceptions
|
|
await asyncio.gather(*tasks, return_exceptions=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.get_event_loop().run_until_complete(main())
|