From c39cd7e2a676ff378390f39eeb9abc9fac3a78f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B6ren=20Beye?= <github@hypfer.de>
Date: Wed, 19 Apr 2023 16:27:28 +0200
Subject: [PATCH] fix(vendor.dreame): Add DreameCarpetModeControlCapability to
 DreameMopValetudoRobots such as the W10

---
 backend/lib/robots/dreame/DreameMopValetudoRobot.js | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/backend/lib/robots/dreame/DreameMopValetudoRobot.js b/backend/lib/robots/dreame/DreameMopValetudoRobot.js
index c941720ff4..adcd7a5355 100644
--- a/backend/lib/robots/dreame/DreameMopValetudoRobot.js
+++ b/backend/lib/robots/dreame/DreameMopValetudoRobot.js
@@ -84,6 +84,12 @@ class DreameMopValetudoRobot extends DreameGen2ValetudoRobot {
             },
         }));
 
+        this.registerCapability(new capabilities.DreameCarpetModeControlCapability({
+            robot: this,
+            siid: DreameGen2ValetudoRobot.MIOT_SERVICES.VACUUM_2.SIID,
+            piid: DreameGen2ValetudoRobot.MIOT_SERVICES.VACUUM_2.PROPERTIES.CARPET_MODE.PIID
+        }));
+
         this.registerCapability(new capabilities.DreameKeyLockCapability({
             robot: this,
             siid: DreameGen2ValetudoRobot.MIOT_SERVICES.VACUUM_2.SIID,