From 95cbf7cca7c9b02f32a9b5a617e90266274149ff Mon Sep 17 00:00:00 2001
From: Matrix <solitudek2@gmail.com>
Date: Wed, 23 Nov 2022 23:27:51 +0800
Subject: [PATCH] Add yolink Dimmer support (#81970)

* Add yolink Dimmer support

* suggest change

* fix suggest

* fix suggest

* fix suggest
---
 .coveragerc                                 |  1 +
 homeassistant/components/yolink/__init__.py |  1 +
 homeassistant/components/yolink/const.py    |  1 +
 homeassistant/components/yolink/light.py    | 75 +++++++++++++++++++++
 4 files changed, 78 insertions(+)
 create mode 100644 homeassistant/components/yolink/light.py

diff --git a/.coveragerc b/.coveragerc
index f3931e47155..bf493a6a0a8 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -1572,6 +1572,7 @@ omit =
     homeassistant/components/yolink/coordinator.py
     homeassistant/components/yolink/cover.py
     homeassistant/components/yolink/entity.py
+    homeassistant/components/yolink/light.py
     homeassistant/components/yolink/lock.py
     homeassistant/components/yolink/sensor.py
     homeassistant/components/yolink/siren.py
diff --git a/homeassistant/components/yolink/__init__.py b/homeassistant/components/yolink/__init__.py
index 06e6fd6472a..58407df38a3 100644
--- a/homeassistant/components/yolink/__init__.py
+++ b/homeassistant/components/yolink/__init__.py
@@ -28,6 +28,7 @@ PLATFORMS = [
     Platform.BINARY_SENSOR,
     Platform.CLIMATE,
     Platform.COVER,
+    Platform.LIGHT,
     Platform.LOCK,
     Platform.SENSOR,
     Platform.SIREN,
diff --git a/homeassistant/components/yolink/const.py b/homeassistant/components/yolink/const.py
index f6add984dc2..14279ebeae0 100644
--- a/homeassistant/components/yolink/const.py
+++ b/homeassistant/components/yolink/const.py
@@ -25,3 +25,4 @@ ATTR_DEVICE_MANIPULATOR = "Manipulator"
 ATTR_DEVICE_CO_SMOKE_SENSOR = "COSmokeSensor"
 ATTR_DEVICE_SWITCH = "Switch"
 ATTR_DEVICE_THERMOSTAT = "Thermostat"
+ATTR_DEVICE_DIMMER = "Dimmer"
diff --git a/homeassistant/components/yolink/light.py b/homeassistant/components/yolink/light.py
new file mode 100644
index 00000000000..fbb8c8d50c6
--- /dev/null
+++ b/homeassistant/components/yolink/light.py
@@ -0,0 +1,75 @@
+"""YoLink Dimmer."""
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import ATTR_COORDINATORS, ATTR_DEVICE_DIMMER, DOMAIN
+from .coordinator import YoLinkCoordinator
+from .entity import YoLinkEntity
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up YoLink Dimmer from a config entry."""
+    device_coordinators = hass.data[DOMAIN][config_entry.entry_id][ATTR_COORDINATORS]
+    entities = [
+        YoLinkDimmerEntity(config_entry, device_coordinator)
+        for device_coordinator in device_coordinators.values()
+        if device_coordinator.device.device_type == ATTR_DEVICE_DIMMER
+    ]
+    async_add_entities(entities)
+
+
+class YoLinkDimmerEntity(YoLinkEntity, LightEntity):
+    """YoLink Dimmer Entity."""
+
+    _attr_color_mode = ColorMode.BRIGHTNESS
+    _attr_has_entity_name = True
+    _attr_name = None
+    _attr_supported_color_modes: set[ColorMode] = {ColorMode.BRIGHTNESS}
+
+    def __init__(
+        self,
+        config_entry: ConfigEntry,
+        coordinator: YoLinkCoordinator,
+    ) -> None:
+        """Init YoLink Dimmer entity."""
+        super().__init__(config_entry, coordinator)
+        self._attr_unique_id = f"{coordinator.device.device_id}"
+
+    @callback
+    def update_entity_state(self, state: dict[str, Any]) -> None:
+        """Update HA Entity State."""
+        if (dimmer_is_on := state.get("state")) is not None:
+            # update _attr_is_on when device report it's state
+            self._attr_is_on = dimmer_is_on
+        if (brightness := state.get("brightness")) is not None:
+            self._attr_brightness = round(255 * brightness / 100)
+        self.async_write_ha_state()
+
+    async def toggle_light_state(self, state: str, brightness: int | None) -> None:
+        """Toggle light state."""
+        params: dict[str, Any] = {"state": state}
+        if brightness is not None:
+            self._attr_brightness = brightness
+            params["brightness"] = round(brightness / 255, 2) * 100
+        await self.call_device_api("setState", params)
+        self._attr_is_on = state == "open"
+        self.async_write_ha_state()
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Turn on light."""
+        brightness = kwargs.get(ATTR_BRIGHTNESS)
+        await self.toggle_light_state("open", brightness)
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off light."""
+        await self.toggle_light_state("close", None)