feat(vendor.dreame): D10s Pro/Plus also support Obstacle Images

Hypfer · Oct 2, 2024 · 63279c4 · 63279c4
1 parent 111b53a
commit 63279c4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/backend/lib/robots/dreame/DreameD10SPlusValetudoRobot.js b/backend/lib/robots/dreame/DreameD10SPlusValetudoRobot.js
@@ -8,6 +8,7 @@ const fs = require("fs");
 const MiioValetudoRobot = require("../MiioValetudoRobot");
 const QuirksCapability = require("../../core/capabilities/QuirksCapability");
 const ValetudoSelectionPreset = require("../../entities/core/ValetudoSelectionPreset");
+const {IMAGE_FILE_FORMAT} = require("../../utils/const");
 
 const stateAttrs = entities.state.attributes;
 
@@ -95,6 +96,15 @@ class DreameD10SPlusValetudoRobot extends DreameGen2LidarValetudoRobot {
             },
         }));
 
+        this.registerCapability(new capabilities.DreameObstacleImagesCapability({
+            robot: this,
+            fileFormat: IMAGE_FILE_FORMAT.JPG,
+            dimensions: {
+                width: 672,
+                height: 504
+            }
+        }));
+
         [
             capabilities.DreameCarpetModeControlCapability,
             capabilities.DreameKeyLockCapability,

diff --git a/backend/lib/robots/dreame/DreameD10SProValetudoRobot.js b/backend/lib/robots/dreame/DreameD10SProValetudoRobot.js
@@ -8,6 +8,7 @@ const fs = require("fs");
 const MiioValetudoRobot = require("../MiioValetudoRobot");
 const QuirksCapability = require("../../core/capabilities/QuirksCapability");
 const ValetudoSelectionPreset = require("../../entities/core/ValetudoSelectionPreset");
+const {IMAGE_FILE_FORMAT} = require("../../utils/const");
 
 const stateAttrs = entities.state.attributes;
 
@@ -95,6 +96,15 @@ class DreameD10SProValetudoRobot extends DreameGen2LidarValetudoRobot {
             },
         }));
 
+        this.registerCapability(new capabilities.DreameObstacleImagesCapability({
+            robot: this,
+            fileFormat: IMAGE_FILE_FORMAT.JPG,
+            dimensions: {
+                width: 672,
+                height: 504
+            }
+        }));
+
         [
             capabilities.DreameCarpetModeControlCapability,
             capabilities.DreameKeyLockCapability,