From e5b90372a666a7a2a6e3808870b2cfe0fd9c9adc Mon Sep 17 00:00:00 2001
From: dylanmay <670374839@qq.com>
Date: Tue, 12 Nov 2024 09:12:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=8C=81=E4=B9=85=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.json                                  |  1 +
 src/store/indexedDB.ts                        | 72 ++++++++++++++++
 src/views/chat/ChatPage/Index.vue             | 15 ++--
 src/views/chat/api/messageApi.ts              |  2 +-
 .../chat/components/ChatMessage/Index.vue     |  2 +-
 .../Index.vue => Conversation/index.vue}      |  8 +-
 .../Index.vue => ConversationItem/index.vue}  |  7 +-
 .../chat/components/InputSection/Index.vue    | 18 +++-
 src/views/chat/components/Message/BaseMsg.vue |  2 +-
 .../chat/components/Message/ImageMsg.vue      |  2 +-
 .../chat/components/ToolSection/Index.vue     |  2 +-
 src/views/chat/model/BaseConversation.ts      |  2 +-
 src/views/chat/model/BaseMessage.ts           |  5 +-
 src/views/chat/model/ImageMessage.ts          |  2 +-
 src/views/chat/model/TextMessage.ts           | 27 +++++-
 src/views/chat/store/chatstore.ts             | 84 ++++++++++++++++++-
 16 files changed, 222 insertions(+), 29 deletions(-)
 create mode 100644 src/store/indexedDB.ts
 rename src/views/chat/components/{Session/Index.vue => Conversation/index.vue} (77%)
 rename src/views/chat/components/{SessionItem/Index.vue => ConversationItem/index.vue} (95%)

diff --git a/package.json b/package.json
index ba5400092..c1b8c358f 100644
--- a/package.json
+++ b/package.json
@@ -50,6 +50,7 @@
     "element-plus": "2.8.4",
     "fast-xml-parser": "^4.3.2",
     "highlight.js": "^11.9.0",
+    "idb": "^8.0.0",
     "jsencrypt": "^3.3.2",
     "lodash-es": "^4.17.21",
     "markdown-it": "^14.1.0",
diff --git a/src/store/indexedDB.ts b/src/store/indexedDB.ts
new file mode 100644
index 000000000..d14278996
--- /dev/null
+++ b/src/store/indexedDB.ts
@@ -0,0 +1,72 @@
+import { ConversationModelType } from '@/views/chat/types/types'
+import { openDB, DBSchema, IDBPDatabase } from 'idb'
+
+// Define your database schema
+interface MyDB extends DBSchema {
+  Conversations: {
+    key: string
+    value: ConversationModelType
+  }
+}
+
+let dbPromise: Promise<IDBPDatabase<MyDB>>
+
+export const initDB = () => {
+  if (!dbPromise) {
+    try {
+      dbPromise = openDB<MyDB>('yudao-im-indexeddb', 1, {
+        upgrade(db) {
+          db.createObjectStore('Conversations', { keyPath: 'conversationNo' })
+        }
+      })
+    } catch (error) {
+      console.log(error)
+    }
+  }
+  return dbPromise
+}
+
+export const addConversation = async (conversation: ConversationModelType) => {
+
+  try {
+    const db = await initDB()
+    await db.put('Conversations', conversation)
+  } catch (error) {
+    console.error(conversation)
+    console.error(error)
+  }
+
+}
+
+export const getConversation = async (conversationNo: string) => {
+
+  try {
+    const db = await initDB()
+    return await db.get('Conversations', conversationNo)
+  } catch (error) {
+    console.error(error)
+  }
+
+}
+
+export const deleteConversation = async (conversationNo: string) => {
+
+  try {
+    const db = await initDB()
+    await db.delete('Conversations', conversationNo)
+  } catch (error) {
+    console.error(error)
+  }
+
+}
+
+export const getAllConversations = async () => {
+
+  try {
+    const db = await initDB()
+    return await db.getAll('Conversations')
+  } catch (error) {
+    console.log(error)
+  }
+
+}
diff --git a/src/views/chat/ChatPage/Index.vue b/src/views/chat/ChatPage/Index.vue
index 3a1a604f2..dae352481 100644
--- a/src/views/chat/ChatPage/Index.vue
+++ b/src/views/chat/ChatPage/Index.vue
@@ -18,17 +18,22 @@
  */
 
 import ToolSection from '../components/ToolSection/Index.vue'
-import Session from '../components/Session/Index.vue'
+import Session from '../components/Conversation/index.vue'
 import Friends from '../components/Friends/Index.vue'
-import ChatHeader from '../components/ChatHeader/Index.vue' // TODO @dylan：为啥这个 index.vue 是大写哈？可以搞成小写哇？
-import ChatMessage from '../components/ChatMessage/Index.vue'
-import InputSection from '../components/InputSection/Index.vue'
+import ChatHeader from '../components/ChatHeader/index.vue'
+import ChatMessage from '../components/ChatMessage/index.vue'
+import InputSection from '../components/InputSection/index.vue'
 import FriendDetail from '../components/FriendDetail/Index.vue'
-import { MENU_LIST_ENUM } from '../types/index.d.ts'
+import { MENU_LIST_ENUM } from '../types/types'
+import { useWebSocketStore } from '../store/websocketStore'
 
 defineOptions({ name: 'ChatPage' })
 
 const bussinessType = ref(1)
+const webSocketStore = useWebSocketStore();
+onMounted(() => {
+  webSocketStore.connect()
+})
 
 const toolMenuSelectChange = (value) => {
   bussinessType.value = value
diff --git a/src/views/chat/api/messageApi.ts b/src/views/chat/api/messageApi.ts
index d086b6396..5a1290fa2 100644
--- a/src/views/chat/api/messageApi.ts
+++ b/src/views/chat/api/messageApi.ts
@@ -6,7 +6,7 @@
  */
 
 import request from '@/config/axios'
-import { MessageModelType } from '../types'
+import { MessageModelType } from '../types/types'
 
 export interface SendMsg {
   clientMessageId: string
diff --git a/src/views/chat/components/ChatMessage/Index.vue b/src/views/chat/components/ChatMessage/Index.vue
index 06027f838..b972b03f6 100644
--- a/src/views/chat/components/ChatMessage/Index.vue
+++ b/src/views/chat/components/ChatMessage/Index.vue
@@ -24,7 +24,7 @@
 import { useChatStore } from '../../store/chatstore'
 import TextMsg from '@/views/chat/components/Message/TextMsg.vue'
 import ImageMsg from '@/views/chat/components/Message/ImageMsg.vue'
-import { ContentType } from '../../types/index.d.ts'
+import { ContentType } from '../../types/types'
 
 defineOptions({ name: 'ChatMessage' })
 
diff --git a/src/views/chat/components/Session/Index.vue b/src/views/chat/components/Conversation/index.vue
similarity index 77%
rename from src/views/chat/components/Session/Index.vue
rename to src/views/chat/components/Conversation/index.vue
index 1db5009fe..6aa3e1945 100644
--- a/src/views/chat/components/Session/Index.vue
+++ b/src/views/chat/components/Conversation/index.vue
@@ -3,7 +3,7 @@
     <view class="flex flex-col w-full">
       <SessionItem
         v-for="(item, index) in chatStore.sessionList"
-        :key="item.id"
+        :key="item.id" 
         :index="index"
         :conversation="item"
         @click="() => onSessionItemClick(index)"
@@ -13,17 +13,17 @@
 </template>
 
 <script lang="ts" setup>
-import SessionItem from '../SessionItem/Index.vue'
+import SessionItem from '@/views/chat/components/ConversationItem/index.vue'
 import { useChatStoreWithOut } from '../../store/chatstore'
 import { onMounted } from 'vue'
 
 defineOptions({ name: 'Session' })
 
 const chatStore = useChatStoreWithOut()
-const { setCurrentConversation, setCurrentSessionIndex, getSession } = useChatStoreWithOut()
+const { setCurrentConversation, setCurrentSessionIndex, getConversationList } = useChatStoreWithOut()
 
 onMounted(() => {
-  getSession()
+  getConversationList()
   // set default conversation
   nextTick(() => {
     setCurrentConversation()
diff --git a/src/views/chat/components/SessionItem/Index.vue b/src/views/chat/components/ConversationItem/index.vue
similarity index 95%
rename from src/views/chat/components/SessionItem/Index.vue
rename to src/views/chat/components/ConversationItem/index.vue
index 75228663f..7baba64b3 100644
--- a/src/views/chat/components/SessionItem/Index.vue
+++ b/src/views/chat/components/ConversationItem/index.vue
@@ -18,10 +18,11 @@
 
 <script lang="ts" setup>
 import { PropType } from 'vue'
-import { ContentType, ConversationModelType } from '../../types/index.d.ts'
+import { ContentType, ConversationModelType } from '../../types/types'
+
 import { formatPast } from '@/utils/formatTime'
-import { useChatStore } from '../../store/chatstore'
-import TextMessage from '../../model/TextMessage';
+import { useChatStore } from '../../store/chatstore.js'
+import TextMessage from '../../model/TextMessage.js';
 
 defineOptions({ name: 'SessionItem' })
 
diff --git a/src/views/chat/components/InputSection/Index.vue b/src/views/chat/components/InputSection/Index.vue
index fe26348e8..03c25a8c9 100644
--- a/src/views/chat/components/InputSection/Index.vue
+++ b/src/views/chat/components/InputSection/Index.vue
@@ -24,18 +24,27 @@
 <script lang="ts" setup>
 import TextMessage from '../../model/TextMessage'
 import { useChatStoreWithOut } from '../../store/chatstore'
-import { CONVERSATION_TYPE } from '../../types/index.d.ts'
-import { SendStatus, MessageRole, ContentType } from '../../types/index.d.ts'
+import { CONVERSATION_TYPE } from '../../types/types'
+import { SendStatus, MessageRole, ContentType } from '../../types/types'
 import { useUserStoreWithOut } from '../../../../store/modules/user';
+import { ElNotification } from 'element-plus';
 
 defineOptions({ name: 'InputSection' })
 
 const chatStore = useChatStoreWithOut()
 const onEnter = () => {
   console.log('enter pressed')
-  const msg = createTextMessage(chatStore.inputText)
+
+  if (!chatStore.inputText.trim())  {
+    ElNotification({
+      title: '温馨提示',
+      message: '请输入内容',
+      type: 'warning'
+    })
+    return 
+  }
+  const msg = createTextMessage(chatStore.inputText.trim())
   chatStore.addMessageToCurrentSession(msg)
-  chatStore.setInputText('')
 }
 
 const createTextMessage = (content: string): TextMessage => {
@@ -53,6 +62,7 @@ const createTextMessage = (content: string): TextMessage => {
     MessageRole.SELF,
     SendStatus.SENDING,
     chatStore.currentSession?.id || '',
+    userStore.user.id,
     chatStore.currentSession ? chatStore.currentSession.targetId : 0,
     chatStore.currentSession?.type || CONVERSATION_TYPE.SINGLE,
     chatStore.currentSession?.senderId || ''
diff --git a/src/views/chat/components/Message/BaseMsg.vue b/src/views/chat/components/Message/BaseMsg.vue
index 78c5e2508..b08fc8170 100644
--- a/src/views/chat/components/Message/BaseMsg.vue
+++ b/src/views/chat/components/Message/BaseMsg.vue
@@ -19,7 +19,7 @@
 <script lang="ts" setup>
 import { PropType } from 'vue'
 import { Loading } from '@element-plus/icons-vue'
-import { MessageModelType, MessageRole, SendStatus } from '../../types/index.d.ts'
+import { MessageModelType, MessageRole, SendStatus } from '../../types/types'
 
 defineOptions({ name: 'BaseMessage' })
 
diff --git a/src/views/chat/components/Message/ImageMsg.vue b/src/views/chat/components/Message/ImageMsg.vue
index 1219629eb..487c076c4 100644
--- a/src/views/chat/components/Message/ImageMsg.vue
+++ b/src/views/chat/components/Message/ImageMsg.vue
@@ -12,7 +12,7 @@
 import { PropType } from 'vue'
 import { useChatStore } from '../../store/chatstore'
 import { onMounted } from 'vue'
-import { MessageModelType } from '../../types'
+import { MessageModelType } from '../../types/types'
 import BaseMesageLayout from './BaseMsg.vue'
 
 defineOptions({ name: 'ImageMessage' })
diff --git a/src/views/chat/components/ToolSection/Index.vue b/src/views/chat/components/ToolSection/Index.vue
index 769e66399..d92455657 100644
--- a/src/views/chat/components/ToolSection/Index.vue
+++ b/src/views/chat/components/ToolSection/Index.vue
@@ -21,7 +21,7 @@
 </template>
 
 <script lang="ts" setup>
-import { MENU_LIST_ENUM } from '../../types/index.d.ts'
+import { MENU_LIST_ENUM } from '../../types/types'
 
 defineOptions({ name: 'ToolSection' })
 
diff --git a/src/views/chat/model/BaseConversation.ts b/src/views/chat/model/BaseConversation.ts
index 962341a18..aaf8b0616 100644
--- a/src/views/chat/model/BaseConversation.ts
+++ b/src/views/chat/model/BaseConversation.ts
@@ -1,4 +1,4 @@
-import { ConversationType, MessageModelType } from '../types'
+import { ConversationType, MessageModelType } from '../types/types'
 
 export default class BaseConversation {
   public id: string
diff --git a/src/views/chat/model/BaseMessage.ts b/src/views/chat/model/BaseMessage.ts
index c65834f87..0333af092 100644
--- a/src/views/chat/model/BaseMessage.ts
+++ b/src/views/chat/model/BaseMessage.ts
@@ -1,4 +1,4 @@
-import { MessageRole, ContentType, SendStatus } from '../types'
+import { MessageRole, ContentType, SendStatus } from '../types/types'
 
 export default class BaseMessage {
   id?: string
@@ -11,6 +11,7 @@ export default class BaseMessage {
   contentType: ContentType
   conversationId: string
   clientMessageId: string
+  senderId: number
   receiverId: number
   conversationType: number
   conversationUserId: number
@@ -24,6 +25,7 @@ export default class BaseMessage {
     sendStauts: SendStatus,
     contentType: ContentType,
     conversationId: string,
+    senderId: number,
     receiverId: number,
     conversationType: number,
     conversationUserId: number
@@ -37,6 +39,7 @@ export default class BaseMessage {
     this.sendStatus = sendStauts
     this.contentType = contentType
     this.conversationId = conversationId
+    this.senderId = senderId
     this.receiverId = receiverId
     this.clientMessageId = this.generateClientMessageId()
     this.conversationType = conversationType
diff --git a/src/views/chat/model/ImageMessage.ts b/src/views/chat/model/ImageMessage.ts
index 357174315..efe665d4f 100644
--- a/src/views/chat/model/ImageMessage.ts
+++ b/src/views/chat/model/ImageMessage.ts
@@ -1,4 +1,4 @@
-import { MessageRole, ContentType, SendStatus } from '@/views/chat/types/index.d.ts'
+import { MessageRole, ContentType, SendStatus } from '@/views/chat/types/types'
 import BaseMessage from './BaseMessage'
 
 export default class ImageMessage extends BaseMessage {
diff --git a/src/views/chat/model/TextMessage.ts b/src/views/chat/model/TextMessage.ts
index 75fb416fb..2bd8d5da2 100644
--- a/src/views/chat/model/TextMessage.ts
+++ b/src/views/chat/model/TextMessage.ts
@@ -1,4 +1,4 @@
-import { MessageRole, ContentType, SendStatus } from '@/views/chat/types/index.d.ts'
+import { MessageRole, ContentType, SendStatus, ImMessageContent } from '@/views/chat/types/types'
 import BaseMessage from './BaseMessage'
 
 export default class TextMessage extends BaseMessage {
@@ -14,6 +14,7 @@ export default class TextMessage extends BaseMessage {
     role: MessageRole,
     sendStatus: SendStatus,
     conversationId: string,
+    senderId: number,
     receiverId: number,
     conversationType: number,
     conversationUserId: number
@@ -28,10 +29,34 @@ export default class TextMessage extends BaseMessage {
       sendStatus,
       ContentType.TEXT,
       conversationId,
+      senderId,
       receiverId,
       conversationType,
       conversationUserId
     )
     this.content = content
   }
+
+  /**
+   * 消息转换
+   * @param websocketMessage 
+   * @returns 
+   */
+  static fromWebsocket(websocketMessage: ImMessageContent): TextMessage {
+    return new TextMessage(
+      websocketMessage.id.toString(), // 服务端也应该返回一个clientMessageId
+      websocketMessage.senderAvatar,
+      websocketMessage.senderNickname,
+      new Date().getTime(), // TODO: 是否合理
+      false,
+      websocketMessage.content,
+      MessageRole.OTHER, // 可以去掉在使用的时候依据逻辑判断
+      SendStatus.SUCCESS,
+      '',   // TODO: [dylan] 
+      websocketMessage.senderId,
+      websocketMessage.receiverId,
+      websocketMessage.conversationType,
+      0
+    )
+  }
 }
diff --git a/src/views/chat/store/chatstore.ts b/src/views/chat/store/chatstore.ts
index c2d1b91fe..65573d489 100644
--- a/src/views/chat/store/chatstore.ts
+++ b/src/views/chat/store/chatstore.ts
@@ -2,11 +2,14 @@ import { store } from '@/store/index'
 import { defineStore } from 'pinia'
 import BaseConversation from '../model/BaseConversation'
 import BaseMessage from '../model/BaseMessage'
-import { ConversationModelType, MessageRole, ContentType, SendStatus } from '../types/index.d.ts'
+import { ConversationModelType, MessageRole, ContentType, SendStatus } from '../types/types'
 import SessionApi from '../api/sessionApi'
 import MessageApi, { SendMsg } from '../api/messageApi'
-import { useUserStoreWithOut } from '@/store/modules/user'
+import { useUserStore, useUserStoreWithOut } from '@/store/modules/user'
 import { formatDate } from '@/utils/formatTime'
+import { addConversation, getAllConversations } from '@/store/indexedDB'
+import { ChatConversation } from '../model/ChatConversation'
+import { generateConversationNo } from './websocketStore'
 
 // TODO @dylan：是不是 chat => im；session => conversation；这样统一一点哈。
 interface ChatStoreModel {
@@ -81,6 +84,8 @@ export const useChatStore = defineStore('chatStore', {
         }
 
         this.updateMsgToCurrentSession(updateMsg)
+      } finally {
+        this.setInputText('')
       }
     },
 
@@ -99,6 +104,42 @@ export const useChatStore = defineStore('chatStore', {
       this.sessionList.splice(conversationIndex, 1, msgConversation)
     },
 
+    /**
+     * 添加消息到会话
+     * @param message
+     */
+    addMessageToConversation<T extends BaseMessage>(message: T): void {
+      // 无论是converstionNo1还是converstionNo2都都需要试一下
+      const converstionNo1 = generateConversationNo(
+        message.senderId,
+        message.receiverId,
+        message.conversationType
+      )
+      const converstionNo2 = generateConversationNo(
+        message.receiverId,
+        message.senderId,
+        message.conversationType
+      )
+
+      const conversationIndex = this.sessionList.findIndex(
+        (item) => item.conversationNo === converstionNo1 || item.conversationNo === converstionNo2
+      )
+
+      if (conversationIndex < 0) {
+        console.log('conversation not exist')
+        return
+      }
+
+      const msgConversation = this.sessionList[conversationIndex]
+      msgConversation.msgList.push(message)
+
+      // replace the old Conversation
+      this.sessionList.splice(conversationIndex, 1, msgConversation)
+
+      // 更新消息到indexeddb
+      addConversation(toRaw(msgConversation) as ChatConversation )
+    },
+
     /**
      * 更新消息到当前会话
      * @param updatedMsg
@@ -112,13 +153,40 @@ export const useChatStore = defineStore('chatStore', {
             return item
           }
         })
+
+        const rawCurrentSesstion = toRaw(this.currentSession)
+        rawCurrentSesstion.msgList =this.currentSession.msgList.map(item => toRaw(item))
+        console.log("raw", rawCurrentSesstion)
+        addConversation(rawCurrentSesstion as ChatConversation)
       }
     },
 
-    async getSession() {
+    async getConversationList() {
+      try {
+        // 从数据库获取数据
+        const _conversationList = await getAllConversations()
+
+        if (_conversationList) {
+          // 加载到内存
+          // TODO:[dylan]处理排序
+          this.sessionList = _conversationList
+        }
+      } catch (error) {
+        console.log(error)
+      } finally{
+
+        // 本地没有数据的时候才请求接口
+        if (this.sessionList.length === 0) {
+          this.getSessionFromServer()
+        }
+      }
+    },
+
+    async getSessionFromServer() {
       try {
         const res = await SessionApi.getSessionList()
-        this.sessionList = res.map((item) => ({
+        
+          this.sessionList = res.map((item) => ({
           ...item,
           updateTime: item.lastReadTime,
           name: item.targetId,
@@ -129,6 +197,12 @@ export const useChatStore = defineStore('chatStore', {
           description: item.lastMessageDescription,
           msgList: []
         }))
+
+        // 同步到数据库
+        this.sessionList.forEach((item) => {
+          console.log(item)
+          addConversation(toRaw(item) as ChatConversation)
+        })
       } catch (error) {
         return error
       }
@@ -157,6 +231,8 @@ export const useChatStore = defineStore('chatStore', {
             avatar: item.senderAvatar
           }
         })
+
+        addConversation(toRaw(this.currentSession) as ChatConversation)
       } catch (error) {
         return error
       }