From 69605dc3f9bb08f0534aa79453c9d74594b12bd7 Mon Sep 17 00:00:00 2001
From: djohnson <mr.demarcus.johnson@gmail.com>
Date: Sun, 24 Nov 2024 07:19:45 +0900
Subject: [PATCH] ChatStore should export ChatViewModels

---
 src/components/OmniBar.tsx                     |  4 ++--
 src/components/chat/ChatListSection.tsx        |  4 ++--
 src/components/chat/ChatSettingsSection.tsx    | 16 +++++++---------
 src/core/chat/ChatStore.ts                     | 18 +++++++++---------
 src/core/chat/ChatViewModel.ts                 |  6 +++++-
 src/features/settings/panels/chat/ChatForm.tsx |  4 ++--
 .../settings/panels/chat/ChatPanel.tsx         | 16 +++++++---------
 src/utils/chatToDateLabel.ts                   |  6 +++---
 8 files changed, 37 insertions(+), 37 deletions(-)

diff --git a/src/components/OmniBar.tsx b/src/components/OmniBar.tsx
index 11e8a98..eafe67f 100644
--- a/src/components/OmniBar.tsx
+++ b/src/components/OmniBar.tsx
@@ -284,7 +284,7 @@ const useRegisterChatActions = () => {
         },
       ]
 
-      _.orderBy(chatStore.chats, 'lastMessageDate', 'desc').forEach(chat => {
+      chatStore.orderedChats.forEach(chat => {
         const name = chat.name || 'new chat'
 
         return nextChatActions.push({
@@ -349,7 +349,7 @@ const useRegisterMessageActions = () => {
     const nextMessageActions: Action[] = []
 
     chatStore.chats.forEach(chat => {
-      const messageCount = countMessagesWithText(chat.messageIds)
+      const messageCount = countMessagesWithText(chat.source.messageIds)
 
       if (messageCount > 0) {
         nextMessageActions.push(
diff --git a/src/components/chat/ChatListSection.tsx b/src/components/chat/ChatListSection.tsx
index 68f7560..ae9f511 100644
--- a/src/components/chat/ChatListSection.tsx
+++ b/src/components/chat/ChatListSection.tsx
@@ -10,13 +10,13 @@ import DocumentArrowUp from '~/icons/DocumentArrowUp'
 import Edit from '~/icons/Edit'
 
 import { chatStore } from '~/core/chat/ChatStore'
-import { ChatModel } from '~/core/chat/ChatModel'
+import { ChatViewModel } from '~/core/chat/ChatViewModel'
 import { chatTable } from '~/core/chat/ChatTable'
 
 export const ChatListSection = observer(({ onChatSelected }: { onChatSelected: () => void }) => {
   const { dateLabelToChatPairs } = chatStore
 
-  const handleChatSelected = async (chat: ChatModel) => {
+  const handleChatSelected = async (chat: ChatViewModel) => {
     await chatStore.selectChat(chat)
 
     onChatSelected()
diff --git a/src/components/chat/ChatSettingsSection.tsx b/src/components/chat/ChatSettingsSection.tsx
index 6e279aa..16d302f 100644
--- a/src/components/chat/ChatSettingsSection.tsx
+++ b/src/components/chat/ChatSettingsSection.tsx
@@ -32,8 +32,6 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
 
   const chat = chatStore.selectedChat!
 
-  const chatModel = chat.source
-
   const onExportClose = () => {
     setIsExportOpen(false)
   }
@@ -43,7 +41,7 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
   }
 
   const handleFormSubmit = handleSubmit(async formData => {
-    await chatTable.put({ ...chatModel, ...formData })
+    await chatTable.put({ ...chat.source, ...formData })
 
     reset(formData)
   })
@@ -60,12 +58,12 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
 
     const link = document.createElement('a')
     link.href = URL.createObjectURL(new Blob([data], { type: 'application/json' }))
-    link.download = `llm-x-chat-${_.snakeCase(chatModel.name).replace('_', '-')}.json`
+    link.download = `llm-x-chat-${_.snakeCase(chat.name).replace('_', '-')}.json`
     link.click()
   }
 
   useEffect(() => {
-    reset({ name: chatModel.name || 'new chat' })
+    reset({ name: chat.name || 'new chat' })
   }, [chat])
 
   useEffect(() => {
@@ -93,12 +91,12 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
           </button>
 
           <span className="flex-shrink-1 line-clamp-1 max-w-[85%] flex-1 text-left md:text-lg">
-            {chatModel.name || 'new chat'}
+            {chat.name || 'new chat'}
           </span>
 
           <NavButton
             to={'/chats/' + chat.id}
-            className="!bg-transparent text-base-content/60 hover:text-base-content p-1 transition-colors duration-250 ease-in-out"
+            className="!bg-transparent p-1 text-base-content/60 transition-colors duration-250 ease-in-out hover:text-base-content"
           >
             <Edit />
           </NavButton>
@@ -130,7 +128,7 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
                   )}
                   control={control}
                   name="name"
-                  defaultValue={chatModel.name || 'new chat'}
+                  defaultValue={chat.name || 'new chat'}
                   rules={{
                     validate: validateName,
                   }}
@@ -152,7 +150,7 @@ export const ChatSettingsSection = observer(({ onBackClicked }: { onBackClicked:
 
             <Tooltip label="Delete Chat">
               <button
-                onClick={() => chatStore.destroyChat(chatModel)}
+                onClick={() => chatStore.destroyChat(chat)}
                 className="btn btn-ghost text-error"
               >
                 <Delete className="h-5 w-5" />
diff --git a/src/core/chat/ChatStore.ts b/src/core/chat/ChatStore.ts
index 50e3f35..64a5d75 100644
--- a/src/core/chat/ChatStore.ts
+++ b/src/core/chat/ChatStore.ts
@@ -20,8 +20,8 @@ class ChatStore {
     makeAutoObservable(this)
   }
 
-  get chats(): ChatModel[] {
-    return chatTable.cache.allValues()
+  get chats() {
+    return _.compact(chatTable.cache.allValues().map(this.chatCache.getOrPut))
   }
 
   get selectedChat() {
@@ -34,12 +34,12 @@ class ChatStore {
     return undefined
   }
 
-  get emptyChat(): ChatModel | undefined {
-    return _.find(this.chats, chat => chat.messageIds.length === 0)
+  get emptyChat(): ChatViewModel | undefined {
+    return _.find(this.chats, chat => chat.source.messageIds.length === 0)
   }
 
   get orderedChats() {
-    return _.orderBy(this.chats, 'lastMessageTimestamp', 'desc') // newest sent message first
+    return _.orderBy(this.chats, 'source.lastMessageTimestamp', 'desc') // newest sent message first
   }
 
   // todo, see if this is used in more than one place?
@@ -51,11 +51,11 @@ class ChatStore {
       .value()
   }
 
-  async destroyChat(chat: ChatModel) {
+  async destroyChat(chat: ChatViewModel) {
     let nextSelectedChat: ChatModel | undefined = undefined
 
     if (this.selectedChat?.id === chat.id) {
-      nextSelectedChat = _.without(chatStore.orderedChats, chat)[0]
+      nextSelectedChat = _.without(chatStore.orderedChats, chat)[0]?.source
     }
 
     // if we deleted the selected chat, and there was nothing to replace it
@@ -67,10 +67,10 @@ class ChatStore {
 
     this.chatCache.remove(chat.id)
 
-    return chatTable.destroy(chat)
+    return chatTable.destroy(chat.source)
   }
 
-  async selectChat(chat: ChatModel) {
+  async selectChat(chat: ChatViewModel) {
     return settingTable.put({ selectedChatId: chat.id })
   }
 }
diff --git a/src/core/chat/ChatViewModel.ts b/src/core/chat/ChatViewModel.ts
index c0a9dd5..70aff9a 100644
--- a/src/core/chat/ChatViewModel.ts
+++ b/src/core/chat/ChatViewModel.ts
@@ -42,6 +42,10 @@ export class ChatViewModel {
     return this.source.id
   }
 
+  get name() {
+    return this.source.name
+  }
+
   get messages() {
     return _.compact(this.source.messageIds.map(this.messageViewModelCache.get))
   }
@@ -224,7 +228,7 @@ export class ChatViewModel {
     this.messageViewModelCache.put(userMessage)
 
     // if there was not a name before, auto make one now
-    const name = this.source.name === 'New Chat' ? content.substring(0, 40) : this.source.name
+    const name = this.name === 'New Chat' ? content.substring(0, 40) : this.name
     // add the message (and new name) to the chat
     await this.update({
       name,
diff --git a/src/features/settings/panels/chat/ChatForm.tsx b/src/features/settings/panels/chat/ChatForm.tsx
index c457752..d2bc88e 100644
--- a/src/features/settings/panels/chat/ChatForm.tsx
+++ b/src/features/settings/panels/chat/ChatForm.tsx
@@ -35,7 +35,7 @@ export const ChatForm = observer(() => {
   const handleDelete = async () => {
     navigate('/chats')
 
-    await chatStore.destroyChat(chat.source)
+    await chatStore.destroyChat(chat)
   }
 
   const validateName = (name: string) => {
@@ -49,7 +49,7 @@ export const ChatForm = observer(() => {
   }, [chat])
 
   return (
-    <Drawer path={chat.id} label={chat.source.name}>
+    <Drawer path={chat.id} label={chat.name}>
       <form onSubmit={handleFormSubmit} className="flex h-full flex-col p-2">
         <Controller
           render={({ field }) => (
diff --git a/src/features/settings/panels/chat/ChatPanel.tsx b/src/features/settings/panels/chat/ChatPanel.tsx
index 71080fb..1cc20ed 100644
--- a/src/features/settings/panels/chat/ChatPanel.tsx
+++ b/src/features/settings/panels/chat/ChatPanel.tsx
@@ -8,7 +8,7 @@ import SettingSection, { SettingSectionItem } from '~/containers/SettingSection'
 
 import Edit from '~/icons/Edit'
 
-import { ChatModel } from '~/core/chat/ChatModel'
+import { ChatViewModel } from '~/core/chat/ChatViewModel'
 import { chatTable } from '~/core/chat/ChatTable'
 import { chatStore } from '~/core/chat/ChatStore'
 import { settingTable } from '~/core/setting/SettingTable'
@@ -17,20 +17,18 @@ export const ChatPanel = observer(() => {
   const { selectedChat, chats } = chatStore
   const navigate = useNavigate()
 
-  const chatToSectionItem = (chat: ChatModel): SettingSectionItem<ChatModel> => ({
+  const chatToSectionItem = (chat: ChatViewModel): SettingSectionItem<ChatViewModel> => ({
     id: chat.id,
     label: chat.name,
-    subLabels: [chat.messageIds.length + ' messages'],
+    subLabels: [chat.source.messageIds.length + ' messages'],
     data: chat,
   })
 
-  const itemFilter = (chat: ChatModel, filterText: string) => {
-    return (
-      chat.name.toLowerCase().includes(filterText)
-    )
+  const itemFilter = (chat: ChatViewModel, filterText: string) => {
+    return chat.name.toLowerCase().includes(filterText)
   }
 
-  const handleChatSelected = async (chat: ChatModel) => {
+  const handleChatSelected = async (chat: ChatViewModel) => {
     await chatStore.selectChat(chat)
   }
 
@@ -65,7 +63,7 @@ export const ChatPanel = observer(() => {
       addButtonProps={{
         label: 'Add New Chat',
         onClick: e => createChat(e),
-        isDisabled: !!chatStore.emptyChat
+        isDisabled: !!chatStore.emptyChat,
       }}
     />
   )
diff --git a/src/utils/chatToDateLabel.ts b/src/utils/chatToDateLabel.ts
index 78f63cf..1b3ecad 100644
--- a/src/utils/chatToDateLabel.ts
+++ b/src/utils/chatToDateLabel.ts
@@ -1,9 +1,9 @@
 import moment from 'moment'
 
-import { ChatModel } from '~/core/chat/ChatModel'
+import { ChatViewModel } from '~/core/chat/ChatViewModel'
 
-export const chatToDateLabel = (chat: ChatModel) => {
-  const sentTime = moment(chat.lastMessageTimestamp)
+export const chatToDateLabel = (chat: ChatViewModel) => {
+  const sentTime = moment(chat.source.lastMessageTimestamp)
 
   const today = moment().startOf('day')