diff --git a/appservice.py b/appservice.py index 93c8431..b442ba9 100644 --- a/appservice.py +++ b/appservice.py @@ -108,21 +108,36 @@ def on_receive_events(transaction): elif event['type'] == 'm.room.member': - if event['content']['membership'] == 'invite' and event['content']['is_direct'] == True: - logging.info('>> GOT INVITE') - txId += 1 - url = app.config['MATRIX_HOST'] - url += '/_matrix/client/r0/join/' + event['room_id'] - url += "?access_token=" + app.config['MATRIX_AS_TOKEN'] - r = requests.post(url, headers={"Content-Type": "application/json"}, data=json.dumps({})) + if event['state_key'] == app.config['MATRIX_AS_ID']: - elif event['membership'] == 'leave': - logging.info('>> GOT LEAVE') - txId += 1 - url = app.config['MATRIX_HOST'] - url += '/_matrix/client/r0/rooms/' + event['room_id'] + "/leave" - url += "?access_token=" + app.config['MATRIX_AS_TOKEN'] - r = requests.post(url, headers={"Content-Type": "application/json"}, data=json.dumps({})) + if event['content']['membership'] == 'invite' + and 'is_direct' in event['content'] + and event['content']['is_direct'] == True: + + logging.info('>> GOT INVITE') + txId += 1 + url = app.config['MATRIX_HOST'] + url += '/_matrix/client/r0/join/' + event['room_id'] + url += "?access_token=" + app.config['MATRIX_AS_TOKEN'] + r = requests.post(url, headers={"Content-Type": "application/json"}, + data=json.dumps({})) + if r.status_code == 200: + addadminrm = MatrixAdminRooms(matrix_id=event['sender'], room_id=event['room_id']) + db.session.add(addadminrm) + db.session.commit() + + elif event['membership'] == 'leave': + logging.info('>> GOT LEAVE') + txId += 1 + url = app.config['MATRIX_HOST'] + url += '/_matrix/client/r0/rooms/' + event['room_id'] + "/leave" + url += "?access_token=" + app.config['MATRIX_AS_TOKEN'] + r = requests.post(url, headers={"Content-Type": "application/json"}, + data=json.dumps({})) + if r.status_code == 200: + adminrm = MatrixAdminRooms.query.filter_by(room_id=event['room_id']).first() + db.session.delete(adminrm) + db.session.commit() return jsonify({})