@bian-womp/spark-remote 0.3.20 → 0.3.21

package/lib/cjs/index.cjs

@@ -520,6 +520,25 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             });
             await originalAdapter.setExtData(payload.extData || {});
         },
+        convertSnapshot: async (converterConfig, options) => {
+            const snapshot = await originalAdapter.snapshotFull();
+            const converter = sparkGraph.buildValueConverter(converterConfig);
+            const converted = sparkGraph.convertSnapshot(snapshot, converter);
+            await originalAdapter.applySnapshotFull(converted, {
+                skipBuild: true,
+            });
+            return converted;
+        },
+        pause: async () => {
+            if (!graphRuntime)
+                return;
+            graphRuntime.pause();
+        },
+        resume: async () => {
+            if (!graphRuntime)
+                return;
+            graphRuntime.resume();
+        },
         cancelNodeRuns: (nodeIds) => {
             if (!graphRuntime)
                 return;
@@ -623,7 +642,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             }
             // Create new engine using shared factory
             engine = new sparkGraph.LocalEngine(graphRuntime, opts?.runMode);
-            engine.launch(opts?.invalidate);
+            engine.launch(opts?.invalidate, opts?.runMode, opts?.startPaused);
         },
         setRunMode: (runMode) => {
             if (!engine)
@@ -696,6 +715,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         snapshot: wrapMethod("snapshot", originalAdapter.snapshot),
         snapshotFull: wrapMethod("snapshotFull", originalAdapter.snapshotFull),
         applySnapshotFull: wrapMethod("applySnapshotFull", originalAdapter.applySnapshotFull),
+        convertSnapshot: wrapMethod("convertSnapshot", originalAdapter.convertSnapshot),
         describeRegistry: wrapMethod("describeRegistry", originalAdapter.describeRegistry),
         update: wrapMethod("update", originalAdapter.update),
         setEnvironment: wrapMethod("setEnvironment", originalAdapter.setEnvironment),
@@ -703,6 +723,8 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         commit: wrapMethod("commit", originalAdapter.commit),
         undo: wrapMethod("undo", originalAdapter.undo),
         redo: wrapMethod("redo", originalAdapter.redo),
+        pause: wrapMethod("pause", originalAdapter.pause),
+        resume: wrapMethod("resume", originalAdapter.resume),
         launch: wrapMethod("launch", originalAdapter.launch),
         // Engine methods
         setInputs: wrapMethod("setInputs", originalAdapter.setInputs),
@@ -926,6 +948,23 @@ class RemoteGraphLifecycleApi {
             },
         });
     }
+    async convertSnapshot(converterConfig, options) {
+        const transport = await this.ensureConnected();
+        const res = await transport.request({
+            message: {
+                type: "ConvertSnapshot",
+                payload: { converterConfig },
+                dry: options?.dry,
+            },
+        });
+        const payload = res?.message || {};
+        return (payload.snapshot || {
+            def: undefined,
+            environment: {},
+            inputs: {},
+            outputs: {},
+        });
+    }
     async describeRegistry() {
         const transport = await this.ensureConnected();
         const res = await transport.request({
@@ -1014,6 +1053,18 @@ class RemoteGraphLifecycleApi {
         const payload = res?.message || {};
         return payload.success ?? false;
     }
+    async pause() {
+        const transport = await this.ensureConnected();
+        await transport.request({
+            message: { type: "Pause" },
+        });
+    }
+    async resume() {
+        const transport = await this.ensureConnected();
+        await transport.request({
+            message: { type: "Resume" },
+        });
+    }
 }
 
 // Node-only transport using a UNIX domain socket.
@@ -1504,6 +1555,14 @@ class ServerCommandHandler {
                 ack();
                 break;
             }
+            case "ConvertSnapshot": {
+                const converterConfig = msg.payload.converterConfig;
+                const dry = msg.dry;
+                this.logCommand("ConvertSnapshot", env, { dry });
+                const converted = await this.adapter.convertSnapshot(converterConfig, { dry });
+                ack({ snapshot: converted });
+                break;
+            }
             case "Coerce": {
                 this.logCommand("Coerce", env, {
                     from: msg.payload.from,
@@ -1559,6 +1618,18 @@ class ServerCommandHandler {
                 ack({ success });
                 break;
             }
+            case "Pause": {
+                this.logCommand("Pause", env);
+                await this.adapter.pause();
+                ack();
+                break;
+            }
+            case "Resume": {
+                this.logCommand("Resume", env);
+                await this.adapter.resume();
+                ack();
+                break;
+            }
             default: {
                 this.logCommand("Unknown type", env);
                 ack();
@@ -1592,7 +1663,7 @@ class ServerCommandHandler {
      * not node execution errors.
      */
     handleError(env, err) {
-        const errorMessage = `error handling command: ${String(err)}`;
+        const errorMessage = `error handling command(${env.message.type}): ${String(err)}`;
         if (this.options?.logHandler) {
             this.options.logHandler("error", errorMessage, {
                 seq: env.seq,