nest/nest-matrix-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Persist credentials

Browse source
This commit is contained in:
Astra Logical 2026-07-23 11:25:21 -04:00
parent f7db8f7fd9
commit f0ce8b0ffa
1 changed files with 28 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
matrix_keycloak_bot.py
View file

@ -35,6 +35,8 @@ KEYCLOAK_REALM = os.getenv("KEYCLOAK_REALM", "master")
KEYCLOAK_CLIENT_ID = os.getenv("KEYCLOAK_CLIENT_ID", "nest-matrix-bot") KEYCLOAK_CLIENT_ID = os.getenv("KEYCLOAK_CLIENT_ID", "nest-matrix-bot")
KEYCLOAK_CLIENT_SECRET = os.getenv("KEYCLOAK_CLIENT_SECRET", "your-client-secret") KEYCLOAK_CLIENT_SECRET = os.getenv("KEYCLOAK_CLIENT_SECRET", "your-client-secret")
CREDENTIALS_FILE = "/app/store/credentials.json"
# In-memory tracking of active requests # In-memory tracking of active requests
pending_requests = {} pending_requests = {}
@ -458,10 +460,33 @@ async def main():
), ),
) )
print(f"Connecting to Matrix homeserver {MATRIX_HOMESERVER}...") # 1. Attempt to restore an existing session
await client.login(MATRIX_BOT_PASSWORD) if os.path.exists(CREDENTIALS_FILE):
print(f"Logged in as {MATRIX_BOT_USER}. Syncing...") print("Restoring existing Matrix session...")
with open(CREDENTIALS_FILE, "r") as f:
creds = json.load(f)
client.access_token = creds["access_token"]
client.user_id = creds["user_id"]
client.device_id = creds["device_id"]
# 2. If no session exists, log in and save the credentials
else:
print(f"No existing session found. Logging in to {MATRIX_HOMESERVER}...")
resp = await client.login(MATRIX_BOT_PASSWORD)
# Create the store directory if it doesn't exist
os.makedirs(os.path.dirname(CREDENTIALS_FILE), exist_ok=True)
with open(CREDENTIALS_FILE, "w") as f:
json.dump({
"access_token": resp.access_token,
"user_id": resp.user_id,
"device_id": resp.device_id,
}, f)
print(f"Logged in and saved credentials for {MATRIX_BOT_USER}.")
print("Syncing...")
await client.sync_forever(timeout=30000, full_state=True) await client.sync_forever(timeout=30000, full_state=True)