@bian-womp/spark-workbench 0.3.53 → 0.3.55

package/lib/cjs/index.cjs

@@ -3930,6 +3930,7 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
     const [registryErrors, setRegistryErrors] = React.useState([]);
     const [inputValidationErrors, setInputValidationErrors] = React.useState([]);
     const [registryVersion, setRegistryVersion] = React.useState(0);
+    const [registryReady, setRegistryReady] = React.useState(false);
     const clearSystemErrors = React.useCallback(() => setSystemErrors([]), []);
     const clearRegistryErrors = React.useCallback(() => setRegistryErrors([]), []);
     const clearInputValidationErrors = React.useCallback(() => setInputValidationErrors([]), []);
@@ -4627,6 +4628,7 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
         // Registry updates: swap registry and refresh graph validation/UI
         const offRunnerRegistry = runner.on("registry", async (registry) => {
             wb.setRegistry(registry);
+            setRegistryReady(true);
         });
         const offFlowViewport = runner.on("viewport", (event) => {
             const viewport = event.viewport;
@@ -4635,6 +4637,17 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
             }
         });
         const offRunnerTransport = runner.on("transport", (t) => {
+            // Remote registry is only considered ready once we've received at least one "registry" event.
+            // Reset readiness whenever transport is not stable.
+            if (t.state === "local") {
+                setRegistryReady(true);
+            }
+            else if (t.state === "connecting" || t.state === "retrying") {
+                setRegistryReady(false);
+            }
+            else if (t.state === "disconnected") {
+                setRegistryReady(false);
+            }
             if (t.state === "disconnected") {
                 console.info("[WorkbenchContext] Transport disconnected, resetting node status");
                 // Reinitialize node status with invalidated=true for all nodes
@@ -4829,6 +4842,7 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
         triggerExternal,
         uiVersion,
         registryVersion,
+        registryReady,
         overrides,
         getNodeDisplayName,
         setNodeName,
@@ -4870,6 +4884,7 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
         runner,
         uiVersion,
         registryVersion,
+        registryReady,
         overrides,
         getNodeDisplayName,
         setNodeName,
@@ -6092,7 +6107,7 @@ const SelectionBoundOverlay = ({ selection, rfInstance, viewportTick }) => {
 };
 
 const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
-    const { showValues, toString, toElement, getDefaultNodeSize, reactFlowProps, } = props;
+    const { showValues, toString, toElement, getDefaultNodeSize, reactFlowProps, children, } = props;
     const { wb, handlesMap, inputsMap, inputDefaultsMap, outputsMap, outputTypesMap, valuesTick, nodeStatus, edgeStatus, validationByNode, validationByEdge, uiVersion, registryVersion, runner, overrides, runNode, runFromHere, runMode, } = useWorkbenchContext();
     const nodeValidation = validationByNode;
     const edgeValidation = validationByEdge.errors;
@@ -6766,7 +6781,7 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
                             if (userOnInit) {
                                 userOnInit(inst);
                             }
-                        }, onConnect: onConnect, onEdgesChange: onEdgesChange, onEdgesDelete: onEdgesDelete, onNodesDelete: onNodesDelete, onNodesChange: onNodesChange, onMove: onMove, onMoveEnd: onMoveEnd, deleteKeyCode: ["Backspace", "Delete"], proOptions: { hideAttribution: true }, noDragClassName: "wb-nodrag", noWheelClassName: "wb-nowheel", noPanClassName: "wb-nopan", children: [BackgroundRenderer ? (jsxRuntime.jsx(BackgroundRenderer, {})) : (jsxRuntime.jsx(react.Background, { id: "workbench-canvas-background", variant: react.BackgroundVariant.Dots, gap: 12, size: 1 })), MinimapRenderer ? jsxRuntime.jsx(MinimapRenderer, {}) : jsxRuntime.jsx(react.MiniMap, {}), ControlsRenderer ? jsxRuntime.jsx(ControlsRenderer, {}) : jsxRuntime.jsx(react.Controls, {}), menuState?.type === "default" &&
+                        }, onConnect: onConnect, onEdgesChange: onEdgesChange, onEdgesDelete: onEdgesDelete, onNodesDelete: onNodesDelete, onNodesChange: onNodesChange, onMove: onMove, onMoveEnd: onMoveEnd, deleteKeyCode: ["Backspace", "Delete"], proOptions: { hideAttribution: true }, noDragClassName: "wb-nodrag", noWheelClassName: "wb-nowheel", noPanClassName: "wb-nopan", children: [children, BackgroundRenderer ? (jsxRuntime.jsx(BackgroundRenderer, {})) : (jsxRuntime.jsx(react.Background, { id: "workbench-canvas-background", variant: react.BackgroundVariant.Dots, gap: 12, size: 1 })), MinimapRenderer ? jsxRuntime.jsx(MinimapRenderer, {}) : jsxRuntime.jsx(react.MiniMap, {}), ControlsRenderer ? jsxRuntime.jsx(ControlsRenderer, {}) : jsxRuntime.jsx(react.Controls, {}), menuState?.type === "default" &&
                                 (DefaultContextMenuRenderer ? (jsxRuntime.jsx(DefaultContextMenuRenderer, { open: true, clientPos: menuState.menuPos, handlers: defaultContextMenuHandlers, registry: wb.registry, nodeIds: nodeIds, keyboardShortcuts: keyboardShortcuts })) : (jsxRuntime.jsx(DefaultContextMenu, { open: true, clientPos: menuState.menuPos, handlers: defaultContextMenuHandlers, registry: wb.registry, nodeIds: nodeIds, keyboardShortcuts: keyboardShortcuts }))), menuState?.type === "node" &&
                                 nodeContextMenuHandlers &&
                                 (NodeContextMenuRenderer ? (jsxRuntime.jsx(NodeContextMenuRenderer, { open: true, clientPos: menuState.menuPos, nodeId: menuState.nodeId, handlers: nodeContextMenuHandlers, bakeableOutputs: bakeableOutputs, runMode: runMode, wb: wb, keyboardShortcuts: keyboardShortcuts })) : (jsxRuntime.jsx(NodeContextMenu, { open: true, clientPos: menuState.menuPos, nodeId: menuState.nodeId, handlers: nodeContextMenuHandlers, bakeableOutputs: bakeableOutputs, runMode: runMode }))), menuState?.type === "selection" &&