@bian-womp/spark-remote 0.3.2 → 0.3.3

package/lib/esm/index.js

@@ -1,5 +1,5 @@
 import { WebSocket } from 'ws';
-import { GraphBuilder, LocalEngine } from '@bian-womp/spark-graph';
+import { GraphBuilder, parseJsonPath, setValueAtPathWithCreation, LocalEngine } from '@bian-womp/spark-graph';
 
 class SeqGenerator {
     constructor() {
@@ -460,8 +460,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 = parseJsonPath(path);
+                setValueAtPathWithCreation(extData, pathSegments, value);
+            }
         },
         snapshot: async () => {
             const inputs = {};
@@ -482,14 +488,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,
             };
         },
@@ -684,7 +688,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),
@@ -962,13 +966,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();
@@ -1520,6 +1522,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();