From 6be119ac95468ebfb88896ff730b670fda80a6c1 Mon Sep 17 00:00:00 2001
From: David Nguyen <davidngu28@gmail.com>
Date: Wed, 3 Jan 2024 20:10:50 +1100
Subject: [PATCH] fix: improve document meta logic

---
 .../server-only/document-meta/upsert-document-meta.ts    | 9 +++++++++
 packages/trpc/server/document-router/router.ts           | 1 +
 2 files changed, 10 insertions(+)

diff --git a/packages/lib/server-only/document-meta/upsert-document-meta.ts b/packages/lib/server-only/document-meta/upsert-document-meta.ts
index c7221cce9..34c33e7cd 100644
--- a/packages/lib/server-only/document-meta/upsert-document-meta.ts
+++ b/packages/lib/server-only/document-meta/upsert-document-meta.ts
@@ -8,6 +8,7 @@ export type CreateDocumentMetaOptions = {
   message: string;
   timezone: string;
   dateFormat: string;
+  userId: number;
 };
 
 export const upsertDocumentMeta = async ({
@@ -16,7 +17,15 @@ export const upsertDocumentMeta = async ({
   timezone,
   dateFormat,
   documentId,
+  userId,
 }: CreateDocumentMetaOptions) => {
+  await prisma.document.findFirstOrThrow({
+    where: {
+      id: documentId,
+      userId,
+    },
+  });
+
   return await prisma.documentMeta.upsert({
     where: {
       documentId,
diff --git a/packages/trpc/server/document-router/router.ts b/packages/trpc/server/document-router/router.ts
index 425f34857..b4a1b60e3 100644
--- a/packages/trpc/server/document-router/router.ts
+++ b/packages/trpc/server/document-router/router.ts
@@ -188,6 +188,7 @@ export const documentRouter = router({
             message: meta.message,
             dateFormat: meta.dateFormat,
             timezone: meta.timezone,
+            userId: ctx.user.id,
           });
         }