Add support for LIFX 26"x13" Ceiling (home-assistant#148459)

Signed-off-by: Avi Miller <[email protected]>

Author: Djelibeybi

homeassistant/components/lifx/const.py

@@ -70,6 +70,7 @@
 }
 
 LIFX_CEILING_PRODUCT_IDS = {176, 177, 201, 202}
+LIFX_128ZONE_CEILING_PRODUCT_IDS = {201, 202}
 
 _LOGGER = logging.getLogger(__package__)
 

homeassistant/components/lifx/coordinator.py

@@ -41,6 +41,7 @@
     DEFAULT_ATTEMPTS,
     DOMAIN,
     IDENTIFY_WAVEFORM,
+    LIFX_128ZONE_CEILING_PRODUCT_IDS,
     MAX_ATTEMPTS_PER_UPDATE_REQUEST_MESSAGE,
     MAX_UPDATE_TIME,
     MESSAGE_RETRIES,
@@ -183,6 +184,11 @@ def is_matrix(self) -> bool:
         """Return true if this is a matrix device."""
         return bool(lifx_features(self.device)["matrix"])
 
+    @cached_property
+    def is_128zone_matrix(self) -> bool:
+        """Return true if this is a 128-zone matrix device."""
+        return bool(self.device.product in LIFX_128ZONE_CEILING_PRODUCT_IDS)
+
     async def diagnostics(self) -> dict[str, Any]:
         """Return diagnostic information about the device."""
         features = lifx_features(self.device)
@@ -216,6 +222,16 @@ async def diagnostics(self) -> dict[str, Any]:
                 "last_result": self.device.last_hev_cycle_result,
             }
 
+        if features["matrix"] is True:
+            device_data["matrix"] = {
+                "effect": self.device.effect,
+                "chain": self.device.chain,
+                "chain_length": self.device.chain_length,
+                "tile_devices": self.device.tile_devices,
+                "tile_devices_count": self.device.tile_devices_count,
+                "tile_device_width": self.device.tile_device_width,
+            }
+
         if features["infrared"] is True:
             device_data["infrared"] = {"brightness": self.device.infrared_brightness}
 
@@ -291,6 +307,37 @@ def _wrapped_callback(
 
         return calls
 
+    @callback
+    def _async_build_get64_update_requests(self) -> list[Callable]:
+        """Build one or more get64 update requests."""
+        if self.device.tile_device_width == 0:
+            return []
+
+        calls: list[Callable] = []
+        calls.append(
+            partial(
+                self.device.get64,
+                tile_index=0,
+                length=1,
+                x=0,
+                y=0,
+                width=self.device.tile_device_width,
+            )
+        )
+        if self.is_128zone_matrix:
+            # For 128-zone ceiling devices, we need another get64 request for the next set of zones
+            calls.append(
+                partial(
+                    self.device.get64,
+                    tile_index=0,
+                    length=1,
+                    x=0,
+                    y=4,
+                    width=self.device.tile_device_width,
+                )
+            )
+        return calls
+
     async def _async_update_data(self) -> None:
         """Fetch all device data from the api."""
         device = self.device
@@ -312,9 +359,9 @@ async def _async_update_data(self) -> None:
                 [
                     self.device.get_tile_effect,
                     self.device.get_device_chain,
-                    self.device.get64,
                 ]
             )
+            methods.extend(self._async_build_get64_update_requests())
         if self.is_extended_multizone:
             methods.append(self.device.get_extended_color_zones)
         elif self.is_legacy_multizone:
@@ -339,6 +386,7 @@ async def _async_update_data(self) -> None:
 
         if self.is_matrix or self.is_extended_multizone or self.is_legacy_multizone:
             self.active_effect = FirmwareEffect[self.device.effect.get("effect", "OFF")]
+
         if self.is_legacy_multizone and num_zones != self.get_number_of_zones():
             # The number of zones has changed so we need
             # to update the zones again. This happens rarely.

tests/components/lifx/__init__.py

@@ -199,6 +199,17 @@ def _mocked_ceiling() -> Light:
     return bulb
 
 
+def _mocked_128zone_ceiling() -> Light:
+    bulb = _mocked_bulb()
+    bulb.product = 201  # LIFX 26"x13" Ceiling
+    bulb.effect = {"effect": "OFF"}
+    bulb.get_tile_effect = MockLifxCommand(bulb)
+    bulb.set_tile_effect = MockLifxCommand(bulb)
+    bulb.get64 = MockLifxCommand(bulb)
+    bulb.get_device_chain = MockLifxCommand(bulb)
+    return bulb
+
+
 def _mocked_bulb_old_firmware() -> Light:
     bulb = _mocked_bulb()
     bulb.host_firmware_version = "2.77"