@cascade-flow/runner 0.2.6 → 0.2.7

package/dist/index.js

@@ -33,6 +33,7 @@ function getFailurePath(dir, requestId) {
 var EMBEDDED_STEP_EXECUTOR = `
 import { pathToFileURL } from "node:url";
 import { join } from "node:path";
+import { AsyncLocalStorage } from "async_hooks";
 
 function serializeError(err) {
   if (err instanceof Error) {
@@ -41,6 +42,9 @@ function serializeError(err) {
   return { message: String(err), name: "Error" };
 }
 
+// AsyncLocalStorage for checkpoint context
+const checkpointStorage = new AsyncLocalStorage();
+
 function createCheckpointFunction(checkpointDir) {
   const checkpointSequences = new Map();
 
@@ -72,9 +76,16 @@ function createCheckpointFunction(checkpointDir) {
       return JSON.parse(response.data);
     }
 
-    // Cache miss: execute function
+    // Cache miss: execute function with parent injection
     try {
-      const result = await fn();
+      const ctx = checkpointStorage.getStore();
+      let result = await fn();
+
+      // Inject parent checkpoint info if in a scope
+      if (ctx?.parent && typeof result === "object" && result !== null) {
+        result = { _parentCheckpoint: ctx.parent, ...result };
+      }
+
       await Bun.write(
         join(checkpointDir, "result-" + requestId + ".json"),
         JSON.stringify({ requestId, data: JSON.stringify(result) })
@@ -96,6 +107,11 @@ function createCheckpointFunction(checkpointDir) {
   };
 }
 
+// Run with checkpoint context
+function runWithCheckpoint(checkpointFn, fn) {
+  return checkpointStorage.run({ checkpoint: checkpointFn }, fn);
+}
+
 async function main() {
   try {
     const inputFile = process.env.CF_STEP_INPUT_FILE;
@@ -123,7 +139,10 @@ async function main() {
       throw new Error("Invalid step module at " + stepPath);
     }
 
-    const result = await stepDef.fn({ dependencies, ctx: reconstructedCtx });
+    // Execute step with checkpoint context
+    const result = await runWithCheckpoint(checkpoint, () =>
+      stepDef.fn({ dependencies, ctx: reconstructedCtx })
+    );
     await Bun.write(outputFile, JSON.stringify(result, null, 2));
     process.exit(0);
   } catch (error) {
@@ -13163,4 +13182,4 @@ export {
   calculateWorkflowHash
 };
 
-//# debugId=922C5B57E7B7118364756E2164756E21
+//# debugId=1B7890ED6FF9B69B64756E2164756E21