"""Number platform for HDFury Integration."""

from collections.abc import Awaitable, Callable
from dataclasses import dataclass
from typing import override

from hdfury import HDFuryAPI, HDFuryError

from homeassistant.components.number import (
    NumberDeviceClass,
    NumberEntity,
    NumberEntityDescription,
    NumberMode,
)
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback

from .const import DOMAIN
from .coordinator import HDFuryConfigEntry
from .entity import HDFuryEntity

PARALLEL_UPDATES = 1


@dataclass(kw_only=True, frozen=True)
class HDFuryNumberEntityDescription(NumberEntityDescription):
    """Description for HDFury number entities."""

    set_value_fn: Callable[[HDFuryAPI, str], Awaitable[None]]


NUMBERS: tuple[HDFuryNumberEntityDescription, ...] = (
    HDFuryNumberEntityDescription(
        key="unmutecnt",
        translation_key="audio_unmute",
        entity_registry_enabled_default=False,
        mode=NumberMode.BOX,
        native_min_value=50,
        native_max_value=1000,
        native_step=1,
        device_class=NumberDeviceClass.DURATION,
        native_unit_of_measurement=UnitOfTime.MILLISECONDS,
        entity_category=EntityCategory.CONFIG,
        set_value_fn=lambda client, value: client.set_audio_unmute(value),
    ),
    HDFuryNumberEntityDescription(
        key="earcunmutecnt",
        translation_key="earc_unmute",
        entity_registry_enabled_default=False,
        mode=NumberMode.BOX,
        native_min_value=0,
        native_max_value=1000,
        native_step=1,
        device_class=NumberDeviceClass.DURATION,
        native_unit_of_measurement=UnitOfTime.MILLISECONDS,
        entity_category=EntityCategory.CONFIG,
        set_value_fn=lambda client, value: client.set_earc_unmute(value),
    ),
    HDFuryNumberEntityDescription(
        key="oledfade",
        translation_key="oled_fade",
        mode=NumberMode.BOX,
        native_min_value=1,
        native_max_value=100,
        native_step=1,
        device_class=NumberDeviceClass.DURATION,
        native_unit_of_measurement=UnitOfTime.SECONDS,
        entity_category=EntityCategory.CONFIG,
        set_value_fn=lambda client, value: client.set_oled_fade(value),
    ),
    HDFuryNumberEntityDescription(
        key="reboottimer",
        translation_key="reboot_timer",
        mode=NumberMode.BOX,
        native_min_value=0,
        native_max_value=100,
        native_step=1,
        device_class=NumberDeviceClass.DURATION,
        native_unit_of_measurement=UnitOfTime.HOURS,
        entity_category=EntityCategory.CONFIG,
        set_value_fn=lambda client, value: client.set_reboot_timer(value),
    ),
)


async def async_setup_entry(
    hass: HomeAssistant,
    entry: HDFuryConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up numbers using the platform schema."""

    coordinator = entry.runtime_data

    async_add_entities(
        HDFuryNumber(coordinator, description)
        for description in NUMBERS
        if description.key in coordinator.data.config
    )


class HDFuryNumber(HDFuryEntity, NumberEntity):
    """Base HDFury Number Class."""

    entity_description: HDFuryNumberEntityDescription

    @property
    @override
    def native_value(self) -> float:
        """Return the current number value."""

        return float(self.coordinator.data.config[self.entity_description.key])

    @override
    async def async_set_native_value(self, value: float) -> None:
        """Set Number Value Event."""

        try:
            await self.entity_description.set_value_fn(
                self.coordinator.client, str(int(value))
            )
        except HDFuryError as error:
            raise HomeAssistantError(
                translation_domain=DOMAIN,
                translation_key="communication_error",
            ) from error

        await self.coordinator.async_request_refresh()