documenso/packages/lib/server-only/document/duplicate-document-by-id.ts

import { z } from 'zod';

import { prisma } from '@documenso/prisma';
import { DocumentSource, type Prisma } from '@documenso/prisma/client';

import { getDocumentWhereInput } from './get-document-by-id';

export interface DuplicateDocumentOptions {
  documentId: number;
  userId: number;
  teamId?: number;
}

export const ZDuplicateDocumentResponseSchema = z.object({
  documentId: z.number(),
});

export type TDuplicateDocumentResponse = z.infer<typeof ZDuplicateDocumentResponseSchema>;

export const duplicateDocument = async ({
  documentId,
  userId,
  teamId,
}: DuplicateDocumentOptions): Promise<TDuplicateDocumentResponse> => {
  const documentWhereInput = await getDocumentWhereInput({
    documentId,
    userId,
    teamId,
  });

  const document = await prisma.document.findUniqueOrThrow({
    where: documentWhereInput,
    select: {
      title: true,
      userId: true,
      documentData: {
        select: {
          data: true,
          initialData: true,
          type: true,
        },
      },
      documentMeta: {
        select: {
          message: true,
          subject: true,
          dateFormat: true,
          password: true,
          timezone: true,
          redirectUrl: true,
        },
      },
    },
  });

  const createDocumentArguments: Prisma.DocumentCreateArgs = {
    data: {
      title: document.title,
      User: {
        connect: {
          id: document.userId,
        },
      },
      documentData: {
        create: {
          ...document.documentData,
          data: document.documentData.initialData,
        },
      },
      documentMeta: {
        create: {
          ...document.documentMeta,
        },
      },
      source: DocumentSource.DOCUMENT,
    },
  };

  if (teamId !== undefined) {
    createDocumentArguments.data.team = {
      connect: {
        id: teamId,
      },
    };
  }

  const createdDocument = await prisma.document.create(createDocumentArguments);

  return {
    documentId: createdDocument.id,
  };
};