shikishikichangchang•2mo ago

Override output schema for certain mutations

Is it possible to overrider output schema's type for certain mutations? Currently I have a base procedure that has an outputschema as such:

import { z } from "zod";

const successSchema = z.object({
    status: z.literal("success"),
    data: z.any(),
});

const errorSchema = z.object({
    status: z.literal("error"),
    errorMessage: z.string(),
});

export const outputSchema = z.discriminatedUnion("status", [successSchema, errorSchema]);
export type Output = z.infer<typeof outputSchema>;

import { z } from "zod";

const successSchema = z.object({
    status: z.literal("success"),
    data: z.any(),
});

const errorSchema = z.object({
    status: z.literal("error"),
    errorMessage: z.string(),
});

export const outputSchema = z.discriminatedUnion("status", [successSchema, errorSchema]);
export type Output = z.infer<typeof outputSchema>;

As you can see, data is z.any(). For most mutations I don't need to return the data so that's not a problem. However, I want to return the data with the correct type for some. The output must have either z.literal("success") or z.literal("error"). I tried to add an extra '.output' but it seems like trpc only allows for one (no chaining to override).

1 Reply

Alex / KATT 🐱•2mo ago

https://github.com/trpc/trpc/issues/5787

GitHub

feat(server): extending .output()s · Issue #5787 · trpc/trpc

Provide environment information System: OS: macOS 14.5 CPU: (10) arm64 Apple M1 Max Memory: 56.36 MB / 32.00 GB Shell: 5.9 - /bin/zsh Binaries: Node: 20.11.1 - ~/.nvm/versions/node/v20.11.1/bin/nod...