@mastra/temporal 0.2.7-alpha.5 → 0.2.7-alpha.7

package/dist/worker.js

@@ -816,6 +816,30 @@ function createTemporalWorkflowHelperStatements() {
     return [statement];
   });
 }
+function getTemporalWorkflowRuntimeOptions(program3) {
+  for (const statement of program3.body) {
+    const declarationStatement = t3.isVariableDeclaration(statement) ? statement : t3.isExportNamedDeclaration(statement) && t3.isVariableDeclaration(statement.declaration) ? statement.declaration : null;
+    if (!declarationStatement) {
+      continue;
+    }
+    for (const declaration of declarationStatement.declarations) {
+      if (!isWorkflowHelperDestructure(declaration) || !t3.isCallExpression(declaration.init)) {
+        continue;
+      }
+      const [temporalParams] = declaration.init.arguments;
+      if (!temporalParams || !t3.isObjectExpression(temporalParams)) {
+        continue;
+      }
+      for (const property of temporalParams.properties) {
+        if (t3.isObjectProperty(property) && getObjectPropertyName(property) === "startToCloseTimeout" && t3.isExpression(property.value)) {
+          return t3.objectExpression([
+            t3.objectProperty(t3.identifier("startToCloseTimeout"), t3.cloneNode(property.value, true))
+          ]);
+        }
+      }
+    }
+  }
+}
 function parseWorkflowChain(node) {
   const methods = [];
   let current = node;
@@ -1011,8 +1035,12 @@ function rewriteChainMethod(method, filePath, workflowName, stepBindings, workfl
 function getExportedName(node) {
   return t3.isIdentifier(node) ? node.name : node.value;
 }
-function createTemporalWorkflowStatements(exportName, workflowId, methods, filePath, includeCommit, stepBindings, workflowBindings) {
-  let expression = t3.callExpression(t3.identifier("createWorkflow"), [t3.cloneNode(workflowId, true)]);
+function createTemporalWorkflowStatements(exportName, workflowId, methods, filePath, includeCommit, stepBindings, workflowBindings, runtimeOptions) {
+  const createWorkflowArgs = [t3.cloneNode(workflowId, true)];
+  if (runtimeOptions) {
+    createWorkflowArgs.push(t3.cloneNode(runtimeOptions, true));
+  }
+  let expression = t3.callExpression(t3.identifier("createWorkflow"), createWorkflowArgs);
   for (const method of methods) {
     const rewrittenMethod = rewriteChainMethod(method, filePath, exportName, stepBindings, workflowBindings);
     expression = t3.callExpression(
@@ -1077,7 +1105,8 @@ function createWorkflowTransformState(program3, filePath) {
     strippedNames: /* @__PURE__ */ new Set(),
     stepBindings: collectStepBindings(program3),
     workflowBindings: collectWorkflowBindings(program3, filePath),
-    workflowExports: []
+    workflowExports: [],
+    runtimeOptions: getTemporalWorkflowRuntimeOptions(program3)
   };
 }
 function collectWorkflowDeclarationMetadata(statement, state) {
@@ -1211,7 +1240,8 @@ function rewriteWorkflowVariableDeclaration(statement, filePath, state) {
         filePath,
         state.committedWorkflowNames.has(declaration.id.name),
         state.stepBindings,
-        state.workflowBindings
+        state.workflowBindings,
+        state.runtimeOptions
       )
     );
   }