"""Provides triggers for moisture."""

from __future__ import annotations

from homeassistant.components.binary_sensor import (
    DOMAIN as BINARY_SENSOR_DOMAIN,
    BinarySensorDeviceClass,
)
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN, SensorDeviceClass
from homeassistant.const import PERCENTAGE, STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.helpers.automation import DomainSpec
from homeassistant.helpers.trigger import (
    Trigger,
    make_entity_numerical_state_changed_trigger,
    make_entity_numerical_state_crossed_threshold_trigger,
    make_entity_target_state_trigger,
)

MOISTURE_BINARY_DOMAIN_SPECS: dict[str, DomainSpec] = {
    BINARY_SENSOR_DOMAIN: DomainSpec(device_class=BinarySensorDeviceClass.MOISTURE),
}

MOISTURE_NUMERICAL_DOMAIN_SPECS: dict[str, DomainSpec] = {
    SENSOR_DOMAIN: DomainSpec(device_class=SensorDeviceClass.MOISTURE),
}


TRIGGERS: dict[str, type[Trigger]] = {
    "detected": make_entity_target_state_trigger(
        MOISTURE_BINARY_DOMAIN_SPECS, STATE_ON
    ),
    "cleared": make_entity_target_state_trigger(
        MOISTURE_BINARY_DOMAIN_SPECS, STATE_OFF
    ),
    "changed": make_entity_numerical_state_changed_trigger(
        MOISTURE_NUMERICAL_DOMAIN_SPECS, valid_unit=PERCENTAGE
    ),
    "crossed_threshold": make_entity_numerical_state_crossed_threshold_trigger(
        MOISTURE_NUMERICAL_DOMAIN_SPECS, valid_unit=PERCENTAGE
    ),
}


async def async_get_triggers(hass: HomeAssistant) -> dict[str, type[Trigger]]:
    """Return the triggers for moisture."""
    return TRIGGERS