@bian-womp/spark-workbench 0.2.56 → 0.2.57

package/lib/cjs/index.cjs

@@ -32,13 +32,6 @@ class DefaultUIExtensionRegistry {
         }
         return result;
     }
-    registerIconProvider(provider) {
-        this.iconProvider = provider;
-        return this;
-    }
-    getIconProvider() {
-        return this.iconProvider;
-    }
     // React Flow renderers
     registerConnectionLineRenderer(renderer) {
         this.connectionLineRenderer = renderer;
@@ -82,6 +75,35 @@ class DefaultUIExtensionRegistry {
     getNodeContextMenuRenderer() {
         return this.nodeContextMenuRenderer;
     }
+    // Layout function overrides
+    registerEstimateNodeSize(override) {
+        this.estimateNodeSizeOverride = override;
+        return this;
+    }
+    getEstimateNodeSize() {
+        return this.estimateNodeSizeOverride;
+    }
+    registerCreateHandleBounds(override) {
+        this.createHandleBoundsOverride = override;
+        return this;
+    }
+    getCreateHandleBounds() {
+        return this.createHandleBoundsOverride;
+    }
+    registerCreateHandleLayout(override) {
+        this.createHandleLayoutOverride = override;
+        return this;
+    }
+    getCreateHandleLayout() {
+        return this.createHandleLayoutOverride;
+    }
+    registerLayoutNode(override) {
+        this.layoutNodeOverride = override;
+        return this;
+    }
+    getLayoutNode() {
+        return this.layoutNodeOverride;
+    }
 }
 
 class AbstractWorkbench {
@@ -1442,7 +1464,7 @@ function estimateNodeSize(args) {
     const baseWidth = showValues ? 320 : 240;
     const width = overrides?.width ?? baseWidth;
     const height = overrides?.height ?? NODE_HEADER_HEIGHT_PX + rows * NODE_ROW_HEIGHT_PX;
-    return { width, height, inputsCount, outputsCount, rowCount: rows };
+    return { width, height };
 }
 /**
  * Calculate the Y position for handle layout (center of row).
@@ -1499,26 +1521,29 @@ function createHandleLayout(args) {
         y: getHandleLayoutY(args.rowIndex),
     };
 }
-function layoutNode(args) {
-    const { node, registry, showValues, overrides } = args;
+function layoutNode(args, overrides) {
+    const { node, registry, showValues, overrides: sizeOverrides } = args;
     const { inputs, outputs } = computeEffectiveHandles(node, registry);
     const inputOrder = Object.keys(inputs).filter((k) => !sparkGraph.isInputPrivate(inputs, k));
     const outputOrder = Object.keys(outputs);
-    const { width, height } = estimateNodeSize({
+    const estimateNodeSizeFn = overrides?.estimateNodeSize ?? estimateNodeSize;
+    const createHandleBoundsFn = overrides?.createHandleBounds ?? createHandleBounds;
+    const createHandleLayoutFn = overrides?.createHandleLayout ?? createHandleLayout;
+    const { width, height } = estimateNodeSizeFn({
         node,
         registry,
         showValues,
-        overrides,
+        overrides: sizeOverrides,
     });
     const handles = [
-        ...inputOrder.map((id, i) => createHandleBounds({
+        ...inputOrder.map((id, i) => createHandleBoundsFn({
             id,
             type: "target",
             position: react.Position.Left,
             rowIndex: i,
             nodeWidth: width,
         })),
-        ...outputOrder.map((id, i) => createHandleBounds({
+        ...outputOrder.map((id, i) => createHandleBoundsFn({
             id,
             type: "source",
             position: react.Position.Right,
@@ -1527,13 +1552,13 @@ function layoutNode(args) {
         })),
     ];
     const handleLayout = [
-        ...inputOrder.map((id, i) => createHandleLayout({
+        ...inputOrder.map((id, i) => createHandleLayoutFn({
             id,
             type: "target",
             position: react.Position.Left,
             rowIndex: i,
         })),
-        ...outputOrder.map((id, i) => createHandleLayout({
+        ...outputOrder.map((id, i) => createHandleLayoutFn({
             id,
             type: "source",
             position: react.Position.Right,
@@ -1826,15 +1851,32 @@ function toReactFlow(def, positions, registry, opts) {
     // This map is still used later for certain checks; align with valid handles
     const nodeHandleMap = {};
     Object.assign(nodeHandleMap, validHandleMap);
+    // Get layout function overrides from UI registry
+    const layoutNodeOverride = opts.ui?.getLayoutNode() ?? layoutNode;
+    const createHandleBoundsFn = opts.ui?.getCreateHandleBounds() ?? createHandleBounds;
+    const createHandleLayoutFn = opts.ui?.getCreateHandleLayout() ?? createHandleLayout;
+    const estimateNodeSizeFn = opts.ui?.getEstimateNodeSize() ?? estimateNodeSize;
     const nodes = def.nodes.map((n) => {
         const { inputs: inputSource, outputs: outputSource } = computeEffectiveHandles(n, registry);
         const overrideSize = opts.getDefaultNodeSize?.(n.typeId);
-        const geom = layoutNode({
-            node: n,
-            registry,
-            showValues: opts.showValues,
-            overrides: overrideSize,
-        });
+        // If layoutNode is overridden, use it directly; otherwise use default with internal overrides
+        const geom = layoutNodeOverride
+            ? layoutNodeOverride({
+                node: n,
+                registry,
+                showValues: opts.showValues,
+                overrides: overrideSize,
+            })
+            : layoutNode({
+                node: n,
+                registry,
+                showValues: opts.showValues,
+                overrides: overrideSize,
+            }, {
+                estimateNodeSize: estimateNodeSizeFn,
+                createHandleBounds: createHandleBoundsFn,
+                createHandleLayout: createHandleLayoutFn,
+            });
         const inputHandles = geom.inputOrder.map((id) => ({
             id,
             typeId: sparkGraph.getInputTypeId(inputSource, id),
@@ -1853,7 +1895,7 @@ function toReactFlow(def, positions, registry, opts) {
         const extraInputs = Array.from(missingInputsByNode[n.nodeId] || []);
         const extraOutputs = Array.from(missingOutputsByNode[n.nodeId] || []);
         const extraHandleLayoutLeft = extraInputs.map((id, i) => ({
-            ...createHandleLayout({
+            ...createHandleLayoutFn({
                 id,
                 type: "target",
                 position: react.Position.Left,
@@ -1862,7 +1904,7 @@ function toReactFlow(def, positions, registry, opts) {
             missing: true,
         }));
         const extraHandleLayoutRight = extraOutputs.map((id, i) => ({
-            ...createHandleLayout({
+            ...createHandleLayoutFn({
                 id,
                 type: "source",
                 position: react.Position.Right,
@@ -1876,14 +1918,14 @@ function toReactFlow(def, positions, registry, opts) {
             ...extraHandleLayoutRight,
         ];
         // Precompute handle bounds (including missing) so edges can render immediately
-        const missingBoundsLeft = extraInputs.map((id, i) => createHandleBounds({
+        const missingBoundsLeft = extraInputs.map((id, i) => createHandleBoundsFn({
             id,
             type: "target",
             position: react.Position.Left,
             rowIndex: baseLeftCount + i,
             nodeWidth: geom.width,
         }));
-        const missingBoundsRight = extraOutputs.map((id, i) => createHandleBounds({
+        const missingBoundsRight = extraOutputs.map((id, i) => createHandleBoundsFn({
             id,
             type: "source",
             position: react.Position.Right,
@@ -3691,6 +3733,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
             selectedNodeIds: new Set(sel.nodes),
             selectedEdgeIds: new Set(sel.edges),
             getDefaultNodeSize,
+            ui,
         });
         // Retain references for unchanged items
         const stableNodes = retainStabilityById(prevNodesRef.current, out.nodes, isSameNode);