Spaces:
Runtime error
Runtime error
Commit ·
07aae89
1
Parent(s): 4d58bc4
add token in query params instead of onopen event
Browse files
src/controllers/_conversation_controller.py
CHANGED
|
@@ -28,16 +28,13 @@ class ConnectionManager:
|
|
| 28 |
|
| 29 |
async def connect(self, websocket: WebSocket):
|
| 30 |
await websocket.accept()
|
| 31 |
-
|
| 32 |
-
token = token_paylod.get("headers").get("Token")
|
| 33 |
|
| 34 |
-
if not re.match(r"Bearer .+", token):
|
| 35 |
-
await websocket.close(code=1008, reason="Invalid token")
|
| 36 |
-
token = token.split(" ")[1]
|
| 37 |
user = self.jwt.validate_jwt(token)
|
| 38 |
if not user:
|
| 39 |
await websocket.close(code=1008, reason="Invalid token")
|
| 40 |
self.active_connections.append(websocket)
|
|
|
|
| 41 |
return user
|
| 42 |
|
| 43 |
def disconnect(self, websocket: WebSocket):
|
|
|
|
| 28 |
|
| 29 |
async def connect(self, websocket: WebSocket):
|
| 30 |
await websocket.accept()
|
| 31 |
+
token = websocket.query_params.get("token")
|
|
|
|
| 32 |
|
|
|
|
|
|
|
|
|
|
| 33 |
user = self.jwt.validate_jwt(token)
|
| 34 |
if not user:
|
| 35 |
await websocket.close(code=1008, reason="Invalid token")
|
| 36 |
self.active_connections.append(websocket)
|
| 37 |
+
|
| 38 |
return user
|
| 39 |
|
| 40 |
def disconnect(self, websocket: WebSocket):
|