feat(vendor.viomi): Implement DND capability (#877)

Hypfer · Apr 29, 2021 · dd05c51 · dd05c51
1 parent 88322b1
commit dd05c51
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 0 deletions.
diff --git a/lib/robots/viomi/ViomiValetudoRobot.js b/lib/robots/viomi/ViomiValetudoRobot.js
@@ -117,6 +117,10 @@ class ViomiValetudoRobot extends MiioValetudoRobot {
             robot: this
         }));
 
+        this.registerCapability(new capabilities.ViomiDoNotDisturbCapability({
+            robot: this
+        }));
+
         this.registerCapability(new capabilities.ViomiManualControlCapability({
             robot: this
         }));

diff --git a/lib/robots/viomi/capabilities/ViomiDoNotDisturbCapability.js b/lib/robots/viomi/capabilities/ViomiDoNotDisturbCapability.js
@@ -0,0 +1,45 @@
+const DoNotDisturbCapability = require("../../../core/capabilities/DoNotDisturbCapability");
+const ValetudoDNDConfiguration = require("../../../entities/core/ValetudoDNDConfiguration");
+
+/**
+ * @extends DoNotDisturbCapability<import("../ViomiValetudoRobot")>
+ */
+class ViomiDoNotDisturbCapability extends DoNotDisturbCapability {
+    /**
+     *
+     * @abstract
+     * @returns {Promise<ValetudoDNDConfiguration>}
+     */
+    async getDndConfiguration() {
+        const res = await this.robot.sendCommand("get_notdisturb", []);
+
+        return new ValetudoDNDConfiguration({
+            enabled: (res[0] === 1),
+            start: {
+                hour: res[1],
+                minute: res[2]
+            },
+            end: {
+                hour: res[3],
+                minute: res[4]
+            }
+        });
+    }
+
+    /**
+     * @abstract
+     * @param {ValetudoDNDConfiguration} dndConfig
+     * @returns {Promise<void>}
+     */
+    async setDndConfiguration(dndConfig) {
+        return this.robot.sendCommand("set_notdisturb", [
+            dndConfig.enabled ? 1 : 0,
+            dndConfig.start.hour,
+            dndConfig.start.minute,
+            dndConfig.end.hour,
+            dndConfig.end.minute
+        ]);
+    }
+}
+
+module.exports = ViomiDoNotDisturbCapability;
diff --git a/lib/robots/viomi/capabilities/index.js b/lib/robots/viomi/capabilities/index.js
@@ -4,6 +4,7 @@ module.exports = {
     ViomiCombinedVirtualRestrictionsCapability: require("./ViomiCombinedVirtualRestrictionsCapability"),
     ViomiConsumableMonitoringCapability: require("./ViomiConsumableMonitoringCapability"),
     ViomiDebugCapability: require("./ViomiDebugCapability"),
+    ViomiDoNotDisturbCapability: require("./ViomiDoNotDisturbCapability"),
     ViomiFanSpeedControlCapability: require("./ViomiFanSpeedControlCapability"),
     ViomiLocateCapability: require("./ViomiLocateCapability"),
     ViomiManualControlCapability: require("./ViomiManualControlCapability"),