narinder1231 commited on
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
- token_paylod = await websocket.receive_json()
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):