Add panel brightness control for Litter-Robot 4 (#86269)

* Add panel brightness control for Litter-Robot 4

* Use translation_key

* Fix test

homeassistant/components/litterrobot/select.py

@@ -3,10 +3,10 @@ from __future__ import annotations
 
 from collections.abc import Callable, Coroutine
 from dataclasses import dataclass
-import itertools
 from typing import Any, Generic, TypeVar
 
-from pylitterbot import FeederRobot, LitterRobot
+from pylitterbot import FeederRobot, LitterRobot, LitterRobot4, Robot
+from pylitterbot.robot.litterrobot4 import BrightnessLevel
 
 from homeassistant.components.select import SelectEntity, SelectEntityDescription
 from homeassistant.config_entries import ConfigEntry
@@ -18,14 +18,21 @@ from .const import DOMAIN
 from .entity import LitterRobotEntity, _RobotT
 from .hub import LitterRobotHub
 
-_CastTypeT = TypeVar("_CastTypeT", int, float)
+_CastTypeT = TypeVar("_CastTypeT", int, float, str)
+
+BRIGHTNESS_LEVEL_ICON_MAP: dict[BrightnessLevel | None, str] = {
+    BrightnessLevel.LOW: "mdi:lightbulb-on-30",
+    BrightnessLevel.MEDIUM: "mdi:lightbulb-on-50",
+    BrightnessLevel.HIGH: "mdi:lightbulb-on",
+    None: "mdi:lightbulb-question",
+}
 
 
 @dataclass
 class RequiredKeysMixin(Generic[_RobotT, _CastTypeT]):
     """A class that describes robot select entity required keys."""
 
-    current_fn: Callable[[_RobotT], _CastTypeT]
+    current_fn: Callable[[_RobotT], _CastTypeT | None]
     options_fn: Callable[[_RobotT], list[_CastTypeT]]
     select_fn: Callable[[_RobotT, str], Coroutine[Any, Any, bool]]
 
@@ -37,26 +44,42 @@ class RobotSelectEntityDescription(
     """A class that describes robot select entities."""
 
     entity_category: EntityCategory = EntityCategory.CONFIG
+    icon_fn: Callable[[_RobotT], str] | None = None
 
 
-LITTER_ROBOT_SELECT = RobotSelectEntityDescription[LitterRobot, int](
-    key="cycle_delay",
-    name="Clean cycle wait time minutes",
-    icon="mdi:timer-outline",
-    unit_of_measurement=UnitOfTime.MINUTES,
-    current_fn=lambda robot: robot.clean_cycle_wait_time_minutes,
-    options_fn=lambda robot: robot.VALID_WAIT_TIMES,
-    select_fn=lambda robot, option: robot.set_wait_time(int(option)),
-)
-FEEDER_ROBOT_SELECT = RobotSelectEntityDescription[FeederRobot, float](
-    key="meal_insert_size",
-    name="Meal insert size",
-    icon="mdi:scale",
-    unit_of_measurement="cups",
-    current_fn=lambda robot: robot.meal_insert_size,
-    options_fn=lambda robot: robot.VALID_MEAL_INSERT_SIZES,
-    select_fn=lambda robot, option: robot.set_meal_insert_size(float(option)),
-)
+ROBOT_SELECT_MAP: dict[type[Robot], RobotSelectEntityDescription] = {
+    LitterRobot: RobotSelectEntityDescription[LitterRobot, int](
+        key="cycle_delay",
+        name="Clean cycle wait time minutes",
+        icon="mdi:timer-outline",
+        unit_of_measurement=UnitOfTime.MINUTES,
+        current_fn=lambda robot: robot.clean_cycle_wait_time_minutes,
+        options_fn=lambda robot: robot.VALID_WAIT_TIMES,
+        select_fn=lambda robot, opt: robot.set_wait_time(int(opt)),
+    ),
+    LitterRobot4: RobotSelectEntityDescription[LitterRobot4, str](
+        key="panel_brightness",
+        name="Panel brightness",
+        translation_key="brightness_level",
+        current_fn=lambda robot: bri.name.lower()
+        if (bri := robot.panel_brightness) is not None
+        else None,
+        options_fn=lambda _: [level.name.lower() for level in BrightnessLevel],
+        select_fn=lambda robot, opt: robot.set_panel_brightness(
+            BrightnessLevel[opt.upper()]
+        ),
+        icon_fn=lambda robot: BRIGHTNESS_LEVEL_ICON_MAP[robot.panel_brightness],
+    ),
+    FeederRobot: RobotSelectEntityDescription[FeederRobot, float](
+        key="meal_insert_size",
+        name="Meal insert size",
+        icon="mdi:scale",
+        unit_of_measurement="cups",
+        current_fn=lambda robot: robot.meal_insert_size,
+        options_fn=lambda robot: robot.VALID_MEAL_INSERT_SIZES,
+        select_fn=lambda robot, opt: robot.set_meal_insert_size(float(opt)),
+    ),
+}
 
 
 async def async_setup_entry(
@@ -66,22 +89,16 @@ async def async_setup_entry(
 ) -> None:
     """Set up Litter-Robot selects using config entry."""
     hub: LitterRobotHub = hass.data[DOMAIN][config_entry.entry_id]
-    entities: list[LitterRobotSelect] = list(
-        itertools.chain(
-            (
-                LitterRobotSelect(robot=robot, hub=hub, description=LITTER_ROBOT_SELECT)
-                for robot in hub.litter_robots()
-            ),
-            (
-                LitterRobotSelect(robot=robot, hub=hub, description=FEEDER_ROBOT_SELECT)
-                for robot in hub.feeder_robots()
-            ),
-        )
-    )
+    entities = [
+        LitterRobotSelectEntity(robot=robot, hub=hub, description=description)
+        for robot in hub.account.robots
+        for robot_type, description in ROBOT_SELECT_MAP.items()
+        if isinstance(robot, robot_type)
+    ]
     async_add_entities(entities)
 
 
-class LitterRobotSelect(
+class LitterRobotSelectEntity(
     LitterRobotEntity[_RobotT], SelectEntity, Generic[_RobotT, _CastTypeT]
 ):
     """Litter-Robot Select."""
@@ -99,6 +116,13 @@ class LitterRobotSelect(
         options = self.entity_description.options_fn(self.robot)
         self._attr_options = list(map(str, options))
 
+    @property
+    def icon(self) -> str | None:
+        """Return the icon to use in the frontend, if any."""
+        if icon_fn := self.entity_description.icon_fn:
+            return str(icon_fn(self.robot))
+        return super().icon
+
     @property
     def current_option(self) -> str | None:
         """Return the selected entity option to represent the entity state."""