@bian-womp/spark-remote 0.3.9 → 0.3.11

package/lib/cjs/index.cjs

@@ -398,7 +398,9 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             return await resolved.convertAsync(value, ac.signal);
         },
         getEnvironment: async () => {
-            return graphRuntime?.getEnvironment?.() ?? {};
+            if (!graphRuntime)
+                return {};
+            return graphRuntime.getEnvironment();
         },
         applyRegistry: async (deltas) => {
             // Pause runtime if exists
@@ -445,8 +447,8 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             });
         },
         build: async (def, opts) => {
-            const env = opts?.environment || {};
-            graphRuntime = builder.build(def, { environment: env });
+            const environment = opts?.environment || {};
+            graphRuntime = builder.build(def, { environment });
             graphRuntime.on("value", (p) => send({ message: { type: "value", payload: p } }));
             graphRuntime.on("invalidate", (p) => send({ message: { type: "invalidate", payload: p } }));
             graphRuntime.on("error", (p) => send({ message: { type: "error", payload: p } }));
@@ -489,15 +491,12 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             return { inputs, outputs };
         },
         snapshotFull: async () => {
+            if (!graphRuntime)
+                return { inputs: {}, outputs: {} };
             const snap = await originalAdapter.snapshot();
-            const env = graphRuntime?.getEnvironment?.() ?? {};
-            const def = graphRuntime?.getGraphDef();
-            return {
-                ...snap,
-                def,
-                environment: env,
-                extData,
-            };
+            const environment = graphRuntime.getEnvironment();
+            const def = graphRuntime.getGraphDef();
+            return { ...snap, def, environment, extData };
         },
         applySnapshotFull: async (payload, options) => {
             const def = payload.def;
@@ -513,7 +512,9 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 await originalAdapter.build(def, { environment: payload.environment });
             }
             // Hydrate inputs/outputs exactly, then re-emit outputs without scheduling runs
-            graphRuntime?.hydrate({
+            if (!graphRuntime)
+                return;
+            graphRuntime.hydrate({
                 inputs: payload.inputs,
                 outputs: payload.outputs,
             });
@@ -715,6 +716,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
         cancelNodeRuns: wrapMethod("cancelNodeRuns", originalAdapter.cancelNodeRuns),
         getOutput: wrapMethod("getOutput", originalAdapter.getOutput),
         on: wrapMethod("on", originalAdapter.on),
+        orignalDispose: originalAdapter.dispose,
     };
     return extendedAdapter;
 }