From 02a9c422fe57e558ee29051e35682be3dc07558c Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Thu, 29 Feb 2024 15:41:45 +0800
Subject: [PATCH] fix: fixed bug image preview im mobile chat session page

---
 api/handler/chatimpl/chat_handler.go    |  1 +
 api/main.go                             |  3 ---
 web/src/components/mobile/ChatReply.vue |  2 +-
 web/src/views/mobile/ChatSession.vue    | 14 +++++++++++++-
 4 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/api/handler/chatimpl/chat_handler.go b/api/handler/chatimpl/chat_handler.go
index 90effac7..58fcaa43 100644
--- a/api/handler/chatimpl/chat_handler.go
+++ b/api/handler/chatimpl/chat_handler.go
@@ -131,6 +131,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 		c.Abort()
 		return
 	}
+	h.Init()
 
 	// 保存会话连接
 	h.App.ChatClients.Put(sessionId, client)
diff --git a/api/main.go b/api/main.go
index cd1e9bda..e1a45534 100644
--- a/api/main.go
+++ b/api/main.go
@@ -393,9 +393,6 @@ func main() {
 				log.Fatal(err)
 			}
 		}),
-		fx.Invoke(func(h *chatimpl.ChatHandler) {
-			h.Init()
-		}),
 		// 注册生命周期回调函数
 		fx.Invoke(func(lifecycle fx.Lifecycle, lc *AppLifecycle) {
 			lifecycle.Append(fx.Hook{
diff --git a/web/src/components/mobile/ChatReply.vue b/web/src/components/mobile/ChatReply.vue
index 2c46e47e..90e0541a 100644
--- a/web/src/components/mobile/ChatReply.vue
+++ b/web/src/components/mobile/ChatReply.vue
@@ -15,7 +15,7 @@
 </template>
 
 <script setup>
-import {onMounted, ref} from "vue"
+import {nextTick, onMounted, ref} from "vue"
 
 import {showImagePreview} from "vant";
 
diff --git a/web/src/views/mobile/ChatSession.vue b/web/src/views/mobile/ChatSession.vue
index e060a066..14b1fb4f 100644
--- a/web/src/views/mobile/ChatSession.vue
+++ b/web/src/views/mobile/ChatSession.vue
@@ -104,7 +104,7 @@
 
 <script setup>
 import {nextTick, onMounted, ref} from "vue";
-import {showNotify, showToast} from "vant";
+import {showImagePreview, showNotify, showToast} from "vant";
 import {onBeforeRouteLeave, useRouter} from "vue-router";
 import {dateFormat, processContent, randString, renderInputText, UUID} from "@/utils/libs";
 import {getChatConfig} from "@/store/chat";
@@ -326,6 +326,18 @@ const connect = function (chat_id, role_id) {
               hl.highlightElement(block)
             })
             scrollListBox()
+
+            const items = document.querySelectorAll('.message-line')
+            const imgs = items[items.length - 1].querySelectorAll('img')
+            for (let i = 0; i < imgs.length; i++) {
+              if (!imgs[i].src) {
+                continue
+              }
+              imgs[i].addEventListener('click', (e) => {
+                e.stopPropagation()
+                showImagePreview([imgs[i].src]);
+              })
+            }
           })
         }