diff --git a/Dockerfile b/Dockerfile index 506a076..c9b7df0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,7 +3,8 @@ FROM python:3.7-alpine EXPOSE 4785 RUN pip3 install --no-cache-dir \ - matrix-client + aiohttp \ + matrix-nio ADD matrix_webhook.py / diff --git a/matrix_webhook.py b/matrix_webhook.py index 0cbfaaf..6bc0e67 100755 --- a/matrix_webhook.py +++ b/matrix_webhook.py @@ -4,10 +4,11 @@ Matrix Webhook Post a message to a matrix room with a simple HTTP POST """ +import asyncio import json import os -from http.server import BaseHTTPRequestHandler, HTTPServer +from aiohttp import web from matrix_client.client import MatrixClient SERVER_ADDRESS = ('', int(os.environ.get('PORT', 4785))) @@ -16,47 +17,52 @@ MATRIX_ID = os.environ.get('MATRIX_ID', 'wwm') MATRIX_PW = os.environ['MATRIX_PW'] API_KEY = os.environ['API_KEY'] +client = MatrixClient(MATRIX_URL) +client.login(username=MATRIX_ID, password=MATRIX_PW) +rooms = client.get_rooms() -class MatrixWebhookServer(HTTPServer): + +async def handler(request): """ - an HTTPServer that embeds a matrix client - """ - - def __init__(self, *args, **kwargs): - super().__init__(*args, **kwargs) - self.client = MatrixClient(MATRIX_URL) - self.client.login(username=MATRIX_ID, password=MATRIX_PW) - self.rooms = self.client.get_rooms() - - -class MatrixWebhookHandler(BaseHTTPRequestHandler): - """ - Class given to the server, st. it knows what to do with an HTTP request. + Coroutine given to the server, st. it knows what to do with an HTTP request. This one handles a POST, checks its content, and forwards it to the matrix room. """ + data = json.loads(request.read().decode()) + status, ret = 400, 'I need a json dict with text & key' + if all(key in data for key in ['text', 'key']): + status, ret = 401, 'I need the good "key"' + if data['key'] == API_KEY: + status, ret = 404, 'I need the id of the room as a path, and to be in this room' + if request.rel_url[1:] not in rooms: + # try to see if this room has been joined recently + rooms = client.get_rooms() + if request.rel_url[1:] in rooms: + status, ret = 200, json.dumps(rooms[request.rel_url[1:]].send_text(data['text'])) - def do_POST(self): - """ - get a json dict from the request, send a message to a matrix room - """ - length = int(self.headers.get('Content-Length')) - data = json.loads(self.rfile.read(length).decode()) - status, ret = 400, 'I need a json dict with text & key' - if all(key in data for key in ['text', 'key']): - status, ret = 401, 'I need the good "key"' - if data['key'] == API_KEY: - status, ret = 404, 'I need the id of the room as a path, and to be in this room' - if self.path[1:] not in self.server.rooms: - # try to see if this room has been joined recently - self.server.rooms = self.server.client.get_rooms() - if self.path[1:] in self.server.rooms: - status, ret = 200, json.dumps(self.server.rooms[self.path[1:]].send_text(data['text'])) + return web.Response(text='{"status": %i, "ret": "%a"}' % (status, ret), + content_type='application/json', + status=status) - self.send_response(status) - self.send_header('Content-Type', 'application/json') - self.end_headers() - self.wfile.write(b'{"status": %i, "ret": "%a"}' % (status, ret)) + +async def main(): + server = web.Server(handler) + runner = web.ServerRunner(server) + await runner.setup() + site = web.TCPSite(runner, *SERVER_ADDRESS) + await site.start() + + print("======= Serving ======") + + # pause here for very long time by serving HTTP requests and + # waiting for keyboard interruption + await asyncio.sleep(100 * 3600) if __name__ == '__main__': - MatrixWebhookServer(SERVER_ADDRESS, MatrixWebhookHandler).serve_forever() + loop = asyncio.get_event_loop() + + try: + loop.run_until_complete(main()) + except KeyboardInterrupt: + pass + loop.close()