@bian-womp/spark-remote 0.3.66 → 0.3.67

package/lib/cjs/index.cjs

@@ -377,6 +377,43 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
     let graphRuntime;
     let engine;
     let extData = {};
+    const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+    const getCustomNodeDataFromExtData = (data) => {
+        if (!data)
+            return undefined;
+        const custom = isRecord(data.custom) ? data.custom : undefined;
+        if (!custom)
+            return undefined;
+        const nodes = isRecord(custom.nodes) ? custom.nodes : undefined;
+        if (!nodes)
+            return undefined;
+        const sanitized = {};
+        for (const [nodeId, nodeData] of Object.entries(nodes)) {
+            if (isRecord(nodeData))
+                sanitized[nodeId] = nodeData;
+        }
+        return Object.keys(sanitized).length > 0 ? sanitized : undefined;
+    };
+    const areCustomNodeDataEqual = (left, right) => {
+        if (!left && !right)
+            return true;
+        if (!left || !right)
+            return false;
+        return JSON.stringify(left) === JSON.stringify(right);
+    };
+    const syncCustomNodeDataToRuntime = () => {
+        if (!graphRuntime)
+            return;
+        const customNodeData = getCustomNodeDataFromExtData(extData);
+        const existingCustomNodeData = graphRuntime.getCustomNodeData();
+        const sanitizedExisting = Object.keys(existingCustomNodeData).length
+            ? existingCustomNodeData
+            : undefined;
+        if (areCustomNodeDataEqual(sanitizedExisting, customNodeData)) {
+            return;
+        }
+        graphRuntime.setCustomNodeData(customNodeData ?? {});
+    };
     // Helper to get current context
     const getContext = () => ({
         registry,
@@ -469,6 +506,7 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                 environment,
                 startPaused: opts?.startPaused,
             });
+            syncCustomNodeDataToRuntime();
             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 } }));
@@ -478,11 +516,13 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
             if (!data || typeof data !== "object") {
                 // Clear all extData when called with non-object (backward-compatible)
                 extData = {};
+                syncCustomNodeDataToRuntime();
                 return;
             }
             // Merge with existing extData so that callers can update a subset of fields
             // (e.g., { ui } or { runtime }) without dropping the other ones.
             extData = { ...extData, ...data };
+            syncCustomNodeDataToRuntime();
         },
         updateExtData: async (updates) => {
             if (!extData || typeof extData !== "object" || Array.isArray(extData)) {
@@ -496,6 +536,9 @@ async function createServerRuntimeAdapter(id, createRegistry, send, extensions)
                     hasCustomUpdate = true;
                 }
             }
+            if (hasCustomUpdate) {
+                syncCustomNodeDataToRuntime();
+            }
             return hasCustomUpdate;
         },
         snapshot: async () => {