ungleich-tools/matrix/matrix-get-messages-from-encrypted-room-via-backup.py

84 lines
2.5 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
# ungleich glarus ag, 2025-01-23
import base58
import base64
import json
import sys
import argparse
import requests
import pprint
class UngleichMatrixClient:
def __init__(self, args):
self.server = args.server_url
self.room_id = args.room_id
self.username = args.login_username
self.password = args.login_password
self.access_token = False
self.matrix_url = {}
self.matrix_url['login'] = f"{args.server_url}/_matrix/client/v3/login"
def ensure_logged_in(self):
if not self.access_token:
self.login_response = self.login_to_server()
self.access_token = self.login_response.json()['access_token']
print(self.access_token)
def login_to_server(self):
login_data = {
'identifier': {
"type": "m.id.user",
"user": f"{self.username}"
},
'type': "m.login.password",
'device_id': "ungleich-matrix-client",
'initial_device_display_name' : "ungleich-matrix-client",
'password': f"{self.password}"
}
r = requests.post(self.matrix_url['login'], json=login_data)
if not r.status_code == 200:
raise Exception("Login Failed")
return r
def get_messages(self):
self.ensure_logged_in()
# Login, get token
# matrix_server=https://ungleich.matrix.ungleich.cloud
# matrix_key_url=https://${matrix_server}/_matrix/client/v3/room_keys/keys
# matrix_user="\@nico:ungleich.ch"
# matrix_password=$(pass ldap/nico)
# matrix_room_id='!pkPjzozYKUNEbcZKKA:unchat.cat'
# matrix_access_token=$(echo $matrix_login | jq .access_token | sed 's/"//g')
# Get room_keys
# Get messages from room
# Decrypt each message:
# Retrieve the session key
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--server-url", required=True, help="Matrix Server URL, i.e. https://your-server ")
parser.add_argument("--room-id", required=True, help="ID of the room to get messages from, i.e. !...:your-matrix-domain ")
parser.add_argument("--login-username", required=True, help="Username for logging into the server, i.e. @you:your-matrix-domain ")
parser.add_argument("--login-password", required=True, help="Password for logging into the server, i.e. your-very-safe-password!! ")
args = parser.parse_args()
client = UngleichMatrixClient(args)
client.get_messages()