Add back promise resolution for transitioning on fitView

2025-03-05 15:00:44 +01:00
parent ee1ef205eb
commit f208223400
6 changed files with 21 additions and 18 deletions
@@ -279,9 +279,11 @@ export function useReactFlow<NodeType extends Node = Node, EdgeType extends Edge
            .connectionLookup.get(`${nodeId}${type ? (handleId ? `-${type}-${handleId}` : `-${type}`) : ''}`)
            ?.values() ?? []
        ),
-      fitView: (options: FitViewOptions<NodeType> | undefined) => {
-        store.setState({ fitViewQueued: true, fitViewOptions: options });
+      fitView: async (options: FitViewOptions<NodeType> | undefined) => {
+        const fitViewResolver = store.getState().fitViewResolver ?? Promise.withResolvers<boolean>();
+        store.setState({ fitViewQueued: true, fitViewOptions: options, fitViewResolver });
        batchContext.nodeQueue.push((nodes) => [...nodes]);
+        return fitViewResolver.promise;
      },
    };
  }, []);