ungleich-public/ungleich-tools
6
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

matrix-client: add basic message getting

Browse source
This commit is contained in:
Nico Schottelius 2025-01-30 20:12:39 +01:00
commit ca3dee2c0e
1 changed files with 59 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

79
matrix/matrix-get-messages-from-encrypted-room-via-backup.py
View file

@ -16,20 +16,18 @@ class UngleichMatrixClient:
self.room_id = args.room_id
self.username = args.login_username
self.password = args.login_password
self.security_key = args.security_key
self.access_token = False
self.room_keys = False
self.room_messages = []
self.matrix_url = {}
self.matrix_url['login'] = f"{args.server_url}/_matrix/client/v3/login"
self.matrix_url['room_keys'] = f"{args.server_url}/_matrix/client/v3/room_keys/keys?version=1"
self.matrix_url['room_messages'] = f"{args.server_url}/_matrix/client/v3/rooms/{self.room_id}/messages"
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': {
@ -47,36 +45,77 @@ class UngleichMatrixClient:
raise Exception("Login Failed")
return r
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']
def get_room_keys(self):
self._ensure_logged_in()
params = {
'version': "1",
'access_token': self.access_token
}
if not self.room_keys:
r = requests.get(self.matrix_url['room_keys'],
params=params)
self.room_keys = r.json()
def get_room_messages(self):
"""
Get messages from a room, requires to use pagination!
Continue until no end property is in the reply anyomer
"""
self._ensure_logged_in()
params = {
'access_token': self.access_token
}
more_messages = True
next_batch = ""
while more_messages:
if next_batch:
params['from'] = next_batch
r = requests.get(self.matrix_url['room_messages'],
params=params)
self.room_messages.append(r.json())
if 'end' in r.json():
next_batch = r.json()['end']
else:
more_messages = False
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')
# self.get_room_keys()
# print(self.room_keys)
self.get_room_messages()
for message in self.room_messages:
print(message)
# Get room_keys
# http 'https://ungleich.matrix.ungleich.cloud/_matrix/client/v3/room_keys/keys' version==1 access_token=="$matrix_access_token"
# jq '.rooms."!pkP......."' < key-backup.json > room-specific-keys.json
# 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!! ")
parser.add_argument("--security-key", required=True, help="Your security backup key, i.e. ABCf defg aaaa - ensure to quote as one argument! ")
args = parser.parse_args()
client = UngleichMatrixClient(args)