chore: update flow hook and store method signatures

alibaba · Dec 31, 2024 · 3cdddff · 3cdddff
1 parent d78ceb6
commit 3cdddff
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/packages/x-flow/src/hooks/useFlow.ts b/packages/x-flow/src/hooks/useFlow.ts
@@ -29,8 +29,8 @@ export const useFlow = () => {
   const setNodes = useMemoizedFn((nodes: FlowNode[]) => {
     storeApi.getState().setNodes(nodes);
   });
-  const addNodes = useMemoizedFn((nodes: FlowNode[]) => {
-    storeApi.getState().addNodes(nodes);
+  const addNodes = useMemoizedFn((nodes: FlowNode[], isVanilla = true) => {
+    storeApi.getState().addNodes(nodes, isVanilla);
   });
   const setEdges = useMemoizedFn((edges: Edge[]) => {
     storeApi.getState().setEdges(edges);

diff --git a/packages/x-flow/src/models/store.ts b/packages/x-flow/src/models/store.ts
@@ -39,7 +39,7 @@ export type FlowState = {
   onConnect: OnConnect;
   setNodes: (nodes: FlowNode[]) => void;
   setEdges: (edges: Edge[]) => void;
-  addNodes: (nodes: FlowNode[]| FlowNode) => void;
+  addNodes: (nodes: FlowNode[]| FlowNode, isVanilla?: boolean) => void;
   addEdges: (edges: Edge[] | Edge) => void;
   deleteNode: (nodeId: string) => void;
   copyNode: (nodeId: string) => void;
@@ -96,9 +96,9 @@ const createStore = (initProps?: Partial<FlowProps>) => {
         getEdges: () => {
           return get().nodes;
         },
-        addNodes: payload => {
+        addNodes: (payload, isVanilla = true) => {
           const newNodes = get().nodes.concat(payload);
-          set({ nodes: newNodes });
+          set({ nodes: isVanilla ? newNodes :  transformNodes(newNodes) });
         },
         addEdges: payload => {
           set({ edges: get().edges.concat(payload) });