feat: improve admin panel

packages/lib/client-only/hooks/use-copy-share-link.ts

@@ -1,5 +1,5 @@
 import { trpc } from '@documenso/trpc/react';
-import { TCreateOrGetShareLinkMutationSchema } from '@documenso/trpc/server/share-link-router/schema';
+import type { TCreateOrGetShareLinkMutationSchema } from '@documenso/trpc/server/share-link-router/schema';
 
 import { useCopyToClipboard } from './use-copy-to-clipboard';
 

packages/lib/server-only/admin/get-entire-document.ts

@@ -0,0 +1,26 @@
+import { prisma } from '@documenso/prisma';
+
+export type GetEntireDocumentOptions = {
+  id: number;
+};
+
+export const getEntireDocument = async ({ id }: GetEntireDocumentOptions) => {
+  const document = await prisma.document.findFirstOrThrow({
+    where: {
+      id,
+    },
+    include: {
+      Recipient: {
+        include: {
+          Field: {
+            include: {
+              Signature: true,
+            },
+          },
+        },
+      },
+    },
+  });
+
+  return document;
+};

packages/lib/server-only/admin/update-recipient.ts

@@ -0,0 +1,30 @@
+import { prisma } from '@documenso/prisma';
+import { SigningStatus } from '@documenso/prisma/client';
+
+export type UpdateRecipientOptions = {
+  id: number;
+  name: string | undefined;
+  email: string | undefined;
+};
+
+export const updateRecipient = async ({ id, name, email }: UpdateRecipientOptions) => {
+  const recipient = await prisma.recipient.findFirstOrThrow({
+    where: {
+      id,
+    },
+  });
+
+  if (recipient.signingStatus === SigningStatus.SIGNED) {
+    throw new Error('Cannot update a recipient that has already signed.');
+  }
+
+  return await prisma.recipient.update({
+    where: {
+      id,
+    },
+    data: {
+      name,
+      email,
+    },
+  });
+};

packages/lib/server-only/document/seal-document.ts

@@ -22,12 +22,14 @@ import { sendCompletedEmail } from './send-completed-email';
 export type SealDocumentOptions = {
   documentId: number;
   sendEmail?: boolean;
+  isResealing?: boolean;
   requestMetadata?: RequestMetadata;
 };
 
 export const sealDocument = async ({
   documentId,
   sendEmail = true,
+  isResealing = false,
   requestMetadata,
 }: SealDocumentOptions) => {
   'use server';
@@ -78,11 +80,20 @@ export const sealDocument = async ({
     throw new Error(`Document ${document.id} has unsigned fields`);
   }
 
+  if (isResealing) {
+    // If we're resealing we want to use the initial data for the document
+    // so we aren't placing fields on top of eachother.
+    documentData.data = documentData.initialData;
+  }
+
   // !: Need to write the fields onto the document as a hard copy
   const pdfData = await getFile(documentData);
 
   const doc = await PDFDocument.load(pdfData);
 
+  // Flatten the form to stop annotation layers from appearing above documenso fields
+  doc.getForm().flatten();
+
   for (const field of fields) {
     await insertFieldInPDF(doc, field);
   }
@@ -134,7 +145,7 @@ export const sealDocument = async ({
     });
   });
 
-  if (sendEmail) {
+  if (sendEmail && !isResealing) {
     await sendCompletedEmail({ documentId, requestMetadata });
   }
 

packages/lib/server-only/user/delete-user.ts

@@ -4,20 +4,18 @@ import { DocumentStatus } from '@documenso/prisma/client';
 import { deletedAccountServiceAccount } from './service-accounts/deleted-account';
 
 export type DeleteUserOptions = {
-  email: string;
+  id: number;
 };
 
-export const deleteUser = async ({ email }: DeleteUserOptions) => {
+export const deleteUser = async ({ id }: DeleteUserOptions) => {
   const user = await prisma.user.findFirst({
     where: {
-      email: {
-        contains: email,
-      },
+      id,
     },
   });
 
   if (!user) {
-    throw new Error(`User with email ${email} not found`);
+    throw new Error(`User with ID ${id} not found`);
   }
 
   const serviceAccount = await deletedAccountServiceAccount();

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

@@ -1,14 +1,39 @@
 import { TRPCError } from '@trpc/server';
 
+import { findDocuments } from '@documenso/lib/server-only/admin/get-all-documents';
+import { updateRecipient } from '@documenso/lib/server-only/admin/update-recipient';
 import { updateUser } from '@documenso/lib/server-only/admin/update-user';
+import { sealDocument } from '@documenso/lib/server-only/document/seal-document';
 import { upsertSiteSetting } from '@documenso/lib/server-only/site-settings/upsert-site-setting';
+import { deleteUser } from '@documenso/lib/server-only/user/delete-user';
+import { getUserById } from '@documenso/lib/server-only/user/get-user-by-id';
 
 import { adminProcedure, router } from '../trpc';
-import { ZUpdateProfileMutationByAdminSchema, ZUpdateSiteSettingMutationSchema } from './schema';
+import {
+  ZAdminDeleteUserMutationSchema,
+  ZAdminFindDocumentsQuerySchema,
+  ZAdminResealDocumentMutationSchema,
+  ZAdminUpdateProfileMutationSchema,
+  ZAdminUpdateRecipientMutationSchema,
+  ZAdminUpdateSiteSettingMutationSchema,
+} from './schema';
 
 export const adminRouter = router({
+  findDocuments: adminProcedure.input(ZAdminFindDocumentsQuerySchema).query(async ({ input }) => {
+    const { term, page, perPage } = input;
+
+    try {
+      return await findDocuments({ term, page, perPage });
+    } catch (err) {
+      throw new TRPCError({
+        code: 'BAD_REQUEST',
+        message: 'We were unable to retrieve the documents. Please try again.',
+      });
+    }
+  }),
+
   updateUser: adminProcedure
-    .input(ZUpdateProfileMutationByAdminSchema)
+    .input(ZAdminUpdateProfileMutationSchema)
     .mutation(async ({ input }) => {
       const { id, name, email, roles } = input;
 
@@ -22,8 +47,23 @@ export const adminRouter = router({
       }
     }),
 
+  updateRecipient: adminProcedure
+    .input(ZAdminUpdateRecipientMutationSchema)
+    .mutation(async ({ input }) => {
+      const { id, name, email } = input;
+
+      try {
+        return await updateRecipient({ id, name, email });
+      } catch (err) {
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'We were unable to update the recipient provided.',
+        });
+      }
+    }),
+
   updateSiteSetting: adminProcedure
-    .input(ZUpdateSiteSettingMutationSchema)
+    .input(ZAdminUpdateSiteSettingMutationSchema)
     .mutation(async ({ ctx, input }) => {
       try {
         const { id, enabled, data } = input;
@@ -41,4 +81,41 @@ export const adminRouter = router({
         });
       }
     }),
+
+  resealDocument: adminProcedure
+    .input(ZAdminResealDocumentMutationSchema)
+    .mutation(async ({ input }) => {
+      const { id } = input;
+
+      try {
+        return await sealDocument({ documentId: id, isResealing: true });
+      } catch (err) {
+        console.log('resealDocument error', err);
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'We were unable to reseal the document provided.',
+        });
+      }
+    }),
+
+  deleteUser: adminProcedure.input(ZAdminDeleteUserMutationSchema).mutation(async ({ input }) => {
+    const { id, email } = input;
+
+    try {
+      const user = await getUserById({ id });
+
+      if (user.email !== email) {
+        throw new Error('Email does not match');
+      }
+
+      return await deleteUser({ id });
+    } catch (err) {
+      console.log(err);
+
+      throw new TRPCError({
+        code: 'BAD_REQUEST',
+        message: 'We were unable to delete the specified account. Please try again.',
+      });
+    }
+  }),
 });

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

@@ -3,17 +3,48 @@ import z from 'zod';
 
 import { ZSiteSettingSchema } from '@documenso/lib/server-only/site-settings/schema';
 
-export const ZUpdateProfileMutationByAdminSchema = z.object({
+export const ZAdminFindDocumentsQuerySchema = z.object({
+  term: z.string().optional(),
+  page: z.number().optional().default(1),
+  perPage: z.number().optional().default(20),
+});
+
+export type TAdminFindDocumentsQuerySchema = z.infer<typeof ZAdminFindDocumentsQuerySchema>;
+
+export const ZAdminUpdateProfileMutationSchema = z.object({
   id: z.number().min(1),
   name: z.string().nullish(),
   email: z.string().email().optional(),
   roles: z.array(z.nativeEnum(Role)).optional(),
 });
 
-export type TUpdateProfileMutationByAdminSchema = z.infer<
-  typeof ZUpdateProfileMutationByAdminSchema
+export type TAdminUpdateProfileMutationSchema = z.infer<typeof ZAdminUpdateProfileMutationSchema>;
+
+export const ZAdminUpdateRecipientMutationSchema = z.object({
+  id: z.number().min(1),
+  name: z.string().optional(),
+  email: z.string().email().optional(),
+});
+
+export type TAdminUpdateRecipientMutationSchema = z.infer<
+  typeof ZAdminUpdateRecipientMutationSchema
 >;
 
-export const ZUpdateSiteSettingMutationSchema = ZSiteSettingSchema;
+export const ZAdminUpdateSiteSettingMutationSchema = ZSiteSettingSchema;
 
-export type TUpdateSiteSettingMutationSchema = z.infer<typeof ZUpdateSiteSettingMutationSchema>;
+export type TAdminUpdateSiteSettingMutationSchema = z.infer<
+  typeof ZAdminUpdateSiteSettingMutationSchema
+>;
+
+export const ZAdminResealDocumentMutationSchema = z.object({
+  id: z.number().min(1),
+});
+
+export type TAdminResealDocumentMutationSchema = z.infer<typeof ZAdminResealDocumentMutationSchema>;
+
+export const ZAdminDeleteUserMutationSchema = z.object({
+  id: z.number().min(1),
+  email: z.string().email(),
+});
+
+export type TAdminDeleteUserMutationSchema = z.infer<typeof ZAdminDeleteUserMutationSchema>;

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

@@ -207,9 +207,9 @@ export const profileRouter = router({
 
   deleteAccount: authenticatedProcedure.mutation(async ({ ctx }) => {
     try {
-      const user = ctx.user;
-
-      return await deleteUser(user);
+      return await deleteUser({
+        id: ctx.user.id,
+      });
     } catch (err) {
       let message = 'We were unable to delete your account. Please try again.';