Refactor remove/hide layers

Signed-off-by: Vijayan Balasubramanian <[email protected]>
opensearch-project · Jan 31, 2023 · ebabf62 · ebabf62
1 parent a53df97
commit ebabf62
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 44 deletions.
diff --git a/public/components/layer_control_panel/layer_control_panel.tsx b/public/components/layer_control_panel/layer_control_panel.tsx
@@ -48,7 +48,7 @@ import {
 } from '../../model/layerRenderController';
 import { MapState } from '../../model/mapState';
 import { ConfigSchema } from '../../../common/config';
-import {moveLayers, updateLayerVisibility} from "../../model/map/layer_operations";
+import {moveLayers, removeLayers, updateLayerVisibility} from "../../model/map/layer_operations";
 
 interface MaplibreRef {
   current: Maplibre | null;
@@ -300,7 +300,7 @@ export const LayerControlPanel = memo(
 
     const onDeleteLayerConfirm = () => {
       if (selectedDeleteLayer) {
-        layersFunctionMap[selectedDeleteLayer.type]?.remove(maplibreRef, selectedDeleteLayer);
+        removeLayers(maplibreRef.current!, selectedDeleteLayer.id, true);
         removeLayer(selectedDeleteLayer.id);
         setIsDeleteLayerModalVisible(false);
         setSelectedDeleteLayer(undefined);

diff --git a/public/model/OSMLayerFunctions.ts b/public/model/OSMLayerFunctions.ts
@@ -1,7 +1,7 @@
 import { Map as Maplibre, LayerSpecification } from 'maplibre-gl';
 import { OSMLayerSpecification } from './mapLayerType';
 import { getMaplibreBeforeLayerId } from './layersFunctions';
-import { hasLayer } from './map/layer_operations';
+import { hasLayer, removeLayers } from './map/layer_operations';
 
 interface MaplibreRef {
   current: Maplibre | null;
@@ -99,26 +99,8 @@ export const OSMLayerFunctions = {
   ) => {
     // If layer already exist in maplibre source, update layer config
     // else add new layer.
-    if (hasLayer(maplibreRef.current!, layerConfig.id)) {
-      updateLayerConfig(layerConfig, maplibreRef);
-    } else {
-      addNewLayer(layerConfig, maplibreRef, beforeLayerId);
-    }
-  },
-  remove: (maplibreRef: MaplibreRef, layerConfig: OSMLayerSpecification) => {
-    const layers = getCurrentStyleLayers(maplibreRef);
-    layers.forEach((mbLayer: { id: any }) => {
-      if (mbLayer.id.includes(layerConfig.id)) {
-        maplibreRef.current?.removeLayer(mbLayer.id);
-      }
-    });
-  },
-  hide: (maplibreRef: MaplibreRef, layerConfig: OSMLayerSpecification) => {
-    const layers = getCurrentStyleLayers(maplibreRef);
-    layers.forEach((mbLayer: { id: any }) => {
-      if (mbLayer.id.includes(layerConfig.id)) {
-        maplibreRef.current?.setLayoutProperty(mbLayer.id, 'visibility', layerConfig.visibility);
-      }
-    });
+    return hasLayer(maplibreRef.current!, layerConfig.id)
+      ? updateLayerConfig(layerConfig, maplibreRef)
+      : addNewLayer(layerConfig, maplibreRef, beforeLayerId);
   },
 };
diff --git a/public/model/customLayerFunctions.ts b/public/model/customLayerFunctions.ts
@@ -1,7 +1,7 @@
 import { Map as Maplibre, AttributionControl, RasterSourceSpecification } from 'maplibre-gl';
 import { CustomLayerSpecification, OSMLayerSpecification } from './mapLayerType';
 import { getMaplibreBeforeLayerId } from './layersFunctions';
-import { hasLayer } from './map/layer_operations';
+import { hasLayer, removeLayers } from './map/layer_operations';
 
 interface MaplibreRef {
   current: Maplibre | null;
@@ -98,26 +98,11 @@ export const CustomLayerFunctions = {
     layerConfig: CustomLayerSpecification,
     beforeLayerId: string | undefined
   ) => {
-    if (hasLayer(maplibreRef.current!, layerConfig.id)) {
-      updateLayerConfig(layerConfig, maplibreRef);
-    } else {
-      addNewLayer(layerConfig, maplibreRef, beforeLayerId);
-    }
+    return hasLayer(maplibreRef.current!, layerConfig.id)
+      ? updateLayerConfig(layerConfig, maplibreRef)
+      : addNewLayer(layerConfig, maplibreRef, beforeLayerId);
   },
   remove: (maplibreRef: MaplibreRef, layerConfig: OSMLayerSpecification) => {
-    const layers = getCurrentStyleLayers(maplibreRef);
-    layers.forEach((mbLayer: { id: any }) => {
-      if (mbLayer.id.includes(layerConfig.id)) {
-        maplibreRef.current?.removeLayer(mbLayer.id);
-      }
-    });
-  },
-  hide: (maplibreRef: MaplibreRef, layerConfig: OSMLayerSpecification) => {
-    const layers = getCurrentStyleLayers(maplibreRef);
-    layers.forEach((mbLayer: { id: any }) => {
-      if (mbLayer.id.includes(layerConfig.id)) {
-        maplibreRef.current?.setLayoutProperty(mbLayer.id, 'visibility', layerConfig.visibility);
-      }
-    });
+    removeLayers(maplibreRef.current!, layerConfig.id, true);
   },
 };