fix: resolve input focus loss issue after sending messages

src/renderer/src/components/ChatInput.vue

@@ -388,10 +388,12 @@ const props = withDefaults(
     contextLength?: number
     maxRows?: number
     rows?: number
+    disabled?: boolean
   }>(),
   {
     maxRows: 10,
-    rows: 1
+    rows: 1,
+    disabled: false
   }
 )
 
@@ -686,7 +688,7 @@ const emitSend = async () => {
 
     emit('send', messageContent)
     inputText.value = ''
-    editor.chain().clearContent().blur().run()
+    editor.chain().clearContent().run()
 
     // 清除历史记录placeholder
     clearHistoryPlaceholder()
@@ -700,6 +702,9 @@ const emitSend = async () => {
         fileInput.value.value = ''
       }
     }
+    nextTick(() => {
+      editor.commands.focus()
+    })
   }
 }
 
@@ -982,6 +987,14 @@ onMounted(() => {
     editor.commands.focus()
   })
 
+  document.addEventListener('visibilitychange', () => {
+    if (!document.hidden) {
+      setTimeout(() => {
+        restoreFocus()
+      }, 100)
+    }
+  })
+
   window.electron.ipcRenderer.on('rate-limit:config-updated', handleRateLimitEvent)
   window.electron.ipcRenderer.on('rate-limit:request-executed', handleRateLimitEvent)
   window.electron.ipcRenderer.on('rate-limit:request-queued', handleRateLimitEvent)
@@ -1090,6 +1103,17 @@ watch(dynamicPlaceholder, () => {
   updatePlaceholder()
 })
 
+watch(
+  () => props.disabled,
+  (newDisabled, oldDisabled) => {
+    if (oldDisabled && !newDisabled) {
+      setTimeout(() => {
+        restoreFocus()
+      }, 100)
+    }
+  }
+)
+
 // 处理历史记录placeholder
 const setHistoryPlaceholder = (text: string) => {
   currentHistoryPlaceholder.value = text
@@ -1157,6 +1181,21 @@ function onKeydown(e: KeyboardEvent) {
   }
 }
 
+const restoreFocus = () => {
+  nextTick(() => {
+    if (editor && !editor.isDestroyed && !props.disabled) {
+      try {
+        const editorElement = editor.view.dom
+        if (editorElement && editorElement.offsetParent !== null) {
+          editor.commands.focus()
+        }
+      } catch (error) {
+        console.warn('恢复焦点时出错:', error)
+      }
+    }
+  })
+}
+
 // 通过名称查找mention数据
 const findMentionByName = (name: string): CategorizedData | null => {
   // 在当前的mentionData中查找匹配的项目
@@ -1261,7 +1300,8 @@ defineExpose({
       console.error('Failed to append mention:', error)
       return false
     }
-  }
+  },
+  restoreFocus
 })
 </script>
 

src/renderer/src/components/ChatView.vue

@@ -11,6 +11,7 @@
     <!-- 输入框区域 -->
     <div class="flex-none px-2 pb-2">
       <ChatInput
+        ref="chatInput"
         :disabled="!chatStore.getActiveThreadId() || isGenerating"
         @send="handleSend"
         @file-upload="handleFileUpload"
@@ -32,6 +33,7 @@ const route = useRoute()
 const settingsStore = useSettingsStore()
 
 const messageList = ref()
+const chatInput = ref()
 
 import { useChatStore } from '@/stores/chat'
 
@@ -48,6 +50,10 @@ const handleSend = async (msg: UserMessageContent) => {
   scrollToBottom()
   await chatStore.sendMessage(msg)
   scrollToBottom()
+
+  setTimeout(() => {
+    chatInput.value?.restoreFocus()
+  }, 100)
 }
 
 const handleFileUpload = () => {
@@ -63,10 +69,16 @@ onMounted(async () => {
 
   window.electron.ipcRenderer.on(STREAM_EVENTS.END, (_, msg) => {
     chatStore.handleStreamEnd(msg)
+    setTimeout(() => {
+      chatInput.value?.restoreFocus()
+    }, 200)
   })
 
   window.electron.ipcRenderer.on(STREAM_EVENTS.ERROR, (_, msg) => {
     chatStore.handleStreamError(msg)
+    setTimeout(() => {
+      chatInput.value?.restoreFocus()
+    }, 200)
   })
 
   if (route.query.modelId && route.query.providerId) {