from fastapi import APIRouter, Depends, HTTPException
from core.database import fake_users_db

router = APIRouter()

@router.post("/logout")
async def logout_handler(
    token: str = Depends(oauth2_scheme),
    db: Session = Depends(get_db)
):
    """Demo logout endpoint"""
    user = get_user_by_token(token, db)
    if not user:
        raise HTTPException(status_code=401, detail="Invalid authentication credentials")
    
    # 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")
    
    return response