From d889f0e8eae950db74955cf679f86958136adcce Mon Sep 17 00:00:00 2001 From: Logan Cusano Date: Sun, 29 Jun 2025 02:37:17 -0400 Subject: [PATCH] Accept active token from node --- app/routers/nodes.py | 3 ++- app/server.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/app/routers/nodes.py b/app/routers/nodes.py index edd20e6..1a335b1 100644 --- a/app/routers/nodes.py +++ b/app/routers/nodes.py @@ -16,11 +16,12 @@ nodes_bp = Blueprint('nodes', __name__) pending_requests = {} -async def register_client(websocket, client_id, client_nickname): +async def register_client(websocket, client_id, client_nickname, active_token): """Registers a new client connection.""" current_app.active_clients[client_id] = ActiveClient() current_app.active_clients[client_id].websocket = websocket current_app.active_clients[client_id].nickname = client_nickname + current_app.active_clients[client_id].active_token = active_token print(f"Client {client_id} connected.") # Create a JWT for the client diff --git a/app/server.py b/app/server.py index 14c1b4e..e0d7279 100644 --- a/app/server.py +++ b/app/server.py @@ -42,7 +42,8 @@ async def websocket_server_handler(websocket): if handshake_data.get("type") == "handshake" and "id" in handshake_data: client_id = handshake_data["id"] client_nickname = handshake_data.get("nickname") - await register_client(websocket, client_id, client_nickname) + client_active_token = handshake_data.get("active_token") + await register_client(websocket, client_id, client_nickname, client_active_token) if not app.active_clients[client_id].access_token: abort(500, "Error retrieving access token") await websocket.send(json.dumps({"type": "handshake_ack", "status": "success", "access_token": app.active_clients[client_id].access_token}))