Add Switch platform to Roborock (#93833)

* init

* add switch platform

* remove stale comments

* remove stale list

* set entity category to config

homeassistant/components/roborock/switch.py

@@ -0,0 +1,110 @@
+"""Support for Roborock switch."""
+from collections.abc import Callable, Coroutine
+from dataclasses import dataclass
+import logging
+from typing import Any
+
+from roborock.roborock_typing import RoborockCommand
+
+from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.util import slugify
+
+from .const import DOMAIN
+from .coordinator import RoborockDataUpdateCoordinator
+from .device import RoborockCoordinatedEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+@dataclass
+class RoborockSwitchDescriptionMixin:
+    """Define an entity description mixin for switch entities."""
+
+    # Gets the status of the switch
+    get_value: Callable[[RoborockCoordinatedEntity], Coroutine[Any, Any, dict]]
+    # Evaluate the result of get_value to determine a bool
+    evaluate_value: Callable[[dict], bool]
+    # Sets the status of the switch
+    set_command: Callable[[RoborockCoordinatedEntity, bool], Coroutine[Any, Any, dict]]
+
+
+@dataclass
+class RoborockSwitchDescription(
+    SwitchEntityDescription, RoborockSwitchDescriptionMixin
+):
+    """Class to describe an Roborock switch entity."""
+
+
+SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
+    RoborockSwitchDescription(
+        set_command=lambda entity, value: entity.send(
+            RoborockCommand.SET_CHILD_LOCK_STATUS, {"lock_status": 1 if value else 0}
+        ),
+        get_value=lambda data: data.send(RoborockCommand.GET_CHILD_LOCK_STATUS),
+        evaluate_value=lambda data: data["lock_status"] == 1,
+        key="child_lock",
+        translation_key="child_lock",
+        icon="mdi:account-lock",
+        entity_category=EntityCategory.CONFIG,
+    )
+]
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up Roborock switch platform."""
+
+    coordinators: dict[str, RoborockDataUpdateCoordinator] = hass.data[DOMAIN][
+        config_entry.entry_id
+    ]
+    async_add_entities(
+        (
+            RoborockSwitchEntity(
+                f"{description.key}_{slugify(device_id)}",
+                coordinator,
+                description,
+            )
+            for device_id, coordinator in coordinators.items()
+            for description in SWITCH_DESCRIPTIONS
+        ),
+        True,
+    )
+
+
+class RoborockSwitchEntity(RoborockCoordinatedEntity, SwitchEntity):
+    """A class to let you turn functionality on Roborock devices on and off."""
+
+    entity_description: RoborockSwitchDescription
+
+    def __init__(
+        self,
+        unique_id: str,
+        coordinator: RoborockDataUpdateCoordinator,
+        entity_description: RoborockSwitchDescription,
+    ) -> None:
+        """Create a switch entity."""
+        self.entity_description = entity_description
+        super().__init__(unique_id, coordinator)
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off the switch."""
+        await self.entity_description.set_command(self, False)
+        return self.async_schedule_update_ha_state(True)
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Turn on the switch."""
+        await self.entity_description.set_command(self, True)
+        return self.async_schedule_update_ha_state(True)
+
+    async def async_update(self) -> None:
+        """Update switch."""
+        self._attr_is_on = self.entity_description.evaluate_value(
+            await self.entity_description.get_value(self)
+        )