From 0442c660c5b434d3d4304d0299ba3eb586d82ac9 Mon Sep 17 00:00:00 2001
From: Jonni Liljamo <jonni@liljamo.com>
Date: Tue, 12 Nov 2024 21:30:57 +0200
Subject: [PATCH] feat: select for operation mode

---
 custom_components/ouman_eh800/__init__.py |  1 +
 custom_components/ouman_eh800/select.py   | 76 +++++++++++++++++++++++
 2 files changed, 77 insertions(+)
 create mode 100644 custom_components/ouman_eh800/select.py

diff --git a/custom_components/ouman_eh800/__init__.py b/custom_components/ouman_eh800/__init__.py
index 8c584ca..5ed4b02 100644
--- a/custom_components/ouman_eh800/__init__.py
+++ b/custom_components/ouman_eh800/__init__.py
@@ -15,6 +15,7 @@ _LOGGER = logging.getLogger(__name__)
 PLATFORMS = [
     Platform.CLIMATE,
     Platform.NUMBER,
+    Platform.SELECT,
     Platform.SENSOR,
     Platform.SWITCH,
     Platform.VALVE,
diff --git a/custom_components/ouman_eh800/select.py b/custom_components/ouman_eh800/select.py
new file mode 100644
index 0000000..64d2ee1
--- /dev/null
+++ b/custom_components/ouman_eh800/select.py
@@ -0,0 +1,76 @@
+import logging
+
+from homeassistant.components.select import (
+    SelectEntity,
+    SelectEntityDescription,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from . import OumanEH800Device
+from .const import DOMAIN
+from .eh800 import OPERATION_MODES
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    """Set up Ouman EH-800 device climate control."""
+    device = hass.data[DOMAIN].get(entry.entry_id)
+
+    entities: list[OumanEH800DeviceSelect] = [
+        OumanEH800DeviceSelect(
+            device,
+            SelectEntityDescription(
+                key="l1_operation_mode",
+            ),
+        )
+    ]
+
+    async_add_entities(entities, True)
+
+
+class OumanEH800DeviceSelect(SelectEntity):
+    entity_description: SelectEntityDescription
+
+    def __init__(
+        self,
+        device: OumanEH800Device,
+        description: SelectEntityDescription,
+    ) -> None:
+        self._device = device
+        self.entity_description = description
+
+        self._attr_name = description.key.replace("_", " ").capitalize()
+        self._attr_unique_id = f"ouman_eh800_{description.key}"
+        self._attr_device_info = device.device_info
+
+    @property
+    def current_option(self) -> str:
+        operation_mode = int(
+            self._device.device.data.get(self.entity_description.key, 0)
+        )
+        return [om.name for om in OPERATION_MODES if om.value == operation_mode][0]
+
+    @property
+    def options(self) -> list[str]:
+        return [om.name for om in OPERATION_MODES]
+
+    async def async_select_option(self, option: str) -> None:
+        operation_mode = [om for om in OPERATION_MODES if om.name == option][0]
+        _LOGGER.debug(
+            "Setting operation mode to '%s' (%s)",
+            operation_mode.name,
+            operation_mode.value,
+        )
+        await self._device.device.update_value(
+            self.entity_description.key,
+            operation_mode.value,
+        )
+        self.async_write_ha_state()
+
+    async def async_update(self) -> None:
+        await self._device.async_update()
-- 
2.44.1