From 5757197fb6a4d38e4fd2db3b62361c80100923bd Mon Sep 17 00:00:00 2001
From: Klaas Neirinck <kneirinck@users.noreply.github.com>
Date: Wed, 26 Oct 2022 12:44:23 +0200
Subject: [PATCH] Add auto preset to Comfoconnect fan (#80697)

---
 homeassistant/components/comfoconnect/fan.py | 50 ++++++++++++++++++--
 1 file changed, 47 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/comfoconnect/fan.py b/homeassistant/components/comfoconnect/fan.py
index dd2c9632d7cd..7577dd77aeb0 100644
--- a/homeassistant/components/comfoconnect/fan.py
+++ b/homeassistant/components/comfoconnect/fan.py
@@ -10,7 +10,10 @@ from pycomfoconnect import (
     CMD_FAN_MODE_HIGH,
     CMD_FAN_MODE_LOW,
     CMD_FAN_MODE_MEDIUM,
+    CMD_MODE_AUTO,
+    CMD_MODE_MANUAL,
     SENSOR_FAN_SPEED_MODE,
+    SENSOR_OPERATING_MODE_BIS,
 )
 
 from homeassistant.components.fan import FanEntity, FanEntityFeature
@@ -37,6 +40,9 @@ CMD_MAPPING = {
 
 SPEED_RANGE = (1, 3)  # away is not included in speeds and instead mapped to off
 
+PRESET_MODE_AUTO = "auto"
+PRESET_MODES = [PRESET_MODE_AUTO]
+
 
 def setup_platform(
     hass: HomeAssistant,
@@ -55,7 +61,8 @@ class ComfoConnectFan(FanEntity):
 
     _attr_icon = "mdi:air-conditioner"
     _attr_should_poll = False
-    _attr_supported_features = FanEntityFeature.SET_SPEED
+    _attr_supported_features = FanEntityFeature.SET_SPEED | FanEntityFeature.PRESET_MODE
+    _attr_preset_modes = PRESET_MODES
     current_speed = None
 
     def __init__(self, ccb: ComfoConnectBridge) -> None:
@@ -63,6 +70,7 @@ class ComfoConnectFan(FanEntity):
         self._ccb = ccb
         self._attr_name = ccb.name
         self._attr_unique_id = ccb.unique_id
+        self._attr_preset_mode = None
 
     async def async_added_to_hass(self) -> None:
         """Register for sensor updates."""
@@ -71,14 +79,24 @@ class ComfoConnectFan(FanEntity):
             async_dispatcher_connect(
                 self.hass,
                 SIGNAL_COMFOCONNECT_UPDATE_RECEIVED.format(SENSOR_FAN_SPEED_MODE),
-                self._handle_update,
+                self._handle_speed_update,
+            )
+        )
+        self.async_on_remove(
+            async_dispatcher_connect(
+                self.hass,
+                SIGNAL_COMFOCONNECT_UPDATE_RECEIVED.format(SENSOR_OPERATING_MODE_BIS),
+                self._handle_mode_update,
             )
         )
         await self.hass.async_add_executor_job(
             self._ccb.comfoconnect.register_sensor, SENSOR_FAN_SPEED_MODE
         )
+        await self.hass.async_add_executor_job(
+            self._ccb.comfoconnect.register_sensor, SENSOR_OPERATING_MODE_BIS
+        )
 
-    def _handle_update(self, value: float) -> None:
+    def _handle_speed_update(self, value: float) -> None:
         """Handle update callbacks."""
         _LOGGER.debug(
             "Handle update for fan speed (%d): %s", SENSOR_FAN_SPEED_MODE, value
@@ -86,6 +104,16 @@ class ComfoConnectFan(FanEntity):
         self.current_speed = value
         self.schedule_update_ha_state()
 
+    def _handle_mode_update(self, value: int) -> None:
+        """Handle update callbacks."""
+        _LOGGER.debug(
+            "Handle update for operating mode (%d): %s",
+            SENSOR_OPERATING_MODE_BIS,
+            value,
+        )
+        self._attr_preset_mode = PRESET_MODE_AUTO if value == -1 else None
+        self.schedule_update_ha_state()
+
     @property
     def percentage(self) -> int | None:
         """Return the current speed percentage."""
@@ -105,6 +133,10 @@ class ComfoConnectFan(FanEntity):
         **kwargs: Any,
     ) -> None:
         """Turn on the fan."""
+        if preset_mode:
+            self.set_preset_mode(preset_mode)
+            return
+
         if percentage is None:
             self.set_percentage(1)  # Set fan speed to low
         else:
@@ -125,3 +157,15 @@ class ComfoConnectFan(FanEntity):
             cmd = CMD_MAPPING[speed]
 
         self._ccb.comfoconnect.cmd_rmi_request(cmd)
+
+    def set_preset_mode(self, preset_mode: str) -> None:
+        """Set new preset mode."""
+        if self.preset_modes and preset_mode in self.preset_modes:
+            _LOGGER.debug("Changing preset mode to %s", preset_mode)
+            if preset_mode == PRESET_MODE_AUTO:
+                # force set it to manual first
+                self._ccb.comfoconnect.cmd_rmi_request(CMD_MODE_MANUAL)
+                # now set it to auto so any previous percentage set gets undone
+                self._ccb.comfoconnect.cmd_rmi_request(CMD_MODE_AUTO)
+        else:
+            raise ValueError(f"Invalid preset mode: {preset_mode}")