import logging
from datetime import timedelta
from typing import Callable, Optional

from homeassistant import config_entries, core
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import CONF_EMAIL
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, HomeAssistantType
from ute_wrapper.ute import UTEClient

from .config_flow import CONF_PHONE_NUMBER, schema
from .const import DOMAIN

_LOGGER = logging.getLogger(__name__)
# Time between updating data from UTE
SCAN_INTERVAL = timedelta(minutes=2)

PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(schema)


async def async_setup_entry(
    hass: core.HomeAssistant,
    config_entry: config_entries.ConfigEntry,
    async_add_entities,
):
    """Setup sensors from a config entry created in the integrations UI."""
    config = hass.data[DOMAIN][config_entry.entry_id]
    ute = UTEClient(
        config[CONF_EMAIL],
        config[CONF_PHONE_NUMBER],
    )
    sensor = UTESensor(ute)
    async_add_entities(sensor, update_before_add=True)


def setup_platform(
    hass: HomeAssistantType,
    config: ConfigType,
    async_add_entities: Callable,
    discovery_info: Optional[DiscoveryInfoType] = None,
) -> None:
    """Set up the sensor platform."""
    ute = UTEClient(
        config[CONF_EMAIL],
        config[CONF_PHONE_NUMBER],
    )
    sensor = UTESensor(ute)
    async_add_entities(sensor, update_before_add=True)


class UTESensor(Entity):
    """Representation of a UTE sensor."""

    def __init__(self, ute: UTEClient):
        super().__init__()
        self.ute = ute
        self._state = None
        self._available = True
        self._name = "Current energy usage"

    async def async_update(self):
        ute_data = await self.ute.get_current_usage_info()
        self._state = ute_data["data"]["power_in_watts"]