@redflow/client 0.0.1

package/src/workflow.ts

@@ -0,0 +1,42 @@
+import type { ZodTypeAny } from "zod";
+import { getDefaultClient } from "./default";
+import { validateInputWithSchema } from "./client";
+import { getDefaultRegistry, type WorkflowHandler, type WorkflowHandlerContext } from "./registry";
+import type { DefineWorkflowOptions, RunHandle, RunOptions } from "./types";
+
+export type InferInput<TSchema extends ZodTypeAny | undefined> = TSchema extends ZodTypeAny ? import("zod").infer<TSchema> : unknown;
+
+export type Workflow<TInput, TOutput> = {
+  name: string;
+  run(input: TInput, options?: RunOptions): Promise<RunHandle<TOutput>>;
+};
+
+export function defineWorkflow<TSchema extends ZodTypeAny | undefined, TOutput>(
+  options: DefineWorkflowOptions<TSchema>,
+  handler: WorkflowHandler<InferInput<TSchema>, TOutput>,
+): Workflow<InferInput<TSchema>, TOutput> {
+  getDefaultRegistry().register({ options, handler });
+
+  const name = options.name;
+  const queue = options.queue ?? "default";
+  const configuredMaxAttempts = options.retries?.maxAttempts;
+  const maxAttemptsOverride =
+    typeof configuredMaxAttempts === "number" && Number.isFinite(configuredMaxAttempts) && configuredMaxAttempts > 0
+      ? Math.floor(configuredMaxAttempts)
+      : undefined;
+
+  return {
+    name,
+    async run(input, runOptions) {
+      const parsedInput = options.schema ? (validateInputWithSchema(options.schema, input) as InferInput<TSchema>) : input;
+      const client = getDefaultClient();
+      return await client.runByName<TOutput>(name, parsedInput, {
+        ...runOptions,
+        queueOverride: runOptions?.queueOverride ?? queue,
+        __maxAttemptsOverride: maxAttemptsOverride,
+      });
+    },
+  };
+}
+
+export type { WorkflowHandler, WorkflowHandlerContext };