pg-workflows 0.13.0 → 0.13.2

package/dist/index.js

@@ -1,6 +1,7 @@
 import {
   DEFAULT_PGBOSS_SCHEMA,
   PAUSE_EVENT_NAME,
+  STEP_BASE_METHOD_TYPES,
   WORKFLOW_RUN_DLQ_QUEUE_NAME,
   WORKFLOW_RUN_QUEUE_NAME,
   WorkflowClient,
@@ -22,7 +23,7 @@ import {
   waitForTimelineKey,
   withPostgresTransaction,
   workflow
-} from "./shared/chunk-5xswmve7.js";
+} from "./shared/chunk-hjycwrsq.js";
 // src/engine.ts
 import { merge } from "es-toolkit";
 import pg from "pg";
@@ -73,11 +74,11 @@ function parseWorkflowHandler(handler) {
       const propertyAccess = node.expression;
       const objectName = propertyAccess.expression.getText(sourceFile);
       const methodName = propertyAccess.name.text;
-      if (objectName === "step" && (methodName === "run" || methodName === "waitFor" || methodName === "pause" || methodName === "waitUntil" || methodName === "delay" || methodName === "sleep" || methodName === "poll" || methodName === "invokeChildWorkflow")) {
+      const stepType = objectName === "step" ? STEP_BASE_METHOD_TYPES.get(methodName) : undefined;
+      if (stepType !== undefined) {
         const firstArg = node.arguments[0];
         if (firstArg) {
           const { id, isDynamic } = extractStepId(firstArg);
-          const stepType = methodName === "sleep" ? "delay" /* DELAY */ : methodName;
           const stepDefinition = {
             id,
             type: stepType,
@@ -243,6 +244,7 @@ class WorkflowEngine {
       for (const workflow2 of this.unregisteredWorkflows.values()) {
         await this.registerWorkflow(workflow2);
       }
+      this.unregisteredWorkflows.clear();
     }
     const mainQueueOptions = {
       retryLimit: 0,
@@ -380,6 +382,9 @@ class WorkflowEngine {
     });
     return run;
   }
+  async createChildWorkflowRun(params) {
+    return this.createWorkflowRun(params);
+  }
   async createWorkflowRun({
     workflowId,
     input,
@@ -766,75 +771,7 @@ class WorkflowEngine {
           }, { db });
         }, this.pool);
       }
-      const baseStep = {
-        run: async (stepId, handler) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          return this.runStep({ stepId, run, handler });
-        },
-        waitFor: async (stepId, { eventName, timeout }) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const timeoutDate = timeout ? new Date(Date.now() + timeout) : undefined;
-          return this.waitStep({ run, stepId, eventName, timeoutDate });
-        },
-        waitUntil: async (stepId, dateOrOptions) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const date = dateOrOptions instanceof Date ? dateOrOptions : typeof dateOrOptions === "string" ? new Date(dateOrOptions) : dateOrOptions.date instanceof Date ? dateOrOptions.date : new Date(dateOrOptions.date);
-          await this.waitStep({ run, stepId, timeoutDate: date });
-        },
-        pause: async (stepId) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          await this.waitStep({ run, stepId, eventName: PAUSE_EVENT_NAME });
-        },
-        delay: async (stepId, duration) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          await this.waitStep({
-            run,
-            stepId,
-            timeoutDate: new Date(Date.now() + parseDuration(duration))
-          });
-        },
-        get sleep() {
-          return this.delay;
-        },
-        poll: async (stepId, conditionFn, options) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const intervalMs = parseDuration(options?.interval ?? "30s");
-          if (intervalMs < 30000) {
-            throw new WorkflowEngineError(`step.poll interval must be at least 30s (got ${intervalMs}ms)`, workflowId, runId);
-          }
-          const timeoutMs = options?.timeout ? parseDuration(options.timeout) : undefined;
-          return this.pollStep({ run, stepId, conditionFn, intervalMs, timeoutMs });
-        },
-        invokeChildWorkflow: async (stepId, refOrParams, inputArg, optionsArg) => {
-          if (!run) {
-            throw new WorkflowEngineError("Missing workflow run", workflowId, runId);
-          }
-          const resolvedChildCall = this.resolveWorkflowRunParameters(refOrParams, inputArg, optionsArg);
-          const childWorkflowInvocation = {
-            run,
-            stepId,
-            workflowId: resolvedChildCall.workflowId,
-            input: resolvedChildCall.input,
-            options: resolvedChildCall.options,
-            resourceId: resolvedChildCall.resourceId,
-            idempotencyKey: resolvedChildCall.idempotencyKey
-          };
-          return this.invokeChildWorkflowStep(childWorkflowInvocation);
-        }
-      };
-      let step = { ...baseStep };
+      let step = this.createBaseStep(run, workflowId, runId);
       const plugins = workflow2.plugins ?? [];
       const context = {
         input: run.input,
@@ -924,6 +861,55 @@ class WorkflowEngine {
       workflowId: run.workflowId
     });
   }
+  createBaseStep(run, workflowId, runId) {
+    return {
+      run: async (stepId, handler) => {
+        return this.runStep({ stepId, run, handler });
+      },
+      waitFor: async (stepId, { eventName, timeout }) => {
+        const timeoutDate = timeout ? new Date(Date.now() + timeout) : undefined;
+        return this.waitStep({ run, stepId, eventName, timeoutDate });
+      },
+      waitUntil: async (stepId, dateOrOptions) => {
+        const date = dateOrOptions instanceof Date ? dateOrOptions : typeof dateOrOptions === "string" ? new Date(dateOrOptions) : dateOrOptions.date instanceof Date ? dateOrOptions.date : new Date(dateOrOptions.date);
+        await this.waitStep({ run, stepId, timeoutDate: date });
+      },
+      pause: async (stepId) => {
+        await this.waitStep({ run, stepId, eventName: PAUSE_EVENT_NAME });
+      },
+      delay: async (stepId, duration) => {
+        await this.waitStep({
+          run,
+          stepId,
+          timeoutDate: new Date(Date.now() + parseDuration(duration))
+        });
+      },
+      get sleep() {
+        return this.delay;
+      },
+      poll: async (stepId, conditionFn, options) => {
+        const intervalMs = parseDuration(options?.interval ?? "30s");
+        if (intervalMs < 30000) {
+          throw new WorkflowEngineError(`step.poll interval must be at least 30s (got ${intervalMs}ms)`, workflowId, runId);
+        }
+        const timeoutMs = options?.timeout ? parseDuration(options.timeout) : undefined;
+        return this.pollStep({ run, stepId, conditionFn, intervalMs, timeoutMs });
+      },
+      invokeChildWorkflow: async (stepId, refOrParams, inputArg, optionsArg) => {
+        const resolvedChildCall = this.resolveWorkflowRunParameters(refOrParams, inputArg, optionsArg);
+        const childWorkflowInvocation = {
+          run,
+          stepId,
+          workflowId: resolvedChildCall.workflowId,
+          input: resolvedChildCall.input,
+          options: resolvedChildCall.options,
+          resourceId: resolvedChildCall.resourceId,
+          idempotencyKey: resolvedChildCall.idempotencyKey
+        };
+        return this.invokeChildWorkflowStep(childWorkflowInvocation);
+      }
+    };
+  }
   getCachedStepEntry(timeline, stepId) {
     const stepEntry = timeline[stepId];
     return stepEntry && typeof stepEntry === "object" && "output" in stepEntry ? stepEntry : null;
@@ -1014,7 +1000,7 @@ class WorkflowEngine {
         });
         return;
       }
-      const result = await this.createWorkflowRun({
+      const result = await this.createChildWorkflowRun({
         workflowId,
         input,
         resourceId: childResourceId,
@@ -1580,5 +1566,5 @@ export {
   WorkflowClient
 };
 
-//# debugId=5BC543336A07527964756E2164756E21
+//# debugId=EE11F9509955D59264756E2164756E21
 //# sourceMappingURL=index.js.map