Add resetOverlayLayers event listener

stilist · Nov 21, 2017 · 189b03b · 189b03b
1 parent 67f524c
commit 189b03b
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 3 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/dist/index.min.js b/dist/index.min.js
diff --git a/dist/index.min.js.map b/dist/index.min.js.map
diff --git a/src/map.ts b/src/map.ts
@@ -1,6 +1,7 @@
 import {
   control,
   icon,
+  Layer,
   map,
   MapOptions,
   Marker,
@@ -55,8 +56,21 @@ function addMap(id: string = "map", options: MapOptions = {}) {
   // This is necessary because `control.layers(...)` always creates a new layer
   // control -- it's a better user experience to add *all* layers to a single
   // control.
+  const addedOverlayLayers: Layer[] = [];
   instance.on("addOverlayLayer", (event: ICustomLeafletEvent) => {
     layers.addOverlay(event.overlayLayer, event.key);
+    addedOverlayLayers.push(event.overlayLayer);
+  });
+  // There doesn't seem to be a way to determine what layers have been added to
+  // the control.
+  instance.on("resetOverlayLayers", () => {
+    for (const layer of addedOverlayLayers) {
+      layers.removeLayer(layer);
+      instance.removeLayer(layer);
+
+      const index = addedOverlayLayers.indexOf(layer);
+      addedOverlayLayers.splice(index, 1);
+    }
   });
 
   return instance;