From c8c13eacaca83ec68eb1d0b02f8e3cb5ff3c8181 Mon Sep 17 00:00:00 2001
From: t8y2 <1156263951@qq.com>
Date: Tue, 5 May 2026 04:18:51 +0800
Subject: [PATCH] feat: add AI assistant SQL handling functions for query tab
 management

---
 src/App.vue | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index cd361157b..e60c37f57 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -168,6 +168,25 @@ function toggleLocale() { setLocale(currentLocale() === "zh-CN" ? "en" : "zh-CN"
 function openGitHub() { openUrl("https://github.com/t8y2/dbx"); }
 function openMcpGuide() { openUrl("https://github.com/t8y2/dbx/blob/main/docs/mcp-guide.md"); }
 
+function ensureQueryTab(): string {
+  const tab = activeTab.value;
+  if (tab && tab.mode === "query") return tab.id;
+  const connId = connectionStore.activeConnectionId || connectionStore.connections[0]?.id || "";
+  const db = tab?.database || "";
+  return queryStore.createTab(connId, db, undefined, "query");
+}
+
+function onAiReplaceSql(sql: string) {
+  const tabId = ensureQueryTab();
+  queryStore.updateSql(tabId, sql);
+}
+
+function onAiExecuteSql(sql: string) {
+  const tabId = ensureQueryTab();
+  queryStore.updateSql(tabId, sql);
+  nextTick(() => tryExecute());
+}
+
 function handleKeydown(e: KeyboardEvent) {
   if (isCloseTabShortcut(e)) {
     e.preventDefault();
@@ -280,8 +299,8 @@ onUnmounted(() => { window.removeEventListener("keydown", handleKeydown, true);
           <div class="panel-resize-handle panel-resize-handle--left" @mousedown="startAiPanelResize" />
           <div class="h-full min-h-0 overflow-hidden">
             <AiAssistant ref="aiAssistantRef" :tab="activeTab" :connection="activeConnection"
-              @replace-sql="(sql: string) => { const tab = activeTab; if (tab) queryStore.updateSql(tab.id, sql) }"
-              @execute-sql="tryExecute" @close="toggleAiPanel"
+              @replace-sql="onAiReplaceSql"
+              @execute-sql="onAiExecuteSql" @close="toggleAiPanel"
             />
           </div>
         </div>