from typing import Any, Dict, List
from fastapi import WebSocket

class WSConnectionManager:
    def __init__(self):
        self.connections: Dict[str, List[WebSocket]] = {}

    async def connect(self, bot_id: str, websocket: WebSocket):
        await websocket.accept()
        self.connections.setdefault(bot_id, []).append(websocket)

    def disconnect(self, bot_id: str, websocket: WebSocket):
        conns = self.connections.get(bot_id, [])
        if websocket in conns:
            conns.remove(websocket)
        if not conns and bot_id in self.connections:
            del self.connections[bot_id]

    async def broadcast(self, bot_id: str, data: Dict[str, Any]):
        conns = list(self.connections.get(bot_id, []))
        for ws in conns:
            try:
                await ws.send_json(data)
            except Exception:
                self.disconnect(bot_id, ws)

manager = WSConnectionManager()