feat: download options for document

packages/trpc/server/document-router/router.ts

@@ -1,5 +1,6 @@
 import { TRPCError } from '@trpc/server';
 import { DateTime } from 'luxon';
+import { PDFDocument } from 'pdf-lib';
 
 import { getServerLimits } from '@documenso/ee/server-only/limits/server';
 import { NEXT_PUBLIC_WEBAPP_URL } from '@documenso/lib/constants/app';
@@ -23,6 +24,7 @@ import { searchDocumentsWithKeyword } from '@documenso/lib/server-only/document/
 import { sendDocument } from '@documenso/lib/server-only/document/send-document';
 import { updateDocument } from '@documenso/lib/server-only/document/update-document';
 import { symmetricEncrypt } from '@documenso/lib/universal/crypto';
+import { getFile } from '@documenso/lib/universal/upload/get-file';
 import { getPresignPostUrl } from '@documenso/lib/universal/upload/server-actions';
 import { DocumentDataType, DocumentStatus } from '@documenso/prisma/client';
 
@@ -65,13 +67,84 @@ export const documentRouter = router({
     .input(ZGetDocumentByIdQuerySchema)
     .query(async ({ input, ctx }) => {
       const { teamId } = ctx;
-      const { documentId } = input;
+      const { documentId, includeCertificate, includeAuditLog } = input;
 
-      return await getDocumentById({
+      const documentWithData = await getDocumentById({
         userId: ctx.user.id,
         teamId,
         documentId,
       });
+
+      if (includeCertificate && includeAuditLog) {
+        return documentWithData;
+      } else if (includeCertificate) {
+        const pdfData = await getFile(documentWithData.documentData);
+        const pdfDoc = await PDFDocument.load(pdfData);
+
+        const totalPages = pdfDoc.getPageCount();
+
+        if (!includeAuditLog) {
+          pdfDoc.removePage(totalPages - 1);
+        }
+
+        const pdfBytes = await pdfDoc.save();
+        const pdfBuffer = Buffer.from(pdfBytes).toString('base64');
+
+        return {
+          ...documentWithData,
+          documentData: {
+            ...documentWithData.documentData,
+            data: pdfBuffer,
+            initialData: documentWithData.documentData.data,
+            type: DocumentDataType.BYTES_64,
+          },
+        };
+      } else if (includeAuditLog) {
+        const pdfData = await getFile(documentWithData.documentData);
+        const pdfDoc = await PDFDocument.load(pdfData);
+
+        const totalPages = pdfDoc.getPageCount();
+
+        if (!includeCertificate) {
+          pdfDoc.removePage(totalPages - 2);
+        }
+
+        const pdfBytes = await pdfDoc.save();
+        const pdfBuffer = Buffer.from(pdfBytes).toString('base64');
+
+        return {
+          ...documentWithData,
+          documentData: {
+            ...documentWithData.documentData,
+            data: pdfBuffer,
+            initialData: documentWithData.documentData.data,
+            type: DocumentDataType.BYTES_64,
+          },
+        };
+      } else if (!includeCertificate && !includeAuditLog) {
+        const pdfData = await getFile(documentWithData.documentData);
+        const pdfDoc = await PDFDocument.load(pdfData);
+
+        const totalPages = pdfDoc.getPageCount();
+
+        pdfDoc.removePage(totalPages - 1);
+        pdfDoc.removePage(totalPages - 2);
+
+        const pdfBytes = await pdfDoc.save();
+        const pdfBuffer = Buffer.from(pdfBytes).toString('base64');
+
+        return {
+          ...documentWithData,
+          documentData: {
+            ...documentWithData.documentData,
+            data: pdfBuffer,
+            initialData: documentWithData.documentData.data,
+            type: DocumentDataType.BYTES_64,
+          },
+        };
+      } else {
+        return documentWithData;
+      }
     }),
 
   /**

packages/trpc/server/document-router/schema.ts

@@ -141,6 +141,8 @@ export const ZFindDocumentAuditLogsQuerySchema = ZFindSearchParamsSchema.extend(
 
 export const ZGetDocumentByIdQuerySchema = z.object({
   documentId: z.number(),
+  includeCertificate: z.boolean().default(true).optional(),
+  includeAuditLog: z.boolean().default(true).optional(),
 });
 
 export const ZDuplicateDocumentRequestSchema = z.object({