remove location to multiple devices

KaplanOpenSource · Dec 17, 2024 · ae2d36f · ae2d36f
1 parent 05825e4
commit ae2d36f
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 10 deletions.
diff --git a/client/src/DeviceTable/DeviceTable.jsx b/client/src/DeviceTable/DeviceTable.jsx
@@ -112,6 +112,7 @@ const DeviceTableSmall = ({ shownDevices }) => {
                         <DeviceItemLocationButton
                             deviceType={deviceType}
                             deviceItem={deviceItem}
+                            surroundingDevices={shownDevices}
                         />
                     </Stack>
                 </Paper>

diff --git a/client/src/Experiment/DeviceItem.jsx b/client/src/Experiment/DeviceItem.jsx
@@ -46,6 +46,7 @@ export const DeviceItem = ({ data, setData, deviceType, showAttributes, devicesE
                         deviceType={deviceType}
                         deviceItem={data}
                         hasLocation={hasLocation}
+                        surroundingDevices={devicesEnclosingList}
                     />
                 </>
             }

diff --git a/client/src/Experiment/DeviceItemLocationButton.jsx b/client/src/Experiment/DeviceItemLocationButton.jsx
@@ -3,8 +3,9 @@ import { ButtonTooltip } from "../Utils/ButtonTooltip"
 import { useContext } from "react";
 import { experimentContext } from "../Context/ExperimentProvider";
 import { RealMapName } from "../constants/constants";
+import { ContextMenu } from "../Utils/ContextMenu";
 
-export const DeviceItemLocationButton = ({ deviceType, deviceItem, hasLocation }) => {
+export const DeviceItemLocationButton = ({ deviceType, deviceItem, hasLocation, surroundingDevices, }) => {
     const { currTrial, setLocationsToDevices } = useContext(experimentContext);
     let hasLocationCalced = hasLocation;
     if (hasLocation === undefined) {
@@ -15,17 +16,39 @@ export const DeviceItemLocationButton = ({ deviceType, deviceItem, hasLocation }
         hasLocationCalced = deviceTrial && deviceTrial.location && deviceTrial.location.coordinates;
     }
 
+    const removeLocation = () => {
+        setLocationsToDevices([{ deviceTypeName: deviceType.name, deviceItemName: deviceItem.name }], [undefined]);
+    }
+
+    const menuItems = [
+        { label: 'Remove location', callback: removeLocation },
+    ];
+
+    if (surroundingDevices) {
+        menuItems.push({
+            label: 'Remove locations to all devices in list', callback: () => {
+                const devicesToRemove = surroundingDevices.map(d => {
+                    return { deviceTypeName: d.deviceType.name, deviceItemName: d.deviceItem.name };
+                });
+                const locationsEmpty = devicesToRemove.map(() => undefined);
+                setLocationsToDevices(devicesToRemove, locationsEmpty);
+            }
+        })
+    }
+
     return (
         <>
-            {currTrial.trial &&
-                <ButtonTooltip
-                    tooltip={hasLocationCalced ? "Remove location" : "Has no location"}
-                    onClick={() => setLocationsToDevices([{ deviceTypeName: deviceType.name, deviceItemName: deviceItem.name }], [undefined])}
-                    disabled={!hasLocationCalced}
-                >
-                    {hasLocationCalced ? <LocationOff /> : <LocationOffOutlined />}
-                </ButtonTooltip>
-            }
+            {currTrial.trial && (
+                <ContextMenu menuItems={menuItems}>
+                    <ButtonTooltip
+                        tooltip={hasLocationCalced ? "Remove location" : "Has no location"}
+                        onClick={removeLocation}
+                        disabled={!hasLocationCalced}
+                    >
+                        {hasLocationCalced ? <LocationOff /> : <LocationOffOutlined />}
+                    </ButtonTooltip>
+                </ContextMenu>
+            )}
         </>
     )
 }