@bian-womp/spark-workbench 0.2.66 → 0.2.68

package/lib/cjs/index.cjs

@@ -131,6 +131,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
             edges: [],
         };
         this.viewport = null;
+        this.copiedData = null;
     }
     setRegistry(registry) {
         this.registry = registry;
@@ -519,6 +520,18 @@ class InMemoryWorkbench extends AbstractWorkbench {
         });
         return { nodeIdMap, edgeIds };
     }
+    /**
+     * Get the currently copied graph data.
+     */
+    getCopiedData() {
+        return this.copiedData;
+    }
+    /**
+     * Set the copied graph data (used for paste functionality).
+     */
+    setCopiedData(data) {
+        this.copiedData = data;
+    }
 }
 
 class CLIWorkbench {
@@ -4380,15 +4393,39 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
     }, [runner]);
     const nodeIds = React.useMemo(() => Array.from(registry.nodes.keys()), [registry, registryVersion]);
     const defaultContextMenuHandlers = React.useMemo(() => {
-        const baseHandlers = createDefaultContextMenuHandlers(addNodeAt, onCloseMenu);
+        // Get storage from override or use workbench's internal storage
+        const storage = overrides?.getCopiedDataStorage
+            ? overrides.getCopiedDataStorage()
+            : {
+                get: () => wb.getCopiedData(),
+                set: (data) => wb.setCopiedData(data),
+            };
+        const baseHandlers = createDefaultContextMenuHandlers(addNodeAt, onCloseMenu, 
+        // Paste handler - checks storage dynamically when called
+        // Only provide handler if storage has data or might have data (for dynamic checking)
+        (position) => {
+            const data = storage.get();
+            if (!data)
+                return;
+            wb.pasteCopiedData(data, position);
+            onCloseMenu();
+        });
         if (overrides?.getDefaultContextMenuHandlers) {
             return overrides.getDefaultContextMenuHandlers(wb, baseHandlers);
         }
         return baseHandlers;
     }, [addNodeAt, onCloseMenu, overrides, wb]);
     const selectionContextMenuHandlers = React.useMemo(() => {
-        const baseHandlers = createSelectionContextMenuHandlers(wb, onCloseSelectionMenu, overrides?.getDefaultNodeSize, undefined, // onCopyResult - will be provided by overrides
-        runner);
+        // Get storage from override or use workbench's internal storage
+        const storage = overrides?.getCopiedDataStorage
+            ? overrides.getCopiedDataStorage()
+            : {
+                get: () => wb.getCopiedData(),
+                set: (data) => wb.setCopiedData(data),
+            };
+        const baseHandlers = createSelectionContextMenuHandlers(wb, onCloseSelectionMenu, overrides?.getDefaultNodeSize, (data) => {
+            storage.set(data);
+        }, runner);
         if (overrides?.getSelectionContextMenuHandlers) {
             const selection = wb.getSelection();
             return overrides.getSelectionContextMenuHandlers(wb, selection, baseHandlers, {
@@ -4400,7 +4437,16 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
     const nodeContextMenuHandlers = React.useMemo(() => {
         if (!nodeAtMenu)
             return null;
-        const baseHandlers = createNodeContextMenuHandlers(nodeAtMenu, wb, runner, registry, outputsMap, outputTypesMap, onCloseNodeMenu, overrides?.getDefaultNodeSize);
+        // Get storage from override or use workbench's internal storage
+        const storage = overrides?.getCopiedDataStorage
+            ? overrides.getCopiedDataStorage()
+            : {
+                get: () => wb.getCopiedData(),
+                set: (data) => wb.setCopiedData(data),
+            };
+        const baseHandlers = createNodeContextMenuHandlers(nodeAtMenu, wb, runner, registry, outputsMap, outputTypesMap, onCloseNodeMenu, overrides?.getDefaultNodeSize, (data) => {
+            storage.set(data);
+        });
         if (overrides?.getNodeContextMenuHandlers) {
             return overrides.getNodeContextMenuHandlers(wb, nodeAtMenu, baseHandlers);
         }
@@ -4415,6 +4461,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         onCloseNodeMenu,
         overrides?.getDefaultNodeSize,
         overrides?.getNodeContextMenuHandlers,
+        overrides?.getCopiedDataStorage,
     ]);
     const canRunPull = React.useMemo(() => engineKind()?.toString() === "pull", [engineKind]);
     const bakeableOutputs = React.useMemo(() => {