Zod with custom type

type media_type = "all" | "movie" | "tv" | "person"

getTrending: publicProcedure
    .input(
      z.object({
        media_type: z.custom() //need the media_type check here)
      })
    )
    .query(({ input }) => {
      return {
        ...input
      };
    }),