feat(mqtt): Optionally expose ObstacleAvoidanceControlCapability

Hypfer · Jun 8, 2024 · 9866d94 · 9866d94
1 parent 572e3ce
commit 9866d94
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 4 deletions.
diff --git a/backend/lib/doc/Configuration.openapi.json b/backend/lib/doc/Configuration.openapi.json
@@ -284,7 +284,8 @@
                 "CurrentStatisticsCapability",
                 "TotalStatisticsCapability",
                 "SpeakerVolumeControlCapability",
-                "KeyLockCapability"
+                "KeyLockCapability",
+                "ObstacleAvoidanceControlCapability"
               ]
             }
           }

diff --git a/backend/lib/mqtt/capabilities/ObstacleAvoidanceControlCapabilityMqttHandle.js b/backend/lib/mqtt/capabilities/ObstacleAvoidanceControlCapabilityMqttHandle.js
@@ -0,0 +1,8 @@
+const SimpleToggleCapabilityMqttHandle = require("./SimpleToggleCapabilityMqttHandle");
+
+
+class ObstacleAvoidanceControlCapabilityMqttHandle extends SimpleToggleCapabilityMqttHandle {}
+
+ObstacleAvoidanceControlCapabilityMqttHandle.OPTIONAL = true;
+
+module.exports = ObstacleAvoidanceControlCapabilityMqttHandle;
diff --git a/backend/lib/mqtt/capabilities/SimpleToggleCapabilityMqttHandle.js b/backend/lib/mqtt/capabilities/SimpleToggleCapabilityMqttHandle.js
@@ -76,11 +76,13 @@ class SimpleToggleCapabilityMqttHandle extends CapabilityMqttHandle {
 }
 
 const CAPABILITIES_TO_FRIENDLY_NAME_MAPPING = {
-    [capabilities.KeyLockCapability.TYPE]: "Lock keys",
+    [capabilities.KeyLockCapability.TYPE]: "Lock Keys",
+    [capabilities.ObstacleAvoidanceControlCapability.TYPE]: "Obstacle Avoidance",
 };
 
 const CAPABILITIES_TO_ICON_MAPPING = {
     [capabilities.KeyLockCapability.TYPE]: "mdi:lock",
+    [capabilities.ObstacleAvoidanceControlCapability.TYPE]: "mdi:cable-data",
 };
 
 module.exports = SimpleToggleCapabilityMqttHandle;
diff --git a/backend/lib/mqtt/capabilities/index.js b/backend/lib/mqtt/capabilities/index.js
@@ -8,6 +8,7 @@ module.exports = {
     KeyLockCapabilityMqttHandle: require("./KeyLockCapabilityMqttHandle"),
     LocateCapabilityMqttHandle: require("./LocateCapabilityMqttHandle"),
     MapSegmentationCapabilityMqttHandle: require("./MapSegmentationCapabilityMqttHandle"),
+    ObstacleAvoidanceControlCapabilityMqttHandle: require("./ObstacleAvoidanceControlCapabilityMqttHandle"),
     PresetSelectionCapabilityMqttHandle: require("./PresetSelectionCapabilityMqttHandle"),
     SpeakerVolumeControlCapabilityMqttHandle: require("./SpeakerVolumeControlCapabilityMqttHandle"),
     TotalStatisticsCapabilityMqttHandle: require("./TotalStatisticsCapabilityMqttHandle"),

diff --git a/backend/lib/mqtt/handles/HandleMappings.js b/backend/lib/mqtt/handles/HandleMappings.js
@@ -18,7 +18,8 @@ const CAPABILITY_TYPE_TO_HANDLE_MAPPING = {
     [capabilities.CurrentStatisticsCapability.TYPE]: capabilityHandles.CurrentStatisticsCapabilityMqttHandle,
     [capabilities.TotalStatisticsCapability.TYPE]: capabilityHandles.TotalStatisticsCapabilityMqttHandle,
     [capabilities.SpeakerVolumeControlCapability.TYPE]: capabilityHandles.SpeakerVolumeControlCapabilityMqttHandle,
-    [capabilities.KeyLockCapability.TYPE]: capabilityHandles.KeyLockCapabilityMqttHandle
+    [capabilities.KeyLockCapability.TYPE]: capabilityHandles.KeyLockCapabilityMqttHandle,
+    [capabilities.ObstacleAvoidanceControlCapability.TYPE]: capabilityHandles.ObstacleAvoidanceControlCapabilityMqttHandle
 };
 
 const STATUS_ATTR_TO_HANDLE_MAPPING = [

diff --git a/frontend/src/robot/RobotOptions.tsx b/frontend/src/robot/RobotOptions.tsx
@@ -86,7 +86,7 @@ const KeyLockCapabilitySwitchListMenuItem = () => {
             }}
             disabled={disabled}
             loadError={isError}
-            primaryLabel={"Lock keys"}
+            primaryLabel={"Lock Keys"}
             secondaryLabel={"Prevents the robot from being operated using its physical buttons."}
             icon={<KeyLockIcon/>}
         />