From 2c97fa3d3f1f670469cde287deb25cb54dc1c99c Mon Sep 17 00:00:00 2001
From: Guilhem Saurel <guilhem.saurel@laas.fr>
Date: Sun, 16 Feb 2020 13:47:00 +0100
Subject: [PATCH] =?UTF-8?q?matrix-client=20=E2=86=92=20matrix-nio?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 matrix_webhook.py | 40 ++++++++++++++++++++++++----------------
 1 file changed, 24 insertions(+), 16 deletions(-)

diff --git a/matrix_webhook.py b/matrix_webhook.py
index 6bc0e67..3780b28 100755
--- a/matrix_webhook.py
+++ b/matrix_webhook.py
@@ -2,6 +2,8 @@
 """
 Matrix Webhook
 Post a message to a matrix room with a simple HTTP POST
+v1: matrix-client & http.server
+v2: matrix-nio & aiohttp
 """
 
 import asyncio
@@ -9,17 +11,15 @@ import json
 import os
 
 from aiohttp import web
-from matrix_client.client import MatrixClient
+from nio import AsyncClient
+from nio.rooms import MatrixRoom
 
 SERVER_ADDRESS = ('', int(os.environ.get('PORT', 4785)))
 MATRIX_URL = os.environ.get('MATRIX_URL', 'https://matrix.org')
-MATRIX_ID = os.environ.get('MATRIX_ID', 'wwm')
+MATRIX_ID = os.environ.get('MATRIX_ID', '@wwm:matrix.org')
 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()
+CLIENT = AsyncClient(MATRIX_URL, MATRIX_ID)
 
 
 async def handler(request):
@@ -27,32 +27,40 @@ async def handler(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())
+    data = await request.read()
+    data = json.loads(data.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']))
+            status, ret = 200, 'OK'
+            await CLIENT.room_send(room_id=str(request.rel_url)[1:],
+                                   message_type="m.room.message",
+                                   content={
+                                       "msgtype": "m.text",
+                                       "body": data['text'],
+                                   })
 
-    return web.Response(text='{"status": %i, "ret": "%a"}' % (status, ret),
+    return web.Response(text='{"status": %i, "ret": "%s"}' % (status, ret),
                         content_type='application/json',
                         status=status)
 
 
 async def main():
+    """
+    Main coroutine
+
+    matrix client login & start web server
+    """
+
+    await CLIENT.login(MATRIX_PW)
+
     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)