"""Sensors for landroid_cloud."""

from __future__ import annotations

import pytz
from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from pyworxcloud import WorxCloud

from .api import LandroidAPI
from .const import ATTR_CLOUD, ATTR_DEVICES, DOMAIN, ERROR_MAP
from .device_base import LandroidSensor, LandroidSensorEntityDescription

SENSORS = [
    LandroidSensorEntityDescription(
        key="battery_state",
        name="Battery",
        entity_category=None,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.BATTERY,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="%",
        value_fn=lambda landroid: landroid.battery.get("percent", None),
        attributes=["charging"],
    ),
    LandroidSensorEntityDescription(
        key="battery_temperature",
        name="Battery Temperature",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.TEMPERATURE,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="°C",
        value_fn=lambda landroid: landroid.battery.get("temperature", None),
    ),
    LandroidSensorEntityDescription(
        key="battery_cycles_total",
        name="Battery Total Charge Cycles",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=None,
        entity_registry_enabled_default=True,
        native_unit_of_measurement=" ",
        value_fn=lambda landroid: (
            landroid.battery["cycles"]["total"]
            if "cycles" in landroid.battery
            else None
        ),
        icon="mdi:battery-sync",
    ),
    LandroidSensorEntityDescription(
        key="battery_voltage",
        name="Battery Voltage",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.VOLTAGE,
        entity_registry_enabled_default=False,
        native_unit_of_measurement="V",
        value_fn=lambda landroid: landroid.battery.get("voltage", None),
    ),
    LandroidSensorEntityDescription(
        key="blades_total_on",
        name="Blades Total On Time",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.DURATION,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="h",
        suggested_display_precision=0,
        value_fn=lambda landroid: (
            round(landroid.blades["total_on"] / 60, 2)
            if "total_on" in landroid.blades
            else None
        ),
        icon="mdi:saw-blade",
        attributes=["total_on"],
    ),
    LandroidSensorEntityDescription(
        key="blades_current_on",
        name="Blades Current On Time",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.DURATION,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="h",
        suggested_display_precision=0,
        value_fn=lambda landroid: (
            round(landroid.blades["current_on"] / 60, 2)
            if "current_on" in landroid.blades
            else None
        ),
        icon="mdi:saw-blade",
        attributes=["current_on"],
    ),
    LandroidSensorEntityDescription(
        key="blades_reset_at",
        name="Blades Reset At Hours",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=None,
        device_class=SensorDeviceClass.DURATION,
        entity_registry_enabled_default=False,
        native_unit_of_measurement="h",
        suggested_display_precision=0,
        value_fn=lambda landroid: (
            round(landroid.blades["reset_at"] / 60, 2)
            if "reset_at" in landroid.blades and landroid.blades["reset_at"] is not None
            else None
        ),
        icon="mdi:history",
        attributes=["reset_at"],
    ),
    LandroidSensorEntityDescription(
        key="blades_reset_time",
        name="Blades Reset At",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=None,
        device_class=SensorDeviceClass.TIMESTAMP,
        entity_registry_enabled_default=False,
        native_unit_of_measurement=None,
        value_fn=lambda landroid: landroid.blades.get("reset_time", None),
    ),
    LandroidSensorEntityDescription(
        key="error",
        name="Error",
        translation_key="landroid_cloud_error",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=None,
        device_class=SensorDeviceClass.ENUM,
        entity_registry_enabled_default=True,
        native_unit_of_measurement=None,
        value_fn=lambda landroid: ERROR_MAP[landroid.error["id"]],
        attributes=["id"],
        icon="mdi:alert-circle",
    ),
    LandroidSensorEntityDescription(
        key="pitch",
        name="Pitch",
        entity_category=None,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=None,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="°",
        value_fn=lambda landroid: landroid.orientation["pitch"],
        suggested_display_precision=1,
        icon="mdi:axis-x-rotate-clockwise",
    ),
    LandroidSensorEntityDescription(
        key="roll",
        name="Roll",
        entity_category=None,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=None,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="°",
        value_fn=lambda landroid: landroid.orientation["roll"],
        suggested_display_precision=1,
        icon="mdi:axis-y-rotate-clockwise",
    ),
    LandroidSensorEntityDescription(
        key="yaw",
        name="Yaw",
        entity_category=None,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=None,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="°",
        value_fn=lambda landroid: landroid.orientation["yaw"],
        suggested_display_precision=1,
        icon="mdi:axis-z-rotate-clockwise",
    ),
    LandroidSensorEntityDescription(
        key="rainsensor_remaining",
        name="Rainsensor Remaining",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=None,
        device_class=SensorDeviceClass.DURATION,
        entity_registry_enabled_default=False,
        native_unit_of_measurement="min",
        value_fn=lambda landroid: landroid.rainsensor.get("remaining", None),
        icon="mdi:weather-rainy",
    ),
    LandroidSensorEntityDescription(
        key="distance",
        name="Distance Driven",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.DISTANCE,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="km",
        suggested_display_precision=2,
        value_fn=lambda landroid: (
            round(landroid.statistics["distance"] / 1000, 3)
            if "distance" in landroid.statistics
            else None
        ),
        attributes=["distance"],
    ),
    LandroidSensorEntityDescription(
        key="worktime_total",
        name="Total Worktime",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.DURATION,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="h",
        suggested_display_precision=0,
        value_fn=lambda landroid: (
            round(landroid.statistics["worktime_total"] / 60, 2)
            if "worktime_total" in landroid.statistics
            else None
        ),
        icon="mdi:update",
        attributes=["worktime_total"],
    ),
    LandroidSensorEntityDescription(
        key="rssi",
        name="Rssi",
        entity_category=EntityCategory.DIAGNOSTIC,
        state_class=SensorStateClass.MEASUREMENT,
        device_class=SensorDeviceClass.SIGNAL_STRENGTH,
        entity_registry_enabled_default=True,
        native_unit_of_measurement="dBm",
        value_fn=lambda landroid: landroid.rssi,
    ),
    LandroidSensorEntityDescription(
        key="last_update",
        name="Last Update",
        entity_category=None,
        state_class=None,
        device_class=SensorDeviceClass.TIMESTAMP,
        entity_registry_enabled_default=True,
        native_unit_of_measurement=None,
        value_fn=lambda landroid: landroid.updated.astimezone(pytz.utc),
        icon="mdi:clock-check",
    ),
    LandroidSensorEntityDescription(
        key="next_start",
        name="Next Scheduled Start",
        entity_category=None,
        state_class=None,
        device_class=SensorDeviceClass.TIMESTAMP,
        entity_registry_enabled_default=True,
        native_unit_of_measurement=None,
        value_fn=lambda landroid: landroid.schedules["next_schedule_start"],
        icon="mdi:clock-start",
        attributes=["schedule"],
    ),
    LandroidSensorEntityDescription(
        key="daily_progress",
        name="Daily Progress",
        entity_category=None,
        state_class=None,
        device_class=None,
        entity_registry_enabled_default=False,
        native_unit_of_measurement="%",
        value_fn=lambda landroid: landroid.schedules["daily_progress"],
        icon="mdi:progress-clock",
    ),
]


async def async_setup_entry(
    hass: HomeAssistant,
    config: ConfigEntry,
    async_add_devices,
) -> None:
    """Set up the sensor platform."""
    sensors = []
    cloud: WorxCloud = hass.data[DOMAIN][config.entry_id][ATTR_CLOUD]
    for _, mower in cloud.devices.items():
        api = LandroidAPI(hass, mower.name, config)
        # for _, info in hass.data[DOMAIN][config.entry_id][ATTR_DEVICES].items():
        # api: LandroidAPI = info["api"]
        for sens in SENSORS:
            entity = LandroidSensor(hass, sens, api, config)

            sensors.append(entity)

    async_add_devices(sensors)