Ryan/documenso
1
0
Fork 1
mirror of https://github.com/documenso/documenso.git synced 2025-11-20 03:32:14 +10:00
Code Issues Packages Projects Releases Wiki Activity

chore: redo changes

Browse Source
This commit is contained in:
Catalin Pit
2025-02-17 15:11:16 +02:00
parent 6a9ae132c7
commit a4f1a138d0
2 changed files with 3 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
packages/trpc/server/document-router/router.ts
View File

@ -1,6 +1,5 @@
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';
@ -24,7 +23,6 @@ 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';
@ -69,82 +67,11 @@ export const documentRouter = router({
const { teamId } = ctx;
const { documentId, includeCertificate, includeAuditLog } = input;
const documentWithData = await getDocumentById({
return 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;
}
}),
/**