feat: add power-reset turn_off for E-heater assist and fix sterilize profile mapping

Author: tazomatalax

custom_components/midea_heatpump_hws/profile_manager.py

@@ -235,6 +235,7 @@ def apply_profile_to_config(self, profile_data: dict[str, Any], user_input: dict
         config["mode_register"] = registers.get("mode", 1)
         config["temp_register"] = registers.get("current_temp", 102)
         config["target_temp_register"] = registers.get("target_temp", 2)
+        config["sterilize_register"] = registers.get("sterilize", 3)
         config["tank_top_temp_register"] = registers.get("tank_top_temp", 101)
         config["tank_bottom_temp_register"] = registers.get("tank_bottom_temp", 102)
         config["condensor_temp_register"] = registers.get("condensor_temp", 103)
@@ -246,6 +247,8 @@ def apply_profile_to_config(self, profile_data: dict[str, Any], user_input: dict
             config["heater_assist_register"] = registers["heater_assist_register"]
         if "sanitize_state_register" in registers:
             config["sanitize_state_register"] = registers["sanitize_state_register"]
+        if "heater_assist_trigger_register" in registers:
+            config["heater_assist_trigger_register"] = registers["heater_assist_trigger_register"]
 
         # Apply mode values
         mode_values = profile_data.get("mode_values", {})

custom_components/midea_heatpump_hws/switch.py

@@ -8,7 +8,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .const import DOMAIN, CONF_MODBUS_UNIT
+from .const import DOMAIN, CONF_MODBUS_UNIT, CONF_HEATER_ASSIST_TRIGGER_REGISTER
 from .coordinator import MideaModbusCoordinator
 
 _LOGGER = logging.getLogger(__name__)
@@ -30,9 +30,17 @@ async def async_setup_entry(
     entities = [MideaPowerSwitch(coordinator, config, host_suffix)]
 
     # Add sterilize switch if register is configured
-    if config.get("sterilize_register") is not None:
+    if coordinator.sterilize_register is not None:
         entities.append(MideaSterilizeSwitch(coordinator, config, host_suffix))
 
+    # Add manual heater assist switch only if the profile exposes both the write
+    # trigger and the R108 diagnostic register it is functionally paired with.
+    if (
+        coordinator.heater_assist_trigger_register is not None
+        and coordinator.heater_assist_register is not None
+    ):
+        entities.append(MideaHeaterAssistSwitch(coordinator, config, host_suffix))
+
     async_add_entities(entities)
 
 
@@ -141,4 +149,55 @@ async def async_turn_off(self, **kwargs: Any) -> None:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        self.async_write_ha_state()
+        self.async_write_ha_state()
+
+
+class MideaHeaterAssistSwitch(CoordinatorEntity, SwitchEntity):
+    """Representation of the manual E-Heater assist trigger switch."""
+
+    def __init__(
+        self,
+        coordinator: MideaModbusCoordinator,
+        config: dict,
+        host_suffix: str
+    ):
+        """Initialize the heater assist switch."""
+        super().__init__(coordinator)
+        self._config = config
+
+        # Entity attributes - include host for uniqueness
+        self._attr_name = f"Manual Heater Assist{host_suffix}"
+        self._attr_unique_id = f"midea_{config['host']}_{config[CONF_MODBUS_UNIT]}_heater_assist_trigger"
+        self._attr_icon = "mdi:fire"
+
+    @property
+    def device_info(self):
+        """Return device info to link this switch to the main device."""
+        return {
+            "identifiers": {(DOMAIN, f"{self._config['host']}_{self._config[CONF_MODBUS_UNIT]}")},
+            "name": f"Midea Heat Pump ({self._config['host']})",
+            "manufacturer": "Midea",
+            "model": "Heat Pump Water Heater",
+        }
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if e-heater is triggered."""
+        if self.coordinator.data:
+            return self.coordinator.data.get("heater_assist_trigger_raw") == 1
+        return False
+
+    @property
+    def available(self) -> bool:
+        """Return if entity is available."""
+        return self.coordinator.last_update_success
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Trigger manual heater assist."""
+        await self.coordinator.write_register("heater_assist_trigger", 1)
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off E-heater assist by resetting power state to clear R4 trigger."""
+        _LOGGER.info("Resetting power state to turn off manual E-heater assist")
+        await self.coordinator.write_register("power_state", False)
+        await self.coordinator.write_register("power_state", True)