import logging
from typing import List

from homeassistant.helpers.entity import Entity
from gehomesdk import (
    ErdCode, 
    ErdApplianceType,
    ErdOnOff
)

from custom_components.ge_home.entities.water_heater.ge_water_heater import GeWaterHeater

from .base import ApplianceApi
from ..entities import (
    GeErdSensor, 
    GeErdBinarySensor,
    GeErdSelect,
    GeErdSwitch, 
    ErdOnOffBoolConverter
)

_LOGGER = logging.getLogger(__name__)


class WaterHeaterApi(ApplianceApi):
    """API class for Water Heater objects"""
    APPLIANCE_TYPE = ErdApplianceType.WATER_HEATER

    def get_all_entities(self) -> List[Entity]:
        base_entities = super().get_all_entities()

        boost_mode: ErdOnOff = self.try_get_erd_value(ErdCode.WH_HEATER_BOOST_STATE)
        active: ErdOnOff = self.try_get_erd_value(ErdCode.WH_HEATER_ACTIVE_STATE)

        wh_entities = [
            GeErdSensor(self, ErdCode.WH_HEATER_TARGET_TEMPERATURE),
            GeErdSensor(self, ErdCode.WH_HEATER_TEMPERATURE),
            GeErdSensor(self, ErdCode.WH_HEATER_MODE_HOURS_REMAINING),
            GeErdSensor(self, ErdCode.WH_HEATER_ELECTRIC_MODE_MAX_TIME),
            GeErdSensor(self, ErdCode.WH_HEATER_VACATION_MODE_MAX_TIME),
            GeWaterHeater(self)
        ]

        if(boost_mode and boost_mode != ErdOnOff.NA):
            wh_entities.append(GeErdSensor(self, ErdCode.WH_HEATER_BOOST_STATE))
            wh_entities.append(GeErdSwitch(self, ErdCode.WH_HEATER_BOOST_CONTROL, ErdOnOffBoolConverter(), icon_on_override="mdi:rocket-launch", icon_off_override="mdi:rocket-launch-outline"))

        if(active and active != ErdOnOff.NA):
            wh_entities.append(GeErdSensor(self, ErdCode.WH_HEATER_ACTIVE_STATE))
            wh_entities.append(GeErdSwitch(self, ErdCode.WH_HEATER_ACTIVE_CONTROL, ErdOnOffBoolConverter(), icon_on_override="mdi:power", icon_off_override="mdi:power-standby"))

        entities = base_entities + wh_entities
        return entities