pg-workflows 0.9.0 → 0.11.0

package/dist/client.entry.cjs

@@ -66,6 +66,8 @@ var exports_client_entry = {};
 __export(exports_client_entry, {
   createWorkflowRef: () => createWorkflowRef,
   WorkflowStatus: () => WorkflowStatus,
+  WorkflowRunNotFoundError: () => WorkflowRunNotFoundError,
+  WorkflowEngineError: () => WorkflowEngineError,
   WorkflowClient: () => WorkflowClient
 });
 module.exports = __toCommonJS(exports_client_entry);
@@ -82,10 +84,15 @@ var WORKFLOW_RUN_DLQ_QUEUE_NAME = "workflow_run_dlq";
 var DEFAULT_PGBOSS_SCHEMA = "pgboss_v12_pgworkflow";
 var MAX_WORKFLOW_ID_LENGTH = 256;
 var MAX_RESOURCE_ID_LENGTH = 256;
+var INVOKE_CHILD_WORKFLOW_TIMELINE_SUFFIX = "invoke-child-workflow";
+var WAIT_FOR_TIMELINE_SUFFIX = "wait-for";
+var invokeChildWorkflowTimelineKey = (stepId) => `${stepId}-${INVOKE_CHILD_WORKFLOW_TIMELINE_SUFFIX}`;
+var waitForTimelineKey = (stepId) => `${stepId}-${WAIT_FOR_TIMELINE_SUFFIX}`;
+var isInvokeChildWorkflowTimelineEntry = (entry) => !!entry && typeof entry === "object" && ("invokeChildWorkflow" in entry);
 
 // src/db/migration.ts
 var MIGRATION_LOCK_ID = 738291645;
-var CURRENT_SCHEMA_VERSION = 3;
+var CURRENT_SCHEMA_VERSION = 4;
 async function runMigrations(db) {
   if (await isSchemaUpToDate(db)) {
     return;
@@ -143,6 +150,11 @@ async function runMigrations(db) {
     commands.push("ALTER TABLE workflow_runs ALTER COLUMN resource_id TYPE varchar(256)");
     commands.push("ALTER TABLE workflow_runs ALTER COLUMN workflow_id TYPE varchar(256)");
   }
+  if (currentVersion < 4) {
+    commands.push("ALTER TABLE workflow_runs ADD COLUMN IF NOT EXISTS parent_run_id varchar(32)");
+    commands.push("ALTER TABLE workflow_runs ADD COLUMN IF NOT EXISTS parent_step_id varchar(256)");
+    commands.push("ALTER TABLE workflow_runs ADD COLUMN IF NOT EXISTS parent_resource_id varchar(256)");
+  }
   if (currentVersion === 0) {
     commands.push(`INSERT INTO workflow_schema_version (version) VALUES (${CURRENT_SCHEMA_VERSION})`);
   } else {
@@ -205,7 +217,10 @@ function mapRowToWorkflowRun(row) {
     retryCount: row.retry_count,
     maxRetries: row.max_retries,
     jobId: row.job_id,
-    idempotencyKey: row.idempotency_key
+    idempotencyKey: row.idempotency_key,
+    parentRunId: row.parent_run_id,
+    parentStepId: row.parent_step_id,
+    parentResourceId: row.parent_resource_id
   };
 }
 async function insertWorkflowRun({
@@ -216,7 +231,10 @@ async function insertWorkflowRun({
   input,
   maxRetries,
   timeoutAt,
-  idempotencyKey
+  idempotencyKey,
+  parentRunId,
+  parentStepId,
+  parentResourceId
 }, db) {
   const runId = generateKSUID("run");
   const now = new Date;
@@ -233,9 +251,12 @@ async function insertWorkflowRun({
       updated_at,
       timeline,
       retry_count,
-      idempotency_key
+      idempotency_key,
+      parent_run_id,
+      parent_step_id,
+      parent_resource_id
     )
-    VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
+    VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16)
     ON CONFLICT (idempotency_key) WHERE idempotency_key IS NOT NULL DO NOTHING
     RETURNING *`, [
     runId,
@@ -250,7 +271,10 @@ async function insertWorkflowRun({
     now,
     "{}",
     0,
-    idempotencyKey ?? null
+    idempotencyKey ?? null,
+    parentRunId ?? null,
+    parentStepId ?? null,
+    parentResourceId ?? null
   ]);
   if (result.rows[0]) {
     return { run: mapRowToWorkflowRun(result.rows[0]), created: true };
@@ -513,15 +537,6 @@ var WorkflowStatus;
   WorkflowStatus2["FAILED"] = "failed";
   WorkflowStatus2["CANCELLED"] = "cancelled";
 })(WorkflowStatus ||= {});
-var StepType;
-((StepType2) => {
-  StepType2["PAUSE"] = "pause";
-  StepType2["RUN"] = "run";
-  StepType2["WAIT_FOR"] = "waitFor";
-  StepType2["WAIT_UNTIL"] = "waitUntil";
-  StepType2["DELAY"] = "delay";
-  StepType2["POLL"] = "poll";
-})(StepType ||= {});
 
 // src/client.ts
 var LOG_PREFIX = "[WorkflowClient]";
@@ -628,7 +643,8 @@ class WorkflowClient {
         };
         await this.boss.send(WORKFLOW_RUN_QUEUE_NAME, job, {
           startAfter: new Date,
-          expireInSeconds: options?.expireInSeconds ?? defaultExpireInSeconds
+          expireInSeconds: options?.expireInSeconds ?? defaultExpireInSeconds,
+          db: _db
         });
       }
       return insertedRun;
@@ -691,6 +707,11 @@ class WorkflowClient {
     if (current.status !== "paused" /* PAUSED */) {
       throw new WorkflowEngineError(`Cannot resume workflow run in '${current.status}' status, must be 'paused'`, current.workflowId, runId);
     }
+    const currentStepId = current.currentStepId;
+    const currentStepTimelineEntry = current.timeline[invokeChildWorkflowTimelineKey(currentStepId)];
+    if (isInvokeChildWorkflowTimelineEntry(currentStepTimelineEntry)) {
+      return current;
+    }
     return this.triggerEvent({
       runId,
       resourceId,
@@ -709,8 +730,12 @@ class WorkflowClient {
     if (run.status !== "paused" /* PAUSED */) {
       return run;
     }
-    const stepId = run.currentStepId;
-    const waitForEntry = run.timeline[`${stepId}-wait-for`];
+    const currentStepId = run.currentStepId;
+    const currentStepTimelineEntry = run.timeline[invokeChildWorkflowTimelineKey(currentStepId)];
+    if (isInvokeChildWorkflowTimelineEntry(currentStepTimelineEntry)) {
+      return run;
+    }
+    const waitForEntry = run.timeline[waitForTimelineKey(currentStepId)];
     if (!waitForEntry || typeof waitForEntry !== "object" || !("waitFor" in waitForEntry)) {
       return run;
     }
@@ -728,7 +753,7 @@ class WorkflowClient {
           resourceId,
           data: {
             timeline: import_es_toolkit.merge(freshRun.timeline, {
-              [stepId]: {
+              [currentStepId]: {
                 output: data ?? {},
                 timestamp: new Date
               }
@@ -828,7 +853,10 @@ function createWorkflowRef(id, options) {
     retries: defineOptions?.retries
   });
   Object.defineProperty(ref, "id", { value: id, enumerable: true });
-  Object.defineProperty(ref, "inputSchema", { value: options?.inputSchema, enumerable: true });
+  Object.defineProperty(ref, "inputSchema", {
+    value: options?.inputSchema,
+    enumerable: true
+  });
   return ref;
 }
 function createWorkflowFactory(plugins = []) {
@@ -849,5 +877,5 @@ function createWorkflowFactory(plugins = []) {
 }
 var workflow = createWorkflowFactory();
 
-//# debugId=5A9A3072C311C6E064756E2164756E21
+//# debugId=AA7D4CB0B3AC99D364756E2164756E21
 //# sourceMappingURL=client.entry.js.map

package/dist/client.entry.d.cts

@@ -20,6 +20,9 @@ type WorkflowRun = {
 	maxRetries: number;
 	jobId: string | null;
 	idempotencyKey: string | null;
+	parentRunId: string | null;
+	parentStepId: string | null;
+	parentResourceId: string | null;
 };
 import { StandardSchemaV1 } from "@standard-schema/spec";
 type DurationObject = {
@@ -40,6 +43,13 @@ declare enum WorkflowStatus {
 }
 type InputParameters = StandardSchemaV1;
 type InferInputParameters<P extends InputParameters> = StandardSchemaV1.InferOutput<P>;
+type StartWorkflowOptions = {
+	resourceId?: string;
+	timeout?: number;
+	retries?: number;
+	expireInSeconds?: number;
+	idempotencyKey?: string;
+};
 type WorkflowOptions<I extends InputParameters> = {
 	timeout?: number;
 	retries?: number;
@@ -79,6 +89,23 @@ type StepBaseContext = {
 	} | {
 		timedOut: true;
 	}>;
+	/**
+	* Invoke a child workflow from inside the current workflow and pause until
+	* the child run reaches a terminal state.
+	*/
+	invokeChildWorkflow: {
+		<
+			TInput extends InputParameters,
+			TOutput = unknown
+		>(stepId: string, ref: WorkflowRef<TInput, TOutput>, input: InferInputParameters<TInput>, options?: StartWorkflowOptions): Promise<TOutput>;
+		<TOutput = unknown>(stepId: string, params: {
+			workflowId: string;
+			input: unknown;
+			resourceId?: string;
+			idempotencyKey?: string;
+			options?: StartWorkflowOptions;
+		}): Promise<TOutput>;
+	};
 };
 /**
 * Plugin that extends the workflow step API with extra methods.
@@ -119,7 +146,10 @@ type WorkflowDefinition<TInput extends InputParameters = InputParameters> = {
 *
 * Callable: pass a handler to create a full WorkflowDefinition.
 */
-interface WorkflowRef<TInput extends InputParameters = InputParameters> {
+interface WorkflowRef<
+	TInput extends InputParameters = InputParameters,
+	TOutput = unknown
+> {
 	(handler: (context: WorkflowContext<TInput, StepBaseContext>) => Promise<unknown>, options?: Omit<WorkflowOptions<TInput>, "inputSchema">): WorkflowDefinition<TInput>;
 	readonly id: string;
 	readonly inputSchema?: TInput;
@@ -149,13 +179,6 @@ type WorkflowClientOptions = {
 	connectionString: string;
 	pool?: never;
 });
-type StartWorkflowOptions = {
-	resourceId?: string;
-	timeout?: number;
-	retries?: number;
-	expireInSeconds?: number;
-	idempotencyKey?: string;
-};
 declare class WorkflowClient {
 	private boss;
 	private db;
@@ -231,7 +254,21 @@ declare class WorkflowClient {
 * Create a lightweight workflow reference.
 * Safe to import from `pg-workflows/client` - no engine or handler code.
 */
-declare function createWorkflowRef<TInput extends InputParameters = InputParameters>(id: string, options?: {
+declare function createWorkflowRef<
+	TOutput = unknown,
+	TInput extends InputParameters = InputParameters
+>(id: string, options?: {
 	inputSchema?: TInput;
-}): WorkflowRef<TInput>;
-export { createWorkflowRef, WorkflowStatus, WorkflowRunProgress, WorkflowRun, WorkflowRef, WorkflowLogger, WorkflowClientOptions, WorkflowClient, StartWorkflowOptions, InputParameters, InferInputParameters };
+}): WorkflowRef<TInput, TOutput>;
+import { StandardSchemaV1 as StandardSchemaV12 } from "@standard-schema/spec";
+declare class WorkflowEngineError extends Error {
+	readonly workflowId?: string | undefined;
+	readonly runId?: string | undefined;
+	readonly cause: Error | undefined;
+	readonly issues?: StandardSchemaV12.FailureResult["issues"] | undefined;
+	constructor(message: string, workflowId?: string | undefined, runId?: string | undefined, cause?: Error | undefined, issues?: StandardSchemaV12.FailureResult["issues"] | undefined);
+}
+declare class WorkflowRunNotFoundError extends WorkflowEngineError {
+	constructor(runId?: string, workflowId?: string);
+}
+export { createWorkflowRef, WorkflowStatus, WorkflowRunProgress, WorkflowRunNotFoundError, WorkflowRun, WorkflowRef, WorkflowLogger, WorkflowEngineError, WorkflowClientOptions, WorkflowClient, StartWorkflowOptions, InputParameters, InferInputParameters };

package/dist/client.entry.d.ts

@@ -20,6 +20,9 @@ type WorkflowRun = {
 	maxRetries: number;
 	jobId: string | null;
 	idempotencyKey: string | null;
+	parentRunId: string | null;
+	parentStepId: string | null;
+	parentResourceId: string | null;
 };
 import { StandardSchemaV1 } from "@standard-schema/spec";
 type DurationObject = {
@@ -40,6 +43,13 @@ declare enum WorkflowStatus {
 }
 type InputParameters = StandardSchemaV1;
 type InferInputParameters<P extends InputParameters> = StandardSchemaV1.InferOutput<P>;
+type StartWorkflowOptions = {
+	resourceId?: string;
+	timeout?: number;
+	retries?: number;
+	expireInSeconds?: number;
+	idempotencyKey?: string;
+};
 type WorkflowOptions<I extends InputParameters> = {
 	timeout?: number;
 	retries?: number;
@@ -79,6 +89,23 @@ type StepBaseContext = {
 	} | {
 		timedOut: true;
 	}>;
+	/**
+	* Invoke a child workflow from inside the current workflow and pause until
+	* the child run reaches a terminal state.
+	*/
+	invokeChildWorkflow: {
+		<
+			TInput extends InputParameters,
+			TOutput = unknown
+		>(stepId: string, ref: WorkflowRef<TInput, TOutput>, input: InferInputParameters<TInput>, options?: StartWorkflowOptions): Promise<TOutput>;
+		<TOutput = unknown>(stepId: string, params: {
+			workflowId: string;
+			input: unknown;
+			resourceId?: string;
+			idempotencyKey?: string;
+			options?: StartWorkflowOptions;
+		}): Promise<TOutput>;
+	};
 };
 /**
 * Plugin that extends the workflow step API with extra methods.
@@ -119,7 +146,10 @@ type WorkflowDefinition<TInput extends InputParameters = InputParameters> = {
 *
 * Callable: pass a handler to create a full WorkflowDefinition.
 */
-interface WorkflowRef<TInput extends InputParameters = InputParameters> {
+interface WorkflowRef<
+	TInput extends InputParameters = InputParameters,
+	TOutput = unknown
+> {
 	(handler: (context: WorkflowContext<TInput, StepBaseContext>) => Promise<unknown>, options?: Omit<WorkflowOptions<TInput>, "inputSchema">): WorkflowDefinition<TInput>;
 	readonly id: string;
 	readonly inputSchema?: TInput;
@@ -149,13 +179,6 @@ type WorkflowClientOptions = {
 	connectionString: string;
 	pool?: never;
 });
-type StartWorkflowOptions = {
-	resourceId?: string;
-	timeout?: number;
-	retries?: number;
-	expireInSeconds?: number;
-	idempotencyKey?: string;
-};
 declare class WorkflowClient {
 	private boss;
 	private db;
@@ -231,7 +254,21 @@ declare class WorkflowClient {
 * Create a lightweight workflow reference.
 * Safe to import from `pg-workflows/client` - no engine or handler code.
 */
-declare function createWorkflowRef<TInput extends InputParameters = InputParameters>(id: string, options?: {
+declare function createWorkflowRef<
+	TOutput = unknown,
+	TInput extends InputParameters = InputParameters
+>(id: string, options?: {
 	inputSchema?: TInput;
-}): WorkflowRef<TInput>;
-export { createWorkflowRef, WorkflowStatus, WorkflowRunProgress, WorkflowRun, WorkflowRef, WorkflowLogger, WorkflowClientOptions, WorkflowClient, StartWorkflowOptions, InputParameters, InferInputParameters };
+}): WorkflowRef<TInput, TOutput>;
+import { StandardSchemaV1 as StandardSchemaV12 } from "@standard-schema/spec";
+declare class WorkflowEngineError extends Error {
+	readonly workflowId?: string | undefined;
+	readonly runId?: string | undefined;
+	readonly cause: Error | undefined;
+	readonly issues?: StandardSchemaV12.FailureResult["issues"] | undefined;
+	constructor(message: string, workflowId?: string | undefined, runId?: string | undefined, cause?: Error | undefined, issues?: StandardSchemaV12.FailureResult["issues"] | undefined);
+}
+declare class WorkflowRunNotFoundError extends WorkflowEngineError {
+	constructor(runId?: string, workflowId?: string);
+}
+export { createWorkflowRef, WorkflowStatus, WorkflowRunProgress, WorkflowRunNotFoundError, WorkflowRun, WorkflowRef, WorkflowLogger, WorkflowEngineError, WorkflowClientOptions, WorkflowClient, StartWorkflowOptions, InputParameters, InferInputParameters };

package/dist/client.entry.js

@@ -1,13 +1,17 @@
 import {
   WorkflowClient,
+  WorkflowEngineError,
+  WorkflowRunNotFoundError,
   WorkflowStatus,
   createWorkflowRef
-} from "./shared/chunk-fr76gdwj.js";
+} from "./shared/chunk-ahxqsytt.js";
 export {
   createWorkflowRef,
   WorkflowStatus,
+  WorkflowRunNotFoundError,
+  WorkflowEngineError,
   WorkflowClient
 };
 
-//# debugId=B0186153DBB6496064756E2164756E21
+//# debugId=64DD8F1F4DBCBD8C64756E2164756E21
 //# sourceMappingURL=client.entry.js.map