"""Yandex.Pogoda triggers."""

import logging

import voluptuous as vol
from homeassistant.components.device_automation import \
    DEVICE_TRIGGER_BASE_SCHEMA as HA_TRIGGER_BASE_SCHEMA
from homeassistant.components.homeassistant.triggers import \
    event as event_trigger
from homeassistant.const import (CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM,
                                 CONF_TYPE)
from homeassistant.core import HomeAssistant

from .const import DOMAIN, WEATHER_STATES_CONVERSION, map_state


def generate_triggers() -> list:
    """Generate triggers list."""
    result = []
    for s in WEATHER_STATES_CONVERSION:
        for d in [True, False]:
            result.append(map_state(src=s, is_day=d, mapping=WEATHER_STATES_CONVERSION))

    return list(set(result))


_LOGGER = logging.getLogger(__name__)

TRIGGERS = generate_triggers()

TRIGGER_SCHEMA = HA_TRIGGER_BASE_SCHEMA.extend(
    {
        vol.Required(CONF_TYPE): vol.In(TRIGGERS),
    }
)


async def async_get_triggers(_hass: HomeAssistant, device_id):
    """Return a list of triggers."""
    triggers = []

    for t in TRIGGERS:
        triggers.append(
            {
                # Required fields of TRIGGER_BASE_SCHEMA
                CONF_PLATFORM: "device",
                CONF_DOMAIN: DOMAIN,
                CONF_DEVICE_ID: device_id,
                # Required fields of TRIGGER_SCHEMA
                CONF_TYPE: t,
            }
        )

    return triggers


async def async_attach_trigger(hass: HomeAssistant, config, action, automation_info):
    """Attach a trigger."""
    config = TRIGGER_SCHEMA(config)
    _LOGGER.debug("Got subscription to trigger: %s", config)
    event_config = event_trigger.TRIGGER_SCHEMA(
        {
            event_trigger.CONF_PLATFORM: "event",
            event_trigger.CONF_EVENT_TYPE: DOMAIN + "_event",
            event_trigger.CONF_EVENT_DATA: {
                CONF_DEVICE_ID: config[CONF_DEVICE_ID],
                CONF_TYPE: config[CONF_TYPE],
            },
        }
    )

    return await event_trigger.async_attach_trigger(
        hass, event_config, action, automation_info, platform_type="device"
    )