import type { z } from 'zod';

import { prisma } from '@documenso/prisma';
import {
  DocumentDataSchema,
  FieldSchema,
  RecipientSchema,
  TemplateDirectLinkSchema,
  TemplateMetaSchema,
  TemplateSchema,
  UserSchema,
} from '@documenso/prisma/generated/zod';

import { AppError, AppErrorCode } from '../../errors/app-error';

export type GetTemplateByIdOptions = {
  id: number;
  userId: number;
  teamId?: number;
};

export const ZGetTemplateByIdResponseSchema = TemplateSchema.extend({
  directLink: TemplateDirectLinkSchema.nullable(),
  templateDocumentData: DocumentDataSchema,
  templateMeta: TemplateMetaSchema.nullable(),
  recipients: RecipientSchema.array(),
  fields: FieldSchema.array(),
  user: UserSchema.pick({
    id: true,
    name: true,
    email: true,
  }),
});

export type TGetTemplateByIdResponse = z.infer<typeof ZGetTemplateByIdResponseSchema>;

export const getTemplateById = async ({
  id,
  userId,
  teamId,
}: GetTemplateByIdOptions): Promise<TGetTemplateByIdResponse> => {
  const template = await prisma.template.findFirst({
    where: {
      id,
      ...(teamId
        ? {
            team: {
              id: teamId,
              members: {
                some: {
                  userId,
                },
              },
            },
          }
        : {
            userId,
            teamId: null,
          }),
    },
    include: {
      directLink: true,
      templateDocumentData: true,
      templateMeta: true,
      recipients: true,
      fields: true,
      user: {
        select: {
          id: true,
          name: true,
          email: true,
        },
      },
    },
  });

  if (!template) {
    throw new AppError(AppErrorCode.NOT_FOUND, {
      message: 'Template not found',
    });
  }

  return template;
};