Add haa vendor extensions (#59750)

Co-authored-by: J. Nick Koston <nick@koston.org>

homeassistant/components/homekit_controller/button.py

@@ -0,0 +1,92 @@
+"""
+Support for Homekit buttons.
+
+These are mostly used where a HomeKit accessory exposes additional non-standard
+characteristics that don't map to a Home Assistant feature.
+"""
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from aiohomekit.model.characteristics import Characteristic, CharacteristicsTypes
+
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.const import ENTITY_CATEGORY_CONFIG
+from homeassistant.core import callback
+
+from . import KNOWN_DEVICES, CharacteristicEntity
+
+
+@dataclass
+class HomeKitButtonEntityDescription(ButtonEntityDescription):
+    """Describes Homekit button."""
+
+    write_value: int | str | None = None
+
+
+BUTTON_ENTITIES: dict[str, HomeKitButtonEntityDescription] = {
+    CharacteristicsTypes.Vendor.HAA_SETUP: HomeKitButtonEntityDescription(
+        key=CharacteristicsTypes.Vendor.HAA_SETUP,
+        name="Setup",
+        icon="mdi:cog",
+        entity_category=ENTITY_CATEGORY_CONFIG,
+        write_value="#HAA@trcmd",
+    ),
+    CharacteristicsTypes.Vendor.HAA_UPDATE: HomeKitButtonEntityDescription(
+        key=CharacteristicsTypes.Vendor.HAA_UPDATE,
+        name="Update",
+        icon="mdi:update",
+        entity_category=ENTITY_CATEGORY_CONFIG,
+        write_value="#HAA@trcmd",
+    ),
+}
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    """Set up Homekit buttons."""
+    hkid = config_entry.data["AccessoryPairingID"]
+    conn = hass.data[KNOWN_DEVICES][hkid]
+
+    @callback
+    def async_add_characteristic(char: Characteristic):
+        if not (description := BUTTON_ENTITIES.get(char.type)):
+            return False
+        info = {"aid": char.service.accessory.aid, "iid": char.service.iid}
+        async_add_entities([HomeKitButton(conn, info, char, description)], True)
+        return True
+
+    conn.add_char_factory(async_add_characteristic)
+
+
+class HomeKitButton(CharacteristicEntity, ButtonEntity):
+    """Representation of a Button control on a homekit accessory."""
+
+    entity_description = HomeKitButtonEntityDescription
+
+    def __init__(
+        self,
+        conn,
+        info,
+        char,
+        description: HomeKitButtonEntityDescription,
+    ):
+        """Initialise a HomeKit button control."""
+        self.entity_description = description
+        super().__init__(conn, info, char)
+
+    def get_characteristic_types(self):
+        """Define the homekit characteristics the entity is tracking."""
+        return [self._char.type]
+
+    @property
+    def name(self) -> str:
+        """Return the name of the device if any."""
+        if name := super().name:
+            return f"{name} - {self.entity_description.name}"
+        return f"{self.entity_description.name}"
+
+    async def async_press(self) -> None:
+        """Press the button."""
+        key = self.entity_description.key
+        val = self.entity_description.write_value
+        return await self.async_put_characteristics({key: val})

homeassistant/components/homekit_controller/const.py

@@ -51,6 +51,8 @@ CHARACTERISTIC_PLATFORMS = {
     CharacteristicsTypes.Vendor.EVE_ENERGY_WATT: "sensor",
     CharacteristicsTypes.Vendor.EVE_DEGREE_AIR_PRESSURE: "sensor",
     CharacteristicsTypes.Vendor.EVE_DEGREE_ELEVATION: "number",
+    CharacteristicsTypes.Vendor.HAA_SETUP: "button",
+    CharacteristicsTypes.Vendor.HAA_UPDATE: "button",
     CharacteristicsTypes.Vendor.KOOGEEK_REALTIME_ENERGY: "sensor",
     CharacteristicsTypes.Vendor.KOOGEEK_REALTIME_ENERGY_2: "sensor",
     CharacteristicsTypes.Vendor.VOCOLINC_HUMIDIFIER_SPRAY_LEVEL: "number",

homeassistant/components/homekit_controller/manifest.json

@@ -3,7 +3,7 @@
   "name": "HomeKit Controller",
   "config_flow": true,
   "documentation": "https://www.home-assistant.io/integrations/homekit_controller",
-  "requirements": ["aiohomekit==0.6.3"],
+  "requirements": ["aiohomekit==0.6.4"],
   "zeroconf": ["_hap._tcp.local."],
   "after_dependencies": ["zeroconf"],
   "codeowners": ["@Jc2k", "@bdraco"],