feat(zeo): add pause(), resume(), shutdown() and split start param DPs by device type

Add pause() (DP 201), shutdown() (DP 202), and update resume() to align with the official app's WasherDpsManager module 727. Split _START_PARAM_DPS into _START_PARAM_DPS_WASHER (10 DPs) and _START_PARAM_DPS_DRYER (7 DPs) so fallback queries only request DPs the device actually supports, avoiding send_decoded_command timeouts on partial responses.

Author: NOisi-X

roborock/devices/traits/a01/command.py

@@ -13,7 +13,7 @@
 
 _LOGGER = logging.getLogger(__name__)
 
-_START_PARAM_DPS: list[RoborockZeoProtocol] = [
+_START_PARAM_DPS_WASHER: list[RoborockZeoProtocol] = [
     RoborockZeoProtocol.MODE,
     RoborockZeoProtocol.PROGRAM,
     RoborockZeoProtocol.TEMP,
@@ -24,9 +24,16 @@
     RoborockZeoProtocol.SOFTENER_SET,
     RoborockZeoProtocol.COUNTDOWN,
     RoborockZeoProtocol.SOAK,
+]
+
+_START_PARAM_DPS_DRYER: list[RoborockZeoProtocol] = [
+    RoborockZeoProtocol.MODE,
+    RoborockZeoProtocol.PROGRAM,
+    RoborockZeoProtocol.DRYING_MODE,
     RoborockZeoProtocol.TOTAL_TIME,
     RoborockZeoProtocol.DRYING_METHOD,
     RoborockZeoProtocol.STEAM_VOLUME,
+    RoborockZeoProtocol.COUNTDOWN,
 ]
 
 _FIELD_TO_DP: dict[str, RoborockZeoProtocol] = {
@@ -50,7 +57,7 @@
 }
 
 class ZeoCommandTrait:
-    """Trait for sending wash-programme commands to Zeo devices."""
+    """Trait for sending commands to Zeo devices."""
 
     def __init__(
         self,
@@ -79,7 +86,6 @@ def _convert_value(self, protocol: RoborockZeoProtocol, value: Any) -> Any:
 
     async def start_program(self) -> dict[RoborockZeoProtocol, Any]:
         """Start the device, bundling the current programme parameters."""
-        _LOGGER.debug("Start command: discovering features and building payload")
         features = self._feature_trait.features
         p = await self._get_start_params()
         dps: dict[RoborockZeoProtocol, Any] = {RoborockZeoProtocol.START: "True"}
@@ -102,33 +108,46 @@ async def start_program(self) -> dict[RoborockZeoProtocol, Any]:
             self._dps_cache[int(dp)] = v
         return {proto: self._convert_value(proto, dps.get(proto)) for proto in dps}
 
+    async def pause(self) -> dict[RoborockZeoProtocol, Any]:
+        """Pause the current programme (DP 201 = "True")."""
+        dps = {RoborockZeoProtocol.PAUSE: "True"}
+        result = await send_decoded_command(self._channel, dps)
+        self._dps_cache[int(RoborockZeoProtocol.PAUSE)] = 1
+        return result
+
     async def resume(self) -> dict[RoborockZeoProtocol, Any]:
-        """Resume a paused programme or start without rebundling).
-        Only works while the device is powered on."""
-        _LOGGER.debug("Resume command")
+        """Start/continue a paused programme (DP 200 = "True").
+           Only works while the device is powered on.
+        """
         dps = {RoborockZeoProtocol.START: "True"}
-        result = await send_decoded_command(
-            self._channel,
-            dps,
-        )
+        result = await send_decoded_command(self._channel, dps)
         self._dps_cache[int(RoborockZeoProtocol.START)] = 1
         return result
 
+    async def shutdown(self) -> dict[RoborockZeoProtocol, Any]:
+        """Power off the device (DP 202 = "True").
+           Only works while the device is powered on.
+        """
+        dps = {RoborockZeoProtocol.SHUTDOWN: "True"}
+        result = await send_decoded_command(self._channel, dps)
+        self._dps_cache[int(RoborockZeoProtocol.SHUTDOWN)] = 1
+        return result
 
     async def _get_start_params(self) -> ZeoStartParams:
         """Read programme settings, querying the device on cache miss."""
         cache = self._dps_cache
+        wanted = _START_PARAM_DPS_DRYER if self._feature_trait.is_dryer else _START_PARAM_DPS_WASHER
         need_refresh = (
             int(RoborockZeoProtocol.MODE) not in cache
             or int(RoborockZeoProtocol.PROGRAM) not in cache
         )
         if need_refresh:
             raw = await send_decoded_command(
                 self._channel,
-                {RoborockZeoProtocol.ID_QUERY: [RoborockZeoProtocol.MODE, RoborockZeoProtocol.PROGRAM]},
+                {RoborockZeoProtocol.ID_QUERY: wanted},
                 value_encoder=json.dumps,
             )
-            for dp in (RoborockZeoProtocol.MODE, RoborockZeoProtocol.PROGRAM):
+            for dp in wanted:
                 if (val := raw.get(dp)) is not None:
                     cache[int(dp)] = val
         return ZeoStartParams(