#!/usr/bin/env python3
from datetime import datetime
import hashlib

from homeassistant.helpers.entity import Entity
from homeassistant.util import Throttle

from .const.const import (
    _LOGGER,
    ATTR_LAST_UPDATE,
    ATTR_YEAR_MONTH_DAY_DATE,
    CONF_DEFAULT_LABEL,
    CONF_ID,
    CONF_POSTAL_CODE,
    CONF_STREET_NUMBER,
    CONF_SUFFIX,
    MIN_TIME_BETWEEN_UPDATES,
    PARALLEL_UPDATES,
    SENSOR_ICON,
    SENSOR_PREFIX,
)


class CustomSensor(Entity):
    def __init__(self, hass, waste_type, fetch_data, config):
        self.hass = hass
        self.waste_type = waste_type
        self.fetch_data = fetch_data
        self.config = config
        self._id_name = self.config.get(CONF_ID)
        self._default_label = self.config.get(CONF_DEFAULT_LABEL)
        self._last_update = None
        self._name = (
            SENSOR_PREFIX + (f"{self._id_name} " if len(self._id_name) > 0 else "")
        ) + self.waste_type

        self._state = self.config.get(CONF_DEFAULT_LABEL)
        self._icon = SENSOR_ICON
        self._year_month_day_date = None
        self._unique_id = hashlib.sha1(
            f"{self.waste_type}{self.config.get(CONF_ID)}{self.config.get(CONF_POSTAL_CODE)}{self.config.get(CONF_STREET_NUMBER)}{self.config.get(CONF_SUFFIX,'')}".encode(
                "utf-8"
            )
        ).hexdigest()

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

    @property
    def unique_id(self):
        return self._unique_id

    @property
    def icon(self):
        return self._icon

    @property
    def state(self):
        return self._state

    @property
    def extra_state_attributes(self):
        if self._year_month_day_date is not None:
            return {
                ATTR_LAST_UPDATE: self._last_update,
                ATTR_YEAR_MONTH_DAY_DATE: self._year_month_day_date,
            }
        else:
            return {
                ATTR_LAST_UPDATE: self._last_update,
            }

    @Throttle(MIN_TIME_BETWEEN_UPDATES)
    async def async_update(self):
        await self.hass.async_add_executor_job(self.fetch_data.update)

        waste_data_custom = self.fetch_data.waste_data_custom

        try:
            # Add attribute, set the last updated status of the sensor
            self._last_update = datetime.now().strftime("%d-%m-%Y %H:%M")

            if isinstance(waste_data_custom[self.waste_type], datetime):
                _LOGGER.debug(
                    f"Generating state via AfvalwijzerCustomSensor for = {self.waste_type} with value {waste_data_custom[self.waste_type].date()}"
                )
                # Add the US date format
                collection_date_us = waste_data_custom[self.waste_type].date()
                self._year_month_day_date = str(collection_date_us)

                # Add the NL date format as default state
                self._state = datetime.strftime(
                    waste_data_custom[self.waste_type].date(), "%d-%m-%Y"
                )
            else:
                _LOGGER.debug(
                    f"Generating state via AfvalwijzerCustomSensor for = {self.waste_type} with value {waste_data_custom[self.waste_type]}"
                )
                # Add non-date as default state
                self._state = str(waste_data_custom[self.waste_type])
        except ValueError:
            _LOGGER.debug("ValueError AfvalwijzerCustomSensor - unable to set value!")
            self._state = self._default_label
            self._year_month_day_date = None
            self._last_update = datetime.now().strftime("%d-%m-%Y %H:%M")