# backend/app/routes/websocket.py
from fastapi import APIRouter, WebSocket, WebSocketDisconnect

from ..core.websocket_manager import manager

router = APIRouter(prefix="/ws", tags=["Websocket"])


@router.websocket("/")
async def websocket_endpoint(websocket: WebSocket):
    await manager.connect(websocket)
    try:
        while True:
            await websocket.receive_text()
    except WebSocketDisconnect:
        manager.disconnect(websocket)