feat: support more input parsers

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,48 @@
-import type { Json, MaybePromise } from "@uploadthing/shared";
+import type { v1 } from "@standard-schema/spec";
+
+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 ParseFn<TType> = (input: unknown) => Promise<TType>;
+
 export type ParserZodEsque<TInput, TParsedInput extends Json> = {
   _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 extends Json = Json> =
+  | ParserZodEsque<In, Out>
+  | ParserStandardSchemaEsque<In, Out>;
+
+export function getParseFn<
+  TOut extends Json,
+  TParser extends JsonParser<any, TOut>,
+>(parser: TParser): ParseFn<TOut> {
+  if ("~standard" in parser) {
+    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;
+    };
+  }
 
-export function getParseFn<TParser extends JsonParser>(
-  parser: TParser,
-): ParseFn<TParser["_output"]> {
-  if (typeof parser.parse === "function") {
-    return parser.parse;
+  if (typeof parser.parseAsync === "function") {
+    return parser.parseAsync;
   }
 
   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 extends Json>(
     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

@@ -42,7 +42,7 @@ function internalCreateBuilder<
     },
 
     inputParser: {
-      parse: () => undefined,
+      parseAsync: () => Promise.resolve(undefined),
       _input: undefined,
       _output: undefined,
     },

packages/uploadthing/test/upload-builder.test.ts

@@ -1,8 +1,10 @@
 /* eslint-disable @typescript-eslint/no-empty-function */
 
+import * as v from "valibot";
 import { expect, expectTypeOf, it } from "vitest";
 import { z } from "zod";
 
+import { getParseFn } from "../src/internal/parser";
 import { UTFiles } from "../src/internal/types";
 import { createBuilder } from "../src/internal/upload-builder";
 
@@ -115,6 +117,13 @@ it("with input", () => {
       expectTypeOf<{ foo: string }>(opts.input);
       return {};
     });
+
+  f(["image"])
+    .input(v.object({ foo: v.string() }))
+    .middleware((opts) => {
+      expectTypeOf<{ foo: string }>(opts.input);
+      return {};
+    });
 });
 
 it("with optional input", () => {
@@ -125,6 +134,13 @@ it("with optional input", () => {
       expectTypeOf<{ foo: string } | undefined>(opts.input);
       return {};
     });
+
+  f(["image"])
+    .input(v.optional(v.object({ foo: v.string() })))
+    .middleware((opts) => {
+      expectTypeOf<{ foo: string } | undefined>(opts.input);
+      return {};
+    });
 });
 
 it("can append a customId", () => {
@@ -140,7 +156,7 @@ it("can append a customId", () => {
     });
 });
 
-it("smoke", async () => {
+it("smoke (zod)", async () => {
   const f = createBuilder<{ req: Request; res: undefined; event: undefined }>();
 
   const uploadable = f(["image", "video"])
@@ -167,11 +183,54 @@ it("smoke", async () => {
 
   expect(uploadable.routerConfig).toEqual(["image", "video"]);
 
+  const parsedInput = await getParseFn(uploadable.inputParser)({ foo: "bar" });
+
+  const metadata = await uploadable.middleware({
+    req: new Request("http://localhost", {
+      headers: { header1: "woohoo" },
+    }),
+    input: parsedInput,
+    res: undefined,
+    event: undefined,
+    files: [{ name: "test.txt", size: 123456, type: "text/plain" }],
+  });
+  expect(metadata).toEqual({ header1: "woohoo", userId: "123" });
+});
+
+it("smoke (valibot)", async () => {
+  const f = createBuilder<{ req: Request; res: undefined; event: undefined }>();
+
+  const uploadable = f(["image", "video"])
+    .input(v.object({ foo: v.string() }))
+    .middleware((opts) => {
+      expect(opts.input).toEqual({ foo: "bar" });
+      expectTypeOf<{ foo: string }>(opts.input);
+      expectTypeOf<readonly { name: string; size: number }[]>(opts.files);
+
+      const header1 = opts.req.headers.get("header1");
+
+      return { header1, userId: "123" as const };
+    })
+    .onUploadComplete(({ file, metadata }) => {
+      // expect(file).toEqual({ name: "file", url: "http://localhost" })
+      expectTypeOf<{ name: string; url: string }>(file);
+
+      expect(metadata).toEqual({ header1: "woohoo", userId: "123" });
+      expectTypeOf<{
+        header1: string | null;
+        userId: "123";
+      }>(metadata);
+    });
+
+  expect(uploadable.routerConfig).toEqual(["image", "video"]);
+
+  const parsedInput = await getParseFn(uploadable.inputParser)({ foo: "bar" });
+
   const metadata = await uploadable.middleware({
     req: new Request("http://localhost", {
       headers: { header1: "woohoo" },
     }),
-    input: { foo: "bar" },
+    input: parsedInput,
     res: undefined,
     event: undefined,
     files: [{ name: "test.txt", size: 123456, type: "text/plain" }],