Add callback support to unfurled websocket

Author: JackJPowell

custom_components/unfoldedcircle/unfurled/unfurled/remote.py

@@ -5,6 +5,8 @@
 import asyncio
 import logging
 import re
+from collections.abc import Callable, Coroutine
+from typing import Any
 from urllib.parse import urljoin, urlparse
 
 import aiohttp
@@ -126,6 +128,7 @@ def __init__(
         # WebSocket
         self._ws_client: RemoteWebSocketClient | None = None
         self._last_update_type: UpdateType = UpdateType.NONE
+        self._state_change_callbacks: list[Callable[[], Coroutine[Any, Any, None]]] = []
 
         # Entities cache (media players referenced by activities)
         self._entities: dict[str, MediaPlayerEntity] = {}
@@ -291,6 +294,10 @@ async def resolve_discovery(
     # WebSocket
     # ------------------------------------------------------------------
 
+    def on_state_change(self, callback: Callable[[], Coroutine[Any, Any, None]]) -> None:
+        """Register a coroutine callback to be called after any WebSocket state update."""
+        self._state_change_callbacks.append(callback)
+
     async def connect_websocket(self, *, reconnect_delay: float = 10.0) -> None:
         """Start a WebSocket connection and keep it alive automatically.
 
@@ -343,6 +350,9 @@ async def _handle_ws_message(self, raw: str) -> None:
             case IRLearningEvent():
                 self._on_ir_learning(event)
 
+        for cb in self._state_change_callbacks:
+            await cb()
+
     # WS event handlers (private, synchronous)
 
     def _on_battery(self, event: BatteryEvent) -> None: