From f374dd4f4aedca94a0243d625c4a0055d6dda6b7 Mon Sep 17 00:00:00 2001
From: Taras Nychko <taras.nychko@sky.uk>
Date: Mon, 14 Jun 2021 23:25:21 +0100
Subject: [PATCH] Add support for paused state

---
 custom_components/localtuya/const.py             | 1 +
 custom_components/localtuya/translations/en.json | 2 ++
 custom_components/localtuya/vacuum.py            | 9 ++++++++-
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/custom_components/localtuya/const.py b/custom_components/localtuya/const.py
index f844609..76739dc 100644
--- a/custom_components/localtuya/const.py
+++ b/custom_components/localtuya/const.py
@@ -53,6 +53,7 @@ CONF_FAN_SPEED_DP = "fan_speed_dp"
 CONF_FAN_SPEEDS = "fan_speeds"
 CONF_CLEAN_TIME_DP = "clean_time_dp"
 CONF_CLEAN_AREA_DP = "clean_area_dp"
+CONF_PAUSED_STATE = "paused_state"
 
 DATA_DISCOVERY = "discovery"
 
diff --git a/custom_components/localtuya/translations/en.json b/custom_components/localtuya/translations/en.json
index d1f9b24..724c2c4 100644
--- a/custom_components/localtuya/translations/en.json
+++ b/custom_components/localtuya/translations/en.json
@@ -71,6 +71,7 @@
                     "fan_speeds": "Fan speeds list (comma-separated)",
                     "clean_time_dp": "Clean Time DP (Usually 33)",
                     "clean_area_dp": "Clean Area DP (Usually 32)",
+                    "paused_state": "Pause state (pause, paused, etc)",
                     "brightness": "Brightness (only for white color)",
                     "brightness_lower": "Brightness Lower Value",
                     "brightness_upper": "Brightness Upper Value",
@@ -134,6 +135,7 @@
                     "fan_speeds": "Fan speeds list (comma-separated)",
                     "clean_time_dp": "Clean Time DP (Usually 33)",
                     "clean_area_dp": "Clean Area DP (Usually 32)",
+                    "paused_state": "Pause state (pause, paused, etc)",
                     "brightness": "Brightness (only for white color)",
                     "brightness_lower": "Brightness Lower Value",
                     "brightness_upper": "Brightness Upper Value",
diff --git a/custom_components/localtuya/vacuum.py b/custom_components/localtuya/vacuum.py
index 94021c0..de91d14 100644
--- a/custom_components/localtuya/vacuum.py
+++ b/custom_components/localtuya/vacuum.py
@@ -9,6 +9,7 @@ from homeassistant.components.vacuum import (
     STATE_DOCKED,
     STATE_IDLE,
     STATE_RETURNING,
+    STATE_PAUSED,
     SUPPORT_BATTERY,
     SUPPORT_FAN_SPEED,
     SUPPORT_PAUSE,
@@ -17,6 +18,7 @@ from homeassistant.components.vacuum import (
     SUPPORT_STATE,
     SUPPORT_STATUS,
     SUPPORT_STOP,
+    SUPPORT_PAUSE,
     StateVacuumEntity,
 )
 
@@ -33,7 +35,8 @@ from .const import (
     CONF_FAN_SPEED_DP,
     CONF_FAN_SPEEDS,
     CONF_CLEAN_TIME_DP,
-    CONF_CLEAN_AREA_DP
+    CONF_CLEAN_AREA_DP,
+    CONF_PAUSED_STATE,
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -48,6 +51,7 @@ DEFAULT_RETURNING_STATUS = "docking"
 DEFAULT_DOCKED_STATUS = "charging,chargecompleted"
 DEFAULT_MODES = "chargego,smart,standby,wall_follow,spiral,single"
 DEFAULT_FAN_SPEEDS = "low,normal,high"
+DEFAULT_PAUSED_STATE = "paused"
 
 def flow_schema(dps):
     """Return schema used in config flow."""
@@ -63,6 +67,7 @@ def flow_schema(dps):
         vol.Optional(CONF_FAN_SPEEDS, default=DEFAULT_FAN_SPEEDS): str,
         vol.Optional(CONF_CLEAN_TIME_DP): vol.In(dps),
         vol.Optional(CONF_CLEAN_AREA_DP): vol.In(dps),
+        vol.Optional(CONF_PAUSED_STATE, default=DEFAULT_PAUSED_STATE): str,
     }
 
 
@@ -193,6 +198,8 @@ class LocaltuyaVacuum(LocalTuyaEntity, StateVacuumEntity):
             self._state = STATE_DOCKED
         elif state_value == self._config[CONF_RETURNING_STATUS_VALUE]:
             self._state = STATE_RETURNING
+        elif state_value == self._config[CONF_PAUSED_STATE]:
+            self._state = STATE_PAUSED
         else:
             self._state = STATE_CLEANING