@bian-womp/spark-graph 0.1.20 → 0.1.21

package/lib/cjs/index.cjs

@@ -1,5 +1,34 @@
 'use strict';
 
+function typed(typeId, value) {
+    return { __spark_type: typeId, __spark_value: value };
+}
+function isTypedOutput(v) {
+    return (!!v &&
+        typeof v === "object" &&
+        Object.prototype.hasOwnProperty.call(v, "__spark_type"));
+}
+function getTypedOutputTypeId(v) {
+    if (isTypedOutput(v))
+        return String(v.__spark_type);
+    return undefined;
+}
+function getTypedOutputValue(v) {
+    if (isTypedOutput(v))
+        return v.__spark_value;
+    return v;
+}
+function getInputTypeId(inputs, handle) {
+    const v = inputs ? inputs[handle] : undefined;
+    if (!v)
+        return undefined;
+    return typeof v === "string" ? v : v.typeId;
+}
+function isInputPrivate(inputs, handle) {
+    const v = inputs ? inputs[handle] : undefined;
+    return !!(v && typeof v === "object" && v.private);
+}
+
 class CategoryRegistry {
     constructor() {
         this.categories = new Map();
@@ -372,25 +401,6 @@ class Registry {
     }
 }
 
-function typed(typeId, value) {
-    return { __spark_type: typeId, __spark_value: value };
-}
-function isTypedOutput(v) {
-    return (!!v &&
-        typeof v === "object" &&
-        Object.prototype.hasOwnProperty.call(v, "__spark_type"));
-}
-function getInputTypeId(inputs, handle) {
-    const v = inputs ? inputs[handle] : undefined;
-    if (!v)
-        return undefined;
-    return typeof v === "string" ? v : v.typeId;
-}
-function isInputPrivate(inputs, handle) {
-    const v = inputs ? inputs[handle] : undefined;
-    return !!(v && typeof v === "object" && v.private);
-}
-
 class GraphRuntime {
     constructor() {
         this.nodes = new Map();
@@ -565,12 +575,12 @@ class GraphRuntime {
                 });
                 if (anyAsync) {
                     convertAsync = async (v, signal) => {
-                        if (!isTypedOutput(v))
+                        const typeId = getTypedOutputTypeId(v);
+                        if (!typeId)
                             throw new Error(`Typed output required for union source; allowed: ${srcTypes.join("|")}`);
-                        const typeId = String(v.__spark_type);
                         if (!srcTypes.includes(typeId))
                             throw new Error(`Invalid typed output ${typeId}; allowed: ${srcTypes.join("|")}`);
-                        const payload = v.__spark_value;
+                        const payload = typeId;
                         const res = registry.resolveCoercion(typeId, dstDeclared);
                         if (!res)
                             return payload;
@@ -581,12 +591,12 @@ class GraphRuntime {
                 }
                 else {
                     convert = (v) => {
-                        if (!isTypedOutput(v))
+                        const typeId = getTypedOutputTypeId(v);
+                        if (!typeId)
                             throw new Error(`Typed output required for union source; allowed: ${srcTypes.join("|")}`);
-                        const typeId = String(v.__spark_type);
                         if (!srcTypes.includes(typeId))
                             throw new Error(`Invalid typed output ${typeId}; allowed: ${srcTypes.join("|")}`);
-                        const payload = v.__spark_value;
+                        const payload = typeId;
                         const res = registry.resolveCoercion(typeId, dstDeclared);
                         if (!res)
                             return payload;
@@ -777,9 +787,7 @@ class GraphRuntime {
             handle: srcHandle,
             value,
             io: "output",
-            runtimeTypeId: isTypedOutput(value)
-                ? String(value.__spark_type)
-                : undefined,
+            runtimeTypeId: getTypedOutputTypeId(value),
         });
         // fan-out along all edges from this output
         const outEdges = this.edges.filter((e) => e.source.nodeId === srcNodeId && e.source.handle === srcHandle);
@@ -811,7 +819,7 @@ class GraphRuntime {
                     handle: e.target.handle,
                     value: v,
                     io: "input",
-                    runtimeTypeId: isTypedOutput(v) ? String(v.__spark_type) : undefined,
+                    runtimeTypeId: getTypedOutputTypeId(v),
                 });
                 if (!this.paused && this.allInboundHaveValue(e.target.nodeId))
                     this.scheduleInputsChanged(e.target.nodeId);
@@ -2487,6 +2495,8 @@ exports.createSimpleGraphRegistry = createSimpleGraphRegistry;
 exports.createValidationGraphDef = createValidationGraphDef;
 exports.createValidationGraphRegistry = createValidationGraphRegistry;
 exports.getInputTypeId = getInputTypeId;
+exports.getTypedOutputTypeId = getTypedOutputTypeId;
+exports.getTypedOutputValue = getTypedOutputValue;
 exports.isInputPrivate = isInputPrivate;
 exports.isTypedOutput = isTypedOutput;
 exports.registerDelayNode = registerDelayNode;