genericity: send any text, to any room

This commit is contained in:
Guilhem Saurel 2019-02-13 01:04:32 +01:00
parent c54e5da19d
commit 675f3855f0

View file

@ -14,7 +14,6 @@ SERVER_ADDRESS = ('', int(os.environ.get('PORT', 4785)))
MATRIX_URL = os.environ.get('MATRIX_URL', 'https://matrix.org') 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_PW = os.environ['MATRIX_PW'] MATRIX_PW = os.environ['MATRIX_PW']
ROOM_ID = os.environ['ROOM_ID']
API_KEY = os.environ['API_KEY'] API_KEY = os.environ['API_KEY']
@ -27,7 +26,7 @@ class WWMBotServer(HTTPServer):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
client = MatrixClient(MATRIX_URL) client = MatrixClient(MATRIX_URL)
client.login(username=MATRIX_ID, password=MATRIX_PW) client.login(username=MATRIX_ID, password=MATRIX_PW)
self.room = client.get_rooms()[ROOM_ID] self.rooms = client.get_rooms()
class WWMBotForwarder(BaseHTTPRequestHandler): class WWMBotForwarder(BaseHTTPRequestHandler):
@ -46,8 +45,10 @@ class WWMBotForwarder(BaseHTTPRequestHandler):
if all(key in data for key in ['text', 'key']): if all(key in data for key in ['text', 'key']):
status = 'wrong key' status = 'wrong key'
if data['key'] == API_KEY: if data['key'] == API_KEY:
status = 'I need the id of the room as a path'
if self.path[1:] in self.server.rooms:
status = 'OK' status = 'OK'
self.server.room.send_text(data['text']) self.server.rooms[self.path[1:]].send_text(data['text'])
self.send_response(200 if status == 'OK' else 401) self.send_response(200 if status == 'OK' else 401)
self.send_header('Content-Type', 'application/json') self.send_header('Content-Type', 'application/json')