db.ts

const fileSchema = z.union([
  z.object({
    type: z.literal('image'),
    width: z.number().int(),
    height: z.number().int(),
  }),
  z.object({
    type: z.literal('pdf'),
    pages: z.number().int(),
  }),
]);

export const prisma = new PrismaClient().$extends({
  model: {
    file: {
      async findUniqueExtended(args: Prisma.FileFindUniqueArgs) {
        const file = await prisma.file.findUnique({
          ...args,
          include: {
            ...args.include,
            fileImage: true,
            filePdf: true,
          },
        });

        return fileSchema.parse({
          ...file.fileImage,
          ...file.filePdf,
          ...file,
        });
      },
    },
  },
});