"""Support for Elgato lights.""" from __future__ import annotations from typing import Any from elgato import Elgato, ElgatoError, Info, Settings, State from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_COLOR_TEMP, ATTR_HS_COLOR, COLOR_MODE_COLOR_TEMP, COLOR_MODE_HS, LightEntity, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_MAC from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import ( AddEntitiesCallback, async_get_current_platform, ) from homeassistant.helpers.update_coordinator import ( CoordinatorEntity, DataUpdateCoordinator, ) from . import HomeAssistantElgatoData from .const import DOMAIN, LOGGER, SERVICE_IDENTIFY from .entity import ElgatoEntity PARALLEL_UPDATES = 1 async def async_setup_entry( hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Set up Elgato Light based on a config entry.""" data: HomeAssistantElgatoData = hass.data[DOMAIN][entry.entry_id] settings = await data.client.settings() async_add_entities( [ ElgatoLight( data.coordinator, data.client, data.info, entry.data.get(CONF_MAC), settings, ) ], True, ) platform = async_get_current_platform() platform.async_register_entity_service( SERVICE_IDENTIFY, {}, ElgatoLight.async_identify.__name__, ) class ElgatoLight(ElgatoEntity, CoordinatorEntity, LightEntity): """Defines an Elgato Light.""" coordinator: DataUpdateCoordinator[State] def __init__( self, coordinator: DataUpdateCoordinator, client: Elgato, info: Info, mac: str | None, settings: Settings, ) -> None: """Initialize Elgato Light.""" super().__init__(client, info, mac) CoordinatorEntity.__init__(self, coordinator) min_mired = 143 max_mired = 344 supported_color_modes = {COLOR_MODE_COLOR_TEMP} # Elgato Light supporting color, have a different temperature range if settings.power_on_hue is not None: supported_color_modes = {COLOR_MODE_COLOR_TEMP, COLOR_MODE_HS} min_mired = 153 max_mired = 285 self._attr_max_mireds = max_mired self._attr_min_mireds = min_mired self._attr_name = info.display_name or info.product_name self._attr_supported_color_modes = supported_color_modes self._attr_unique_id = info.serial_number @property def brightness(self) -> int | None: """Return the brightness of this light between 1..255.""" return round((self.coordinator.data.brightness * 255) / 100) @property def color_temp(self) -> int | None: """Return the CT color value in mireds.""" return self.coordinator.data.temperature @property def color_mode(self) -> str | None: """Return the color mode of the light.""" if self.coordinator.data.hue is not None: return COLOR_MODE_HS return COLOR_MODE_COLOR_TEMP @property def hs_color(self) -> tuple[float, float] | None: """Return the hue and saturation color value [float, float].""" return (self.coordinator.data.hue or 0, self.coordinator.data.saturation or 0) @property def is_on(self) -> bool: """Return the state of the light.""" return self.coordinator.data.on async def async_turn_off(self, **kwargs: Any) -> None: """Turn off the light.""" try: await self.client.light(on=False) except ElgatoError: LOGGER.error("An error occurred while updating the Elgato Light") await self.coordinator.async_refresh() async def async_turn_on(self, **kwargs: Any) -> None: """Turn on the light.""" temperature = kwargs.get(ATTR_COLOR_TEMP) hue = None saturation = None if ATTR_HS_COLOR in kwargs: hue, saturation = kwargs[ATTR_HS_COLOR] brightness = None if ATTR_BRIGHTNESS in kwargs: brightness = round((kwargs[ATTR_BRIGHTNESS] / 255) * 100) # For Elgato lights supporting color mode, but in temperature mode; # adjusting only brightness make them jump back to color mode. # Resending temperature prevents that. if ( brightness and ATTR_HS_COLOR not in kwargs and ATTR_COLOR_TEMP not in kwargs and self.supported_color_modes and COLOR_MODE_HS in self.supported_color_modes and self.color_mode == COLOR_MODE_COLOR_TEMP ): temperature = self.color_temp try: await self.client.light( on=True, brightness=brightness, hue=hue, saturation=saturation, temperature=temperature, ) except ElgatoError: LOGGER.error("An error occurred while updating the Elgato Light") await self.coordinator.async_refresh() async def async_identify(self) -> None: """Identify the light, will make it blink.""" try: await self.client.identify() except ElgatoError: LOGGER.exception("An error occurred while identifying the Elgato Light") await self.coordinator.async_refresh()