feat: refactoring bunch of code

dahal · dahal · commit 38f78a6503eb · 2024-04-17T01:54:18.000-05:00
diff --git a/prisma/seeds/index.ts b/prisma/seeds/index.ts
@@ -1,11 +1,11 @@
 import { db } from "@/server/db";
-import inquirer from "inquirer";
 import colors from "colors";
+import inquirer from "inquirer";
 colors.enable();
 
-import seedTeam from "./team";
-import seedCompanies from "./companies";
 import type { QuestionCollection } from "inquirer";
+import seedCompanies from "./companies";
+import seedTeam from "./team";
 
 if (process.env.NODE_ENV === "production") {
   console.log("❌ You cannot run this command on production".red);
@@ -44,6 +44,7 @@ const nuke = async () => {
     await db.shareClass.deleteMany();
     await db.equityPlan.deleteMany();
     await db.document.deleteMany();
+    await db.bucket.deleteMany();
     await db.audit.deleteMany();
     await db.session.deleteMany();
   });
diff --git a/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataRoomPublicId]/page.tsx b/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataRoomPublicId]/page.tsx
@@ -1,7 +1,8 @@
 "use server";
 
 import { getServerAuthSession } from "@/server/auth";
-import { db } from "@/server/db";
+import { api } from "@/trpc/server";
+import { notFound } from "next/navigation";
 import DataRoomFiles from "../components/data-room-files";
 
 const DataRoomSettinsPage = async ({
@@ -10,24 +11,24 @@ const DataRoomSettinsPage = async ({
   params: { publicId: string; dataRoomPublicId: string };
 }) => {
   const session = await getServerAuthSession();
-  const companyId = session?.user.companyId;
+  const { dataRoom, documents, recipients } =
+    await api.dataRoom.getDataRoom.query({
+      dataRoomPublicId,
+      getRecipients: true,
+    });
 
-  const dataRoom = await db.dataRoom.findFirstOrThrow({
-    where: {
-      publicId: dataRoomPublicId,
-      companyId,
-    },
+  if (!dataRoom) {
+    return notFound();
+  }
 
-    include: {
-      documents: {
-        orderBy: {
-          createdAt: "desc",
-        },
-      },
-    },
-  });
-
-  return <DataRoomFiles dataRoom={dataRoom} companyPublicId={publicId} />;
+  return (
+    <DataRoomFiles
+      dataRoom={dataRoom}
+      documents={documents}
+      recipients={recipients}
+      companyPublicId={publicId}
+    />
+  );
 };
 
 export default DataRoomSettinsPage;
diff --git a/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataroomPublicId]/page.tsx b/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/[dataroomPublicId]/page.tsx
@@ -1,7 +1,8 @@
 "use server";
 
 import { getServerAuthSession } from "@/server/auth";
-import { db } from "@/server/db";
+import { api } from "@/trpc/server";
+import { notFound } from "next/navigation";
 import DataRoomFiles from "../components/data-room-files";
 
 const DataRoomSettinsPage = async ({
@@ -10,24 +11,24 @@ const DataRoomSettinsPage = async ({
   params: { publicId: string; dataRoomPublicId: string };
 }) => {
   const session = await getServerAuthSession();
-  const companyId = session?.user.companyId;
+  const { dataRoom, documents, recipients } =
+    await api.dataRoom.getDataRoom.query({
+      dataRoomPublicId,
+      getRecipients: true,
+    });
 
-  const dataRoom = await db.dataRoom.findFirstOrThrow({
-    where: {
-      publicId: dataRoomPublicId,
-      companyId,
-    },
+  if (!dataRoom) {
+    return notFound();
+  }
 
-    include: {
-      documents: {
-        orderBy: {
-          createdAt: "desc",
-        },
-      },
-    },
-  });
-
-  return <DataRoomFiles dataRoom={dataRoom} companyPublicId={publicId} />;
+  return (
+    <DataRoomFiles
+      dataRoom={dataRoom}
+      documents={documents}
+      recipients={recipients}
+      companyPublicId={publicId}
+    />
+  );
 };
 
 export default DataRoomSettinsPage;
diff --git a/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/components/data-room-files.tsx b/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/components/data-room-files.tsx
@@ -2,20 +2,24 @@
 
 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 DataRoomUploader from "./data-room-uploader";
-// import { useToast } from "@/components/ui/use-toast";
-import type { DataRoom } from "@prisma/client";
+import { api } from "@/trpc/react";
+import type {
+  DataRoom,
+  DataRoomDocument,
+  DataRoomRecipient,
+} from "@prisma/client";
 import {
   RiFolder3Fill as FolderIcon,
   RiAddFill,
   RiArrowDownSLine,
   RiUploadCloudLine,
 } from "@remixicon/react";
 import Link from "next/link";
-// import { useRouter } from "next/navigation";
 import { Fragment, useState } from "react";
+import DataRoomUploader from "./data-room-uploader";
 
 interface DataRoomType extends DataRoom {
   documents: {
@@ -29,12 +33,17 @@ interface DataRoomType extends DataRoom {
 type DataRoomFilesProps = {
   dataRoom: DataRoomType;
   companyPublicId: string;
+  documents: DataRoomDocument[];
+  recipients: DataRoomRecipient[];
 };
 
-const DataRoomFiles = ({ dataRoom, companyPublicId }: DataRoomFilesProps) => {
-  // const router = useRouter();
-  // const { toast } = useToast();
-
+const DataRoomFiles = ({
+  dataRoom,
+  documents,
+  recipients,
+  companyPublicId,
+}: DataRoomFilesProps) => {
+  const { mutateAsync } = api.dataRoom.save.useMutation();
   const [loading, setLoading] = useState<boolean>(false);
 
   return (
@@ -60,14 +69,20 @@ const DataRoomFiles = ({ dataRoom, companyPublicId }: DataRoomFilesProps) => {
               className="h4 min-w-[300px] bg-transparent px-2 text-gray-800 outline-none focus:ring-0	focus:ring-offset-0"
               placeholder={`Data room's folder name`}
               defaultValue={dataRoom.name}
-              onChange={(e) => {
-                console.log("TODO: save with debounce", e.target.value);
+              onChange={async (e) => {
+                // TODO - debounce this
+
+                const name = e.target.value;
+                await mutateAsync({
+                  name,
+                  publicId: dataRoom.publicId,
+                });
               }}
             />
           </div>
         </div>
 
-        {dataRoom?.documents.length > 0 && (
+        {documents.length > 0 && (
           <div>
             <DropdownButton
               buttonSlot={
@@ -109,22 +124,26 @@ const DataRoomFiles = ({ dataRoom, companyPublicId }: DataRoomFilesProps) => {
       </form>
 
       <div>
-        <EmptyState
-          icon={<RiUploadCloudLine />}
-          title="Data room is empty!"
-          subtitle="Upload one or many documents to get started."
-        >
-          <DataRoomUploader
-            dataRoom={dataRoom}
-            companyPublicId={companyPublicId}
-            trigger={
-              <Button size="lg">
-                <RiAddFill className="mr-2 h-5 w-5" />
-                Upload documents
-              </Button>
-            }
-          />
-        </EmptyState>
+        {documents.length > 0 ? (
+          <DataRoomFileExplorer documents={documents} />
+        ) : (
+          <EmptyState
+            icon={<RiUploadCloudLine />}
+            title="Data room is empty!"
+            subtitle="Upload one or many documents to get started."
+          >
+            <DataRoomUploader
+              dataRoom={dataRoom}
+              companyPublicId={companyPublicId}
+              trigger={
+                <Button size="lg">
+                  <RiAddFill className="mr-2 h-5 w-5" />
+                  Upload documents
+                </Button>
+              }
+            />
+          </EmptyState>
+        )}
       </div>
 
       {loading && <Loading />}
diff --git a/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/components/data-room-uploader.tsx b/src/app/(authenticated)/(dashboard)/[publicId]/documents/data-rooms/components/data-room-uploader.tsx
@@ -21,7 +21,8 @@ const DataRoomUploader = ({
   const router = useRouter();
   const [open, setOpen] = useState(false);
 
-  const { mutateAsync } = api.dataRoom.save.useMutation();
+  const dataRoomMutation = api.dataRoom.save.useMutation();
+  const documentMutation = api.document.create.useMutation();
 
   return (
     <Modal
@@ -39,13 +40,18 @@ const DataRoomUploader = ({
         multiple={true}
         identifier={companyPublicId}
         keyPrefix={`data-room/${dataRoom.publicId}/file`}
-        onSuccess={async (document: UploadReturn) => {
-          await mutateAsync({
+        onSuccess={async (upload: UploadReturn) => {
+          const document = await documentMutation.mutateAsync({
+            name: upload.name,
+            bucketId: upload.id,
+          });
+
+          await dataRoomMutation.mutateAsync({
             name: dataRoom.name,
             publicId: dataRoom.publicId,
             documents: [
               {
-                documentId: document.key,
+                documentId: document.id,
               },
             ],
           });
diff --git a/src/components/documents/data-room/explorer.tsx b/src/components/documents/data-room/explorer.tsx
@@ -0,0 +1,9 @@
+import type { Document } from "@prisma/client";
+
+const DataRoomFileExplorer = (documents: Document[]) => {
+  <div className="flex flex-col gap-y-8">
+    Documents JSON.stringify(documents, null, 2)
+  </div>;
+};
+
+export default DataRoomFileExplorer;
diff --git a/src/components/documents/data-room/preview.tsx b/src/components/documents/data-room/preview.tsx
diff --git a/src/trpc/routers/data-room-router/router.ts b/src/trpc/routers/data-room-router/router.ts
@@ -1,9 +1,99 @@
 import { generatePublicId } from "@/common/id";
 import { createTRPCRouter, withAuth } from "@/trpc/api/trpc";
-import type { DataRoom } from "@prisma/client";
+import type {
+  DataRoom,
+  DataRoomDocument,
+  DataRoomRecipient,
+} from "@prisma/client";
+import { z } from "zod";
 import { DataRoomSchema } from "./schema";
 
 export const dataRoomRouter = createTRPCRouter({
+  getDataRoom: withAuth
+    .input(
+      z.object({
+        dataRoomPublicId: z.string(),
+        getRecipients: z.boolean().optional().default(false),
+      }),
+    )
+    .query(async ({ ctx, input }) => {
+      const { db, session } = ctx;
+      const user = session.user;
+      const companyId = user.companyId;
+      const { dataRoomPublicId, getRecipients } = input;
+
+      const dataRoom = await db.dataRoom.findUniqueOrThrow({
+        where: {
+          publicId: dataRoomPublicId,
+          companyId,
+        },
+
+        include: {
+          documents: true,
+
+          ...(getRecipients && { recipients: true }),
+        },
+      });
+
+      const documentIds = dataRoom.documents.map((doc) => doc.id);
+
+      const documents: DataRoomDocument[] = await db.dataRoomDocument.findMany({
+        where: {
+          id: { in: documentIds },
+        },
+
+        include: {
+          document: {
+            select: {
+              id: true,
+              bucket: true,
+            },
+          },
+        },
+      });
+
+      if (getRecipients) {
+        const recipientIds = dataRoom.recipients.map(
+          (recipient) => recipient.id,
+        );
+
+        const recipients: DataRoomRecipient[] =
+          await db.dataRoomRecipient.findMany({
+            where: {
+              id: { in: recipientIds },
+            },
+
+            include: {
+              member: {
+                select: {
+                  id: true,
+                  workEmail: true,
+                },
+              },
+
+              stakeholder: {
+                select: {
+                  id: true,
+                  name: true,
+                  email: true,
+                },
+              },
+            },
+          });
+
+        return {
+          dataRoom,
+          documents,
+          recipients,
+        };
+      } else {
+        return {
+          dataRoom,
+          documents,
+        };
+      }
+    }),
+
   save: withAuth.input(DataRoomSchema).mutation(async ({ ctx, input }) => {
     try {
       let room = {} as DataRoom;
