From 810d194cfd3f5e227e818d13c58322f21c342027 Mon Sep 17 00:00:00 2001
From: Ephraim Atta-Duncan <ephraimduncan68@gmail.com>
Date: Thu, 30 Jan 2025 10:29:22 +0000
Subject: [PATCH] feat: return the necessary information for document preview
 and download

---
 .../q/[token]/document-download-button.tsx    |  2 +-
 .../document/get-document-by-access-token.ts  | 20 +++++++++++++++----
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/apps/web/src/app/(share)/q/[token]/document-download-button.tsx b/apps/web/src/app/(share)/q/[token]/document-download-button.tsx
index d45c21f02..fc4ef111d 100644
--- a/apps/web/src/app/(share)/q/[token]/document-download-button.tsx
+++ b/apps/web/src/app/(share)/q/[token]/document-download-button.tsx
@@ -10,7 +10,7 @@ import { Button } from '@documenso/ui/primitives/button';
 import { useToast } from '@documenso/ui/primitives/use-toast';
 
 export type DocumentDownloadButtonProps = {
-  document: Document & {
+  document: Pick<Document, 'title'> & {
     documentData: DocumentData;
   };
 };
diff --git a/packages/lib/server-only/document/get-document-by-access-token.ts b/packages/lib/server-only/document/get-document-by-access-token.ts
index 078fd3012..3911a86ed 100644
--- a/packages/lib/server-only/document/get-document-by-access-token.ts
+++ b/packages/lib/server-only/document/get-document-by-access-token.ts
@@ -13,11 +13,23 @@ export const getDocumentByAccessToken = async ({ token }: GetDocumentByAccessTok
     where: {
       token,
     },
-    include: {
+    select: {
       document: {
-        include: {
-          documentData: true,
-          documentMeta: true,
+        select: {
+          title: true,
+          documentData: {
+            select: {
+              id: true,
+              type: true,
+              data: true,
+              initialData: true,
+            },
+          },
+          documentMeta: {
+            select: {
+              password: true,
+            },
+          },
         },
       },
     },