from .midea_entity import MideaEntity
from .midea_devices import MIDEA_DEVICES
from homeassistant.components.number import NumberEntity
from homeassistant.const import (
    Platform,
    CONF_DEVICE_ID,
    CONF_SWITCHES
)
from .const import (
    DOMAIN,
    DEVICES,
)


async def async_setup_entry(hass, config_entry, async_add_entities):
    device_id = config_entry.data.get(CONF_DEVICE_ID)
    device = hass.data[DOMAIN][DEVICES].get(device_id)
    extra_switches = config_entry.options.get(
        CONF_SWITCHES, []
    )
    numbers = []
    for entity_key, config in MIDEA_DEVICES[device.device_type]["entities"].items():
        if config["type"] == Platform.NUMBER and entity_key in extra_switches:
            dev = MideaNumber(device, entity_key)
            numbers.append(dev)
    async_add_entities(numbers)


class MideaNumber(MideaEntity, NumberEntity):
    def __init__(self, device, entity_key: str):
        super().__init__(device, entity_key)
        self._max_value = self._config.get("max")
        self._min_value = self._config.get("min")
        self._step_value = self._config.get("step")

    @property
    def native_min_value(self):
        return self._min_value if (type(self._min_value) is int) else \
            self._device.get_attribute(attr=self._min_value) \
            if self._device.get_attribute(attr=self._min_value) else \
            getattr(self._device, self._min_value)

    @property
    def native_max_value(self):
        return self._max_value if (type(self._max_value) is int) else \
            self._device.get_attribute(attr=self._max_value) \
            if self._device.get_attribute(attr=self._max_value) else \
            getattr(self._device, self._max_value)

    @property
    def native_step(self):
        return self._step_value if (type(self._step_value) is int) else \
            self._device.get_attribute(attr=self._step_value) \
            if self._device.get_attribute(attr=self._step_value) else \
            getattr(self._device, self._step_value)

    @property
    def native_value(self):
        return self._device.get_attribute(self._entity_key)

    def set_native_value(self, value):
        self._device.set_attribute(self._entity_key, value)