# services/comms/guest-name-job.yaml apiVersion: batch/v1 kind: CronJob metadata: name: guest-name-randomizer namespace: comms spec: schedule: "*/1 * * * *" suspend: false successfulJobsHistoryLimit: 1 failedJobsHistoryLimit: 1 jobTemplate: spec: backoffLimit: 0 template: spec: restartPolicy: Never volumes: - name: mas-admin-client secret: secretName: mas-admin-client-runtime items: - key: client_secret path: client_secret containers: - name: rename image: python:3.11-slim volumeMounts: - name: mas-admin-client mountPath: /etc/mas-admin-client readOnly: true env: - name: SYNAPSE_BASE value: http://othrys-synapse-matrix-synapse:8008 - name: MAS_ADMIN_CLIENT_ID value: 01KDXMVQBQ5JNY6SEJPZW6Z8BM - name: MAS_ADMIN_CLIENT_SECRET_FILE value: /etc/mas-admin-client/client_secret - name: MAS_ADMIN_API_BASE value: http://matrix-authentication-service:8081/api/admin/v1 - name: MAS_TOKEN_URL value: http://matrix-authentication-service:8080/oauth2/token - name: SEEDER_USER value: othrys-seeder - name: PGHOST value: postgres-service.postgres.svc.cluster.local - name: PGPORT value: "5432" - name: PGDATABASE value: synapse - name: PGUSER value: synapse - name: PGPASSWORD valueFrom: secretKeyRef: name: synapse-db key: POSTGRES_PASSWORD command: - /bin/sh - -c - | set -euo pipefail pip install --no-cache-dir requests psycopg2-binary >/dev/null python - <<'PY' import base64 import os import random import requests import time import urllib.parse import psycopg2 ADJ = [ "brisk","calm","eager","gentle","merry","nifty","rapid","sunny","witty","zesty", "amber","bold","bright","crisp","daring","frosty","glad","jolly","lively","mellow", "quiet","ripe","serene","spry","tidy","vivid","warm","wild","clever","kind", ] NOUN = [ "otter","falcon","comet","ember","grove","harbor","meadow","raven","river","summit", "breeze","cedar","cinder","cove","delta","forest","glade","lark","marsh","peak", "pine","quartz","reef","ridge","sable","sage","shore","thunder","vale","zephyr", ] BASE = os.environ["SYNAPSE_BASE"] MAS_ADMIN_CLIENT_ID = os.environ["MAS_ADMIN_CLIENT_ID"] MAS_ADMIN_CLIENT_SECRET_FILE = os.environ["MAS_ADMIN_CLIENT_SECRET_FILE"] MAS_ADMIN_API_BASE = os.environ["MAS_ADMIN_API_BASE"].rstrip("/") MAS_TOKEN_URL = os.environ["MAS_TOKEN_URL"] SEEDER_USER = os.environ["SEEDER_USER"] ROOM_ALIAS = "#othrys:live.bstein.dev" SERVER_NAME = "live.bstein.dev" def mas_admin_token(): with open(MAS_ADMIN_CLIENT_SECRET_FILE, "r", encoding="utf-8") as f: secret = f.read().strip() basic = base64.b64encode(f"{MAS_ADMIN_CLIENT_ID}:{secret}".encode()).decode() last_err = None for attempt in range(5): try: r = requests.post( MAS_TOKEN_URL, headers={"Authorization": f"Basic {basic}"}, data={"grant_type": "client_credentials", "scope": "urn:mas:admin"}, timeout=30, ) r.raise_for_status() return r.json()["access_token"] except Exception as exc: # noqa: BLE001 last_err = exc time.sleep(2 ** attempt) raise last_err def mas_user_id(token, username): r = requests.get( f"{MAS_ADMIN_API_BASE}/users/by-username/{urllib.parse.quote(username)}", headers={"Authorization": f"Bearer {token}"}, timeout=30, ) r.raise_for_status() return r.json()["data"]["id"] def mas_personal_session(token, user_id): r = requests.post( f"{MAS_ADMIN_API_BASE}/personal-sessions", headers={"Authorization": f"Bearer {token}"}, json={ "actor_user_id": user_id, "human_name": "guest-name-randomizer", "scope": "urn:matrix:client:api:*", "expires_in": 300, }, timeout=30, ) r.raise_for_status() data = r.json().get("data", {}).get("attributes", {}) or {} return data["access_token"], r.json()["data"]["id"] def mas_revoke_session(token, session_id): requests.post( f"{MAS_ADMIN_API_BASE}/personal-sessions/{urllib.parse.quote(session_id)}/revoke", headers={"Authorization": f"Bearer {token}"}, json={}, timeout=30, ) def resolve_alias(token, alias): headers = {"Authorization": f"Bearer {token}"} enc = urllib.parse.quote(alias) r = requests.get(f"{BASE}/_matrix/client/v3/directory/room/{enc}", headers=headers) r.raise_for_status() return r.json()["room_id"] def room_members(token, room_id): headers = {"Authorization": f"Bearer {token}"} r = requests.get(f"{BASE}/_matrix/client/v3/rooms/{urllib.parse.quote(room_id)}/members", headers=headers) r.raise_for_status() members = set() existing_names = set() for ev in r.json().get("chunk", []): user_id = ev.get("state_key") if user_id: members.add(user_id) disp = (ev.get("content") or {}).get("displayname") if disp: existing_names.add(disp) return members, existing_names def mas_list_users(token): headers = {"Authorization": f"Bearer {token}"} users = [] cursor = None while True: url = f"{MAS_ADMIN_API_BASE}/users?page[size]=100" if cursor: url += f"&page[after]={urllib.parse.quote(cursor)}" r = requests.get(url, headers=headers, timeout=30) r.raise_for_status() data = r.json().get("data", []) if not data: break users.extend(data) cursor = data[-1].get("meta", {}).get("page", {}).get("cursor") if not cursor: break return users def synapse_list_users(token): headers = {"Authorization": f"Bearer {token}"} users = [] from_token = None while True: url = f"{BASE}/_synapse/admin/v2/users?local=true&deactivated=false&limit=100" if from_token: url += f"&from={urllib.parse.quote(from_token)}" r = requests.get(url, headers=headers, timeout=30) r.raise_for_status() payload = r.json() users.extend(payload.get("users", [])) from_token = payload.get("next_token") if not from_token: break return users def user_id_for_username(username): return f"@{username}:live.bstein.dev" def get_displayname(token, user_id): headers = {"Authorization": f"Bearer {token}"} r = requests.get(f"{BASE}/_matrix/client/v3/profile/{urllib.parse.quote(user_id)}", headers=headers) r.raise_for_status() return r.json().get("displayname") def get_displayname_admin(token, user_id): headers = {"Authorization": f"Bearer {token}"} r = requests.get( f"{BASE}/_synapse/admin/v2/users/{urllib.parse.quote(user_id)}", headers=headers, timeout=30, ) if r.status_code == 404: return None r.raise_for_status() return r.json().get("displayname") def set_displayname(token, room_id, user_id, name, in_room): headers = {"Authorization": f"Bearer {token}"} payload = {"displayname": name} r = requests.put( f"{BASE}/_matrix/client/v3/profile/{urllib.parse.quote(user_id)}/displayname", headers=headers, json=payload, ) r.raise_for_status() if not in_room: return state_url = f"{BASE}/_matrix/client/v3/rooms/{urllib.parse.quote(room_id)}/state/m.room.member/{urllib.parse.quote(user_id)}" content = {"membership": "join", "displayname": name} requests.put(state_url, headers=headers, json=content, timeout=30) def set_displayname_admin(token, user_id, name): headers = {"Authorization": f"Bearer {token}"} payload = {"displayname": name} r = requests.put( f"{BASE}/_synapse/admin/v2/users/{urllib.parse.quote(user_id)}", headers=headers, json=payload, timeout=30, ) if r.status_code in (200, 201, 204): return True return False def needs_rename_username(username): return username.isdigit() or username.startswith("guest-") def needs_rename_display(display): return not display or display.isdigit() or display.startswith("guest-") def db_rename_numeric(existing_names): user_ids = [] profiles = {} conn = psycopg2.connect( host=os.environ["PGHOST"], port=int(os.environ["PGPORT"]), dbname=os.environ["PGDATABASE"], user=os.environ["PGUSER"], password=os.environ["PGPASSWORD"], ) try: with conn: with conn.cursor() as cur: cur.execute( "SELECT name FROM users WHERE name ~ %s", (f"^@\\d+:{SERVER_NAME}$",), ) user_ids = [row[0] for row in cur.fetchall()] if not user_ids: return cur.execute( "SELECT user_id, displayname FROM profiles WHERE user_id = ANY(%s)", (user_ids,), ) profiles = {row[0]: row[1] for row in cur.fetchall()} for user_id in user_ids: display = profiles.get(user_id) if display and not needs_rename_display(display): continue new = None for _ in range(30): candidate = f"{random.choice(ADJ)}-{random.choice(NOUN)}" if candidate not in existing_names: new = candidate existing_names.add(candidate) break if not new: continue cur.execute( "INSERT INTO profiles (user_id, displayname, full_user_id) VALUES (%s, %s, %s) ON CONFLICT (user_id) DO UPDATE SET displayname = EXCLUDED.displayname", (user_id, new, user_id), ) finally: conn.close() admin_token = mas_admin_token() seeder_id = mas_user_id(admin_token, SEEDER_USER) seeder_token, seeder_session = mas_personal_session(admin_token, seeder_id) try: room_id = resolve_alias(seeder_token, ROOM_ALIAS) members, existing = room_members(seeder_token, room_id) users = mas_list_users(admin_token) mas_usernames = set() for user in users: attrs = user.get("attributes") or {} username = attrs.get("username") or "" if username: mas_usernames.add(username) legacy_guest = attrs.get("legacy_guest") if not username: continue if not (legacy_guest or needs_rename_username(username)): continue user_id = user_id_for_username(username) access_token, session_id = mas_personal_session(admin_token, user["id"]) try: display = get_displayname(access_token, user_id) if display and not needs_rename_display(display): continue new = None for _ in range(30): candidate = f"{random.choice(ADJ)}-{random.choice(NOUN)}" if candidate not in existing: new = candidate existing.add(candidate) break if not new: continue set_displayname(access_token, room_id, user_id, new, user_id in members) finally: mas_revoke_session(admin_token, session_id) try: entries = synapse_list_users(seeder_token) except Exception as exc: # noqa: BLE001 print(f"synapse admin list skipped: {exc}") entries = [] for entry in entries: user_id = entry.get("name") or "" if not user_id.startswith("@"): continue localpart = user_id.split(":", 1)[0].lstrip("@") if localpart in mas_usernames: continue is_guest = entry.get("is_guest") if not (is_guest or needs_rename_username(localpart)): continue display = get_displayname_admin(seeder_token, user_id) if display and not needs_rename_display(display): continue new = None for _ in range(30): candidate = f"{random.choice(ADJ)}-{random.choice(NOUN)}" if candidate not in existing: new = candidate existing.add(candidate) break if not new: continue if not set_displayname_admin(seeder_token, user_id, new): continue db_rename_numeric(existing) finally: mas_revoke_session(admin_token, seeder_session) PY