@mastra/temporal 0.5.0 → 0.6.0-alpha.0

package/dist/worker.js

@@ -403,24 +403,55 @@ function hasLocalMastraBinding(ast) {
 }
 function createTemporalActivitiesHelperStatements(mastraImportPath, hasMastraBinding, requestContextClassName) {
 	const helperSource = `
-    function withRequestContext(params) {
+    function withRequestContext(params, args) {
       const { requestContext, initData, ...rest } = params;
+      let suspended;
       return {
         ...rest,
         requestContext: new ${requestContextClassName}(requestContext),
         getInitData: () => initData,
+        suspend: async (suspendPayload, options) => {
+          const parsedSuspendPayload = await parseWithSchema(args.suspendSchema, suspendPayload, 'Suspend payload');
+          suspended = {
+            __temporalSuspension: true,
+            suspendPayload: parsedSuspendPayload,
+            ...(options?.resumeLabel !== undefined && { resumeLabel: options.resumeLabel }),
+          };
+        },
+        __getTemporalSuspension: () => suspended,
       };
     }
 
+    async function parseWithSchema(schema, value, label) {
+      if (!schema) return value;
+      if (schema['~standard']?.validate) {
+        const result = await schema['~standard'].validate(value);
+        if ('issues' in result && result.issues?.length) {
+          throw new Error(label + ' validation failed: ' + result.issues.map(issue => issue.message).join(', '));
+        }
+        return result.value;
+      }
+      if (typeof schema.parseAsync === 'function') return schema.parseAsync(value);
+      if (typeof schema.parse === 'function') return schema.parse(value);
+      return value;
+    }
+
+    async function executeTemporalStep(args, context) {
+      const result = await args.execute(context);
+      return context.__getTemporalSuspension() ?? result;
+    }
+
     function createStep(args) {
       return {
         inputSchema: args.inputSchema,
         requestContextSchema: args.requestContextSchema,
+        suspendSchema: args.suspendSchema,
+        resumeSchema: args.resumeSchema,
         execute: async (params) => {
           ${mastraImportPath ? `
       const { mastra } = await import(${JSON.stringify(mastraImportPath)});
-      return args.execute({ ...withRequestContext(params), mastra });
-    ` : hasMastraBinding ? "return args.execute({ ...withRequestContext(params), mastra });" : "return args.execute(withRequestContext(params));"}
+      return executeTemporalStep(args, { ...withRequestContext(params, args), mastra });
+    ` : hasMastraBinding ? "return executeTemporalStep(args, { ...withRequestContext(params, args), mastra });" : "return executeTemporalStep(args, withRequestContext(params, args));"}
         },
       };
     }