logan/drb-core-server
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Accept active token from node
All checks were successful
release-image / release-image (push) Successful in 2m8s
Details

Browse Source
This commit is contained in:
Logan Cusano
2025-06-29 02:37:17 -04:00
parent 7820e87989
commit d889f0e8ea
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/routers/nodes.py
View File

@@ -16,11 +16,12 @@ nodes_bp = Blueprint('nodes', __name__)
pending_requests = {} 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.""" """Registers a new client connection."""
current_app.active_clients[client_id] = ActiveClient() current_app.active_clients[client_id] = ActiveClient()
current_app.active_clients[client_id].websocket = websocket current_app.active_clients[client_id].websocket = websocket
current_app.active_clients[client_id].nickname = client_nickname 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.") print(f"Client {client_id} connected.")
# Create a JWT for the client # Create a JWT for the client

3
app/server.py
View File

@@ -42,7 +42,8 @@ async def websocket_server_handler(websocket):
if handshake_data.get("type") == "handshake" and "id" in handshake_data: if handshake_data.get("type") == "handshake" and "id" in handshake_data:
client_id = handshake_data["id"] client_id = handshake_data["id"]
client_nickname = handshake_data.get("nickname") 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: if not app.active_clients[client_id].access_token:
abort(500, "Error retrieving 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})) await websocket.send(json.dumps({"type": "handshake_ack", "status": "success", "access_token": app.active_clients[client_id].access_token}))