@workflow/world-testing 4.0.1-beta.6 → 4.0.1-beta.7

package/dist/.well-known/workflow/v1/manifest.debug.json

@@ -11,6 +11,11 @@
                 "stepId": "__builtin_response_text"
             }
         },
+        "workflows/addition.ts": {
+            "add": {
+                "stepId": "step//workflows/addition.ts//add"
+            }
+        },
         "workflows/noop.ts": {
             "noop": {
                 "stepId": "step//workflows/noop.ts//noop"
@@ -19,14 +24,6 @@
         "workflow/dist/stdlib.js": {
             "fetch": {
                 "stepId": "step//workflow/dist/stdlib.js//fetch"
-            },
-            "sleep": {
-                "stepId": "step//workflow/dist/stdlib.js//sleep"
-            }
-        },
-        "workflows/addition.ts": {
-            "add": {
-                "stepId": "step//workflows/addition.ts//add"
             }
         }
     },

package/dist/.well-known/workflow/v1/step.js

@@ -684,7 +684,7 @@ var require_build_client = __commonJS({
             BuildCache: /* @__PURE__ */ __name(() => BuildCache, "BuildCache")
         });
         module2.exports = __toCommonJS2(build_client_exports);
-        var import_index2 = require_cache();
+        var import_index = require_cache();
         var BuildCache = class {
             static {
                 __name(this, "BuildCache");
@@ -704,8 +704,8 @@ var require_build_client = __commonJS({
                             return null;
                         }
                         if (res.status === 200) {
-                            const cacheState = res.headers.get(import_index2.HEADERS_VERCEL_CACHE_STATE);
-                            if (cacheState !== import_index2.PkgCacheState.Fresh) {
+                            const cacheState = res.headers.get(import_index.HEADERS_VERCEL_CACHE_STATE);
+                            if (cacheState !== import_index.PkgCacheState.Fresh) {
                                 res.body?.cancel?.();
                                 clearTimeout(timeoutId);
                                 return null;
@@ -738,13 +738,13 @@ var require_build_client = __commonJS({
                     try {
                         const optionalHeaders = {};
                         if (options?.ttl) {
-                            optionalHeaders[import_index2.HEADERS_VERCEL_REVALIDATE] = options.ttl.toString();
+                            optionalHeaders[import_index.HEADERS_VERCEL_REVALIDATE] = options.ttl.toString();
                         }
                         if (options?.tags && options.tags.length > 0) {
-                            optionalHeaders[import_index2.HEADERS_VERCEL_CACHE_TAGS] = options.tags.join(",");
+                            optionalHeaders[import_index.HEADERS_VERCEL_CACHE_TAGS] = options.tags.join(",");
                         }
                         if (options?.name) {
-                            optionalHeaders[import_index2.HEADERS_VERCEL_CACHE_ITEM_NAME] = options.name;
+                            optionalHeaders[import_index.HEADERS_VERCEL_CACHE_ITEM_NAME] = options.name;
                         }
                         const res = await fetch(`${this.endpoint}${key}`, {
                             method: "POST",
@@ -5257,17 +5257,13 @@ __name(__builtin_response_text, "__builtin_response_text");
 registerStepFunction("__builtin_response_array_buffer", __builtin_response_array_buffer);
 registerStepFunction("__builtin_response_json", __builtin_response_json);
 registerStepFunction("__builtin_response_text", __builtin_response_text);
-// workflows/noop.ts
-var count = 0;
-async function noop(_i) {
-    count++;
-    return count;
-}
-__name(noop, "noop");
-registerStepFunction("step//workflows/noop.ts//noop", noop);
 // ../errors/dist/index.js
 var import_ms = __toESM(require_ms(), 1);
 var BASE_URL = "https://useworkflow.dev/err";
+function isError(value) {
+    return typeof value === "object" && value !== null && "name" in value && "message" in value;
+}
+__name(isError, "isError");
 var WorkflowError = class extends Error {
     static {
         __name(this, "WorkflowError");
@@ -5286,6 +5282,9 @@ Learn more: ${BASE_URL}/${options.slug}` : message;
 Caused by: ${options.cause.stack}`;
         }
     }
+    static is(value) {
+        return isError(value) && value.name === "WorkflowError";
+    }
 };
 var WorkflowAPIError = class extends WorkflowError {
     static {
@@ -5303,6 +5302,9 @@ var WorkflowAPIError = class extends WorkflowError {
         this.code = options?.code;
         this.url = options?.url;
     }
+    static is(value) {
+        return isError(value) && value.name === "WorkflowAPIError";
+    }
 };
 var WorkflowRuntimeError = class extends WorkflowError {
     static {
@@ -5314,6 +5316,9 @@ var WorkflowRuntimeError = class extends WorkflowError {
         });
         this.name = "WorkflowRuntimeError";
     }
+    static is(value) {
+        return isError(value) && value.name === "WorkflowRuntimeError";
+    }
 };
 var WorkflowRunNotFoundError = class extends WorkflowError {
     static {
@@ -5325,6 +5330,9 @@ var WorkflowRunNotFoundError = class extends WorkflowError {
         this.name = "WorkflowRunNotFoundError";
         this.runId = runId;
     }
+    static is(value) {
+        return isError(value) && value.name === "WorkflowRunNotFoundError";
+    }
 };
 var FatalError = class extends Error {
     static {
@@ -5335,6 +5343,9 @@ var FatalError = class extends Error {
         super(message);
         this.name = "FatalError";
     }
+    static is(value) {
+        return isError(value) && value.name === "FatalError";
+    }
 };
 var RetryableError = class extends Error {
     static {
@@ -5362,6 +5373,9 @@ var RetryableError = class extends Error {
         }
         this.retryAfter = new Date(Date.now() + retryAfterSeconds * 1e3);
     }
+    static is(value) {
+        return isError(value) && value.name === "RetryableError";
+    }
 };
 // ../core/dist/runtime/resume-hook.js
 var import_functions = __toESM(require_functions(), 1);
@@ -19808,6 +19822,8 @@ var EventTypeSchema = external_exports.enum([
     "hook_created",
     "hook_received",
     "hook_disposed",
+    "wait_created",
+    "wait_completed",
     "workflow_completed",
     "workflow_failed",
     "workflow_started"
@@ -19858,6 +19874,17 @@ var HookDisposedEventSchema = BaseEventSchema.extend({
     eventType: external_exports.literal("hook_disposed"),
     correlationId: external_exports.string()
 });
+var WaitCreatedEventSchema = BaseEventSchema.extend({
+    eventType: external_exports.literal("wait_created"),
+    correlationId: external_exports.string(),
+    eventData: external_exports.object({
+        resumeAt: external_exports.coerce.date()
+    })
+});
+var WaitCompletedEventSchema = BaseEventSchema.extend({
+    eventType: external_exports.literal("wait_completed"),
+    correlationId: external_exports.string()
+});
 var WorkflowCompletedEventSchema = BaseEventSchema.extend({
     eventType: external_exports.literal("workflow_completed")
 });
@@ -19878,6 +19905,8 @@ var CreateEventSchema = external_exports.discriminatedUnion("eventType", [
     HookCreatedEventSchema,
     HookReceivedEventSchema,
     HookDisposedEventSchema,
+    WaitCreatedEventSchema,
+    WaitCompletedEventSchema,
     WorkflowCompletedEventSchema,
     WorkflowFailedEventSchema,
     WorkflowStartedEventSchema
@@ -19985,7 +20014,8 @@ var StepSchema = external_exports.object({
     startedAt: external_exports.coerce.date().optional(),
     completedAt: external_exports.coerce.date().optional(),
     createdAt: external_exports.coerce.date(),
-    updatedAt: external_exports.coerce.date()
+    updatedAt: external_exports.coerce.date(),
+    retryAfter: external_exports.coerce.date().optional()
 });
 // ../../node_modules/.pnpm/ulid@3.0.1/node_modules/ulid/dist/node/index.js
 var import_node_crypto = __toESM(require("node:crypto"), 1);
@@ -20151,6 +20181,7 @@ function monotonicFactory(prng) {
 }
 __name(monotonicFactory, "monotonicFactory");
 // ../world-local/dist/queue.js
+var LOCAL_QUEUE_MAX_VISIBILITY = parseInt(process.env.WORKFLOW_LOCAL_QUEUE_MAX_VISIBILITY ?? "0", 10) || Infinity;
 function createQueue(port) {
     const transport = new JsonTransport();
     const generateId = monotonicFactory();
@@ -20204,8 +20235,8 @@ function createQueue(port) {
                 const text = await response.text();
                 if (response.status === 503) {
                     try {
-                        const retryIn = Number(JSON.parse(text).retryIn);
-                        await (0, import_promises.setTimeout)(retryIn * 1e3);
+                        const timeoutSeconds = Number(JSON.parse(text).timeoutSeconds);
+                        await (0, import_promises.setTimeout)(timeoutSeconds * 1e3);
                         defaultRetriesLeft++;
                         continue;
                     }
@@ -20257,15 +20288,18 @@ function createQueue(port) {
             }
             const body = await new JsonTransport().deserialize(req.body);
             try {
-                const response = await handler(body, {
+                const result = await handler(body, {
                     attempt,
                     queueName,
                     messageId
                 });
-                const retryIn = typeof response === "undefined" ? null : response.timeoutSeconds;
-                if (retryIn) {
+                let timeoutSeconds = null;
+                if (typeof result?.timeoutSeconds === "number") {
+                    timeoutSeconds = Math.min(result.timeoutSeconds, LOCAL_QUEUE_MAX_VISIBILITY);
+                }
+                if (timeoutSeconds) {
                     return Response.json({
-                        retryIn
+                        timeoutSeconds
                     }, {
                         status: 503
                     });
@@ -22670,6 +22704,7 @@ var MessageWrapper = object({
     payload: QueuePayloadSchema,
     queueName: ValidQueueName
 });
+var VERCEL_QUEUE_MAX_VISIBILITY = 82800;
 function createQueue2(config3) {
     const { baseUrl, usingProxy } = getHttpUrl(config3);
     const headers = getHeaders(config3);
@@ -22700,13 +22735,20 @@ function createQueue2(config3) {
     const createQueueHandler = /* @__PURE__ */ __name((prefix, handler) => {
         return handleCallback({
             [`${prefix}*`]: {
-                default: /* @__PURE__ */ __name((body, meta) => {
+                default: /* @__PURE__ */ __name(async (body, meta) => {
                     const { payload, queueName } = MessageWrapper.parse(body);
-                    return handler(payload, {
+                    const result = await handler(payload, {
                         queueName,
                         messageId: MessageId.parse(meta.messageId),
                         attempt: meta.deliveryCount
                     });
+                    if (typeof result?.timeoutSeconds === "number") {
+                        const adjustedTimeoutSeconds = Math.min(result.timeoutSeconds, VERCEL_QUEUE_MAX_VISIBILITY);
+                        if (adjustedTimeoutSeconds !== result.timeoutSeconds) {
+                            result.timeoutSeconds = adjustedTimeoutSeconds;
+                        }
+                    }
+                    return result;
                 }, "default")
             }
         });
@@ -23318,6 +23360,7 @@ var getWorld = /* @__PURE__ */ __name(() => {
 // ../core/dist/symbols.js
 var WORKFLOW_USE_STEP = Symbol.for("WORKFLOW_USE_STEP");
 var WORKFLOW_CREATE_HOOK = Symbol.for("WORKFLOW_CREATE_HOOK");
+var WORKFLOW_SLEEP = Symbol.for("WORKFLOW_SLEEP");
 var WORKFLOW_CONTEXT = Symbol.for("WORKFLOW_CONTEXT");
 var WORKFLOW_GET_STREAM_ID = Symbol.for("WORKFLOW_GET_STREAM_ID");
 var STREAM_NAME_SYMBOL = Symbol.for("WORKFLOW_STREAM_NAME");
@@ -23866,44 +23909,11 @@ __name(getActiveSpan, "getActiveSpan");
 // ../core/dist/step/context-storage.js
 var import_node_async_hooks = require("node:async_hooks");
 var contextStorage = /* @__PURE__ */ new import_node_async_hooks.AsyncLocalStorage();
-// ../core/dist/step/get-step-metadata.js
-function getStepMetadata() {
-    const ctx = contextStorage.getStore();
-    if (!ctx) {
-        throw new Error("`getStepMetadata()` can only be called inside a step function");
-    }
-    return ctx.stepMetadata;
-}
-__name(getStepMetadata, "getStepMetadata");
 // ../workflow/dist/stdlib.js
-var import_ms3 = __toESM(require_ms(), 1);
-var MAX_SLEEP_DURATION_SECONDS = (0, import_ms3.default)("23h") / 1e3;
-async function sleep(param) {
-    const { stepStartedAt } = getStepMetadata();
-    const durationMs = typeof param === "string" ? (0, import_ms3.default)(param) : param.getTime() - Number(stepStartedAt);
-    if (typeof durationMs !== "number" || durationMs < 0) {
-        const message = param instanceof Date ? `Invalid sleep date: "${param}". Expected a future date.` : `Invalid sleep duration: "${param}". Expected a valid duration string like "1s", "1m", "1h", etc.`;
-        throw new Error(message);
-    }
-    const endAt = +stepStartedAt + durationMs;
-    const now = Date.now();
-    if (now < endAt) {
-        const remainingSeconds = (endAt - now) / 1e3;
-        const retryAfter = Math.min(remainingSeconds, MAX_SLEEP_DURATION_SECONDS);
-        throw new RetryableError(`Sleeping for ${(0, import_ms3.default)(retryAfter * 1e3, {
-            long: true
-        })}`, {
-            retryAfter
-        });
-    }
-}
-__name(sleep, "sleep");
-sleep.maxRetries = Infinity;
 async function fetch2(...args) {
     return globalThis.fetch(...args);
 }
 __name(fetch2, "fetch");
-registerStepFunction("step//workflow/dist/stdlib.js//sleep", sleep);
 registerStepFunction("step//workflow/dist/stdlib.js//fetch", fetch2);
 // workflows/addition.ts
 async function add(num, num2) {
@@ -23914,6 +23924,14 @@ async function add(num, num2) {
 }
 __name(add, "add");
 registerStepFunction("step//workflows/addition.ts//add", add);
+// workflows/noop.ts
+var count = 0;
+async function noop(_i) {
+    count++;
+    return count;
+}
+__name(noop, "noop");
+registerStepFunction("step//workflows/noop.ts//noop", noop);
 // ../core/dist/runtime.js
 var import_functions3 = __toESM(require_functions(), 1);
 // ../core/dist/logger.js
@@ -23963,33 +23981,6 @@ var StepInvokePayloadSchema2 = external_exports.object({
 });
 // ../core/dist/types.js
 var import_node_util = require("node:util");
-function getConstructorName(obj) {
-    if (obj === null || obj === void 0) {
-        return null;
-    }
-    const ctor = obj.constructor;
-    if (!ctor || ctor.name === "Object") {
-        return null;
-    }
-    return ctor.name;
-}
-__name(getConstructorName, "getConstructorName");
-function getConstructorNames(obj) {
-    const proto = Object.getPrototypeOf(obj);
-    const name = getConstructorName(proto);
-    if (name === null) {
-        return [];
-    }
-    return [
-        name,
-        ...getConstructorNames(proto)
-    ];
-}
-__name(getConstructorNames, "getConstructorNames");
-function isInstanceOf(v, ctor) {
-    return getConstructorNames(v).includes(ctor.name);
-}
-__name(isInstanceOf, "isInstanceOf");
 function getErrorName(v) {
     if (import_node_util.types.isNativeError(v)) {
         return v.name;
@@ -24013,6 +24004,8 @@ var EventConsumerResult;
 })(EventConsumerResult || (EventConsumerResult = {}));
 // ../core/dist/vm/index.js
 var import_seedrandom = __toESM(require_seedrandom2(), 1);
+// ../core/dist/workflow/sleep.js
+var import_ms3 = __toESM(require_ms(), 1);
 // ../core/dist/runtime/start.js
 var import_functions2 = __toESM(require_functions(), 1);
 // ../core/dist/runtime.js
@@ -24051,6 +24044,22 @@ var stepEntrypoint = /* @__PURE__ */ getWorldHandlers().createQueueHandler("__wk
             span?.setAttributes({
                 ...StepStatus(step.status)
             });
+            const now = Date.now();
+            if (step.retryAfter && step.retryAfter.getTime() > now) {
+                const timeoutSeconds = Math.ceil((step.retryAfter.getTime() - now) / 1e3);
+                span?.setAttributes({
+                    ...StepRetryTimeoutSeconds(timeoutSeconds)
+                });
+                runtimeLogger.debug("Step retryAfter timestamp not yet reached", {
+                    stepName,
+                    stepId: step.stepId,
+                    retryAfter: step.retryAfter,
+                    timeoutSeconds
+                });
+                return {
+                    timeoutSeconds
+                };
+            }
             let result;
             const attempt = step.attempt + 1;
             try {
@@ -24115,13 +24124,13 @@ var stepEntrypoint = /* @__PURE__ */ getWorldHandlers().createQueueHandler("__wk
                     ...StepErrorName(getErrorName(err)),
                     ...StepErrorMessage(String(err))
                 });
-                if (isInstanceOf(err, WorkflowAPIError)) {
+                if (WorkflowAPIError.is(err)) {
                     if (err.status === 410) {
                         console.warn(`Workflow run "${workflowRunId}" has already completed, skipping step "${stepId}": ${err.message}`);
                         return;
                     }
                 }
-                if (isInstanceOf(err, FatalError)) {
+                if (FatalError.is(err)) {
                     const stackLines = getErrorStack(err).split("\n").slice(0, 4);
                     console.error(`[Workflows] "${workflowRunId}" - Encountered \`FatalError\` while executing step "${stepName}":
   > ${stackLines.join("\n    > ")}
@@ -24178,7 +24187,7 @@ Bubbling up error to parent workflow`);
                         });
                     }
                     else {
-                        if (isInstanceOf(err, RetryableError)) {
+                        if (RetryableError.is(err)) {
                             console.warn(`[Workflows] "${workflowRunId}" - Encountered \`RetryableError\` while executing step "${stepName}" (attempt ${attempt}):
   > ${String(err.message)}
 
@@ -24200,9 +24209,12 @@ Bubbling up error to parent workflow`);
                             }
                         });
                         await world.steps.update(workflowRunId, stepId, {
-                            status: "pending"
+                            status: "pending",
+                            ...RetryableError.is(err) && {
+                                retryAfter: err.retryAfter
+                            }
                         });
-                        const timeoutSeconds = Math.max(1, isInstanceOf(err, RetryableError) ? Math.floor((+err.retryAfter.getTime() - Date.now()) / 1e3) : 1);
+                        const timeoutSeconds = Math.max(1, RetryableError.is(err) ? Math.ceil((+err.retryAfter.getTime() - Date.now()) / 1e3) : 1);
                         span?.setAttributes({
                             ...StepRetryTimeoutSeconds(timeoutSeconds),
                             ...StepRetryWillRetry(true)