refoss

homeassistant/components/refoss/config_flow.py

@@ -12,6 +12,7 @@
 from homeassistant.helpers.typing import DiscoveryInfoType
 
 from .const import DOMAIN
+from .models import HomeAssistantRefossData
 from .util import get_refoss_socket_server
 
 
@@ -40,6 +41,9 @@ def message_received(self, data: dict):
                 data=data,
             )
 
+        elif "header" in data and "payload" in data:
+            self.hass.create_task(self.async_update_push_state(data=data))
+
     async def async_step_discovery(
         self, discovery_info: DiscoveryInfoType
     ) -> FlowResult:
@@ -56,3 +60,17 @@ async def async_step_discovery(
         self._abort_if_unique_id_configured(updates=discovery_info)
 
         return self.async_create_entry(title=device_name, data=discovery_info)
+
+    async def async_update_push_state(self, data: dict) -> None:
+        """Async update the status of device."""
+        header = data["header"]
+        uuid = header["uuid"]
+        if entry := await self.async_set_unique_id(uuid):
+            namespace = header["namespace"]
+            payload = data["payload"]
+            refoss_data: HomeAssistantRefossData = self.hass.data[DOMAIN][
+                entry.entry_id
+            ]
+            await refoss_data.base_device.async_handle_push_notification(
+                namespace, payload, uuid
+            )

homeassistant/components/refoss/device.py

@@ -25,6 +25,7 @@ def __init__(self, device: BaseDevice, channel: int) -> None:
         else:
             self._attr_name = str(channel)
         self.device = device
+        self.channel = channel
         self._attr_device_info = DeviceInfo(
             identifiers={(DOMAIN, device.uuid)},
             manufacturer="refoss",
@@ -41,4 +42,25 @@ def available(self) -> bool:
 
     async def async_device_update(self, warning: bool = True) -> None:
         """Async update device status."""
-        await self.device.async_handle_update()
+        await self.device.async_handle_update(self.channel)
+
+    async def async_added_to_hass(self) -> None:
+        """Call when entity is added to hass."""
+        self.device.register_push_notification_handler_coroutine(
+            self._async_push_notification_received
+        )
+
+    async def async_will_remove_from_hass(self) -> None:
+        """Call when entity is remove from hass."""
+        self.device.unregister_push_notification_handler_coroutine(
+            self._async_push_notification_received
+        )
+
+    async def _async_push_notification_received(
+        self, namespace: str, data: dict, uuid: str
+    ) -> None:
+        """Synchronize the status of device push."""
+        await self.device.async_update_push_state(
+            namespace=namespace, data=data, uuid=uuid
+        )
+        self.async_write_ha_state()

homeassistant/components/refoss/manifest.json

@@ -6,5 +6,5 @@
   "documentation": "https://www.home-assistant.io/integrations/refoss",
   "integration_type": "device",
   "iot_class": "local_polling",
-  "requirements": ["refoss-ha==1.1.3"]
+  "requirements": ["refoss-ha==1.1.6"]
 }

homeassistant/components/refoss/util.py

@@ -23,3 +23,14 @@ def shutdown_listener(ev: Event) -> None:
 
     hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, shutdown_listener)
     return socket_server
+
+
+def verify_msg(data: dict) -> str | None:
+    """Verify push msg."""
+    header = data.get("header", {})
+    namespace = header.get("namespace", None)
+    uuid = header.get("uuid", None)
+    payload = data.get("payload", None)
+    if namespace is None or uuid is None or payload is None:
+        return None
+    return uuid

requirements_all.txt

@@ -2303,7 +2303,7 @@ rapt-ble==0.1.2
 raspyrfm-client==1.2.8
 
 # homeassistant.components.refoss
-refoss-ha==1.1.3
+refoss-ha==1.1.6
 
 # homeassistant.components.rainmachine
 regenmaschine==2023.06.0

requirements_test_all.txt

@@ -1714,7 +1714,7 @@ radiotherm==2.1.0
 rapt-ble==0.1.2
 
 # homeassistant.components.refoss
-refoss-ha==1.1.3
+refoss-ha==1.1.6
 
 # homeassistant.components.rainmachine
 regenmaschine==2023.06.0