@bian-womp/spark-remote 0.3.0 → 0.3.2

package/lib/cjs/index.cjs

@@ -557,9 +557,8 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         update: async (def, options) => {
             if (!graphRuntime)
                 return;
-            const wasPaused = graphRuntime.isPaused();
-            if (options?.dry && !wasPaused)
-                graphRuntime.pause();
+            // Use requestPause for dry mode to temporarily pause without affecting base run mode
+            const releasePause = options?.dry ? graphRuntime.requestPause() : null;
             try {
                 graphRuntime.update(def, registry);
                 send({
@@ -570,16 +569,14 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 });
             }
             finally {
-                if (options?.dry && !wasPaused)
-                    graphRuntime.resume();
+                releasePause?.();
             }
         },
         setEnvironment: async (env, opts) => {
             if (!graphRuntime)
                 return;
-            const wasPaused = graphRuntime.isPaused();
-            if (opts?.dry && !wasPaused)
-                graphRuntime.pause();
+            // Use requestPause for dry mode to temporarily pause without affecting base run mode
+            const releasePause = opts?.dry ? graphRuntime.requestPause() : null;
             try {
                 if (opts?.merge) {
                     const current = graphRuntime.getEnvironment();
@@ -591,8 +588,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 }
             }
             finally {
-                if (opts?.dry && !wasPaused)
-                    graphRuntime.resume();
+                releasePause?.();
             }
         },
         setInputs: (nodeId, inputs, options) => {
@@ -620,7 +616,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 engine = undefined;
             }
             // Create new engine using shared factory
-            engine = new sparkGraph.UnifiedEngine(graphRuntime, opts?.runMode);
+            engine = new sparkGraph.LocalEngine(graphRuntime, opts?.runMode);
             engine.launch(opts?.invalidate);
         },
         setRunMode: (runMode) => {
@@ -769,13 +765,8 @@ class RemoteEngine {
             this.emit("stats", msg.payload);
         }
     }
-    launch(invalidate, runMode) {
-        this.transport.send({
-            message: {
-                type: "Launch",
-                payload: { runMode, invalidate },
-            },
-        });
+    launch() {
+        throw new Error("RemoteEngine.launch() is not supported. Use GraphLifecycleApi.launch() instead.");
     }
     setRunMode(runMode) {
         this.transport.send({
@@ -1387,7 +1378,7 @@ class ServerCommandHandler {
             case "SetEnvironment": {
                 const dry = msg.dry;
                 this.logCommand("SetEnvironment", env, dry ? { dry: true } : undefined);
-                this.adapter.setEnvironment(msg.payload.environment, {
+                await this.adapter.setEnvironment(msg.payload.environment, {
                     merge: Boolean(msg.payload.merge),
                     dry,
                 });
@@ -1403,7 +1394,7 @@ class ServerCommandHandler {
                 });
                 const nodeId = msg.payload.nodeId;
                 const inputs = decodeInputsFromTransport(msg.payload.inputs);
-                this.adapter.setInputs(nodeId, inputs, { dry });
+                await this.adapter.setInputs(nodeId, inputs, { dry });
                 ack();
                 break;
             }
@@ -1414,7 +1405,7 @@ class ServerCommandHandler {
                     toNodeId: msg.payload.toNodeId,
                     dry: !!dry,
                 });
-                this.adapter.copyOutputs(msg.payload.fromNodeId, msg.payload.toNodeId, {
+                await this.adapter.copyOutputs(msg.payload.fromNodeId, msg.payload.toNodeId, {
                     dry,
                 });
                 ack();
@@ -1427,7 +1418,7 @@ class ServerCommandHandler {
                     event: summarize(msg.payload.event),
                     dry: !!dry,
                 });
-                this.adapter.triggerExternal(msg.payload.nodeId, msg.payload.event, {
+                await this.adapter.triggerExternal(msg.payload.nodeId, msg.payload.event, {
                     dry,
                 });
                 ack();
@@ -1441,14 +1432,14 @@ class ServerCommandHandler {
                     runMode: launchPayload.runMode,
                     invalidate: launchPayload.invalidate,
                 });
-                this.adapter.launch(launchPayload);
+                await this.adapter.launch(launchPayload);
                 ack();
                 break;
             }
             case "SetRunMode": {
                 const setRunModePayload = msg.payload;
                 this.logCommand("SetRunMode", env, setRunModePayload);
-                this.adapter.setRunMode(setRunModePayload.runMode);
+                await this.adapter.setRunMode(setRunModePayload.runMode);
                 ack();
                 break;
             }
@@ -1470,7 +1461,7 @@ class ServerCommandHandler {
             case "CancelNodeRuns": {
                 const cancelPayload = msg.payload;
                 this.logCommand("CancelNodeRuns", env, cancelPayload);
-                this.adapter.cancelNodeRuns(cancelPayload.nodeIds);
+                await this.adapter.cancelNodeRuns(cancelPayload.nodeIds);
                 ack();
                 break;
             }
@@ -1482,19 +1473,19 @@ class ServerCommandHandler {
             }
             case "Snapshot": {
                 this.logCommand("Snapshot", env);
-                const snap = this.adapter.snapshot();
+                const snap = await this.adapter.snapshot();
                 ack({ snapshot: snap });
                 break;
             }
             case "SnapshotFull": {
                 this.logCommand("SnapshotFull", env);
-                const snap = this.adapter.snapshotFull();
+                const snap = await this.adapter.snapshotFull();
                 ack({ snapshot: snap });
                 break;
             }
             case "GetEnvironment": {
                 this.logCommand("GetEnvironment", env);
-                const environment = this.adapter.getEnvironment();
+                const environment = await this.adapter.getEnvironment();
                 ack({ environment });
                 break;
             }
@@ -1516,7 +1507,7 @@ class ServerCommandHandler {
             }
             case "DescribeRegistry": {
                 this.logCommand("DescribeRegistry", env);
-                const desc = this.adapter.describeRegistry();
+                const desc = await this.adapter.describeRegistry();
                 ack({ registry: desc });
                 break;
             }
@@ -1527,13 +1518,13 @@ class ServerCommandHandler {
                 break;
             }
             case "SetExtData": {
-                this.adapter.setExtData(msg.payload);
+                await this.adapter.setExtData(msg.payload);
                 ack();
                 break;
             }
             case "Dispose": {
                 this.logCommand("Dispose", env);
-                this.adapter.dispose();
+                await this.adapter.dispose();
                 ack();
                 break;
             }