From 0eee57078101fc31cfb656f768187aa25413514e Mon Sep 17 00:00:00 2001
From: Catalin Pit <25515812+catalinpit@users.noreply.github.com>
Date: Tue, 23 Apr 2024 12:33:40 +0300
Subject: [PATCH] fix: complete document when all recipients are CC

---
 .../documents/[id]/edit-document.tsx          | 13 ++++++++++++-
 .../trpc/server/document-router/router.ts     | 19 +++++++++++++++++++
 .../trpc/server/document-router/schema.ts     |  6 ++++++
 3 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/apps/web/src/app/(dashboard)/documents/[id]/edit-document.tsx b/apps/web/src/app/(dashboard)/documents/[id]/edit-document.tsx
index 2e2f0c889..fbc700219 100644
--- a/apps/web/src/app/(dashboard)/documents/[id]/edit-document.tsx
+++ b/apps/web/src/app/(dashboard)/documents/[id]/edit-document.tsx
@@ -8,6 +8,7 @@ import {
   DO_NOT_INVALIDATE_QUERY_ON_MUTATION,
   SKIP_QUERY_BATCH_META,
 } from '@documenso/lib/constants/trpc';
+import { DocumentStatus, RecipientRole } from '@documenso/prisma/client';
 import type { DocumentWithDetails } from '@documenso/prisma/types/document';
 import { trpc } from '@documenso/trpc/react';
 import { cn } from '@documenso/ui/lib/utils';
@@ -68,6 +69,10 @@ export const EditDocumentForm = ({
 
   const { Recipient: recipients, Field: fields } = document;
 
+  const allRecipientsAreCC = recipients.every((recipient) => recipient.role === RecipientRole.CC);
+
+  const { mutateAsync: updateDocumentStatus } = trpc.document.updateDocument.useMutation();
+
   const { mutateAsync: setSettingsForDocument } = trpc.document.setSettingsForDocument.useMutation({
     ...DO_NOT_INVALIDATE_QUERY_ON_MUTATION,
     onSuccess: (newData) => {
@@ -248,8 +253,14 @@ export const EditDocumentForm = ({
 
   const onAddSubjectFormSubmit = async (data: TAddSubjectFormSchema) => {
     const { subject, message } = data.meta;
-
     try {
+      if (allRecipientsAreCC) {
+        await updateDocumentStatus({
+          documentId: document.id,
+          data: { status: DocumentStatus.COMPLETED },
+        });
+      }
+
       await sendDocument({
         documentId: document.id,
         teamId: team?.id,
diff --git a/packages/trpc/server/document-router/router.ts b/packages/trpc/server/document-router/router.ts
index d12002674..fa9d95262 100644
--- a/packages/trpc/server/document-router/router.ts
+++ b/packages/trpc/server/document-router/router.ts
@@ -16,6 +16,7 @@ import { getDocumentWithDetailsById } from '@documenso/lib/server-only/document/
 import { resendDocument } from '@documenso/lib/server-only/document/resend-document';
 import { searchDocumentsWithKeyword } from '@documenso/lib/server-only/document/search-documents-with-keyword';
 import { sendDocument } from '@documenso/lib/server-only/document/send-document';
+import { updateDocument } from '@documenso/lib/server-only/document/update-document';
 import { updateDocumentSettings } from '@documenso/lib/server-only/document/update-document-settings';
 import { updateTitle } from '@documenso/lib/server-only/document/update-title';
 import { symmetricEncrypt } from '@documenso/lib/universal/crypto';
@@ -36,6 +37,7 @@ import {
   ZSetPasswordForDocumentMutationSchema,
   ZSetSettingsForDocumentMutationSchema,
   ZSetTitleForDocumentMutationSchema,
+  ZUpdateDocumentMutationSchema,
 } from './schema';
 
 export const documentRouter = router({
@@ -132,6 +134,23 @@ export const documentRouter = router({
       }
     }),
 
+  updateDocument: authenticatedProcedure
+    .input(ZUpdateDocumentMutationSchema)
+    .mutation(async ({ input, ctx }) => {
+      try {
+        const { documentId, data } = input;
+
+        await updateDocument({ documentId, data, userId: ctx.user.id });
+      } catch (err) {
+        console.error(err);
+
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'We were unable to delete this document. Please try again later.',
+        });
+      }
+    }),
+
   deleteDocument: authenticatedProcedure
     .input(ZDeleteDocumentMutationSchema)
     .mutation(async ({ input, ctx }) => {
diff --git a/packages/trpc/server/document-router/schema.ts b/packages/trpc/server/document-router/schema.ts
index 483d32e50..4936aae37 100644
--- a/packages/trpc/server/document-router/schema.ts
+++ b/packages/trpc/server/document-router/schema.ts
@@ -48,6 +48,12 @@ export const ZCreateDocumentMutationSchema = z.object({
   teamId: z.number().optional(),
 });
 
+export const ZUpdateDocumentMutationSchema = z.object({
+  documentId: z.number().min(1),
+  teamId: z.number().optional(),
+  data: z.any(),
+});
+
 export type TCreateDocumentMutationSchema = z.infer<typeof ZCreateDocumentMutationSchema>;
 
 export const ZSetSettingsForDocumentMutationSchema = z.object({