@upstash/workflow 0.2.8 → 0.2.10-hono-generics

package/h3.js

@@ -20,7 +20,9 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // platforms/h3.ts
 var h3_exports = {};
 __export(h3_exports, {
-  serve: () => serve
+  createWorkflow: () => createWorkflow,
+  serve: () => serve,
+  serveMany: () => serveMany
 });
 module.exports = __toCommonJS(h3_exports);
 
@@ -395,12 +397,13 @@ var WORKFLOW_INIT_HEADER = "Upstash-Workflow-Init";
 var WORKFLOW_URL_HEADER = "Upstash-Workflow-Url";
 var WORKFLOW_FAILURE_HEADER = "Upstash-Workflow-Is-Failure";
 var WORKFLOW_FEATURE_HEADER = "Upstash-Feature-Set";
+var WORKFLOW_INVOKE_COUNT_HEADER = "Upstash-Workflow-Invoke-Count";
 var WORKFLOW_PROTOCOL_VERSION = "1";
 var WORKFLOW_PROTOCOL_VERSION_HEADER = "Upstash-Workflow-Sdk-Version";
 var DEFAULT_CONTENT_TYPE = "application/json";
 var NO_CONCURRENCY = 1;
 var DEFAULT_RETRIES = 3;
-var VERSION = "v0.2.3";
+var VERSION = "v0.2.7";
 var SDK_TELEMETRY = `@upstash/workflow@${VERSION}`;
 var TELEMETRY_HEADER_SDK = "Upstash-Telemetry-Sdk";
 var TELEMETRY_HEADER_FRAMEWORK = "Upstash-Telemetry-Framework";
@@ -447,6 +450,31 @@ var formatWorkflowError = (error) => {
   };
 };
 
+// src/utils.ts
+var NANOID_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
+var NANOID_LENGTH = 21;
+function getRandomInt() {
+  return Math.floor(Math.random() * NANOID_CHARS.length);
+}
+function nanoid() {
+  return Array.from({ length: NANOID_LENGTH }).map(() => NANOID_CHARS[getRandomInt()]).join("");
+}
+function getWorkflowRunId(id) {
+  return `wfr_${id ?? nanoid()}`;
+}
+function decodeBase64(base64) {
+  try {
+    const binString = atob(base64);
+    const intArray = Uint8Array.from(binString, (m) => m.codePointAt(0));
+    return new TextDecoder().decode(intArray);
+  } catch (error) {
+    console.warn(
+      `Upstash Qstash: Failed while decoding base64 "${base64}". Decoding with atob and returning it instead. ${error}`
+    );
+    return atob(base64);
+  }
+}
+
 // src/context/steps.ts
 var BaseLazyStep = class {
   stepName;
@@ -551,8 +579,9 @@ var LazyCallStep = class extends BaseLazyStep {
   headers;
   retries;
   timeout;
+  flowControl;
   stepType = "Call";
-  constructor(stepName, url, method, body, headers, retries, timeout) {
+  constructor(stepName, url, method, body, headers, retries, timeout, flowControl) {
     super(stepName);
     this.url = url;
     this.method = method;
@@ -560,6 +589,7 @@ var LazyCallStep = class extends BaseLazyStep {
     this.headers = headers;
     this.retries = retries;
     this.timeout = timeout;
+    this.flowControl = flowControl;
   }
   getPlanStep(concurrent, targetStep) {
     return {
@@ -627,6 +657,49 @@ var LazyNotifyStep = class extends LazyFunctionStep {
     });
   }
 };
+var LazyInvokeStep = class extends BaseLazyStep {
+  stepType = "Invoke";
+  params;
+  constructor(stepName, {
+    workflow,
+    body,
+    headers = {},
+    workflowRunId,
+    retries,
+    flowControl
+  }) {
+    super(stepName);
+    this.params = {
+      workflow,
+      body,
+      headers,
+      workflowRunId: getWorkflowRunId(workflowRunId),
+      retries,
+      flowControl
+    };
+  }
+  getPlanStep(concurrent, targetStep) {
+    return {
+      stepId: 0,
+      stepName: this.stepName,
+      stepType: this.stepType,
+      concurrent,
+      targetStep
+    };
+  }
+  /**
+   * won't be used as it's the server who will add the result step
+   * in Invoke step.
+   */
+  getResultStep(concurrent, stepId) {
+    return Promise.resolve({
+      stepId,
+      stepName: this.stepName,
+      stepType: this.stepType,
+      concurrent
+    });
+  }
+};
 
 // node_modules/neverthrow/dist/index.es.js
 var defaultErrorConfig = {
@@ -1051,7 +1124,8 @@ var StepTypes = [
   "SleepUntil",
   "Call",
   "Wait",
-  "Notify"
+  "Notify",
+  "Invoke"
 ];
 
 // src/workflow-requests.ts
@@ -1059,8 +1133,9 @@ var import_qstash3 = require("@upstash/qstash");
 var triggerFirstInvocation = async ({
   workflowContext,
   useJSONContent,
-  telemetry,
-  debug
+  telemetry: telemetry2,
+  debug,
+  invokeCount
 }) => {
   const { headers } = getHeaders({
     initHeaderValue: "true",
@@ -1069,7 +1144,9 @@ var triggerFirstInvocation = async ({
     userHeaders: workflowContext.headers,
     failureUrl: workflowContext.failureUrl,
     retries: workflowContext.retries,
-    telemetry
+    telemetry: telemetry2,
+    invokeCount,
+    flowControl: workflowContext.flowControl
   });
   if (workflowContext.headers.get("content-type")) {
     headers["content-type"] = workflowContext.headers.get("content-type");
@@ -1115,8 +1192,8 @@ var triggerRouteFunction = async ({
   debug
 }) => {
   try {
-    await onStep();
-    await onCleanup();
+    const result = await onStep();
+    await onCleanup(result);
     return ok("workflow-finished");
   } catch (error) {
     const error_ = error;
@@ -1137,14 +1214,15 @@ var triggerRouteFunction = async ({
     }
   }
 };
-var triggerWorkflowDelete = async (workflowContext, debug, cancel = false) => {
+var triggerWorkflowDelete = async (workflowContext, result, debug, cancel = false) => {
   await debug?.log("SUBMIT", "SUBMIT_CLEANUP", {
     deletedWorkflowRunId: workflowContext.workflowRunId
   });
   await workflowContext.qstashClient.http.request({
     path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
     method: "DELETE",
-    parseResponseAsJson: false
+    parseResponseAsJson: false,
+    body: JSON.stringify(result)
   });
   await debug?.log(
     "SUBMIT",
@@ -1173,7 +1251,8 @@ var handleThirdPartyCallResult = async ({
   workflowUrl,
   failureUrl,
   retries,
-  telemetry,
+  telemetry: telemetry2,
+  flowControl,
   debug
 }) => {
   try {
@@ -1221,6 +1300,7 @@ ${atob(callbackMessage.body ?? "")}`
       const stepType = request.headers.get("Upstash-Workflow-StepType");
       const concurrentString = request.headers.get("Upstash-Workflow-Concurrent");
       const contentType = request.headers.get("Upstash-Workflow-ContentType");
+      const invokeCount = request.headers.get(WORKFLOW_INVOKE_COUNT_HEADER);
       if (!(workflowRunId && stepIdString && stepName && StepTypes.includes(stepType) && concurrentString && contentType)) {
         throw new Error(
           `Missing info in callback message source header: ${JSON.stringify({
@@ -1241,7 +1321,9 @@ ${atob(callbackMessage.body ?? "")}`
         userHeaders,
         failureUrl,
         retries,
-        telemetry
+        telemetry: telemetry2,
+        invokeCount: Number(invokeCount),
+        flowControl
       });
       const callResponse = {
         status: callbackMessage.status,
@@ -1280,11 +1362,11 @@ ${atob(callbackMessage.body ?? "")}`
     );
   }
 };
-var getTelemetryHeaders = (telemetry) => {
+var getTelemetryHeaders = (telemetry2) => {
   return {
-    [TELEMETRY_HEADER_SDK]: telemetry.sdk,
-    [TELEMETRY_HEADER_FRAMEWORK]: telemetry.framework,
-    [TELEMETRY_HEADER_RUNTIME]: telemetry.runtime ?? "unknown"
+    [TELEMETRY_HEADER_SDK]: telemetry2.sdk,
+    [TELEMETRY_HEADER_FRAMEWORK]: telemetry2.framework,
+    [TELEMETRY_HEADER_RUNTIME]: telemetry2.runtime ?? "unknown"
   };
 };
 var getHeaders = ({
@@ -1297,15 +1379,24 @@ var getHeaders = ({
   step,
   callRetries,
   callTimeout,
-  telemetry
+  telemetry: telemetry2,
+  invokeCount,
+  flowControl,
+  callFlowControl
 }) => {
+  const contentType = (userHeaders ? userHeaders.get("Content-Type") : void 0) ?? DEFAULT_CONTENT_TYPE;
   const baseHeaders = {
     [WORKFLOW_INIT_HEADER]: initHeaderValue,
     [WORKFLOW_ID_HEADER]: workflowRunId,
     [WORKFLOW_URL_HEADER]: workflowUrl,
     [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody",
-    ...telemetry ? getTelemetryHeaders(telemetry) : {}
+    [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION,
+    "content-type": contentType,
+    ...telemetry2 ? getTelemetryHeaders(telemetry2) : {}
   };
+  if (invokeCount !== void 0 && !step?.callUrl) {
+    baseHeaders[`Upstash-Forward-${WORKFLOW_INVOKE_COUNT_HEADER}`] = invokeCount.toString();
+  }
   if (!step?.callUrl) {
     baseHeaders[`Upstash-Forward-${WORKFLOW_PROTOCOL_VERSION_HEADER}`] = WORKFLOW_PROTOCOL_VERSION;
   }
@@ -1322,6 +1413,11 @@ var getHeaders = ({
     if (retries !== void 0) {
       baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
     }
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Failure-Callback-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Failure-Callback-Flow-Control-Value"] = flowControlValue;
+    }
     if (!step?.callUrl) {
       baseHeaders["Upstash-Failure-Callback"] = failureUrl;
     }
@@ -1333,9 +1429,26 @@ var getHeaders = ({
       baseHeaders["Upstash-Callback-Retries"] = retries.toString();
       baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
     }
-  } else if (retries !== void 0) {
-    baseHeaders["Upstash-Retries"] = retries.toString();
-    baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
+    if (callFlowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(callFlowControl);
+      baseHeaders["Upstash-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Flow-Control-Value"] = flowControlValue;
+    }
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Callback-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Callback-Flow-Control-Value"] = flowControlValue;
+    }
+  } else {
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Flow-Control-Value"] = flowControlValue;
+    }
+    if (retries !== void 0) {
+      baseHeaders["Upstash-Retries"] = retries.toString();
+      baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
+    }
   }
   if (userHeaders) {
     for (const header of userHeaders.keys()) {
@@ -1347,7 +1460,6 @@ var getHeaders = ({
       baseHeaders[`Upstash-Failure-Callback-Forward-${header}`] = userHeaders.get(header);
     }
   }
-  const contentType = (userHeaders ? userHeaders.get("Content-Type") : void 0) ?? DEFAULT_CONTENT_TYPE;
   if (step?.callHeaders) {
     const forwardedHeaders = Object.fromEntries(
       Object.entries(step.callHeaders).map(([header, value]) => [
@@ -1371,6 +1483,7 @@ var getHeaders = ({
         "Upstash-Callback-Forward-Upstash-Workflow-StepType": step.stepType,
         "Upstash-Callback-Forward-Upstash-Workflow-Concurrent": step.concurrent.toString(),
         "Upstash-Callback-Forward-Upstash-Workflow-ContentType": contentType,
+        [`Upstash-Callback-Forward-${WORKFLOW_INVOKE_COUNT_HEADER}`]: (invokeCount ?? 0).toString(),
         "Upstash-Workflow-CallType": "toCallback"
       }
     };
@@ -1387,8 +1500,8 @@ var getHeaders = ({
           Object.entries(baseHeaders).map(([header, value]) => [header, [value]])
         ),
         // to include telemetry headers:
-        ...telemetry ? Object.fromEntries(
-          Object.entries(getTelemetryHeaders(telemetry)).map(([header, value]) => [
+        ...telemetry2 ? Object.fromEntries(
+          Object.entries(getTelemetryHeaders(telemetry2)).map(([header, value]) => [
             header,
             [value]
           ])
@@ -1397,8 +1510,7 @@ var getHeaders = ({
         "Upstash-Workflow-Runid": [workflowRunId],
         [WORKFLOW_INIT_HEADER]: ["false"],
         [WORKFLOW_URL_HEADER]: [workflowUrl],
-        "Upstash-Workflow-CallType": ["step"],
-        "Content-Type": [contentType]
+        "Upstash-Workflow-CallType": ["step"]
       }
     };
   }
@@ -1429,9 +1541,159 @@ If you want to disable QStash Verification, you should clear env variables QSTAS
     );
   }
 };
+var prepareFlowControl = (flowControl) => {
+  const parallelism = flowControl.parallelism?.toString();
+  const rate = flowControl.ratePerSecond?.toString();
+  const controlValue = [
+    parallelism ? `parallelism=${parallelism}` : void 0,
+    rate ? `rate=${rate}` : void 0
+  ].filter(Boolean);
+  if (controlValue.length === 0) {
+    throw new import_qstash3.QstashError("Provide at least one of parallelism or ratePerSecond for flowControl");
+  }
+  return {
+    flowControlKey: flowControl.key,
+    flowControlValue: controlValue.join(", ")
+  };
+};
 
 // src/context/auto-executor.ts
 var import_qstash4 = require("@upstash/qstash");
+
+// src/serve/serve-many.ts
+var getWorkflowId = (url) => {
+  const components = url.split("/");
+  const lastComponent = components[components.length - 1];
+  return lastComponent.split("?")[0];
+};
+var serveManyBase = ({
+  workflows,
+  getUrl: getUrl2,
+  serveMethod,
+  options
+}) => {
+  const workflowIds = [];
+  const workflowMap = Object.fromEntries(
+    Object.entries(workflows).map((workflow) => {
+      const workflowId = workflow[0];
+      if (workflowIds.includes(workflowId)) {
+        throw new WorkflowError(
+          `Duplicate workflow name found: '${workflowId}'. Please set different workflow names in serveMany.`
+        );
+      }
+      if (workflowId.includes("/")) {
+        throw new WorkflowError(
+          `Invalid workflow name found: '${workflowId}'. Workflow name cannot contain '/'.`
+        );
+      }
+      workflowIds.push(workflowId);
+      workflow[1].workflowId = workflowId;
+      workflow[1].options = {
+        ...options,
+        ...workflow[1].options
+      };
+      const params = [workflow[1].routeFunction, workflow[1].options];
+      const handler = serveMethod(...params);
+      return [workflowId, handler];
+    })
+  );
+  return {
+    handler: async (...params) => {
+      const url = getUrl2(...params);
+      const pickedWorkflowId = getWorkflowId(url);
+      if (!pickedWorkflowId) {
+        return new Response(
+          `Unexpected request in serveMany. workflowId not set. Please update the URL of your request.`,
+          {
+            status: 404
+          }
+        );
+      }
+      const workflow = workflowMap[pickedWorkflowId];
+      if (!workflow) {
+        return new Response(
+          `No workflows in serveMany found for '${pickedWorkflowId}'. Please update the URL of your request.`,
+          {
+            status: 404
+          }
+        );
+      }
+      return await workflow(...params);
+    }
+  };
+};
+var invokeWorkflow = async ({
+  settings,
+  invokeStep,
+  context,
+  invokeCount,
+  telemetry: telemetry2
+}) => {
+  const {
+    body,
+    workflow,
+    headers = {},
+    workflowRunId = getWorkflowRunId(),
+    retries,
+    flowControl
+  } = settings;
+  const { workflowId } = workflow;
+  const {
+    retries: workflowRetries,
+    failureFunction,
+    failureUrl,
+    useJSONContent,
+    flowControl: workflowFlowControl
+  } = workflow.options;
+  if (!workflowId) {
+    throw new WorkflowError("You can only invoke workflow which has a workflowId");
+  }
+  const { headers: invokerHeaders } = getHeaders({
+    initHeaderValue: "false",
+    workflowRunId: context.workflowRunId,
+    workflowUrl: context.url,
+    userHeaders: context.headers,
+    failureUrl: context.failureUrl,
+    retries: context.retries,
+    telemetry: telemetry2,
+    invokeCount,
+    flowControl: context.flowControl
+  });
+  invokerHeaders["Upstash-Workflow-Runid"] = context.workflowRunId;
+  const newUrl = context.url.replace(/[^/]+$/, workflowId);
+  const { headers: triggerHeaders } = getHeaders({
+    initHeaderValue: "true",
+    workflowRunId,
+    workflowUrl: newUrl,
+    userHeaders: new Headers(headers),
+    retries: retries ?? workflowRetries,
+    telemetry: telemetry2,
+    failureUrl: failureFunction ? newUrl : failureUrl,
+    invokeCount: invokeCount + 1,
+    flowControl: flowControl ?? workflowFlowControl
+  });
+  triggerHeaders["Upstash-Workflow-Invoke"] = "true";
+  if (useJSONContent) {
+    triggerHeaders["content-type"] = "application/json";
+  }
+  const request = {
+    body: JSON.stringify(body),
+    headers: Object.fromEntries(
+      Object.entries(invokerHeaders).map((pairs) => [pairs[0], [pairs[1]]])
+    ),
+    workflowRunId,
+    workflowUrl: context.url,
+    step: invokeStep
+  };
+  await context.qstashClient.publish({
+    headers: triggerHeaders,
+    method: "POST",
+    body: JSON.stringify(request),
+    url: newUrl
+  });
+};
+
+// src/context/auto-executor.ts
 var AutoExecutor = class _AutoExecutor {
   context;
   promises = /* @__PURE__ */ new WeakMap();
@@ -1440,14 +1702,16 @@ var AutoExecutor = class _AutoExecutor {
   nonPlanStepCount;
   steps;
   indexInCurrentList = 0;
+  invokeCount;
   telemetry;
   stepCount = 0;
   planStepCount = 0;
   executingStep = false;
-  constructor(context, steps, telemetry, debug) {
+  constructor(context, steps, telemetry2, invokeCount, debug) {
     this.context = context;
     this.steps = steps;
-    this.telemetry = telemetry;
+    this.telemetry = telemetry2;
+    this.invokeCount = invokeCount ?? 0;
     this.debug = debug;
     this.nonPlanStepCount = this.steps.filter((step) => !step.targetStep).length;
   }
@@ -1670,7 +1934,9 @@ var AutoExecutor = class _AutoExecutor {
         step: waitStep,
         failureUrl: this.context.failureUrl,
         retries: this.context.retries,
-        telemetry: this.telemetry
+        telemetry: this.telemetry,
+        invokeCount: this.invokeCount,
+        flowControl: this.context.flowControl
       });
       const waitBody = {
         url: this.context.url,
@@ -1693,7 +1959,19 @@ var AutoExecutor = class _AutoExecutor {
         method: "POST",
         parseResponseAsJson: false
       });
-      throw new WorkflowAbort(steps[0].stepName, steps[0]);
+      throw new WorkflowAbort(waitStep.stepName, waitStep);
+    }
+    if (steps.length === 1 && lazySteps[0] instanceof LazyInvokeStep) {
+      const invokeStep = steps[0];
+      const lazyInvokeStep = lazySteps[0];
+      await invokeWorkflow({
+        settings: lazyInvokeStep.params,
+        invokeStep,
+        context: this.context,
+        invokeCount: this.invokeCount,
+        telemetry: this.telemetry
+      });
+      throw new WorkflowAbort(invokeStep.stepName, invokeStep);
     }
     const result = await this.context.qstashClient.batchJSON(
       steps.map((singleStep, index) => {
@@ -1708,11 +1986,14 @@ var AutoExecutor = class _AutoExecutor {
           retries: this.context.retries,
           callRetries: lazyStep instanceof LazyCallStep ? lazyStep.retries : void 0,
           callTimeout: lazyStep instanceof LazyCallStep ? lazyStep.timeout : void 0,
-          telemetry: this.telemetry
+          telemetry: this.telemetry,
+          invokeCount: this.invokeCount,
+          flowControl: this.context.flowControl,
+          callFlowControl: lazyStep instanceof LazyCallStep ? lazyStep.flowControl : void 0
         });
         const willWait = singleStep.concurrent === NO_CONCURRENCY || singleStep.stepId === 0;
         singleStep.out = JSON.stringify(singleStep.out);
-        return singleStep.callUrl ? (
+        return singleStep.callUrl && lazyStep instanceof LazyCallStep ? (
           // if the step is a third party call, we call the third party
           // url (singleStep.callUrl) and pass information about the workflow
           // in the headers (handled in getHeaders). QStash makes the request
@@ -2369,6 +2650,11 @@ var WorkflowContext = class {
    * Number of retries
    */
   retries;
+  /**
+   * Settings for controlling the number of active requests
+   * and number of requests per second with the same key.
+   */
+  flowControl;
   constructor({
     qstashClient,
     workflowRunId,
@@ -2380,7 +2666,9 @@ var WorkflowContext = class {
     initialPayload,
     env,
     retries,
-    telemetry
+    telemetry: telemetry2,
+    invokeCount,
+    flowControl
   }) {
     this.qstashClient = qstashClient;
     this.workflowRunId = workflowRunId;
@@ -2391,7 +2679,8 @@ var WorkflowContext = class {
     this.requestPayload = initialPayload;
     this.env = env ?? {};
     this.retries = retries ?? DEFAULT_RETRIES;
-    this.executor = new AutoExecutor(this, this.steps, telemetry, debug);
+    this.flowControl = flowControl;
+    this.executor = new AutoExecutor(this, this.steps, telemetry2, invokeCount, debug);
   }
   /**
    * Executes a workflow step
@@ -2493,7 +2782,7 @@ var WorkflowContext = class {
    *   }
    */
   async call(stepName, settings) {
-    const { url, method = "GET", body, headers = {}, retries = 0, timeout } = settings;
+    const { url, method = "GET", body, headers = {}, retries = 0, timeout, flowControl } = settings;
     const result = await this.addStep(
       new LazyCallStep(
         stepName,
@@ -2502,7 +2791,8 @@ var WorkflowContext = class {
         body,
         headers,
         retries,
-        timeout
+        timeout,
+        flowControl
       )
     );
     if (typeof result === "string") {
@@ -2611,6 +2901,13 @@ var WorkflowContext = class {
       return result;
     }
   }
+  async invoke(stepName, settings) {
+    const result = await this.addStep(new LazyInvokeStep(stepName, settings));
+    return {
+      ...result,
+      body: result.body ? JSON.parse(result.body) : void 0
+    };
+  }
   /**
    * Cancel the current workflow run
    *
@@ -2688,31 +2985,6 @@ var WorkflowLogger = class _WorkflowLogger {
   }
 };
 
-// src/utils.ts
-var NANOID_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
-var NANOID_LENGTH = 21;
-function getRandomInt() {
-  return Math.floor(Math.random() * NANOID_CHARS.length);
-}
-function nanoid() {
-  return Array.from({ length: NANOID_LENGTH }).map(() => NANOID_CHARS[getRandomInt()]).join("");
-}
-function getWorkflowRunId(id) {
-  return `wfr_${id ?? nanoid()}`;
-}
-function decodeBase64(base64) {
-  try {
-    const binString = atob(base64);
-    const intArray = Uint8Array.from(binString, (m) => m.codePointAt(0));
-    return new TextDecoder().decode(intArray);
-  } catch (error) {
-    console.warn(
-      `Upstash Qstash: Failed while decoding base64 "${base64}". Decoding with atob and returning it instead. ${error}`
-    );
-    return atob(base64);
-  }
-}
-
 // src/serve/authorization.ts
 var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
@@ -2757,7 +3029,8 @@ var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowCon
       failureUrl: context.failureUrl,
       initialPayload: context.requestPayload,
       env: context.env,
-      retries: context.retries
+      retries: context.retries,
+      flowControl: context.flowControl
     });
     try {
       await routeFunction(disabledContext);
@@ -2910,7 +3183,7 @@ var parseRequest = async (requestPayload, isFirstInvocation, workflowRunId, requ
     };
   }
 };
-var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, debug) => {
+var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, flowControl, debug) => {
   if (request.headers.get(WORKFLOW_FAILURE_HEADER) !== "true") {
     return ok("not-failure-callback");
   }
@@ -2922,22 +3195,21 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
     );
   }
   try {
-    const { status, header, body, url, sourceHeader, sourceBody, workflowRunId } = JSON.parse(
-      requestPayload
-    );
+    const { status, header, body, url, sourceBody, workflowRunId } = JSON.parse(requestPayload);
     const decodedBody = body ? decodeBase64(body) : "{}";
     const errorPayload = JSON.parse(decodedBody);
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
       initialPayload: sourceBody ? initialPayloadParser(decodeBase64(sourceBody)) : void 0,
-      headers: recreateUserHeaders(new Headers(sourceHeader)),
+      headers: recreateUserHeaders(request.headers),
       steps: [],
       url,
       failureUrl: url,
       debug,
       env,
       retries,
+      flowControl,
       telemetry: void 0
       // not going to make requests in authentication check
     });
@@ -3050,7 +3322,7 @@ var determineUrls = async (request, url, baseUrl, failureFunction, failureUrl, d
 var AUTH_FAIL_MESSAGE = `Failed to authenticate Workflow request. If this is unexpected, see the caveat https://upstash.com/docs/workflow/basics/caveats#avoid-non-deterministic-code-outside-context-run`;
 
 // src/serve/index.ts
-var serveBase = (routeFunction, telemetry, options) => {
+var serveBase = (routeFunction, telemetry2, options) => {
   const {
     qstashClient,
     onStepFinish,
@@ -3064,9 +3336,10 @@ var serveBase = (routeFunction, telemetry, options) => {
     env,
     retries,
     useJSONContent,
-    disableTelemetry
+    disableTelemetry,
+    flowControl
   } = processOptions(options);
-  telemetry = disableTelemetry ? void 0 : telemetry;
+  telemetry2 = disableTelemetry ? void 0 : telemetry2;
   const debug = WorkflowLogger.getLogger(verbose);
   const handler = async (request) => {
     await debug?.log("INFO", "ENDPOINT_START");
@@ -3105,6 +3378,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       failureFunction,
       env,
       retries,
+      flowControl,
       debug
     );
     if (failureCheck.isErr()) {
@@ -3113,6 +3387,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       await debug?.log("WARN", "RESPONSE_DEFAULT", "failureFunction executed");
       return onStepFinish(workflowRunId, "failure-callback");
     }
+    const invokeCount = Number(request.headers.get(WORKFLOW_INVOKE_COUNT_HEADER) ?? "0");
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
@@ -3124,7 +3399,9 @@ var serveBase = (routeFunction, telemetry, options) => {
       debug,
       env,
       retries,
-      telemetry
+      telemetry: telemetry2,
+      invokeCount,
+      flowControl
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -3147,7 +3424,8 @@ var serveBase = (routeFunction, telemetry, options) => {
       workflowUrl,
       failureUrl: workflowFailureUrl,
       retries,
-      telemetry,
+      flowControl,
+      telemetry: telemetry2,
       debug
     });
     if (callReturnCheck.isErr()) {
@@ -3156,10 +3434,16 @@ var serveBase = (routeFunction, telemetry, options) => {
       });
       throw callReturnCheck.error;
     } else if (callReturnCheck.value === "continue-workflow") {
-      const result = isFirstInvocation ? await triggerFirstInvocation({ workflowContext, useJSONContent, telemetry, debug }) : await triggerRouteFunction({
+      const result = isFirstInvocation ? await triggerFirstInvocation({
+        workflowContext,
+        useJSONContent,
+        telemetry: telemetry2,
+        debug,
+        invokeCount
+      }) : await triggerRouteFunction({
         onStep: async () => routeFunction(workflowContext),
-        onCleanup: async () => {
-          await triggerWorkflowDelete(workflowContext, debug);
+        onCleanup: async (result2) => {
+          await triggerWorkflowDelete(workflowContext, result2, debug);
         },
         onCancel: async () => {
           await makeCancelRequest(workflowContext.qstashClient.http, workflowRunId);
@@ -3199,39 +3483,57 @@ function transformHeaders(headers) {
   ]);
   return formattedHeaders;
 }
+function getUrl(event) {
+  const request_ = event.node.req;
+  const protocol = request_.headers["x-forwarded-proto"];
+  const host = request_.headers.host;
+  const url = `${protocol}://${host}${event.path}`;
+  return url;
+}
+var telemetry = {
+  sdk: SDK_TELEMETRY,
+  framework: "h3",
+  runtime: process.versions.bun ? `bun@${process.versions.bun}/node@${process.version}` : `node@${process.version}`
+};
 var serve = (routeFunction, options) => {
   const handler = defineEventHandler(async (event) => {
     const method = event.node.req.method;
     if (method?.toUpperCase() !== "POST") {
-      return {
-        status: 405,
-        body: "Only POST requests are allowed in worklfows"
-      };
+      return new Response("Only POST requests are allowed in worklfows", {
+        status: 405
+      });
     }
-    const request_ = event.node.req;
-    const protocol = request_.headers["x-forwarded-proto"];
-    const host = request_.headers.host;
-    const url = `${protocol}://${host}${event.path}`;
-    const headers = transformHeaders(request_.headers);
+    const url = getUrl(event);
+    const headers = transformHeaders(event.node.req.headers);
     const request = new Request(url, {
       headers,
       body: await readRawBody(event),
       method: "POST"
     });
-    const { handler: serveHandler } = serveBase(
-      routeFunction,
-      {
-        sdk: SDK_TELEMETRY,
-        framework: "h3",
-        runtime: process.versions.bun ? `bun@${process.versions.bun}/node@${process.version}` : `node@${process.version}`
-      },
-      options
-    );
+    const { handler: serveHandler } = serveBase(routeFunction, telemetry, options);
     return await serveHandler(request);
   });
   return { handler };
 };
+var createWorkflow = (...params) => {
+  const [routeFunction, options = {}] = params;
+  return {
+    routeFunction,
+    options,
+    workflowId: void 0
+  };
+};
+var serveMany = (workflows, options) => {
+  return serveManyBase({
+    workflows,
+    getUrl,
+    serveMethod: (...params) => serve(...params).handler,
+    options
+  });
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  serve
+  createWorkflow,
+  serve,
+  serveMany
 });