from fastapi import APIRouter
from starlette.websockets import WebSocket, WebSocketDisconnect

from db.postgres.ws_manager import ConnectionManager

ws_manager = ConnectionManager()


iot_ws_router = APIRouter()


@iot_ws_router.websocket("/device-status")
async def websocket_device_status(websocket: WebSocket):
    await ws_manager.connect(websocket)
    print("[WS] client connected")

    try:
        while True:
            # 这里不需要接收任何消息
            await websocket.receive_text()
    except WebSocketDisconnect:
        await ws_manager.disconnect(websocket)
        print("[WS] client disconnected")