first commit

backend/utils/ws_manager.py

@@ -0,0 +1,45 @@
+"""Gestionnaire de connexions WebSocket."""
+
+import json
+import logging
+from typing import Any
+
+from fastapi import WebSocket
+
+logger = logging.getLogger(__name__)
+
+
+class WSManager:
+    def __init__(self):
+        self.active: list[WebSocket] = []
+
+    async def connect(self, ws: WebSocket):
+        await ws.accept()
+        self.active.append(ws)
+        logger.debug("WS connecté (%d total)", len(self.active))
+
+    def disconnect(self, ws: WebSocket):
+        self.active.discard(ws) if hasattr(self.active, "discard") else None
+        if ws in self.active:
+            self.active.remove(ws)
+        logger.debug("WS déconnecté (%d restants)", len(self.active))
+
+    async def broadcast(self, data: dict[str, Any]):
+        if not self.active:
+            return
+        msg = json.dumps(data, default=str)
+        dead = []
+        for ws in self.active:
+            try:
+                await ws.send_text(msg)
+            except Exception:
+                dead.append(ws)
+        for ws in dead:
+            self.disconnect(ws)
+
+    async def send(self, ws: WebSocket, data: dict[str, Any]):
+        try:
+            await ws.send_text(json.dumps(data, default=str))
+        except Exception as e:
+            logger.debug("Erreur envoi WS: %s", e)
+            self.disconnect(ws)