feat: add config and status for BLU TRV

aioshelly/const.py

@@ -144,10 +144,13 @@
 # Firmware 1.0.99 release date
 GEN3_MIN_FIRMWARE_DATE = 20231102
 
+# Firmware 1.5.x release date
+# Temporary use beta release to allow BluTrv support
+GEN3_GATEWAY_MIN_FIRMWARE_DATE = 20241219
+
 # Firmware 1.4.x release date
 GEN4_MIN_FIRMWARE_DATE = 20240902
 
-
 # Fallback for unknown devices
 MIN_FIRMWARE_DATES = {
     GEN1: GEN1_MIN_FIRMWARE_DATE,
@@ -473,7 +476,7 @@ class ShellyDevice:
     MODEL_BLU_GATEWAY_GEN3: ShellyDevice(
         model="S3GW-1DBT001",
         name="Shelly BLU Gateway Gen3",
-        min_fw_date=GEN3_MIN_FIRMWARE_DATE,
+        min_fw_date=GEN3_GATEWAY_MIN_FIRMWARE_DATE,
         gen=GEN3,
         supported=True,
     ),
@@ -929,6 +932,10 @@ class ShellyDevice:
     MODEL_VINTAGE_V2,
 }
 
+BLU_TRV_IDENTIFIER = "blutrv"
+BLU_TRV_MODEL_ID = {8: "SBTR-001AEU"}
+BLU_TRV_MODEL_NAME = {"SBTR-001AEU": "Shelly BLU TRV"}
+
 
 class UndefinedType(Enum):
     """Singleton type for use with not set sentinel values."""

aioshelly/rpc_device/device.py

@@ -18,11 +18,14 @@
     process_ip_or_options,
 )
 from ..const import (
+    BLU_TRV_IDENTIFIER,
+    BLU_TRV_MODEL_ID,
     CONNECT_ERRORS,
     DEVICE_INIT_TIMEOUT,
     DEVICE_IO_TIMEOUT,
     DEVICE_POLL_TIMEOUT,
     FIRMWARE_PATTERN,
+    MODEL_BLU_GATEWAY_GEN3,
     NOTIFY_WS_CLOSED,
     VIRTUAL_COMPONENTS,
     VIRTUAL_COMPONENTS_MIN_FIRMWARE,
@@ -280,6 +283,7 @@
         self._status = results[0]
         if has_dynamic:
             self._parse_dynamic_components(results[1])
+            await self._retrieve_blutrv_components(results[1])
 
     async def _init_calls(self) -> None:
         """Make calls needed to initialize the device."""
@@ -309,7 +313,9 @@
         if fetch_status:
             self._status = results.pop(0)
         if fetch_dynamic:
-            self._parse_dynamic_components(results.pop(0))
+            components = results.pop(0)
+            self._parse_dynamic_components(components)
+            await self._retrieve_blutrv_components(components)
 
     async def script_list(self) -> list[ShellyScript]:
         """Get a list of scripts from 'Script.List'."""
@@ -509,6 +515,7 @@
             return
         components = await self.call_rpc("Shelly.GetComponents", {"dynamic_only": True})
         self._parse_dynamic_components(components)
+        await self._retrieve_blutrv_components(components)
 
     def _supports_dynamic_components(self) -> bool:
         """Return True if device supports dynamic components."""
@@ -538,3 +545,32 @@
                 for item in self._dynamic_components
             }
         )
+
+    async def _retrieve_blutrv_components(self, components: dict[str, Any]) -> None:
+        """Retrieve BLU TRV components."""
+        if self.model != MODEL_BLU_GATEWAY_GEN3:
+            return
+
+        if not self._config or not self._status:
+            raise NotInitialized
+
+        for component in components.get("components", []):
+            _key = component["key"].split(":")
+
+            if _key[0] != BLU_TRV_IDENTIFIER:
+                continue
+
+            calls = [
+                ("BluTrv.GetRemoteConfig", {"id": _key[1]}),
+                ("BluTrv.GetRemoteStatus", {"id": _key[1]}),
+            ]
+            results = await self.call_rpc_multiple(calls)
+
+            cfg: dict[str, Any] = results[0]["config"]["trv:0"]
+            # addr, name and model_id must be added from Shelly.GetComponents call
+            _attrs = component.get("attrs", {})
+            cfg.update({"addr": component["config"]["addr"]})
+            cfg.update({"name": component["config"]["name"]})
+            cfg.update({"local_name": BLU_TRV_MODEL_ID.get(_attrs.get("model_id"))})
+            self._config.update({component["key"]: cfg})
+            self._status.update({component["key"]: results[1]["status"]["trv:0"]})