"""
Routes WebSocket pour les mises à jour en temps réel.
"""

from fastapi import APIRouter, WebSocket, WebSocketDisconnect

from app.services import ws_manager

router = APIRouter()


@router.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    """Endpoint WebSocket pour les mises à jour en temps réel."""
    await ws_manager.connect(websocket)
    try:
        while True:
            # Garder la connexion ouverte
            data = await websocket.receive_text()
            # Traiter les messages entrants si nécessaire
    except WebSocketDisconnect:
        ws_manager.disconnect(websocket)