From 5f3a5871f1ca5621375613721889cc5d15473bbc Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Sun, 25 Jun 2023 18:21:38 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20vue-mobile=20=3D>=20=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E8=81=8A=E5=A4=A9=E8=AE=B0=E5=BD=95=E6=8B=8D=E7=89=88=E6=A0=B7?=
 =?UTF-8?q?=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/go/handler/admin/admin_handler.go         |  11 +-
 web/src/components/black/ChatReply.vue        | 136 ------------------
 .../{black => mobile}/ChatPrompt.vue          |  30 ++--
 web/src/components/mobile/ChatReply.vue       | 134 +++++++++++++++++
 web/src/main.js                               |   2 +
 web/src/views/mobile/ChatList.vue             |  66 +++++++--
 web/src/views/mobile/ChatSession.vue          |  81 ++++++++++-
 web/src/views/mobile/Home.vue                 |   2 +-
 8 files changed, 284 insertions(+), 178 deletions(-)
 delete mode 100644 web/src/components/black/ChatReply.vue
 rename web/src/components/{black => mobile}/ChatPrompt.vue (69%)
 create mode 100644 web/src/components/mobile/ChatReply.vue

diff --git a/api/go/handler/admin/admin_handler.go b/api/go/handler/admin/admin_handler.go
index b64ede19..9155b4f1 100644
--- a/api/go/handler/admin/admin_handler.go
+++ b/api/go/handler/admin/admin_handler.go
@@ -97,7 +97,7 @@ func (h *ManagerHandler) Migrate(c *gin.Context) {
 		}
 		break
 	case "role":
-		// TestRole 修改角色图片，改成绝对路径
+		// 修改角色图片，改成绝对路径
 		var roles []model.ChatRole
 		h.db.Find(&roles)
 		for _, r := range roles {
@@ -105,6 +105,15 @@ func (h *ManagerHandler) Migrate(c *gin.Context) {
 			h.db.Updates(&r)
 		}
 		break
+	case "history":
+		// 修改角色图片，改成绝对路径
+		var message []model.HistoryMessage
+		h.db.Find(&message)
+		for _, r := range message {
+			r.Icon = "/" + r.Icon
+			h.db.Updates(&r)
+		}
+		break
 	}
 
 	resp.SUCCESS(c, "SUCCESS")
diff --git a/web/src/components/black/ChatReply.vue b/web/src/components/black/ChatReply.vue
deleted file mode 100644
index 55156f16..00000000
--- a/web/src/components/black/ChatReply.vue
+++ /dev/null
@@ -1,136 +0,0 @@
-<template>
-  <div class="chat-line chat-line-left">
-    <div class="chat-icon">
-      <img :src="icon" alt="ChatGPT">
-    </div>
-
-    <div class="chat-item">
-      <div class="triangle"></div>
-      <div class="content-box">
-        <div class="content" v-html="content"></div>
-        <div class="tool-box">
-          <el-tooltip
-              class="box-item"
-              effect="light"
-              content="复制回答"
-              placement="bottom"
-          >
-            <el-button type="info" class="copy-reply" :data-clipboard-text="orgContent">
-              <el-icon>
-                <DocumentCopy/>
-              </el-icon>
-            </el-button>
-          </el-tooltip>
-        </div>
-      </div>
-    </div>
-  </div>
-</template>
-
-<script>
-import {defineComponent} from "vue"
-import {DocumentCopy} from "@element-plus/icons-vue";
-
-export default defineComponent({
-  name: 'ChatReply',
-  components: {DocumentCopy},
-  props: {
-    content: {
-      type: String,
-      default: '',
-    },
-    orgContent: {
-      type: String,
-      default: '',
-    },
-    icon: {
-      type: String,
-      default: 'images/gpt-icon.png',
-    }
-  },
-  data() {
-    return {}
-  },
-})
-</script>
-
-<style lang="stylus">
-.common-layout {
-  .chat-line-left {
-    justify-content: flex-start;
-
-    .chat-icon {
-      margin-right 5px;
-
-      img {
-        border-radius 50%;
-      }
-    }
-
-    .chat-item {
-      display: inline-block;
-      position: relative;
-      padding: 0 0 0 5px;
-      overflow: hidden;
-
-      .triangle {
-        width: 0;
-        height: 0;
-        border-top: 6px solid transparent;
-        border-bottom: 6px solid transparent;
-        border-right: 6px solid #404042;
-        position: absolute;
-        left: 0;
-        top: 10px;
-      }
-
-      .content-box {
-
-        display flex
-        flex-direction row
-
-        .content {
-          min-height 20px;
-          word-break break-word;
-          padding: 6px 10px;
-          color var(--content-color)
-          background-color: #404042;
-          font-size: var(--content-font-size);
-          border-radius: 5px;
-          overflow auto;
-
-          p {
-            line-height 1.5
-
-            code {
-              color #f1f1f1
-              background-color #202121
-              padding 0 3px;
-              border-radius 5px;
-            }
-          }
-
-          p:last-child {
-            margin-bottom: 0
-          }
-
-          p:first-child {
-            margin-top 0
-          }
-        }
-
-        .tool-box {
-          padding-left 10px;
-          font-size 16px;
-
-          .el-button {
-            height 20px
-            padding 5px 2px;
-          }
-        }
-      }
-    }
-  }
-}
-
-</style>
diff --git a/web/src/components/black/ChatPrompt.vue b/web/src/components/mobile/ChatPrompt.vue
similarity index 69%
rename from web/src/components/black/ChatPrompt.vue
rename to web/src/components/mobile/ChatPrompt.vue
index 9b0925ec..f41814f2 100644
--- a/web/src/components/black/ChatPrompt.vue
+++ b/web/src/components/mobile/ChatPrompt.vue
@@ -1,5 +1,5 @@
 <template>
-  <div class="chat-line chat-line-right">
+  <div class="message-reply">
     <div class="chat-item">
       <div class="content" v-html="content"></div>
       <div class="triangle"></div>
@@ -32,15 +32,15 @@ export default defineComponent({
 })
 </script>
 
-<style lang="stylus" scoped>
-.chat-line-right {
+<style lang="stylus">
+.message-reply {
   justify-content: flex-end;
 
   .chat-icon {
     margin-left 5px;
 
     img {
-      border-radius 50%;
+      border-radius 5px;
     }
   }
 
@@ -52,9 +52,9 @@ export default defineComponent({
     .triangle {
       width: 0;
       height: 0;
-      border-top: 6px solid transparent;
-      border-bottom: 6px solid transparent;
-      border-left: 6px solid #223A34;
+      border-top: 5px solid transparent;
+      border-bottom: 5px solid transparent;
+      border-left: 5px solid #98E165;
       position: absolute;
       right: 0;
       top: 10px;
@@ -63,23 +63,11 @@ export default defineComponent({
     .content {
       word-break break-word;
       padding: 6px 10px;
-      background-color: #223A34;
+      background-color: #98E165;
       color var(--content-color);
       font-size: var(--content-font-size);
       border-radius: 5px;
-      overflow: auto;
-
-      p {
-        line-height 1.5
-      }
-
-      p:last-child {
-        margin-bottom: 0
-      }
-
-      p:first-child {
-        margin-top 0
-      }
+      line-height 1.5
     }
   }
 }
diff --git a/web/src/components/mobile/ChatReply.vue b/web/src/components/mobile/ChatReply.vue
new file mode 100644
index 00000000..dd5ee081
--- /dev/null
+++ b/web/src/components/mobile/ChatReply.vue
@@ -0,0 +1,134 @@
+<template>
+  <div class="message-prompt">
+    <div class="chat-icon">
+      <img :src="icon" alt="ChatGPT">
+    </div>
+
+    <div class="chat-item">
+      <div class="triangle"></div>
+      <div class="content-box">
+        <div class="content" v-html="content"></div>
+        <div class="tool-box">
+          <el-tooltip
+              class="box-item"
+              effect="dark"
+              content="复制回答"
+              placement="bottom"
+          >
+            <el-button type="info" class="copy-reply" :data-clipboard-text="orgContent" plain>
+              <el-icon>
+                <DocumentCopy/>
+              </el-icon>
+            </el-button>
+          </el-tooltip>
+        </div>
+      </div>
+
+    </div>
+  </div>
+</template>
+
+<script>
+import {defineComponent} from "vue"
+import {randString} from "@/utils/libs";
+import {DocumentCopy} from "@element-plus/icons-vue";
+
+export default defineComponent({
+  name: 'ChatReply',
+  components: {DocumentCopy},
+  props: {
+    content: {
+      type: String,
+      default: '',
+    },
+    orgContent: {
+      type: String,
+      default: '',
+    },
+    icon: {
+      type: String,
+      default: 'images/gpt-icon.png',
+    }
+  },
+  data() {
+    return {
+      id: randString(32),
+      clipboard: null,
+    }
+  },
+
+})
+</script>
+
+<style lang="stylus">
+.message-prompt {
+  justify-content: flex-start;
+
+  .chat-icon {
+    margin-right 5px;
+
+    img {
+      border-radius 5px;
+    }
+  }
+
+  .chat-item {
+    display: inline-block;
+    position: relative;
+    padding: 0 0 0 5px;
+    overflow: hidden;
+
+    .triangle {
+      width: 0;
+      height: 0;
+      border-top: 5px solid transparent;
+      border-bottom: 5px solid transparent;
+      border-right: 5px solid #fff;
+      position: absolute;
+      left: 0;
+      top: 13px;
+    }
+
+    .content-box {
+
+      display flex
+      flex-direction row
+
+      .content {
+        min-height 20px;
+        word-break break-word;
+        padding: 6px 10px;
+        color var(--content-color)
+        background-color: #fff;
+        font-size: var(--content-font-size);
+        border-radius: 5px;
+
+        p:last-child {
+          margin-bottom: 0
+        }
+
+        p:first-child {
+          margin-top 0
+        }
+
+        p > code {
+          color #cc0000
+          background-color #f1f1f1
+        }
+      }
+
+      .tool-box {
+        padding-left 10px;
+        font-size 16px;
+
+        .el-button {
+          height 20px
+          padding 5px 2px;
+        }
+      }
+    }
+
+  }
+}
+
+</style>
\ No newline at end of file
diff --git a/web/src/main.js b/web/src/main.js
index f55099a6..3388c472 100644
--- a/web/src/main.js
+++ b/web/src/main.js
@@ -9,6 +9,7 @@ import {
     Cell,
     CellGroup,
     ConfigProvider,
+    Dialog,
     DropdownItem,
     DropdownMenu,
     Field,
@@ -52,6 +53,7 @@ app.use(Icon);
 app.use(DropdownItem);
 app.use(Sticky);
 app.use(SwipeCell);
+app.use(Dialog);
 app.use(router).use(ElementPlus).mount('#app')
 
 
diff --git a/web/src/views/mobile/ChatList.vue b/web/src/views/mobile/ChatList.vue
index 5ebc0e90..e8d272ce 100644
--- a/web/src/views/mobile/ChatList.vue
+++ b/web/src/views/mobile/ChatList.vue
@@ -26,7 +26,7 @@
           @load="onLoad"
       >
         <van-swipe-cell v-for="item in chats" :key="item.id">
-          <van-cell>
+          <van-cell @click="changeChat(item)">
             <div class="chat-list-item">
               <van-image
                   round
@@ -70,10 +70,12 @@
 import {ref} from "vue";
 import {httpGet} from "@/utils/http";
 import {getLoginUser} from "@/store/session";
-import {showFailToast, showSuccessToast, showToast} from "vant";
+import {showConfirmDialog, showFailToast, showSuccessToast, showToast} from "vant";
 import {checkSession} from "@/action/session";
 import router from "@/router";
 import {setChatConfig} from "@/store/chat";
+import {removeArrayItem, UUID} from "@/utils/libs";
+import {ElMessage} from "element-plus";
 
 const title = ref("会话列表")
 const chatName = ref("")
@@ -84,9 +86,10 @@ const finished = ref(false)
 const error = ref(false)
 const user = getLoginUser()
 const isLogin = ref(false)
-
+const roles = ref([])
+const models = ref([])
 const showPicker = ref(false)
-const columns = ref([])
+const columns = ref([roles.value, models.value])
 
 checkSession().then(() => {
   isLogin.value = true
@@ -94,12 +97,10 @@ checkSession().then(() => {
   httpGet(`/api/role/list?user_id=${user.id}`).then((res) => {
     if (res.data) {
       const items = res.data
-      const roles = []
       for (let i = 0; i < items.length; i++) {
         // console.log(items[i])
-        roles.push({text: items[i].name, value: items[i].id, icon: items[i].icon})
+        roles.value.push({text: items[i].name, value: items[i].id, icon: items[i].icon})
       }
-      columns.value[0] = roles
     }
   }).catch(() => {
     showFailToast("加载聊天角色失败")
@@ -109,11 +110,9 @@ checkSession().then(() => {
   httpGet('/api/admin/config/get?key=system').then(res => {
     if (res.data) {
       const items = res.data.models
-      const models = []
       for (let i = 0; i < items.length; i++) {
-        models.push({text: items[i].toUpperCase(), value: items[i]})
+        models.value.push({text: items[i].toUpperCase(), value: items[i]})
       }
-      columns.value[1] = models
     }
   }).catch(() => {
     showFailToast("加载系统配置失败")
@@ -151,7 +150,19 @@ const search = () => {
 }
 
 const clearAllChatHistory = () => {
-  showSuccessToast('所有聊天记录已清空')
+  showConfirmDialog({
+    title: '操作提示',
+    message: '确定要删除所有的会话记录吗？'
+  }).then(() => {
+    httpGet("/api/chat/clear").then(() => {
+      showSuccessToast('所有聊天记录已清空')
+      chats.value = [];
+    }).catch(e => {
+      showFailToast("操作失败：" + e.message)
+    })
+  }).catch(() => {
+    // on cancel
+  })
 }
 
 const newChat = (item) => {
@@ -164,16 +175,47 @@ const newChat = (item) => {
       icon: options[0].icon
     },
     model: options[1].value,
+    title: '新建会话',
+    chatId: UUID()
+  })
+  router.push('/mobile/chat/session')
+}
+
+const changeChat = (chat) => {
+  let role = {}
+  for (let i = 0; i < roles.value.length; i++) {
+    if (roles.value[i].value === chat.role_id) {
+      role = roles.value[i]
+      break
+    }
+  }
+  setChatConfig({
+    role: {
+      id: chat.role_id,
+      name: role.text,
+      icon: role.icon
+    },
+    model: chat.model,
+    title: chat.title,
+    chatId: chat.chat_id
   })
   router.push('/mobile/chat/session')
 }
 
 const editChat = (item) => {
   showToast('修改会话标题')
+
 }
 
 const removeChat = (item) => {
-  showToast('删除当前会话')
+  httpGet('/api/chat/remove?chat_id=' + item.chat_id).then(() => {
+    chats.value = removeArrayItem(chats.value, item, function (e1, e2) {
+      return e1.id === e2.id
+    })
+  }).catch(e => {
+    showFailToast('操作失败：' + e.message);
+  })
+
 }
 
 </script>
diff --git a/web/src/views/mobile/ChatSession.vue b/web/src/views/mobile/ChatSession.vue
index 98bcbadf..2db60eaf 100644
--- a/web/src/views/mobile/ChatSession.vue
+++ b/web/src/views/mobile/ChatSession.vue
@@ -18,8 +18,30 @@
     </van-sticky>
 
 
-    <div class="message-list-box" :style="{height: winHeight+'px'}">
-
+    <div class="message-list-box" id="message-list-box" :style="{height: winHeight+'px'}">
+      <van-list
+          v-model:loading="loading"
+          :finished="finished"
+          v-model:error="error"
+          error-text="请求失败，点击重新加载"
+          @load="onLoad"
+      >
+        <van-cell v-for="item in chatData" :key="item">
+          <chat-prompt
+              v-if="item.type==='prompt'"
+              :icon="item.icon"
+              :created-at="dateFormat(item['created_at'])"
+              :tokens="item['tokens']"
+              :model="model"
+              :content="item.content"/>
+          <chat-reply v-else-if="item.type==='reply'"
+                      :icon="item.icon"
+                      :org-content="item.orgContent"
+                      :created-at="dateFormat(item['created_at'])"
+                      :tokens="item['tokens']"
+                      :content="item.content"/>
+        </van-cell>
+      </van-list>
     </div>
 
     <van-sticky :offset-bottom="0" position="bottom" ref="bottomBarRef">
@@ -49,13 +71,17 @@
 </template>
 
 <script setup>
-import {onMounted, ref} from "vue";
+import {nextTick, onMounted, ref} from "vue";
 import {showToast} from "vant";
 import {useRouter} from "vue-router";
-import {UUID} from "@/utils/libs";
+import {dateFormat, UUID} from "@/utils/libs";
 import {getChatConfig} from "@/store/chat";
+import {httpGet} from "@/utils/http";
+import hl from "highlight.js";
+import 'highlight.js/styles/a11y-dark.css'
+import ChatPrompt from "@/components/mobile/ChatPrompt.vue";
+import ChatReply from "@/components/mobile/ChatReply.vue";
 
-const title = ref('简单介绍一下高更的艺术思想')
 const winHeight = ref(0)
 const navBarRef = ref(null)
 const bottomBarRef = ref(null)
@@ -64,11 +90,52 @@ const router = useRouter()
 const chatConfig = getChatConfig()
 const role = chatConfig.role
 const model = chatConfig.model
+const title = chatConfig.title
+const chatId = chatConfig.chatId
 
 onMounted(() => {
-  winHeight.value = window.innerHeight - navBarRef.value.$el.offsetHeight - bottomBarRef.value.$el.offsetHeight
+  winHeight.value = document.body.offsetHeight - navBarRef.value.$el.offsetHeight - bottomBarRef.value.$el.offsetHeight
 })
 
+const chatData = ref([])
+const loading = ref(false)
+const finished = ref(false)
+const error = ref(false)
+const onLoad = () => {
+  httpGet('/api/chat/history?chat_id=' + chatId).then(res => {
+    // 加载状态结束
+    loading.value = false;
+    finished.value = true;
+    const data = res.data
+    if (!data || data.length === 0) {
+      return
+    }
+
+    const md = require('markdown-it')();
+    for (let i = 0; i < data.length; i++) {
+      if (data[i].type === "prompt") {
+        chatData.value.push(data[i]);
+        continue;
+      }
+
+      data[i].orgContent = data[i].content;
+      data[i].content = md.render(data[i].content);
+      chatData.value.push(data[i]);
+    }
+
+    nextTick(() => {
+      hl.configure({ignoreUnescapedHTML: true})
+      const blocks = document.querySelector("#message-list-box").querySelectorAll('pre code');
+      blocks.forEach((block) => {
+        hl.highlightElement(block)
+      })
+    })
+  }).catch(() => {
+    error.value = true
+  })
+
+};
+
 // 创建 socket 连接
 const prompt = ref('');
 const showStopGenerate = ref(false); // 停止生成
@@ -229,7 +296,7 @@ const reGenerate = () => {
 <style scoped lang="stylus">
 .mobile-chat {
   .message-list-box {
-
+    overflow-x auto
   }
 
   .chat-box {
diff --git a/web/src/views/mobile/Home.vue b/web/src/views/mobile/Home.vue
index c161bc64..227cc8fe 100644
--- a/web/src/views/mobile/Home.vue
+++ b/web/src/views/mobile/Home.vue
@@ -4,7 +4,7 @@
       <router-view/>
 
       <van-tabbar route v-model="active" @change="onChange">
-        <van-tabbar-item to="/mobile/chat" name="home" icon="chat-o"></van-tabbar-item>
+        <van-tabbar-item to="/mobile/chat/list" name="home" icon="chat-o"></van-tabbar-item>
         <van-tabbar-item to="/mobile/setting" name="setting" icon="setting-o"></van-tabbar-item>
         <van-tabbar-item to="/mobile/profile" name="profile" icon="user-o"></van-tabbar-item>
       </van-tabbar>