@bian-womp/spark-graph 0.1.19 → 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);
@@ -2158,6 +2166,57 @@ function setupBasicGraphRegistry() {
             return { Values: out };
         },
     });
+    // Timer
+    registry.registerNode({
+        id: "base.timer",
+        categoryId: "compute",
+        inputs: {
+            Enabled: "base.bool",
+            IntervalMs: "base.float",
+            Immediate: "base.bool",
+        },
+        outputs: { Now: "base.float", Count: "base.float" },
+        inputDefaults: { Enabled: true, IntervalMs: 1000, Immediate: true },
+        impl: (ins, ctx) => {
+            const enabled = Boolean(ins.Enabled);
+            const intervalMs = Math.max(1, Math.trunc(Number(ins.IntervalMs ?? 1000)));
+            const immediate = Boolean(ins.Immediate);
+            const stop = () => {
+                const id = ctx.state.timerId;
+                if (id !== undefined) {
+                    clearInterval(id);
+                    ctx.setState({ timerId: undefined });
+                }
+            };
+            if (!enabled) {
+                stop();
+                return;
+            }
+            // restart timer with new settings
+            stop();
+            let count = 0;
+            if (immediate) {
+                ctx.emit("Now", Date.now());
+                ctx.emit("Count", count);
+                count += 1;
+            }
+            const id = setInterval(() => {
+                ctx.emit("Now", Date.now());
+                ctx.emit("Count", count);
+                count += 1;
+            }, intervalMs);
+            ctx.setState({ timerId: id });
+        },
+        lifecycle: {
+            dispose: (ctx) => {
+                const id = ctx.state.timerId;
+                if (id !== undefined) {
+                    clearInterval(id);
+                    ctx.setState({ timerId: undefined });
+                }
+            },
+        },
+    });
     return registry;
 }
 function registerDelayNode(registry) {
@@ -2436,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;