API key working, CORS working, SocketIO working (but no JWT), Chat client v1, Session implemented (server side)

eveai_chat/socket_handlers/chat_handler.py

@@ -0,0 +1,52 @@
+from flask_jwt_extended import verify_jwt_in_request, get_jwt_identity, verify_jwt_in_request, decode_token
+from flask_socketio import emit, disconnect
+from flask import current_app, request
+from common.extensions import socketio
+
+
+@socketio.on('connect')
+def handle_connect():
+    try:
+        # Extract token from the auth object
+        token = request.args.get('token')
+        if not token:
+            raise Exception("Missing Authorization Token")
+        current_app.logger.debug(f'SocketIO: Received token: {token}')
+        # Verify token
+        decoded_token = decode_token(token.split(" ")[1])  # Split to remove "Bearer " prefix
+        tenant_id = decoded_token["identity"]["tenant_id"]
+        current_app.logger.info(f'SocketIO: Tenant {decoded_token["identity"]["tenant_id"]} connected')
+        # communicate connection to client
+        emit('connect', {'status': 'Connected', 'tenant_id': tenant_id})
+    except Exception as e:
+        current_app.logger.error(f'SocketIO: Connection failed: {e}')
+        # communicate connection problem to client
+        emit('connect', {'status': 'Connection Failed'})
+        disconnect()
+
+
+@socketio.on('disconnect')
+def handle_disconnect():
+    current_app.logger.debug('SocketIO: Client disconnected')
+
+
+@socketio.on('user_message')
+def handle_message(data):
+    try:
+        current_app.logger.debug(f"SocketIO: Received message from tenant {data['tenantId']}: {data['message']}")
+        verify_jwt_in_request()
+        current_tenant = get_jwt_identity()
+        print(f'Tenant {current_tenant["tenant_id"]} sent a message: {data}')
+        # Store interaction in the database
+        response = {
+            'tenantId': data['tenantId'],
+            'message': 'This is a bot response. Actual implementation still required.',
+            'messageId': 'bot-message-id',
+            'algorithm': 'alg1'
+        }
+        current_app.logger.debug(f"SocketIO: Bot response: {response}")
+        emit('bot_response', response, broadcast=True)
+    except Exception as e:
+        current_app.logger.error(f'SocketIO: Message handling failed: {e}')
+        disconnect()
+