From 9565689dfc76b4b3aaf713163744a2707925f32d Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Wed, 6 Mar 2024 18:06:34 +0100
Subject: [PATCH] fix(react): select nodes with selection in uncontrolled flows

---
 packages/react/src/container/Pane/index.tsx | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/packages/react/src/container/Pane/index.tsx b/packages/react/src/container/Pane/index.tsx
index 414a4745..dfb240d0 100644
--- a/packages/react/src/container/Pane/index.tsx
+++ b/packages/react/src/container/Pane/index.tsx
@@ -128,7 +128,8 @@ export function Pane({
   };
 
   const onMouseMove = (event: ReactMouseEvent): void => {
-    const { userSelectionRect, edges, transform, nodeOrigin, nodes, onNodesChange, onEdgesChange } = store.getState();
+    const { userSelectionRect, edges, transform, nodeOrigin, nodes, triggerNodeChanges, triggerEdgeChanges } =
+      store.getState();
     if (!isSelecting || !containerBounds.current || !userSelectionRect) {
       return;
     }
@@ -172,17 +173,13 @@ export function Pane({
     if (prevSelectedNodesCount.current !== selectedNodeIds.size) {
       prevSelectedNodesCount.current = selectedNodeIds.size;
       const changes = getSelectionChanges(nodes, selectedNodeIds, true) as NodeChange[];
-      if (changes.length) {
-        onNodesChange?.(changes);
-      }
+      triggerNodeChanges(changes);
     }
 
     if (prevSelectedEdgesCount.current !== selectedEdgeIds.size) {
       prevSelectedEdgesCount.current = selectedEdgeIds.size;
       const changes = getSelectionChanges(edges, selectedEdgeIds) as EdgeChange[];
-      if (changes.length) {
-        onEdgesChange?.(changes);
-      }
+      triggerEdgeChanges(changes);
     }
 
     store.setState({