Add homekit_controller support for ecobee vendor extensions (#60914)

Co-authored-by: josephnad <>

homeassistant/components/homekit_controller/button.py

@@ -53,6 +53,7 @@ BUTTON_ENTITIES: dict[str, HomeKitButtonEntityDescription] = {
     ),
 }
 
+
 # For legacy reasons, "built-in" characteristic types are in their short form
 # And vendor types don't have a short form
 # This means long and short forms get mixed up in this dict, and comparisons
@@ -74,10 +75,17 @@ async def async_setup_entry(
 
     @callback
     def async_add_characteristic(char: Characteristic):
-        if not (description := BUTTON_ENTITIES.get(char.type)):
-            return False
+        entities = []
         info = {"aid": char.service.accessory.aid, "iid": char.service.iid}
-        async_add_entities([HomeKitButton(conn, info, char, description)], True)
+
+        if description := BUTTON_ENTITIES.get(char.type):
+            entities.append(HomeKitButton(conn, info, char, description))
+        elif entity_type := BUTTON_ENTITY_CLASSES.get(char.type):
+            entities.append(entity_type(conn, info, char))
+        else:
+            return False
+
+        async_add_entities(entities, True)
         return True
 
     conn.add_char_factory(async_add_characteristic)
@@ -115,3 +123,37 @@ class HomeKitButton(CharacteristicEntity, ButtonEntity):
         key = self.entity_description.key
         val = self.entity_description.write_value
         return await self.async_put_characteristics({key: val})
+
+
+class HomeKitEcobeeClearHoldButton(CharacteristicEntity, ButtonEntity):
+    """Representation of a Button control for Ecobee clear hold request."""
+
+    def get_characteristic_types(self):
+        """Define the homekit characteristics the entity is tracking."""
+        return []
+
+    @property
+    def name(self) -> str:
+        """Return the name of the device if any."""
+        prefix = ""
+        if name := super().name:
+            prefix = name
+        return f"{prefix} Clear Hold"
+
+    async def async_press(self) -> None:
+        """Press the button."""
+        key = self._char.type
+
+        # If we just send true, the request doesn't always get executed by ecobee.
+        # Sending false value then true value will ensure that the hold gets cleared
+        # and schedule resumed.
+        # Ecobee seems to cache the state and not update it correctly, which
+        # causes the request to be ignored if it thinks it has no effect.
+
+        for val in (False, True):
+            await self.async_put_characteristics({key: val})
+
+
+BUTTON_ENTITY_CLASSES: dict[str, type] = {
+    CharacteristicsTypes.Vendor.ECOBEE_CLEAR_HOLD: HomeKitEcobeeClearHoldButton,
+}

homeassistant/components/homekit_controller/const.py

@@ -74,6 +74,9 @@ CHARACTERISTIC_PLATFORMS = {
     CharacteristicsTypes.Vendor.KOOGEEK_REALTIME_ENERGY_2: "sensor",
     CharacteristicsTypes.Vendor.VOCOLINC_HUMIDIFIER_SPRAY_LEVEL: "number",
     CharacteristicsTypes.Vendor.VOCOLINC_OUTLET_ENERGY: "sensor",
+    CharacteristicsTypes.Vendor.ECOBEE_CLEAR_HOLD: "button",
+    CharacteristicsTypes.Vendor.ECOBEE_FAN_WRITE_SPEED: "number",
+    CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE: "number",
     CharacteristicsTypes.TEMPERATURE_CURRENT: "sensor",
     CharacteristicsTypes.RELATIVE_HUMIDITY_CURRENT: "sensor",
     CharacteristicsTypes.AIR_QUALITY: "sensor",

homeassistant/components/homekit_controller/number.py

@@ -91,10 +91,17 @@ async def async_setup_entry(
 
     @callback
     def async_add_characteristic(char: Characteristic):
-        if not (description := NUMBER_ENTITIES.get(char.type)):
-            return False
+        entities = []
         info = {"aid": char.service.accessory.aid, "iid": char.service.iid}
-        async_add_entities([HomeKitNumber(conn, info, char, description)], True)
+
+        if description := NUMBER_ENTITIES.get(char.type):
+            entities.append(HomeKitNumber(conn, info, char, description))
+        elif entity_type := NUMBER_ENTITY_CLASSES.get(char.type):
+            entities.append(entity_type(conn, info, char))
+        else:
+            return False
+
+        async_add_entities(entities, True)
         return True
 
     conn.add_char_factory(async_add_characteristic)
@@ -152,3 +159,72 @@ class HomeKitNumber(CharacteristicEntity, NumberEntity):
                 self._char.type: value,
             }
         )
+
+
+class HomeKitEcobeeFanModeNumber(CharacteristicEntity, NumberEntity):
+    """Representation of a Number control for Ecobee Fan Mode request."""
+
+    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."""
+        prefix = ""
+        if name := super().name:
+            prefix = name
+        return f"{prefix} Fan Mode"
+
+    @property
+    def min_value(self) -> float:
+        """Return the minimum value."""
+        return self._char.minValue
+
+    @property
+    def max_value(self) -> float:
+        """Return the maximum value."""
+        return self._char.maxValue
+
+    @property
+    def step(self) -> float:
+        """Return the increment/decrement step."""
+        return self._char.minStep
+
+    @property
+    def value(self) -> float:
+        """Return the current characteristic value."""
+        return self._char.value
+
+    async def async_set_value(self, value: float):
+        """Set the characteristic to this value."""
+
+        # Sending the fan mode request sometimes ends up getting ignored by ecobee
+        # and this might be because it the older value instead of newer, and ecobee
+        # thinks there is nothing to do.
+        # So in order to make sure that the request is executed by ecobee, we need
+        # to send a different value before sending the target value.
+        # Fan mode value is a value from 0 to 100. We send a value off by 1 first.
+
+        if value > self.min_value:
+            other_value = value - 1
+        else:
+            other_value = self.min_value + 1
+
+        if value != other_value:
+            await self.async_put_characteristics(
+                {
+                    self._char.type: other_value,
+                }
+            )
+
+        await self.async_put_characteristics(
+            {
+                self._char.type: value,
+            }
+        )
+
+
+NUMBER_ENTITY_CLASSES: dict[str, type] = {
+    CharacteristicsTypes.Vendor.ECOBEE_FAN_WRITE_SPEED: HomeKitEcobeeFanModeNumber,
+}