feat: reusable trpc endpoint to fetch documents

Author: dahal

.eslintrc.cjs

@@ -35,6 +35,7 @@ const config = {
         destructuredArrayIgnorePattern: "^_",
       },
     ],
+    "@typescript-eslint/prefer-optional-chain": "off",
     "@typescript-eslint/require-await": "off",
     "@typescript-eslint/no-unsafe-assignment": "warn",
     "@typescript-eslint/no-unsafe-member-access": "warn",

src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataRoomPublicId]/page.tsx

@@ -12,10 +12,7 @@ const DataRoomSettinsPage = async ({
 }) => {
   const session = await getServerAuthSession();
   const { dataRoom, documents, recipients } =
-    await api.dataRoom.getDataRoom.query({
-      dataRoomPublicId,
-      getRecipients: true,
-    });
+    await api.dataRoom.getDataRoom.query({ dataRoomPublicId });
 
   if (!dataRoom) {
     return notFound();

src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataroomPublicId]/page.tsx

@@ -12,10 +12,7 @@ const DataRoomSettinsPage = async ({
 }) => {
   const session = await getServerAuthSession();
   const { dataRoom, documents, recipients } =
-    await api.dataRoom.getDataRoom.query({
-      dataRoomPublicId,
-      getRecipients: true,
-    });
+    await api.dataRoom.getDataRoom.query({ dataRoomPublicId });
 
   if (!dataRoom) {
     return notFound();

src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/components/data-room-files.tsx

@@ -2,15 +2,13 @@
 
 import EmptyState from "@/components/common/empty-state";
 import Loading from "@/components/common/loading";
-import DataRoomFileExplorer from "@/components/documents/data-room/explorer";
 import { Button } from "@/components/ui/button";
 import { DropdownButton } from "@/components/ui/dropdown-button";
 import { api } from "@/trpc/react";
-import type {
-  DataRoom,
-  DataRoomDocument,
-  DataRoomRecipient,
-} from "@prisma/client";
+import type { Bucket, DataRoom } from "@prisma/client";
+
+import { type DataRoomRecipientType } from "@/types/documents/data-room";
+
 import {
   RiFolder3Fill as FolderIcon,
   RiAddFill,
@@ -31,10 +29,10 @@ interface DataRoomType extends DataRoom {
 }
 
 type DataRoomFilesProps = {
-  dataRoom: DataRoomType;
+  dataRoom: DataRoom;
+  documents: Bucket[];
   companyPublicId: string;
-  documents: DataRoomDocument[];
-  recipients: DataRoomRecipient[];
+  recipients: DataRoomRecipientType[];
 };
 
 const DataRoomFiles = ({
@@ -125,7 +123,8 @@ const DataRoomFiles = ({
 
       <div>
         {documents.length > 0 ? (
-          <DataRoomFileExplorer documents={documents} />
+          // <DataRoomFileExplorer documents={documents} />
+          <pre>{JSON.stringify(documents, null, 2)}</pre>
         ) : (
           <EmptyState
             icon={<RiUploadCloudLine />}

src/components/documents/data-room/explorer.tsx

@@ -1,6 +1,6 @@
-import type { Document } from "@prisma/client";
+import type { DataRoomDocument } from "@prisma/client";
 
-const DataRoomFileExplorer = (documents: Document[]) => {
+const DataRoomFileExplorer = (documents: DataRoomDocument[]) => {
   <div className="flex flex-col gap-y-8">
     Documents JSON.stringify(documents, null, 2)
   </div>;

src/trpc/routers/data-room-router/router.ts

@@ -1,26 +1,26 @@
 import { generatePublicId } from "@/common/id";
 import { createTRPCRouter, withAuth } from "@/trpc/api/trpc";
-import type {
-  DataRoom,
-  DataRoomDocument,
-  DataRoomRecipient,
-} from "@prisma/client";
+import type { Bucket, DataRoom, DataRoomDocument } from "@prisma/client";
 import { z } from "zod";
 import { DataRoomSchema } from "./schema";
 
+import { type DataRoomRecipientType } from "@/types/documents/data-room";
+
+interface DataRoomDocumentType extends DataRoomDocument {
+  document: {
+    id: string;
+    bucket: Bucket;
+  };
+}
+
 export const dataRoomRouter = createTRPCRouter({
   getDataRoom: withAuth
-    .input(
-      z.object({
-        dataRoomPublicId: z.string(),
-        getRecipients: z.boolean().optional().default(false),
-      }),
-    )
+    .input(z.object({ dataRoomPublicId: z.string() }))
     .query(async ({ ctx, input }) => {
       const { db, session } = ctx;
       const user = session.user;
       const companyId = user.companyId;
-      const { dataRoomPublicId, getRecipients } = input;
+      const { dataRoomPublicId } = input;
 
       const dataRoom = await db.dataRoom.findUniqueOrThrow({
         where: {
@@ -30,68 +30,94 @@ export const dataRoomRouter = createTRPCRouter({
 
         include: {
           documents: true,
-
-          ...(getRecipients && { recipients: true }),
+          recipients: true,
         },
       });
 
       const documentIds = dataRoom.documents.map((doc) => doc.id);
 
-      const documents: DataRoomDocument[] = await db.dataRoomDocument.findMany({
-        where: {
-          id: { in: documentIds },
-        },
+      const dataRoomDocument: DataRoomDocumentType[] =
+        await db.dataRoomDocument.findMany({
+          where: {
+            id: { in: documentIds },
+          },
 
-        include: {
-          document: {
-            select: {
-              id: true,
-              bucket: true,
+          include: {
+            document: {
+              select: {
+                id: true,
+                bucket: true,
+              },
             },
           },
-        },
-      });
+        });
 
-      if (getRecipients) {
-        const recipientIds = dataRoom.recipients.map(
-          (recipient) => recipient.id,
-        );
+      const documents: Bucket[] = dataRoomDocument.map((doc) => ({
+        id: doc.document.bucket.id,
+        name: doc.document.bucket.name,
+        key: doc.document.bucket.key,
+        mimeType: doc.document.bucket.mimeType,
+        size: doc.document.bucket.size,
+        createdAt: doc.document.bucket.createdAt,
+        updatedAt: doc.document.bucket.updatedAt,
+      }));
 
-        const recipients: DataRoomRecipient[] =
-          await db.dataRoomRecipient.findMany({
-            where: {
-              id: { in: recipientIds },
-            },
+      const recipientIds = dataRoom.recipients.map((recipient) => recipient.id);
 
-            include: {
-              member: {
-                select: {
-                  id: true,
-                  workEmail: true,
-                },
-              },
+      const dataRoomRecipient = await db.dataRoomRecipient.findMany({
+        where: {
+          id: { in: recipientIds },
+        },
 
-              stakeholder: {
+        include: {
+          member: {
+            select: {
+              id: true,
+              user: {
                 select: {
-                  id: true,
-                  name: true,
                   email: true,
+                  name: true,
                 },
               },
             },
-          });
+          },
+          stakeholder: true,
+        },
+      });
 
-        return {
-          dataRoom,
-          documents,
-          recipients,
-        };
-      } else {
-        return {
-          dataRoom,
-          documents,
-        };
-      }
+      const recipients: DataRoomRecipientType[] = dataRoomRecipient.map(
+        (recipient) => {
+          const r = {
+            id: recipient.id,
+            email: recipient.email,
+          } as DataRoomRecipientType;
+
+          if (recipient.member && recipient.member.user) {
+            r.member = {
+              id: recipient.member.id,
+              email: recipient.member.user.email ?? "",
+              name: recipient.member.user.name ?? "",
+            };
+          }
+
+          if (recipient.stakeholder) {
+            r.stakeholder = {
+              id: recipient.stakeholder.id,
+              email: recipient.stakeholder.email,
+              name: recipient.stakeholder.name,
+              institutionName: recipient.stakeholder.institutionName ?? "",
+            };
+          }
+
+          return r;
+        },
+      );
+
+      return {
+        dataRoom,
+        documents,
+        recipients,
+      };
     }),
 
   save: withAuth.input(DataRoomSchema).mutation(async ({ ctx, input }) => {

src/types/documents/data-room.ts

@@ -0,0 +1,15 @@
+export type DataRoomRecipientType = {
+  id: string;
+  email?: string | null;
+  member?: {
+    id: string;
+    name: string;
+    email: string;
+  };
+  stakeholder?: {
+    id: string;
+    name: string;
+    email: string;
+    institutionName: string;
+  };
+};