import type { z } from 'zod';

import { prisma } from '@documenso/prisma';
import type { Prisma, Template } from '@documenso/prisma/client';
import {
  DocumentDataSchema,
  FieldSchema,
  RecipientSchema,
  TeamSchema,
  TemplateDirectLinkSchema,
  TemplateMetaSchema,
  TemplateSchema,
} from '@documenso/prisma/generated/zod';

import { type FindResultResponse, ZFindResultResponse } from '../../types/search-params';

export type FindTemplatesOptions = {
  userId: number;
  teamId?: number;
  type?: Template['type'];
  page?: number;
  perPage?: number;
};

export const ZFindTemplatesResponseSchema = ZFindResultResponse.extend({
  data: TemplateSchema.extend({
    templateDocumentData: DocumentDataSchema,
    team: TeamSchema.pick({
      id: true,
      url: true,
    }).nullable(),
    Field: FieldSchema.array(),
    Recipient: RecipientSchema.array(),
    templateMeta: TemplateMetaSchema.pick({
      signingOrder: true,
      distributionMethod: true,
    }).nullable(),
    directLink: TemplateDirectLinkSchema.pick({
      token: true,
      enabled: true,
    }).nullable(),
  }).array(), // Todo: openapi.
});

export type TFindTemplatesResponse = z.infer<typeof ZFindTemplatesResponseSchema>;
export type FindTemplateRow = TFindTemplatesResponse['data'][number];

export const findTemplates = async ({
  userId,
  teamId,
  type,
  page = 1,
  perPage = 10,
}: FindTemplatesOptions): Promise<TFindTemplatesResponse> => {
  let whereFilter: Prisma.TemplateWhereInput = {
    userId,
    teamId: null,
    type,
  };

  if (teamId !== undefined) {
    whereFilter = {
      team: {
        id: teamId,
        members: {
          some: {
            userId,
          },
        },
      },
    };
  }

  const [data, count] = await Promise.all([
    prisma.template.findMany({
      where: whereFilter,
      include: {
        templateDocumentData: true,
        team: {
          select: {
            id: true,
            url: true,
          },
        },
        Field: true,
        Recipient: true,
        templateMeta: {
          select: {
            signingOrder: true,
            distributionMethod: true,
          },
        },
        directLink: {
          select: {
            token: true,
            enabled: true,
          },
        },
      },
      skip: Math.max(page - 1, 0) * perPage,
      orderBy: {
        createdAt: 'desc',
      },
    }),
    prisma.template.count({
      where: whereFilter,
    }),
  ]);

  return {
    data,
    count,
    currentPage: Math.max(page, 1),
    perPage,
    totalPages: Math.ceil(count / perPage),
  } satisfies FindResultResponse<typeof data>;
};