init

custom_components/alarmo/card.py

@@ -0,0 +1,34 @@
+"""WebSocket handler and registration for Alarmo card update events."""
+
+import voluptuous as vol
+from homeassistant.core import callback
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.components.websocket_api import decorators, async_register_command
+
+
+@decorators.websocket_command(
+    {
+        vol.Required("type"): "alarmo_updated",
+    }
+)
+@decorators.async_response
+async def handle_subscribe_updates(hass, connection, msg):
+    """Handle subscribe updates."""
+
+    @callback
+    def handle_event(event: str, area_id: str, args: dict = {}):
+        """Forward events to websocket."""
+        data = dict(**args, **{"event": event, "area_id": area_id})
+        connection.send_message(
+            {"id": msg["id"], "type": "event", "event": {"data": data}}
+        )
+
+    connection.subscriptions[msg["id"]] = async_dispatcher_connect(
+        hass, "alarmo_event", handle_event
+    )
+    connection.send_result(msg["id"])
+
+
+async def async_register_card(hass):
+    """Publish event to lovelace when alarm changes."""
+    async_register_command(hass, handle_subscribe_updates)