From 98f68b8b08dc15b039c43e1caafed9de5362f12b Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Tue, 2 Apr 2024 06:33:59 -0700
Subject: [PATCH] refac: shared chat user id

---
 backend/apps/web/models/chats.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/backend/apps/web/models/chats.py b/backend/apps/web/models/chats.py
index 5a49354ba..598494c81 100644
--- a/backend/apps/web/models/chats.py
+++ b/backend/apps/web/models/chats.py
@@ -108,7 +108,7 @@ class ChatTable:
         shared_chat = ChatModel(
             **{
                 "id": str(uuid.uuid4()),
-                "user_id": "shared",
+                "user_id": f"shared-{chat_id}",
                 "title": chat.title,
                 "chat": chat.chat,
                 "timestamp": int(time.time()),
@@ -122,6 +122,15 @@ class ChatTable:
 
         return shared_chat if (shared_result and result) else None
 
+    def delete_shared_chat_by_chat_id(self, chat_id: str) -> bool:
+        try:
+            query = Chat.delete().where(Chat.user_id == f"shared-{chat_id}")
+            query.execute()  # Remove the rows, return number of rows removed.
+
+            return True
+        except:
+            return False
+
     def update_chat_share_id_by_id(
         self, od: str, share_id: Optional[str]
     ) -> Optional[ChatModel]: