Add bare bones homekit module for iot devices (#1566)

Based on the existing smart HomeKit module, this has been tested with a real device that supports this module.


---------

Co-authored-by: Teemu Rytilahti <tpr@iki.fi>

kasa/iot/modules/__init__.py

@@ -6,6 +6,7 @@ from .cloud import Cloud
 from .countdown import Countdown
 from .dimmer import Dimmer
 from .emeter import Emeter
+from .homekit import HomeKit
 from .led import Led
 from .light import Light
 from .lighteffect import LightEffect
@@ -34,4 +35,5 @@ __all__ = [
     "Schedule",
     "Time",
     "Usage",
+    "HomeKit",
 ]

kasa/iot/modules/homekit.py

@@ -0,0 +1,53 @@
+"""Implementation of HomeKit module for IOT devices that natively support HomeKit."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from ...feature import Feature
+from ..iotmodule import IotModule
+
+
+class HomeKit(IotModule):
+    """Implementation of HomeKit module for IOT devices."""
+
+    def query(self) -> dict:
+        """Request HomeKit setup info."""
+        return {"smartlife.iot.homekit": {"setup_info_get": {}}}
+
+    @property
+    def info(self) -> dict[str, Any]:
+        """Return the HomeKit setup info."""
+        # Only return info if the module has data
+        if self._module not in self._device._last_update:
+            return {}
+        return self.data.get("setup_info_get", {})
+
+    @property
+    def setup_code(self) -> str:
+        """Return the HomeKit setup code."""
+        return self.info["setup_code"]
+
+    @property
+    def setup_payload(self) -> str:
+        """Return the HomeKit setup payload."""
+        return self.info["setup_payload"]
+
+    def _initialize_features(self) -> None:
+        """Initialize features after the initial update."""
+        # Only add features if the device supports the module
+        data = self._device._last_update.get(self._module, {})
+        if not data or "setup_info_get" not in data:
+            return
+
+        self._add_feature(
+            Feature(
+                self._device,
+                container=self,
+                id="homekit_setup_code",
+                name="HomeKit setup code",
+                attribute_getter="setup_code",
+                type=Feature.Type.Sensor,
+                category=Feature.Category.Debug,
+            )
+        )