fix(react): uncontrolled flows with Provider

packages/react/src/components/ReactFlowProvider/index.tsx

@@ -26,7 +26,6 @@ export function ReactFlowProvider({
   fitView?: boolean;
 }) {
   const storeRef = useRef<UseBoundStoreWithEqualityFn<StoreApi<ReactFlowState>> | null>(null);
-
   if (!storeRef.current) {
     storeRef.current = createRFStore({
       nodes: initialNodes,

packages/react/src/components/StoreUpdater/index.tsx

@@ -15,6 +15,8 @@ import { initNodeOrigin } from '../../container/ReactFlow';
 const reactFlowFieldsToTrack = [
   'nodes',
   'edges',
+  'defaultNodes',
+  'defaultEdges',
   'onConnect',
   'onConnectStart',
   'onConnectEnd',
@@ -79,16 +81,24 @@ const selector = (s: ReactFlowState) => ({
   setTranslateExtent: s.setTranslateExtent,
   setNodeExtent: s.setNodeExtent,
   reset: s.reset,
+  setDefaultNodesAndEdges: s.setDefaultNodesAndEdges,
 });
 
 export function StoreUpdater(props: StoreUpdaterProps) {
-  const { setNodes, setEdges, setMinZoom, setMaxZoom, setTranslateExtent, setNodeExtent, reset } = useStore(
+  const {
-    selector,
+    setNodes,
-    shallow
+    setEdges,
-  );
+    setMinZoom,
+    setMaxZoom,
+    setTranslateExtent,
+    setNodeExtent,
+    reset,
+    setDefaultNodesAndEdges,
+  } = useStore(selector, shallow);
   const store = useStoreApi();
 
   useEffect(() => {
+    setDefaultNodesAndEdges(props.defaultNodes, props.defaultEdges);
     return () => {
       reset();
     };

packages/react/src/container/ReactFlow/index.tsx

@@ -242,6 +242,8 @@ const ReactFlow = forwardRef<ReactFlowRefType, ReactFlowProps>(
           <StoreUpdater
             nodes={nodes}
             edges={edges}
+            defaultNodes={defaultNodes}
+            defaultEdges={defaultEdges}
             onConnect={onConnect}
             onConnectStart={onConnectStart}
             onConnectEnd={onConnectEnd}

packages/react/src/store/index.ts

@@ -63,6 +63,18 @@ const createRFStore = ({
 
         set({ edges });
       },
+      setDefaultNodesAndEdges: (nodes?: Node[], edges?: Edge[]) => {
+        if (nodes) {
+          const { setNodes } = get();
+          setNodes(nodes);
+          set({ hasDefaultNodes: true });
+        }
+        if (edges) {
+          const { setEdges } = get();
+          setEdges(edges);
+          set({ hasDefaultEdges: true });
+        }
+      },
       // Every node gets registerd at a ResizeObserver. Whenever a node
       // changes its dimensions, this function is called to measure the
       // new dimensions and update the nodes.

packages/react/src/types/store.ts

@@ -168,6 +168,7 @@ export type ReactFlowActions = {
   triggerEdgeChanges: (changes: EdgeChange[] | null) => void;
   panBy: PanBy;
   fitView: (nodes: Node[], options?: FitViewOptions) => boolean;
+  setDefaultNodesAndEdges: (nodes?: Node[], edges?: Edge[]) => void;
 };
 
 export type ReactFlowState = ReactFlowStore & ReactFlowActions;