hOn/custom_components/hon/switch.py

import logging
from dataclasses import dataclass
from typing import Any

from homeassistant.components.switch import SwitchEntityDescription, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from pyhon import Hon
from pyhon.appliance import HonAppliance
from pyhon.parameter.range import HonParameterRange

from .const import DOMAIN
from .hon import HonCoordinator, HonEntity, unique_entities

_LOGGER = logging.getLogger(__name__)


@dataclass
class HonSwitchEntityDescriptionMixin:
    turn_on_key: str = ""
    turn_off_key: str = ""


@dataclass
class HonSwitchEntityDescription(
    HonSwitchEntityDescriptionMixin, SwitchEntityDescription
):
    pass


SWITCHES: dict[str, tuple[HonSwitchEntityDescription, ...]] = {
    "WM": (
        HonSwitchEntityDescription(
            key="active",
            name="Washing Machine",
            icon="mdi:washing-machine",
            turn_on_key="startProgram",
            turn_off_key="stopProgram",
            translation_key="washing_machine",
        ),
        HonSwitchEntityDescription(
            key="pause",
            name="Pause Washing Machine",
            icon="mdi:pause",
            turn_on_key="pauseProgram",
            turn_off_key="resumeProgram",
            translation_key="pause",
        ),
        HonSwitchEntityDescription(
            key="startProgram.delayStatus",
            name="Delay Status",
            icon="mdi:timer-check",
            entity_category=EntityCategory.CONFIG,
            translation_key="delay_time",
        ),
        HonSwitchEntityDescription(
            key="startProgram.haier_SoakPrewashSelection",
            name="Soak Prewash Selection",
            icon="mdi:tshirt-crew",
            entity_category=EntityCategory.CONFIG,
            translation_key="prewash",
        ),
        HonSwitchEntityDescription(
            key="startProgram.autoSoftenerStatus",
            name="Keep Fresh",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:refresh-circle",
            translation_key="keep_fresh",
        ),
        HonSwitchEntityDescription(
            key="startProgram.autoDetergentStatus",
            name="Auto Dose",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:cup",
            translation_key="auto_dose",
        ),
        HonSwitchEntityDescription(
            key="startProgram.acquaplus",
            name="Acqua Plus",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:water-plus",
            translation_key="acqua_plus",
        ),
        HonSwitchEntityDescription(
            key="extraRinse1",
            name="Extra Rinse 1",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:numeric-1-box-multiple-outline",
            translation_key="extra_rinse_1",
        ),
        HonSwitchEntityDescription(
            key="extraRinse2",
            name="Extra Rinse 2",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:numeric-2-box-multiple-outline",
            translation_key="extra_rinse_2",
        ),
        HonSwitchEntityDescription(
            key="extraRinse3",
            name="Extra Rinse 3",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:numeric-3-box-multiple-outline",
            translation_key="extra_rinse_3",
        ),
        HonSwitchEntityDescription(
            key="goodNight",
            name="Good Night",
            icon="mdi:weather-night",
            entity_category=EntityCategory.CONFIG,
            translation_key="good_night",
        ),
    ),
    "TD": (
        HonSwitchEntityDescription(
            key="active",
            name="Tumble Dryer",
            icon="mdi:tumble-dryer",
            turn_on_key="startProgram",
            turn_off_key="stopProgram",
            translation_key="tumble_dryer",
        ),
        HonSwitchEntityDescription(
            key="pause",
            name="Pause Tumble Dryer",
            icon="mdi:pause",
            turn_on_key="pauseProgram",
            turn_off_key="resumeProgram",
            translation_key="pause",
        ),
        HonSwitchEntityDescription(
            key="startProgram.sterilizationStatus",
            name="Sterilization",
            icon="mdi:clock-start",
            entity_category=EntityCategory.CONFIG,
        ),
        HonSwitchEntityDescription(
            key="startProgram.antiCreaseTime",
            name="Anti-Crease",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:timer",
            translation_key="anti_crease",
        ),
        HonSwitchEntityDescription(
            key="startProgram.anticrease",
            name="Anti-Crease",
            entity_category=EntityCategory.CONFIG,
            icon="mdi:timer",
            translation_key="anti_crease",
        ),
    ),
    "OV": (
        HonSwitchEntityDescription(
            key="active",
            name="Oven",
            icon="mdi:toaster-oven",
            turn_on_key="startProgram",
            turn_off_key="stopProgram",
            translation_key="oven",
        ),
        HonSwitchEntityDescription(
            key="startProgram.preheatStatus",
            name="Preheat",
            icon="mdi:thermometer-chevron-up",
            entity_category=EntityCategory.CONFIG,
            translation_key="preheat",
        ),
    ),
    "WD": (
        HonSwitchEntityDescription(
            key="active",
            name="Washer Dryer",
            icon="mdi:washing-machine",
            turn_on_key="startProgram",
            turn_off_key="stopProgram",
            translation_key="washer_dryer",
        ),
        HonSwitchEntityDescription(
            key="pause",
            name="Pause Washer Dryer",
            icon="mdi:pause",
            turn_on_key="pauseProgram",
            turn_off_key="resumeProgram",
            translation_key="pause",
        ),
    ),
    "DW": (
        HonSwitchEntityDescription(
            key="active",
            name="Dish Washer",
            icon="mdi:dishwasher",
            turn_on_key="startProgram",
            turn_off_key="stopProgram",
            translation_key="dish_washer",
        ),
        HonSwitchEntityDescription(
            key="startProgram.extraDry",
            name="Extra Dry",
            icon="mdi:hair-dryer",
            entity_category=EntityCategory.CONFIG,
            translation_key="extra_dry",
        ),
        HonSwitchEntityDescription(
            key="startProgram.halfLoad",
            name="Half Load",
            icon="mdi:fraction-one-half",
            entity_category=EntityCategory.CONFIG,
            translation_key="half_load",
        ),
        HonSwitchEntityDescription(
            key="startProgram.openDoor",
            name="Open Door",
            icon="mdi:door-open",
            entity_category=EntityCategory.CONFIG,
            translation_key="open_door",
        ),
        HonSwitchEntityDescription(
            key="startProgram.threeInOne",
            name="Three in One",
            icon="mdi:numeric-3-box-outline",
            entity_category=EntityCategory.CONFIG,
            translation_key="three_in_one",
        ),
        HonSwitchEntityDescription(
            key="startProgram.ecoExpress",
            name="Eco Express",
            icon="mdi:sprout",
            entity_category=EntityCategory.CONFIG,
            translation_key="eco",
        ),
        HonSwitchEntityDescription(
            key="startProgram.addDish",
            name="Add Dish",
            icon="mdi:silverware-fork-knife",
            entity_category=EntityCategory.CONFIG,
            translation_key="add_dish",
        ),
    ),
    "AC": (
        HonSwitchEntityDescription(
            key="settings.10degreeHeatingStatus",
            name="10° Heating",
            icon="mdi:heat-wave",
            entity_category=EntityCategory.CONFIG,
            translation_key="10_degree_heating",
        ),
        HonSwitchEntityDescription(
            key="settings.echoStatus",
            name="Echo",
            icon="mdi:account-voice",
            entity_category=EntityCategory.CONFIG,
        ),
        HonSwitchEntityDescription(
            key="settings.ecoMode",
            name="Eco Mode",
            entity_category=EntityCategory.CONFIG,
            translation_key="eco_mode",
        ),
        HonSwitchEntityDescription(
            key="settings.healthMode",
            name="Health Mode",
            icon="mdi:medication-outline",
            entity_category=EntityCategory.CONFIG,
        ),
        HonSwitchEntityDescription(
            key="settings.muteStatus",
            name="Mute",
            icon="mdi:volume-off",
            entity_category=EntityCategory.CONFIG,
            translation_key="mute_mode",
        ),
        HonSwitchEntityDescription(
            key="settings.rapidMode",
            name="Rapid Mode",
            icon="mdi:run-fast",
            entity_category=EntityCategory.CONFIG,
            translation_key="rapid_mode",
        ),
        HonSwitchEntityDescription(
            key="settings.screenDisplayStatus",
            name="Screen Display",
            icon="mdi:monitor-small",
            entity_category=EntityCategory.CONFIG,
        ),
        HonSwitchEntityDescription(
            key="settings.selfCleaning56Status",
            name="Self Cleaning 56",
            icon="mdi:air-filter",
            entity_category=EntityCategory.CONFIG,
            translation_key="self_clean_56",
        ),
        HonSwitchEntityDescription(
            key="settings.selfCleaningStatus",
            name="Self Cleaning",
            icon="mdi:air-filter",
            entity_category=EntityCategory.CONFIG,
            translation_key="self_clean",
        ),
        HonSwitchEntityDescription(
            key="settings.silentSleepStatus",
            name="Silent Sleep",
            icon="mdi:bed",
            entity_category=EntityCategory.CONFIG,
            translation_key="silent_mode",
        ),
    ),
}

SWITCHES["WD"] = unique_entities(SWITCHES["WD"], SWITCHES["WM"])
SWITCHES["WD"] = unique_entities(SWITCHES["WD"], SWITCHES["TD"])


async def async_setup_entry(hass, entry: ConfigEntry, async_add_entities) -> None:
    hon: Hon = hass.data[DOMAIN][entry.unique_id]
    coordinators = hass.data[DOMAIN]["coordinators"]
    appliances = []
    for device in hon.appliances:
        if device.unique_id in coordinators:
            coordinator = hass.data[DOMAIN]["coordinators"][device.unique_id]
        else:
            coordinator = HonCoordinator(hass, device)
            hass.data[DOMAIN]["coordinators"][device.unique_id] = coordinator
        await coordinator.async_config_entry_first_refresh()

        if descriptions := SWITCHES.get(device.appliance_type):
            for description in descriptions:
                if (
                    device.get(description.key) is not None
                    or description.key in device.available_settings
                    or description.turn_on_key in list(device.commands)
                    or description.turn_off_key in list(device.commands)
                ):
                    appliances.extend(
                        [HonSwitchEntity(hass, coordinator, entry, device, description)]
                    )

    async_add_entities(appliances)


class HonSwitchEntity(HonEntity, SwitchEntity):
    entity_description: HonSwitchEntityDescription

    def __init__(
        self,
        hass,
        coordinator,
        entry,
        device: HonAppliance,
        description: HonSwitchEntityDescription,
    ) -> None:
        super().__init__(hass, entry, coordinator, device)
        self._coordinator = coordinator
        self._device = device
        self.entity_description = description
        self._attr_unique_id = f"{super().unique_id}{description.key}"

    @property
    def is_on(self) -> bool | None:
        """Return True if entity is on."""
        if self.entity_category == EntityCategory.CONFIG:
            setting = self._device.settings[self.entity_description.key]
            return (
                setting.value == "1"
                or hasattr(setting, "min")
                and setting.value != setting.min
            )
        return self._device.get(self.entity_description.key, False)

    async def async_turn_on(self, **kwargs: Any) -> None:
        if self.entity_category == EntityCategory.CONFIG:
            setting = self._device.settings[self.entity_description.key]
            setting.value = (
                setting.max if isinstance(setting, HonParameterRange) else "1"
            )
            self.async_write_ha_state()
            if self._device.appliance_type in ["AC"]:
                self._device.commands["startProgram"].send()
            await self.coordinator.async_refresh()
        else:
            await self._device.commands[self.entity_description.turn_on_key].send()

    async def async_turn_off(self, **kwargs: Any) -> None:
        if self.entity_category == EntityCategory.CONFIG:
            setting = self._device.settings[self.entity_description.key]
            setting.value = (
                setting.min if isinstance(setting, HonParameterRange) else "0"
            )
            self.async_write_ha_state()
            if self._device.appliance_type in ["AC"]:
                self._device.commands["startProgram"].send()
            await self.coordinator.async_refresh()
        else:
            await self._device.commands[self.entity_description.turn_off_key].send()

    @property
    def available(self) -> bool:
        """Return True if entity is available."""
        if self.entity_category == EntityCategory.CONFIG:
            return super().available
        else:
            return super().available and self._device.get("remoteCtrValid") == "1"