From d2f06b060758f0b4675a38970228bb65246e0e8a Mon Sep 17 00:00:00 2001
From: Vincent Le Bourlot <vlebourl@gmail.com>
Date: Fri, 1 Apr 2022 16:49:02 +0200
Subject: [PATCH] fix mode selection and display for humidifier Dual 200S (#2)

---
 custom_components/vesync/humidifier.py | 29 +++++++++++++++++++-------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/custom_components/vesync/humidifier.py b/custom_components/vesync/humidifier.py
index e7cf08f..6f19ff7 100644
--- a/custom_components/vesync/humidifier.py
+++ b/custom_components/vesync/humidifier.py
@@ -2,7 +2,12 @@
 import logging
 
 from homeassistant.components.humidifier import HumidifierEntity
-from homeassistant.components.humidifier.const import SUPPORT_MODES
+from homeassistant.components.humidifier.const import (
+    MODE_AUTO,
+    MODE_BOOST,
+    MODE_SLEEP,
+    SUPPORT_MODES,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
@@ -16,6 +21,10 @@ _LOGGER = logging.getLogger(__name__)
 MAX_HUMIDITY = 80
 MIN_HUMIDITY = 30
 
+MODE_MANUAL = "manual"
+
+MODES = [MODE_AUTO, MODE_BOOST, MODE_SLEEP]
+
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -68,7 +77,7 @@ class VeSyncHumidifierHA(VeSyncDevice, HumidifierEntity):
     @property
     def available_modes(self):
         """Return the available mist modes."""
-        return self.device.config_dict["mist_modes"]
+        return MODES
 
     @property
     def supported_features(self):
@@ -83,8 +92,14 @@ class VeSyncHumidifierHA(VeSyncDevice, HumidifierEntity):
     @property
     def mode(self):
         """Get the current preset mode."""
-        if self.smarthumidifier.details["mode"] in self.available_modes:
-            return self.smarthumidifier.details["mode"]
+        if self.smarthumidifier.details["mode"] == MODE_AUTO:
+            return MODE_AUTO
+        if self.smarthumidifier.details["mode"] == MODE_MANUAL:
+            return (
+                MODE_SLEEP
+                if self.smarthumidifier.details["mist_level"] == 1
+                else MODE_BOOST
+            )
         return None
 
     @property
@@ -138,10 +153,8 @@ class VeSyncHumidifierHA(VeSyncDevice, HumidifierEntity):
             raise ValueError(
                 "{mode} is not one of the valid available modes: {self.available_modes}"
             )
-        if mode == "manual":
-            self.smarthumidifier.set_mist_level(
-                self.smarthumidifier.details["mist_level"]
-            )
+        if mode != MODE_AUTO:
+            self.smarthumidifier.set_mist_level(1 if mode == MODE_SLEEP else 2)
         else:
             self.smarthumidifier.set_humidity_mode(mode)
         self.schedule_update_ha_state()