@bian-womp/spark-remote 0.3.1 → 0.3.3

package/lib/cjs/index.cjs

@@ -462,8 +462,14 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             // (e.g., { ui } or { runtime }) without dropping the other ones.
             extData = { ...extData, ...data };
         },
-        getExtData: async () => {
-            return extData;
+        updateExtData: async (updates) => {
+            if (!extData || typeof extData !== "object" || Array.isArray(extData)) {
+                extData = {};
+            }
+            for (const { path, value } of updates) {
+                const pathSegments = sparkGraph.parseJsonPath(path);
+                sparkGraph.setValueAtPathWithCreation(extData, pathSegments, value);
+            }
         },
         snapshot: async () => {
             const inputs = {};
@@ -484,14 +490,12 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         },
         snapshotFull: async () => {
             const snap = await originalAdapter.snapshot();
-            const extData = await originalAdapter.getExtData();
             const env = graphRuntime?.getEnvironment?.() ?? {};
             const def = graphRuntime?.getGraphDef();
             return {
+                ...snap,
                 def,
                 environment: env,
-                inputs: snap.inputs,
-                outputs: snap.outputs,
                 extData,
             };
         },
@@ -557,9 +561,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 +573,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 +592,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 }
             }
             finally {
-                if (opts?.dry && !wasPaused)
-                    graphRuntime.resume();
+                releasePause?.();
             }
         },
         setInputs: (nodeId, inputs, options) => {
@@ -620,7 +620,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) => {
@@ -690,7 +690,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         applyRegistry: wrapMethod("applyRegistry", originalAdapter.applyRegistry),
         build: wrapMethod("build", originalAdapter.build),
         setExtData: wrapMethod("setExtData", originalAdapter.setExtData),
-        getExtData: wrapMethod("getExtData", originalAdapter.getExtData),
+        updateExtData: wrapMethod("updateExtData", originalAdapter.updateExtData),
         snapshot: wrapMethod("snapshot", originalAdapter.snapshot),
         snapshotFull: wrapMethod("snapshotFull", originalAdapter.snapshotFull),
         applySnapshotFull: wrapMethod("applySnapshotFull", originalAdapter.applySnapshotFull),
@@ -769,13 +769,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({
@@ -973,13 +968,11 @@ class RemoteGraphLifecycleApi {
             message: { type: "SetExtData", payload: data },
         });
     }
-    async getExtData() {
+    async updateExtData(updates) {
         const transport = await this.ensureConnected();
-        const res = await transport.request({
-            message: { type: "GetExtData" },
+        await transport.request({
+            message: { type: "UpdateExtData", payload: updates },
         });
-        const payload = res?.message || {};
-        return payload.extData || {};
     }
     async setViewport(viewport) {
         const transport = await this.ensureConnected();
@@ -1531,6 +1524,11 @@ class ServerCommandHandler {
                 ack();
                 break;
             }
+            case "UpdateExtData": {
+                await this.adapter.updateExtData(msg.payload);
+                ack();
+                break;
+            }
             case "Dispose": {
                 this.logCommand("Dispose", env);
                 await this.adapter.dispose();