From 164d2f891f3c74af689d3898ddc949516eb480ff Mon Sep 17 00:00:00 2001
From: Xun Li <lixun910@gmail.com>
Date: Tue, 7 May 2024 11:32:07 -0700
Subject: [PATCH] Support BrushLink part2

---
 src/components/src/editor/editor.tsx | 2 ++
 src/components/src/map-container.tsx | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/components/src/editor/editor.tsx b/src/components/src/editor/editor.tsx
index 340d671a46..ba2d705425 100644
--- a/src/components/src/editor/editor.tsx
+++ b/src/components/src/editor/editor.tsx
@@ -45,6 +45,7 @@ interface EditorProps {
   onSetEditorMode: (m: any) => void;
   onDeleteFeature: (f: Feature) => any;
   onTogglePolygonFilter: (l: Layer, f: Feature) => any;
+  setBrushAndLink?: (layers: Layer[], editFeature: Feature) => any;
 }
 
 export type PortalEditorProps = FeatureActionPanelProps & {
@@ -165,6 +166,7 @@ export default function EditorFactory(
       const {selectedFeature} = editor;
       if (selectedFeature) {
         this.props.onDeleteFeature(selectedFeature);
+        this.props.setBrushAndLink?.(this.props.layers, selectedFeature);
       }
     };
 
diff --git a/src/components/src/map-container.tsx b/src/components/src/map-container.tsx
index e5eb2ba9f4..f9048a4c69 100644
--- a/src/components/src/map-container.tsx
+++ b/src/components/src/map-container.tsx
@@ -1047,6 +1047,7 @@ export default function MapContainerFactory(
             filters={this.polygonFiltersSelector(this.props)}
             layers={layers}
             onDeleteFeature={visStateActions.deleteFeature}
+            setBrushAndLink={visStateActions.setBrushAndLink}
             onSelect={visStateActions.setSelectedFeature}
             onTogglePolygonFilter={visStateActions.setPolygonFilterLayer}
             onSetEditorMode={visStateActions.setEditorMode}