"""Number platform for Remote Media Player integration (display brightness).""" from __future__ import annotations import logging from typing import Any from homeassistant.components.number import NumberEntity, NumberMode from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from .api_client import MediaServerClient, MediaServerError from .const import DOMAIN _LOGGER = logging.getLogger(__name__) async def async_setup_entry( hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Set up display brightness number entities from a config entry.""" client: MediaServerClient = hass.data[DOMAIN][entry.entry_id]["client"] try: monitors = await client.get_display_monitors() except MediaServerError as err: _LOGGER.error("Failed to fetch display monitors: %s", err) return entities = [ DisplayBrightnessNumber( client=client, entry=entry, monitor=monitor, ) for monitor in monitors if monitor.get("brightness") is not None ] if entities: async_add_entities(entities) _LOGGER.info("Added %d display brightness entities", len(entities)) class DisplayBrightnessNumber(NumberEntity): """Number entity for controlling display brightness.""" _attr_has_entity_name = True _attr_native_min_value = 0 _attr_native_max_value = 100 _attr_native_step = 1 _attr_native_unit_of_measurement = "%" _attr_mode = NumberMode.SLIDER _attr_icon = "mdi:brightness-6" def __init__( self, client: MediaServerClient, entry: ConfigEntry, monitor: dict[str, Any], ) -> None: """Initialize the display brightness entity.""" self._client = client self._entry = entry self._monitor_id: int = monitor["id"] self._monitor_name: str = monitor.get("name", f"Monitor {monitor['id']}") self._resolution: str | None = monitor.get("resolution") self._attr_native_value = monitor.get("brightness") # Use resolution in name to disambiguate same-name monitors display_name = self._monitor_name if self._resolution: display_name = f"{self._monitor_name} ({self._resolution})" self._attr_unique_id = f"{entry.entry_id}_display_brightness_{self._monitor_id}" self._attr_name = f"Display {display_name} Brightness" @property def device_info(self) -> DeviceInfo: """Return device info.""" return DeviceInfo( identifiers={(DOMAIN, self._entry.entry_id)}, name=self._entry.title, manufacturer="Remote Media Player", model="Media Server", ) async def async_set_native_value(self, value: float) -> None: """Set the brightness value.""" try: await self._client.set_display_brightness(self._monitor_id, int(value)) self._attr_native_value = int(value) self.async_write_ha_state() except MediaServerError as err: _LOGGER.error("Failed to set brightness for monitor %d: %s", self._monitor_id, err) async def async_update(self) -> None: """Fetch updated brightness from the server.""" try: monitors = await self._client.get_display_monitors() for monitor in monitors: if monitor["id"] == self._monitor_id: self._attr_native_value = monitor.get("brightness") break except MediaServerError as err: _LOGGER.error("Failed to update brightness for monitor %d: %s", self._monitor_id, err)