"""Writable zone setpoint numbers for GREE Controller."""

from __future__ import annotations

from typing import Any

from homeassistant.components.number import NumberDeviceClass, NumberEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity

from . import GreeControllerRuntimeData
from .const import DOMAIN
from .coordinator import GreeControllerCoordinator


async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Create one writable target-temperature number per zone."""
    runtime: GreeControllerRuntimeData = entry.runtime_data
    entities = [
        GreeControllerZoneTargetNumber(runtime.coordinator, str(zone["zone_id"]))
        for zone in runtime.coordinator.plan.get("zones", [])
        if zone.get("zone_id")
    ]
    async_add_entities(entities)


class GreeControllerZoneTargetNumber(CoordinatorEntity[GreeControllerCoordinator], NumberEntity):
    """Zone target override controlled through the standalone service."""

    _attr_has_entity_name = True
    _attr_name = "Target temperature"
    _attr_device_class = NumberDeviceClass.TEMPERATURE
    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
    _attr_native_min_value = 8.0
    _attr_native_max_value = 30.0
    _attr_native_step = 0.5

    def __init__(self, coordinator: GreeControllerCoordinator, zone_id: str) -> None:
        super().__init__(coordinator)
        self._zone_id = zone_id
        self._attr_unique_id = f"{zone_id}-target-temperature"

    @property
    def _zone(self) -> dict[str, Any]:
        for zone in self.coordinator.plan.get("zones", []):
            if str(zone.get("zone_id")) == self._zone_id:
                return zone
        return {}

    @property
    def available(self) -> bool:
        return super().available and bool(self._zone)

    @property
    def native_value(self) -> float | None:
        value = self._zone.get("target_temperature")
        if value is None:
            value = self._zone.get("setpoint")
        return float(value) if value is not None else None

    @property
    def device_info(self) -> DeviceInfo:
        zone = self._zone
        return DeviceInfo(
            identifiers={(DOMAIN, f"zone:{self._zone_id}")},
            name=str(zone.get("zone_name") or self._zone_id),
            manufacturer="GREE Controller",
            model="Zone thermostat",
            via_device=(DOMAIN, "controller"),
        )

    async def async_set_native_value(self, value: float) -> None:
        await self.coordinator.async_zone_control(self._zone_id, {"setpoint": float(value)})