adds .local/share/millennium and easyeffects

.local/share/millennium/plugins/extendium/backend/websocket/client_manager.py

@@ -0,0 +1,94 @@
+"""
+Client manager for WebSocket connections.
+Handles tracking and managing connected clients.
+"""
+
+from typing import Any, Dict, List, Optional
+
+
+class ClientManager:
+    """
+    Manages WebSocket clients.
+    Tracks frontend and webkit clients separately.
+    """
+
+    def __init__(self):
+        """Initialize the client manager."""
+        self._frontend_client = None
+        self._webkit_clients: List[Dict[str, Any]] = []
+
+    @property
+    def frontend_client(self) -> Optional[Dict[str, Any]]:
+        """Get the frontend client."""
+        return self._frontend_client
+
+    @frontend_client.setter
+    def frontend_client(self, client: Dict[str, Any]):
+        """Set the frontend client."""
+        self._frontend_client = client
+
+    @property
+    def webkit_clients(self) -> List[Dict[str, Any]]:
+        """Get all webkit clients."""
+        return self._webkit_clients
+
+    def add_webkit_client(self, client: Dict[str, Any]):
+        """
+        Add a webkit client.
+
+        Args:
+            client: The WebSocket client object
+        """
+        self._webkit_clients.append(client)
+
+    def remove_client(self, client_id: str) -> bool:
+        """
+        Remove a client by its ID.
+
+        Args:
+            client_id: The ID of the client to remove
+
+        Returns:
+            bool: True if client was removed, False otherwise
+        """
+        # Check if it's the frontend client
+        if self._frontend_client and self._frontend_client.get('id') == client_id:
+            self._frontend_client = None
+            return True
+
+        # Check if it's a webkit client
+        for i, client in enumerate(self._webkit_clients):
+            if client.get('id') == client_id:
+                del self._webkit_clients[i]
+                return True
+
+        return False
+
+    def has_frontend_client(self) -> bool:
+        """Check if a frontend client is connected."""
+        return self._frontend_client is not None
+
+    def has_webkit_clients(self) -> bool:
+        """Check if any webkit clients are connected."""
+        return len(self._webkit_clients) > 0
+
+    def disconnect_all_clients(self):
+        if self._frontend_client:
+            try:
+                websocket = getattr(self._frontend_client, 'websocket', None)
+                if websocket and hasattr(websocket, 'close'):
+                    websocket.close()
+            except Exception:
+                pass
+            finally:
+                self._frontend_client = None
+
+        for client in self._webkit_clients:
+            try:
+                websocket = getattr(client, 'websocket', None)
+                if websocket and hasattr(websocket, 'close'):
+                    websocket.close()
+            except Exception:
+                pass
+
+        self._webkit_clients.clear()