"""Oasis Mini number entity."""

from __future__ import annotations

from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback

from .const import DOMAIN
from .coordinator import OasisMiniCoordinator
from .entity import OasisMiniEntity


class OasisMiniNumberEntity(OasisMiniEntity, NumberEntity):
    """Oasis Mini number entity."""

    @property
    def native_value(self) -> str | None:
        """Return the value reported by the number."""
        return getattr(self.device, self.entity_description.key)

    async def async_set_native_value(self, value: float) -> None:
        """Set new value."""
        if self.entity_description.key == "ball_speed":
            await self.device.async_set_ball_speed(value)
        elif self.entity_description.key == "led_speed":
            await self.device.async_set_led(led_speed=value)
        await self.coordinator.async_request_refresh()


DESCRIPTORS = {
    NumberEntityDescription(
        key="ball_speed",
        name="Ball speed",
        native_max_value=800,
        native_min_value=200,
    ),
    NumberEntityDescription(
        key="led_speed",
        name="LED speed",
        native_max_value=90,
        native_min_value=-90,
    ),
}


async def async_setup_entry(
    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
    """Set up Oasis Mini numbers using config entry."""
    coordinator: OasisMiniCoordinator = hass.data[DOMAIN][entry.entry_id]
    async_add_entities(
        [
            OasisMiniNumberEntity(coordinator, entry, descriptor)
            for descriptor in DESCRIPTORS
        ]
    )