From de34af8747e3669f6f2379bb96817f72a6be38ac Mon Sep 17 00:00:00 2001
From: ChenZhaoYu <790348264@qq.com>
Date: Tue, 14 Feb 2023 15:07:50 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A4=9A=E4=BC=9A=E8=AF=9D=E5=9F=BA?=
 =?UTF-8?q?=E7=A1=80=E9=80=BB=E8=BE=91=E6=A2=B3=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/business/Chat/hooks/useChat.ts | 30 +++++++++
 src/components/business/Chat/index.vue        | 40 +++++-------
 .../business/Chat/layout/sider/List.vue       | 63 ++++++++++++-------
 .../business/Chat/layout/sider/index.vue      | 22 ++-----
 src/components/business/Chat/types.ts         | 18 ------
 src/store/modules/app/helper.ts               | 12 ++--
 src/store/modules/app/index.ts                |  6 +-
 src/store/modules/history/helper.ts           | 21 +++++++
 src/store/modules/history/index.ts            | 59 ++++++++++++++---
 src/typings/chat.d.ts                         | 20 ++++++
 src/utils/storage/local.ts                    | 29 ++++++---
 11 files changed, 213 insertions(+), 107 deletions(-)
 create mode 100644 src/components/business/Chat/hooks/useChat.ts
 delete mode 100644 src/components/business/Chat/types.ts
 create mode 100644 src/store/modules/history/helper.ts
 create mode 100644 src/typings/chat.d.ts

diff --git a/src/components/business/Chat/hooks/useChat.ts b/src/components/business/Chat/hooks/useChat.ts
new file mode 100644
index 0000000..f6ebd50
--- /dev/null
+++ b/src/components/business/Chat/hooks/useChat.ts
@@ -0,0 +1,30 @@
+import { useHistoryStore } from '@/store'
+
+export function useChat() {
+  const historyStore = useHistoryStore()
+
+  function addChat(message: string, args?: { reversal?: boolean; error?: boolean; options?: Chat.ChatOptions }) {
+    if (historyStore.historyChat.length === 0) {
+      historyStore.addHistory({
+        title: message,
+        isEdit: false,
+        data: [],
+      })
+      historyStore.chooseHistory(historyStore.historyChat.length - 1)
+    }
+
+    historyStore.addChat({
+      dateTime: new Date().toLocaleString(),
+      message,
+      reversal: args?.reversal ?? false,
+      error: args?.error ?? false,
+      options: args?.options ?? undefined,
+    })
+  }
+
+  function clearChat() {
+    historyStore.clearChat()
+  }
+
+  return { addChat, clearChat }
+}
diff --git a/src/components/business/Chat/index.vue b/src/components/business/Chat/index.vue
index 279ea44..c22942a 100644
--- a/src/components/business/Chat/index.vue
+++ b/src/components/business/Chat/index.vue
@@ -1,27 +1,26 @@
 <script setup lang='ts'>
-import { computed, nextTick, onMounted, ref } from 'vue'
+import { computed, nextTick, ref } from 'vue'
 import { NButton, NInput, useMessage } from 'naive-ui'
-import type { ChatOptions, ChatProps } from './types'
 import { Message } from './components'
 import { Layout } from './layout'
+import { useChat } from './hooks/useChat'
 import { fetchChatAPI } from '@/api'
 import { HoverButton, SvgIcon } from '@/components/common'
+import { useHistoryStore } from '@/store'
+
+const ms = useMessage()
+
+const historyStore = useHistoryStore()
 
 const scrollRef = ref<HTMLDivElement>()
 
-const ms = useMessage()
+const { addChat, clearChat } = useChat()
 
 const prompt = ref('')
 const loading = ref(false)
 
-const list = ref<ChatProps[]>([])
-const chatList = computed(() => list.value.filter(item => (!item.reversal && !item.error)))
-
-function initChat() {
-  addMessage('Hi, I am ChatGPT, a chatbot based on GPT-3.')
-}
-
-onMounted(initChat)
+const list = computed<Chat.Chat[]>(() => historyStore.getCurrentChat)
+const chatList = computed<Chat.Chat[]>(() => list.value.filter(item => (!item.reversal && !item.error)))
 
 async function handleSubmit() {
   if (loading.value)
@@ -37,7 +36,7 @@ async function handleSubmit() {
   addMessage(message, { reversal: true })
   prompt.value = ''
 
-  let options: ChatOptions = {}
+  let options: Chat.ChatOptions = {}
   const lastContext = chatList.value[chatList.value.length - 1]?.options
 
   if (lastContext)
@@ -63,21 +62,14 @@ function handleEnter(event: KeyboardEvent) {
 
 function addMessage(
   message: string,
-  args?: { reversal?: boolean; error?: boolean; options?: ChatOptions },
+  args?: { reversal?: boolean; error?: boolean; options?: Chat.ChatOptions },
 ) {
-  list.value.push({
-    dateTime: new Date().toLocaleString(),
-    message,
-    reversal: args?.reversal ?? false,
-    error: args?.error ?? false,
-    options: args?.options ?? undefined,
-  })
+  addChat(message, args)
   nextTick(() => scrollRef.value && (scrollRef.value.scrollTop = scrollRef.value.scrollHeight))
 }
 
 function handleClear() {
-  list.value = []
-  setTimeout(initChat, 100)
+  clearChat()
 }
 </script>
 
@@ -96,8 +88,8 @@ function handleClear() {
       </main>
       <footer class="p-4">
         <div class="flex items-center justify-between space-x-2">
-          <HoverButton tooltip="Clear conversations" @click="handleClear">
-            <span class="text-xl text-[#4f555e]">
+          <HoverButton tooltip="Clear conversations">
+            <span class="text-xl text-[#4f555e]" @click="handleClear">
               <SvgIcon icon="ri:delete-bin-line" />
             </span>
           </HoverButton>
diff --git a/src/components/business/Chat/layout/sider/List.vue b/src/components/business/Chat/layout/sider/List.vue
index caf99a7..5e55079 100644
--- a/src/components/business/Chat/layout/sider/List.vue
+++ b/src/components/business/Chat/layout/sider/List.vue
@@ -1,50 +1,65 @@
 <script setup lang='ts'>
-import { NScrollbar } from 'naive-ui'
-import type { HistoryChatProps } from '../../types'
+import { ref } from 'vue'
+import { NInput, NScrollbar } from 'naive-ui'
 import { SvgIcon } from '@/components/common'
+import { useHistoryStore } from '@/store'
 
-interface Props {
-  data: HistoryChatProps[]
-}
+const historyStore = useHistoryStore()
 
-interface Emit {
-  (ev: 'delete', index: number): void
-  (ev: 'edit', index: number): void
-}
+const dataSources = ref(historyStore.historyChat)
 
-defineProps<Props>()
+function handleSelect(index: number) {
+  historyStore.chooseHistory(index)
+}
 
-const emit = defineEmits<Emit>()
+function handleEdit(index: number, isEdit: boolean) {
+  historyStore.editHistory(index, isEdit)
+}
 
-function handleEdit(index: number) {
-  emit('delete', index)
+function handleRemove(index: number) {
+  historyStore.removeHistory(index)
 }
 
-function handleDelete(index: number) {
-  emit('delete', index)
+function handleEnter(index: number, isEdit: boolean, event: KeyboardEvent) {
+  if (event.key === 'Enter')
+    handleEdit(index, isEdit)
 }
 </script>
 
 <template>
   <NScrollbar class="px-4">
     <div class="flex flex-col gap-2 text-sm">
-      <div v-for="(item, index) of data" :key="index">
+      <div v-for="(item, index) of dataSources" :key="index">
         <a
           class="relative flex items-center gap-3 px-3 py-3 break-all rounded-md cursor-pointer bg-neutral-50 pr-14 hover:bg-neutral-100 group"
+          @click="handleSelect(index)"
         >
           <span>
             <SvgIcon icon="ri:message-3-line" />
           </span>
-          <div class="relative flex-1 overflow-hidden break-all text-ellipsis whitespace-nowrap max-h-5">
-            <span>{{ item.title }}</span>
+          <div class="relative flex-1 overflow-hidden break-all text-ellipsis whitespace-nowrap">
+            <NInput
+              v-if="item.isEdit"
+              v-model:value="item.title"
+              size="tiny"
+              @keypress="handleEnter(index, false, $event)"
+            />
+            <span v-else>{{ item.title }}</span>
           </div>
           <div class="absolute z-10 flex visible right-1">
-            <button class="p-1">
-              <SvgIcon icon="ri:edit-line" @click="handleEdit(index)" />
-            </button>
-            <button class="p-1" @click="handleDelete(index)">
-              <SvgIcon icon="ri:delete-bin-line" />
-            </button>
+            <template v-if="item.isEdit">
+              <button class="p-1" @click="handleEdit(index, false)">
+                <SvgIcon icon="ri:save-line" />
+              </button>
+            </template>
+            <template v-else>
+              <button class="p-1">
+                <SvgIcon icon="ri:edit-line" @click="handleEdit(index, true)" />
+              </button>
+              <button class="p-1" @click="handleRemove(index)">
+                <SvgIcon icon="ri:delete-bin-line" />
+              </button>
+            </template>
           </div>
         </a>
       </div>
diff --git a/src/components/business/Chat/layout/sider/index.vue b/src/components/business/Chat/layout/sider/index.vue
index ee24c3e..3e248ed 100644
--- a/src/components/business/Chat/layout/sider/index.vue
+++ b/src/components/business/Chat/layout/sider/index.vue
@@ -1,33 +1,23 @@
 <script setup lang='ts'>
 import { ref } from 'vue'
 import { NButton, NLayoutSider } from 'naive-ui'
-import type { HistoryChatProps } from '../../types'
 import List from './List.vue'
 import Footer from './Footer.vue'
-import { useAppStore } from '@/store'
+import { useAppStore, useHistoryStore } from '@/store'
 
 const appStore = useAppStore()
+const historyStore = useHistoryStore()
 
 const collapsed = ref(appStore.siderCollapsed ?? false)
 
-const history = ref<HistoryChatProps[]>([])
-
 function handleAdd() {
-  history.value.push({
-    title: 'New chat',
-    edit: false,
+  historyStore.addHistory({
+    title: '',
+    isEdit: false,
     data: [],
   })
 }
 
-function handleEdit(index: number) {
-  history.value[index].edit = true
-}
-
-function handleDelete(index: number) {
-  history.value.splice(index, 1)
-}
-
 function handleCollapsed() {
   collapsed.value = !collapsed.value
   appStore.setSiderCollapsed(collapsed.value)
@@ -51,7 +41,7 @@ function handleCollapsed() {
             New chat
           </NButton>
         </div>
-        <List :data="history" @edit="handleEdit" @delete="handleDelete" />
+        <List />
       </main>
       <Footer />
     </div>
diff --git a/src/components/business/Chat/types.ts b/src/components/business/Chat/types.ts
deleted file mode 100644
index 1201899..0000000
--- a/src/components/business/Chat/types.ts
+++ /dev/null
@@ -1,18 +0,0 @@
-export interface ChatOptions {
-  conversationId?: string
-  parentMessageId?: string
-}
-
-export interface ChatProps {
-  dateTime: string
-  message: string
-  reversal?: boolean
-  error?: boolean
-  options?: ChatOptions
-}
-
-export interface HistoryChatProps {
-  title: string
-  edit: boolean
-  data: ChatProps[]
-}
diff --git a/src/store/modules/app/helper.ts b/src/store/modules/app/helper.ts
index 55f9b22..7f71ff7 100644
--- a/src/store/modules/app/helper.ts
+++ b/src/store/modules/app/helper.ts
@@ -1,4 +1,6 @@
-import { ls } from '@/utils/storage'
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'appSetting'
 
 export interface AppState {
   siderCollapsed: boolean
@@ -8,11 +10,11 @@ export function defaultSetting() {
   return { siderCollapsed: false }
 }
 
-export function getAppSetting() {
-  const localSetting: AppState = ls.get('appSetting')
+export function getLocalSetting() {
+  const localSetting: AppState | undefined = ss.get(LOCAL_NAME)
   return localSetting ?? defaultSetting()
 }
 
-export function setAppSetting(setting: AppState) {
-  ls.set('appSetting', setting)
+export function setLocalSetting(setting: AppState) {
+  ss.set(LOCAL_NAME, setting)
 }
diff --git a/src/store/modules/app/index.ts b/src/store/modules/app/index.ts
index 8935e67..bbe2829 100644
--- a/src/store/modules/app/index.ts
+++ b/src/store/modules/app/index.ts
@@ -1,13 +1,13 @@
 import { defineStore } from 'pinia'
 import type { AppState } from './helper'
-import { getAppSetting, setAppSetting } from './helper'
+import { getLocalSetting, setLocalSetting } from './helper'
 
 export const useAppStore = defineStore('app-store', {
-  state: (): AppState => getAppSetting(),
+  state: (): AppState => getLocalSetting(),
   actions: {
     setSiderCollapsed(collapsed: boolean) {
       this.siderCollapsed = collapsed
-      setAppSetting(this.$state)
+      setLocalSetting(this.$state)
     },
     toggleSiderCollapse() {
       this.setSiderCollapsed(!this.siderCollapsed)
diff --git a/src/store/modules/history/helper.ts b/src/store/modules/history/helper.ts
new file mode 100644
index 0000000..635ba9c
--- /dev/null
+++ b/src/store/modules/history/helper.ts
@@ -0,0 +1,21 @@
+import { ss } from '@/utils/storage'
+
+const LOCAL_NAME = 'historyChat'
+
+export interface HistoryState {
+  historyChat: Chat.HistoryChat[]
+  active: number | null
+}
+
+export function defaultSetting() {
+  return { historyChat: [], active: null }
+}
+
+export function getLocalHistory() {
+  const localSetting: HistoryState | undefined = ss.get(LOCAL_NAME)
+  return localSetting ?? defaultSetting()
+}
+
+export function setLocalHistory(data: HistoryState) {
+  ss.set(LOCAL_NAME, data)
+}
diff --git a/src/store/modules/history/index.ts b/src/store/modules/history/index.ts
index 3f1c2dd..5b6df0f 100644
--- a/src/store/modules/history/index.ts
+++ b/src/store/modules/history/index.ts
@@ -1,12 +1,55 @@
 import { defineStore } from 'pinia'
-
-interface HistoryState {
-  list: any[]
-}
+import type { HistoryState } from './helper'
+import { getLocalHistory, setLocalHistory } from './helper'
 
 export const useHistoryStore = defineStore('history-store', {
-  state: (): HistoryState => ({
-    list: [],
-  }),
-  actions: {},
+  state: (): HistoryState => getLocalHistory(),
+  getters: {
+    getCurrentChat(state): Chat.Chat[] {
+      if (state.historyChat.length === 0)
+        return []
+
+      if (state.active === null)
+        state.active = state.historyChat.length - 1
+
+      return state.historyChat[state.active].data
+    },
+  },
+  actions: {
+    addChat(data: Chat.Chat) {
+      if (this.active !== null) {
+        this.historyChat[this.active].data.push(data)
+        this.active = this.historyChat.length - 1
+        setLocalHistory(this.$state)
+      }
+    },
+
+    clearChat() {
+      if (this.active !== null) {
+        this.historyChat[this.active].data = []
+        setLocalHistory(this.$state)
+      }
+    },
+
+    chooseHistory(index: number) {
+      this.active = index
+      setLocalHistory(this.$state)
+    },
+
+    addHistory(data: Chat.HistoryChat) {
+      this.historyChat.push(data)
+      this.active = this.historyChat.length - 1
+      setLocalHistory(this.$state)
+    },
+
+    editHistory(index: number, isEdit: boolean) {
+      this.historyChat[index].isEdit = isEdit
+      setLocalHistory(this.$state)
+    },
+
+    removeHistory(index: number) {
+      this.historyChat.splice(index, 1)
+      setLocalHistory(this.$state)
+    },
+  },
 })
diff --git a/src/typings/chat.d.ts b/src/typings/chat.d.ts
new file mode 100644
index 0000000..2a82b0f
--- /dev/null
+++ b/src/typings/chat.d.ts
@@ -0,0 +1,20 @@
+declare namespace Chat{
+	 interface ChatOptions {
+		conversationId?: string
+		parentMessageId?: string
+	}
+
+	 interface Chat {
+		dateTime: string
+		message: string
+		reversal?: boolean
+		error?: boolean
+		options?: ChatOptions
+	}
+
+	 interface HistoryChat {
+		title: string
+		isEdit: boolean
+		data: Chat[]
+	}
+}
diff --git a/src/utils/storage/local.ts b/src/utils/storage/local.ts
index fabfd23..215d22c 100644
--- a/src/utils/storage/local.ts
+++ b/src/utils/storage/local.ts
@@ -1,19 +1,28 @@
 import { deCrypto, enCrypto } from '../crypto'
 
 interface StorageData<T = any> {
-  value: T
+  data: T
   expire: number | null
 }
 
-function createLocalStorage() {
-  const DEFAULT_CACHE_TIME = 60 * 60 * 24 * 7 // 7 days
+export function createLocalStorage(options?: { expire?: number | null; crypto?: boolean }) {
+  const DEFAULT_CACHE_TIME = 60 * 60 * 24 * 7
 
-  function set<T = any>(key: string, value: T, expire: number | null = DEFAULT_CACHE_TIME) {
+  const { expire, crypto } = Object.assign(
+    {
+      expire: DEFAULT_CACHE_TIME,
+      crypto: true,
+    },
+    options,
+  )
+
+  function set<T = any>(key: string, data: T) {
     const storageData: StorageData<T> = {
-      value,
+      data,
       expire: expire !== null ? new Date().getTime() + expire * 1000 : null,
     }
-    const json = enCrypto(storageData)
+
+    const json = crypto ? enCrypto(storageData) : JSON.stringify(storageData)
     window.localStorage.setItem(key, json)
   }
 
@@ -23,16 +32,16 @@ function createLocalStorage() {
       let storageData: StorageData | null = null
 
       try {
-        storageData = deCrypto(json)
+        storageData = crypto ? deCrypto(json) : JSON.parse(json)
       }
       catch {
         // Prevent failure
       }
 
       if (storageData) {
-        const { value, expire } = storageData
+        const { data, expire } = storageData
         if (expire === null || expire >= Date.now())
-          return value
+          return data
       }
 
       remove(key)
@@ -57,3 +66,5 @@ function createLocalStorage() {
 }
 
 export const ls = createLocalStorage()
+
+export const ss = createLocalStorage({ expire: null, crypto: false })