import logging

from homeassistant.helpers.entity import Entity, async_generate_entity_id
from homeassistant.helpers.update_coordinator import CoordinatorEntity

from .const import (
    DOMAIN,
    DATA_DEVICES,
    DATA_ALIASES,
    DATA_ADDERS,
    CONFIG_DEVICES,
    DATA_CONFIG,
    CONFIG_PREFIX,
    CONFIG_DISABLE,
    CONFIG_DISABLE_ALL,
    DATA_SETUP_COMPLETE,
)

from .coordinator import Coordinator

_LOGGER = logging.getLogger(__name__)


def get_devices(hass):
    return hass.data[DOMAIN][DATA_DEVICES]


def get_alias(hass, deviceID):
    for k, v in hass.data[DOMAIN][DATA_ALIASES].items():
        if v == deviceID:
            return k
    return None


def get_config(hass, deviceID):
    config = hass.data[DOMAIN][DATA_CONFIG].get(CONFIG_DEVICES, {})
    return config.get(deviceID, config.get(deviceID.replace("-", "_"), {}))


def create_entity(hass, platform, deviceID, connection):
    conf = get_config(hass, deviceID)
    if conf and (
        platform in conf.get(CONFIG_DISABLE, [])
        or CONFIG_DISABLE_ALL in conf.get(CONFIG_DISABLE, [])
    ):
        return None
    if not conf and (
        platform in hass.data[DOMAIN][DATA_CONFIG].get(CONFIG_DISABLE, [])
        or CONFIG_DISABLE_ALL in hass.data[DOMAIN][DATA_CONFIG].get(CONFIG_DISABLE, [])
    ):
        return None
    adder = hass.data[DOMAIN][DATA_ADDERS].get(platform)
    if not adder:
        return None
    entity = adder(hass, deviceID, connection, get_alias(hass, deviceID))
    return entity


def setup_platform(hass, config, async_add_devices, platform, cls):
    if platform in hass.data[DOMAIN][DATA_ADDERS]:
        return True

    def adder(hass, deviceID, connection, alias=None):
        entity = cls(hass, connection, deviceID, alias)
        async_add_devices([entity])
        return entity

    hass.data[DOMAIN][DATA_ADDERS][platform] = adder
    return True


def is_setup_complete(hass):
    return hass.data[DOMAIN][DATA_SETUP_COMPLETE]


class BrowserModEntity(CoordinatorEntity):
    def __init__(self, coordinator, deviceID, name):
        super().__init__(coordinator)
        self.deviceID = deviceID
        self._name = name

    @property
    def _data(self):
        return self.coordinator.data or {}

    @property
    def device_info(self):
        return {
            "identifiers": {(DOMAIN, self.deviceID)},
            "name": self.deviceID,
            "manufacturer": "Browser Mod",
        }

    @property
    def extra_state_attributes(self):
        return {
            "type": "browser_mod",
            "deviceID": self.deviceID,
        }

    @property
    def name(self):
        return self._name

    @property
    def has_entity_name(self):
        return True

    @property
    def entity_registry_visible_default(self):
        return False

    @property
    def unique_id(self):
        return f"{self.deviceID}-{self._name.replace(' ','_')}"