feat: JWT auth in WebSocket handler

src/tracker/ws/handler.py

@@ -16,12 +16,18 @@ router = APIRouter()
 
 
 @router.websocket("/ws")
-async def websocket_endpoint(ws: WebSocket):
+async def websocket_endpoint(ws: WebSocket, token: str = ""):
     await ws.accept()
     session_id = None
 
     try:
-        # Wait for auth message
+        # Try query param token first (for direct JWT auth)
+        if token:
+            session_id = await _authenticate(ws, token)
+            if not session_id:
+                return
+        else:
+            # Wait for auth message (backward compatibility with agents)
             auth_msg = await ws.receive_json()
             if auth_msg.get("type") != "auth":
                 await ws.send_json({"type": "auth.error", "message": "First message must be auth"})
@@ -86,22 +92,28 @@ async def websocket_endpoint(ws: WebSocket):
 async def _authenticate(ws: WebSocket, token: str, on_behalf_of: str | None = None) -> str | None:
     """Authenticate and register session. Returns session_id or None."""
     async with async_session() as db:
+        member = None
+        
+        # Check if it's an agent token (starts with 'tb-')
+        if token.startswith("tb-"):
             result = await db.execute(
                 select(Member).where(Member.token == token).options(selectinload(Member.agent_config))
             )
             member = result.scalar_one_or_none()
-
+        else:
-        if not member:
+            # Try JWT decode
             from tracker.api.auth import decode_jwt
             try:
                 payload = decode_jwt(token)
+                member_id = payload["sub"]
                 result = await db.execute(
-                    select(Member).where(Member.id == payload["sub"])
+                    select(Member).where(Member.id == member_id)
                     .options(selectinload(Member.agent_config))
                 )
                 member = result.scalar_one_or_none()
-            except Exception:
+                logger.info("JWT auth successful for member_id=%s", member_id)
-                pass
+            except Exception as e:
+                logger.warning("JWT decode failed: %s", e)
 
         if not member:
             await ws.send_json({"type": "auth.error", "message": "Invalid token"})