Update code in endpoints/logout.post.py

endpoints/logout.post.py

@@ -5,18 +5,28 @@ router = APIRouter()
 
 @router.post("/logout")
 async def logout_handler(
-    token: str = Depends(oauth2_scheme),
-    db: Session = Depends(get_db)
+    session_id: str = Depends(get_session),
+    user_id: str = Depends(get_current_user)
 ):
     """Demo logout endpoint"""
-    user = get_user_by_token(token, db)
+    user = fake_users_db.get(user_id)
     if not user:
-        raise HTTPException(status_code=401, detail="Invalid authentication credentials")
+        raise HTTPException(status_code=404, detail="User not found")
     
-    # Clear access token, refresh token and session from cookies
-    response = RedirectResponse(url="/")
-    response.delete_cookie("access_token")
-    response.delete_cookie("refresh_token")
-    response.delete_cookie("session_id")
+    # Clear session data
+    session_data = fake_sessions_db.pop(session_id, None)
+    if session_data:
+        session_data["active"] = False
+        session_data["logout_time"] = datetime.utcnow()
     
-    return response
+    # Clear access and refresh tokens
+    user["access_token"] = None
+    user["refresh_token"] = None
+    
+    return {
+        "message": "Logout successful",
+        "user_id": user_id,
+        "next_steps": [
+            "Clear client-side cookies/storage"
+        ]
+    }