feat: support more input parsers

docs/src/app/(docs)/file-routes/page.mdx

@@ -171,10 +171,26 @@ f(["image"])
 
 ## `input` {{ since: '5.0' }}
 
-You can pass a [Zod](https://github.com/colinhacks/zod) schema to validate user
-input from the client. This data comes from the client when the upload starts.
-If validation here fails, an error will be thrown and none of your `middleware`
-n'or `onUploadComplete` functions will be executed.
+You can pass a schema validator to validate user input from the client. This
+data comes from the client when the upload starts. If validation here fails, an
+error will be thrown and none of your `middleware` n'or `onUploadComplete`
+functions will be executed.
+
+<Note>
+  The schema's **input type** must only contain JSON serializable types as
+  UploadThing does no special data transforming to handle non-JSON types. For
+  example, `z.date()` is invalid but `z.string().transform(Date)` is valid.
+</Note>
+
+Historically only [Zod](https://github.com/colinhacks/zod) was supported, but
+since `7.4` the following validators are supported:
+
+- [Zod](https://github.com/colinhacks/zod)
+- [Effect/Schema](https://effect.website/docs/datatypes/schema)
+- Any validator that implements the
+  [Standard Schema specification](https://github.com/standard-schema/standard-schema),
+  for example [Valibot >=1](https://github.com/fabian-hiller/valibot) and
+  [ArkType](https://github.com/arktypeio/arktype)
 
 The input is validated on **your** server and only leaves your server if you
 pass it along from the `.middleware` to the `.onUploadComplete`. If you only use

packages/uploadthing/package.json

@@ -151,6 +151,7 @@
   },
   "dependencies": {
     "@effect/platform": "0.69.24",
+    "@standard-schema/spec": "1.0.0-beta.3",
     "@uploadthing/mime-types": "workspace:*",
     "@uploadthing/shared": "workspace:*",
     "effect": "3.10.15"
@@ -176,6 +177,7 @@
     "type-fest": "^4.10.3",
     "typescript": "^5.5.2",
     "undici": "^6.6.2",
+    "valibot": "1.0.0-beta.7",
     "vue": "^3.4.21",
     "wait-on": "^7.2.0",
     "zod": "^3.23.8"

packages/uploadthing/src/internal/handler.ts

@@ -460,7 +460,7 @@ const handleUploadAction = (opts: {
     // validate the input
     yield* Effect.logDebug("Parsing user input");
     const parsedInput = yield* Effect.tryPromise({
-      try: async () => getParseFn(uploadable.inputParser)(json.input),
+      try: () => getParseFn(uploadable.inputParser)(json.input),
       catch: (error) =>
         new UploadThingError({
           code: "BAD_REQUEST",

packages/uploadthing/src/internal/parser.ts

@@ -1,25 +1,72 @@
-import type { Json, MaybePromise } from "@uploadthing/shared";
+import type { v1 } from "@standard-schema/spec";
+import * as Cause from "effect/Cause";
+import * as Runtime from "effect/Runtime";
+import * as Schema from "effect/Schema";
+
+import type { Json } from "@uploadthing/shared";
 
 /**
  * TODO: Do we wanna support effect/schema parsers now??
  */
 
-// Don't want to use Zod cause it's an optional dependency
-export type ParseFn<TType> = (input: unknown) => MaybePromise<TType>;
-export type ParserZodEsque<TInput, TParsedInput extends Json> = {
+export type ParseFn<TType> = (input: unknown) => Promise<TType>;
+
+export type ParserZodEsque<TInput extends Json, TParsedInput> = {
   _input: TInput;
   _output: TParsedInput; // if using .transform etc
-  parse: ParseFn<TParsedInput>;
+  parseAsync: ParseFn<TParsedInput>;
 };
 
+export type ParserStandardSchemaEsque<TInput, TParsedInput> = v1.StandardSchema<
+  TInput,
+  TParsedInput
+>;
+
 // In case we add support for more parsers later
-export type JsonParser = ParserZodEsque<Json, Json>;
+export type JsonParser<In extends Json = Json, Out = In> =
+  | ParserZodEsque<In, Out>
+  | ParserStandardSchemaEsque<In, Out>
+  | Schema.Schema<Out, In>;
+
+export function getParseFn<
+  TOut extends Json,
+  TParser extends JsonParser<any, TOut>,
+>(parser: TParser): ParseFn<TOut> {
+  if ("~standard" in parser) {
+    /**
+     * Standard Schema
+     */
+    return async (value) => {
+      const result = await parser["~standard"].validate(value);
+      if (result.issues) {
+        throw new Error(
+          "Input validation failed. See validation issues in the error cause.",
+          { cause: result.issues },
+        );
+      }
+      return result.value;
+    };
+  }
+
+  if ("parseAsync" in parser && typeof parser.parseAsync === "function") {
+    /**
+     * Zod
+     */
+    return parser.parseAsync;
+  }
 
-export function getParseFn<TParser extends JsonParser>(
-  parser: TParser,
-): ParseFn<TParser["_output"]> {
-  if (typeof parser.parse === "function") {
-    return parser.parse;
+  if (Schema.isSchema(parser)) {
+    /**
+     * Effect Schema
+     */
+    return (value) =>
+      Schema.decodeUnknownPromise(parser as Schema.Schema<any, TOut>)(
+        value,
+      ).catch((error) => {
+        throw Cause.squash(
+          (error as Runtime.FiberFailure)[Runtime.FiberFailureCauseId],
+        );
+      });
   }
 
   throw new Error("Invalid parser");

packages/uploadthing/src/internal/types.ts

@@ -97,13 +97,13 @@ type UploadErrorFn<TArgs extends AdapterFnArgs<any, any, any>> = (
 ) => MaybePromise<void>;
 
 export interface UploadBuilder<TParams extends AnyParams> {
-  input: <TParser extends JsonParser>(
+  input: <TIn extends Json, TOut>(
     parser: TParams["_input"]["in"] extends UnsetMarker
-      ? TParser
+      ? JsonParser<TIn, TOut>
       : ErrorMessage<"input is already set">,
   ) => UploadBuilder<{
     _routeOptions: TParams["_routeOptions"];
-    _input: { in: TParser["_input"]; out: TParser["_output"] };
+    _input: { in: TIn; out: TOut };
     _metadata: TParams["_metadata"];
     _adapterFnArgs: TParams["_adapterFnArgs"];
     _errorShape: TParams["_errorShape"];

packages/uploadthing/src/internal/upload-builder.ts

@@ -6,6 +6,7 @@ import type {
 } from "@uploadthing/shared";
 
 import { defaultErrorFormatter } from "./error-formatter";
+import type { JsonParser } from "./parser";
 import type {
   AdapterFnArgs,
   AnyBuiltUploaderTypes,
@@ -42,7 +43,7 @@ function internalCreateBuilder<
     },
 
     inputParser: {
-      parse: () => undefined,
+      parseAsync: () => Promise.resolve(undefined),
       _input: undefined,
       _output: undefined,
     },
@@ -63,7 +64,7 @@ function internalCreateBuilder<
     input(userParser) {
       return internalCreateBuilder({
         ..._def,
-        inputParser: userParser,
+        inputParser: userParser as JsonParser,
       }) as UploadBuilder<any>;
     },
     middleware(userMiddleware) {

packages/uploadthing/test/input.test.ts

@@ -0,0 +1,208 @@
+import { ParseError } from "effect/ParseResult";
+import * as Schema from "effect/Schema";
+import * as v from "valibot";
+import { expect, expectTypeOf, it } from "vitest";
+import * as z from "zod";
+
+import { noop } from "@uploadthing/shared";
+
+import { getParseFn } from "../src/internal/parser";
+import { createBuilder } from "../src/internal/upload-builder";
+import type { inferEndpointInput } from "../src/types";
+
+const f = createBuilder<{ req: Request; res: undefined; event: undefined }>();
+
+it.each([
+  ["zod", z.string()],
+  ["valibot", v.string()],
+  ["effect/schema", Schema.String],
+])("primitive string schema (%s)", async (_, input) => {
+  const fileRoute = f(["image"])
+    .input(input)
+    .middleware((opts) => {
+      expectTypeOf<string>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<string>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)("bar");
+  expect(parsedInput).toEqual("bar");
+});
+
+it.each([
+  ["zod", z.object({ foo: z.string() })],
+  ["valibot", v.object({ foo: v.string() })],
+  ["effect/schema", Schema.Struct({ foo: Schema.String })],
+])("object input (%s)", async (_, input) => {
+  const fileRoute = f(["image"])
+    .input(input)
+    .middleware((opts) => {
+      expectTypeOf<{ foo: string }>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<{ foo: string }>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)({ foo: "bar" });
+  expect(parsedInput).toEqual({ foo: "bar" });
+});
+
+it.each([
+  ["zod", z.object({ foo: z.string() }).optional()],
+  ["valibot", v.optional(v.object({ foo: v.string() }))],
+  // FIXME: Effect's optional schema wraps the entire type which makes it incompatible with the current approach
+  //   [
+  //     "effect/schema",
+  //     Schema.Struct({ foo: Schema.String }).pipe(Schema.optional),
+  //   ],
+])("optional input (%s)", async (_, input) => {
+  const fileRoute = f(["image"])
+    .input(input)
+    .middleware((opts) => {
+      expectTypeOf<{ foo: string } | undefined>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<{ foo: string } | undefined>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)({ foo: "bar" });
+  expect(parsedInput).toEqual({ foo: "bar" });
+});
+
+it("validation fails when input is invalid (zod)", async () => {
+  const fileRoute = f(["image"]).input(z.string()).onUploadComplete(noop);
+  const err = await getParseFn(fileRoute.inputParser)(123).catch((e) => e);
+  expect(err).toBeInstanceOf(z.ZodError);
+});
+it("validation fails when input is invalid (valibot)", async () => {
+  const fileRoute = f(["image"]).input(v.string()).onUploadComplete(noop);
+  const err = await getParseFn(fileRoute.inputParser)(123).catch((e) => e);
+  expect(err).toBeInstanceOf(Error);
+  expect(Array.isArray(err.cause)).toBe(true);
+});
+it("validation fails when input is invalid (effect/schema)", async () => {
+  const fileRoute = f(["image"]).input(Schema.String).onUploadComplete(noop);
+  const err = await getParseFn(fileRoute.inputParser)(123).catch((e) => e);
+  expect(err).toBeInstanceOf(ParseError);
+});
+
+it("with data transforming (zod)", async () => {
+  f(["image"]).input(
+    // @ts-expect-error - Date -> string is not JSON serializable
+    z.object({
+      date: z.date().transform((s) => s.toISOString()),
+    }),
+  );
+
+  // string -> Date should work
+  const fileRoute = f(["image"])
+    .input(
+      z.object({
+        date: z.string().transform((s) => new Date(s)),
+      }),
+    )
+    .middleware((opts) => {
+      expectTypeOf<{ date: Date }>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<{ date: string }>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)({
+    date: "2024-01-01",
+  });
+  expect(parsedInput).toEqual({ date: new Date("2024-01-01") });
+});
+it("with data transforming (valibot)", async () => {
+  f(["image"]).input(
+    // @ts-expect-error - Date -> string is not JSON serializable
+    v.object({
+      date: v.pipe(
+        v.date(),
+        v.transform((d) => d.toISOString()),
+      ),
+    }),
+  );
+
+  // string -> Date should work
+  const fileRoute = f(["image"])
+    .input(
+      v.object({
+        date: v.pipe(
+          v.string(),
+          v.transform((s) => new Date(s)),
+        ),
+      }),
+    )
+    .middleware((opts) => {
+      expectTypeOf<{ date: Date }>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<{ date: string }>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)({
+    date: "2024-01-01",
+  });
+  expect(parsedInput).toEqual({ date: new Date("2024-01-01") });
+});
+it("with data transforming (effect/schema)", async () => {
+  const fileRoute = f(["image"])
+    .input(
+      Schema.Struct({
+        date: Schema.Date,
+      }),
+    )
+    .middleware((opts) => {
+      expectTypeOf<{ date: Date }>(opts.input);
+      return {};
+    })
+    .onUploadComplete(noop);
+
+  type Input = inferEndpointInput<typeof fileRoute>;
+  expectTypeOf<Input>().toMatchTypeOf<{ date: string }>();
+
+  const parsedInput = await getParseFn(fileRoute.inputParser)({
+    date: "2024-01-01",
+  });
+  expect(parsedInput).toEqual({ date: new Date("2024-01-01") });
+});
+
+it("type errors for non-JSON data types (zod)", () => {
+  f(["image"])
+    // @ts-expect-error - Set is not a valid JSON type
+    .input(z.object({ foo: z.set(z.string()) }))
+    .middleware((opts) => {
+      return {};
+    })
+    .onUploadComplete(noop);
+});
+it("type errors for non-JSON data types (valibot)", () => {
+  f(["image"])
+    // @ts-expect-error - Set is not a valid JSON type
+    .input(v.object({ foo: v.set(v.string()) }))
+    .middleware((opts) => {
+      return {};
+    })
+    .onUploadComplete(noop);
+});
+it("type errors for non-JSON data types (effect/schema)", () => {
+  f(["image"])
+    // @ts-expect-error - Set is not a valid JSON type
+    .input(Schema.Struct({ foo: Schema.Set(Schema.String) }))
+    .middleware((opts) => {
+      return {};
+    })
+    .onUploadComplete(noop);
+});