terminal-bridge-setup 3.0.0 → 3.2.0

package/files/extension/background.js

@@ -135,6 +135,11 @@ chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
     });
     return true;  // 异步
   }
+  if (msg.type === "CSV_CLEAR") {
+    chrome.storage.session.remove(CSV_STORAGE_KEY).catch(() => {});
+    sendResponse({ ok: true });
+    return false;
+  }
   if (msg.type === "CSV_DOWNLOAD") {
     getCsvExports().then(list => {
       const record = list.find(e => e.id === msg.id);
@@ -143,13 +148,38 @@ chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
         return;
       }
       chrome.downloads.download(
-        { url: record.dataUrl, filename: record.name, saveAs: true },
+        { url: record.dataUrl, filename: "yearning-csv/" + record.name, saveAs: false },
         (downloadId) => {
           if (chrome.runtime.lastError) {
             sendResponse({ ok: false, msg: chrome.runtime.lastError.message });
             return;
           }
-          sendResponse({ ok: true, downloadId });
+          sendResponse({ ok: true, downloadId, path: "~/Downloads/yearning-csv/" + record.name });
+        }
+      );
+    });
+    return true;  // 异步
+  }
+  if (msg.type === "CSV_PROMPT") {
+    // 🤖 按钮：静默下载该文件到固定目录，返回让 Agent 读文件的 prompt
+    getCsvExports().then(list => {
+      const record = list.find(e => e.id === msg.id);
+      if (!record) {
+        sendResponse({ ok: false, msg: "导出记录不存在" });
+        return;
+      }
+      chrome.downloads.download(
+        { url: record.dataUrl, filename: "yearning-csv/" + record.name, saveAs: false },
+        (downloadId) => {
+          if (chrome.runtime.lastError) {
+            sendResponse({ ok: false, msg: chrome.runtime.lastError.message });
+            return;
+          }
+          const table = record.table || record.name.split("_")[0];
+          sendResponse({
+            ok: true,
+            prompt: `表名${table}的查询结果请读这个文件 ~/Downloads/yearning-csv/${record.name}`,
+          });
         }
       );
     });
@@ -672,20 +702,47 @@ function csvCell(value) {
   return s;
 }
 
-// Yearning 结果 JSON → CSV 文本。多结果集时只取第一个非空表
-// （Yearning 对 SHOW INDEX 会推两份相同结果，取一即可）
+// Yearning 结果 JSON → CSV 文本。多结果集全部写入同一个 CSV，不能只取第一张表：
+// 一次执行多条 SQL 时 results 有多项，popup 行数是总和，文件也必须包含总和。
+// 每组结果独立表头，组间用 Result Set 标记和空行分隔，兼容不同 SQL 的列结构。
 function resultJsonToCsv(jsonText) {
   let obj;
   try { obj = JSON.parse(jsonText); } catch { return null; }
   if (!Array.isArray(obj.results)) return null;
-  const table = obj.results.find(t => t && Array.isArray(t.field) && t.field.length > 0)
-    || obj.results[0];
-  if (!table || !Array.isArray(table.field)) return null;
-  const headers = table.field.map(f => csvCell(f.title || f.dataIndex || ""));
-  const rows = (table.data || []).map(row =>
-    table.field.map(f => csvCell(row[f.dataIndex])).join(",")
-  );
-  return "\uFEFF" + [headers.join(","), ...rows].join("\r\n") + "\r\n";
+
+  const blocks = [];
+  obj.results.forEach((table, index) => {
+    if (!table || !Array.isArray(table.field) || table.field.length === 0) return;
+    const headers = table.field.map(f => csvCell(f.title || f.dataIndex || ""));
+    const rows = (table.data || []).map(row =>
+      table.field.map(f => csvCell(row[f.dataIndex])).join(",")
+    );
+    // 多结果集时保留组标记；单结果集不额外增加噪音
+    if (obj.results.filter(t => t && Array.isArray(t.field) && t.field.length > 0).length > 1) {
+      blocks.push(`-- Result Set ${index + 1} --`);
+    }
+    blocks.push(headers.join(","), ...rows, "");
+  });
+  return blocks.length > 0 ? "\uFEFF" + blocks.join("\r\n") : null;
+}
+
+// 文件名：表名_库_数据源_日期。表名从 SQL 提取（from/into/update 后的词），
+// 库和数据源从目标 tab 的 meta（tapTabMeta）取。
+// 手动查询（sql 为 manual-query）时先读编辑器里的实际 SQL 再提取表名。
+async function buildCsvFileName(sql, tabId) {
+  let sqlText = sql || "";
+  if (!sqlText || sqlText === "manual-query") {
+    const editor = await sendFrameMessage(tabId, { type: "yr-sql-get" }, 0);
+    if (editor?.ok && editor.sql) sqlText = editor.sql;  // 手动查询后编辑器里就是刚执行的 SQL
+  }
+  const tableMatch = sqlText.match(/\b(?:from|into|update|join)\s+[`"]?(\w+)[`"]?/i);
+  const table = (tableMatch ? tableMatch[1] : "").slice(0, 40) || "query";
+  const meta = tapTabMeta.get(tabId) || {};
+  const database = (meta.database || "").replace(/[^\w-]+/g, "") || "nodb";
+  const source = (meta.dataSource || "").split(" · ")[0].replace(/[^\w.-]+/g, "") || "nosrc";
+  const date = new Date().toISOString().slice(0, 10);
+  const time = new Date().toTimeString().slice(0, 5).replace(":", "");
+  return { name: `${table}_${database}_${source}_${date}_${time}.csv`, table };
 }
 
 async function handleYrExportCsv(frame) {
@@ -694,13 +751,12 @@ async function handleYrExportCsv(frame) {
     console.warn("[bg] yr-export-csv: 结果 JSON 解析失败或无表结构");
     return;
   }
-  const stamp = new Date().toISOString().replace(/[:.]/g, "-").slice(0, 19);
-  const sqlHead = (frame.sql || "query").slice(0, 30).replace(/[^\w-]+/g, "_");
-  const name = `yearning-${sqlHead}-${stamp}.csv`;
+  const { name, table } = await buildCsvFileName(frame.sql, frame.tabId);
   const dataUrl = "data:text/csv;charset=utf-8," + encodeURIComponent(csv);
   await addCsvExport({
     id: Date.now(),
     name,
+    table,
     rows: frame.rows || 0,
     sql: frame.sql || "",
     time: Date.now(),

package/files/extension/content-yearning.js

@@ -107,7 +107,8 @@
       if (idc && idc !== dataSource) dataSource = dataSource ? `${dataSource} · ${idc}` : idc;
     } catch {}
 
-    // 3. 兜底：form 启发式
+    // 3. 兜底：form 启发式（排除与 dataSource 重叠/相似的值，防止把
+    //    数据源名误认成数据库名——文件名第二段曾因此错成 dk_shard）
     if (!database || !dataSource) {
       const form = document.querySelector("form") || xpathNode(FORM_XPATH);
       if (form) {
@@ -117,9 +118,11 @@
           const value = el.tagName === "SELECT" ? el.options[el.selectedIndex]?.textContent : el.value;
           if (value?.trim()) values.push(value.trim());
         });
-        const unique = [...new Set(values)].filter(v => !/^(查询|执行|取消|确定|SQL)$/i.test(v));
+        const unique = [...new Set(values)].filter(v =>
+          !/^(查询|执行|取消|确定|SQL)$/i.test(v) &&
+          v !== dataSource && !dataSource.includes(v) && !v.includes("shard"));
         if (!dataSource) dataSource = unique.find(v => /source|实例|数据源|tdsql|mysql|prod|test/i.test(v)) || "";
-        if (!database) database = unique.find(v => /database|db|库|schema/i.test(v)) || "";
+        if (!database) database = unique.find(v => /database|schema|^\w+_dk\b/i.test(v)) || "";
       }
     }
 
@@ -293,6 +296,11 @@
       sendResponse({ ok: true });
       return true;
     }
+    if (msg.type === "yr-sql-get") {
+      // 读当前编辑器完整 SQL（手动查询结果帧到达时，编辑器里就是刚执行的 SQL）
+      sendResponse({ ok: true, sql: readMonacoText() });
+      return true;
+    }
     if (msg.type === "yr-verify-sql") {
       // CDP 注入后验证：读回 monaco view-lines 内容比对
       const current = readMonacoText();

package/files/extension/popup.html

@@ -131,6 +131,10 @@
     .csv-item:hover { background: #dcebfd; }
     .csv-item .csv-name { flex: 1; min-width: 0; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
     .csv-item .csv-rows { color: #888; font-size: 10px; flex-shrink: 0; }
+    .csv-prompt-btn {
+      width: auto !important; padding: 2px 7px !important; margin: 0 !important;
+      font-size: 11px; background: #7c4dff; flex-shrink: 0;
+    }
     .guide {
       font-size: 11px;
       color: #666;
@@ -200,10 +204,14 @@
     <div class="hint" id="wsTapStatus">未监听 Yearning 页面</div>
     <div id="yearningTabList" style="margin-top:6px;"></div>
     <div id="csvSection" style="margin-top:8px;display:none;">
-      <div class="section-title" style="margin-bottom:4px;">CSV 导出记录</div>
+      <div style="display:flex;align-items:center;justify-content:space-between;margin-bottom:4px;">
+        <span class="section-title" style="margin:0;">CSV 导出记录</span>
+        <button class="copy-btn" id="btnCsvClear" style="background:#e53935;">清空</button>
+      </div>
       <div id="csvList"></div>
-      <div class="hint">点击记录可重新下载（仅保留最近 20 条，插件刷新后清空）</div>
+      <div class="hint">点击记录下载 · 表名_库_数据源_日期.csv</div>
     </div>
+    <button class="scan" id="btnCopyPrompt" style="margin-top:8px;background:#7c4dff;">📋 复制 Agent 使用 Prompt</button>
   </div>
 
   <div class="section">

package/files/extension/popup.js

@@ -144,17 +144,22 @@ const csvList = document.getElementById('csvList');
 function refreshCsvList() {
   chrome.runtime.sendMessage({ type: 'CSV_LIST' }, (res) => {
     if (chrome.runtime.lastError || !res || !res.ok) return;
-    const exports = res.exports || [];
-    csvSection.style.display = exports.length ? 'block' : 'none';
+    const all = res.exports || [];
+    csvSection.style.display = all.length ? 'block' : 'none';
+    // 只显示最近 5 条，popup 不超长（存储仍保留 20 条）
+    const exports = all.slice(0, 5);
     csvList.innerHTML = exports.map(e => {
       const time = new Date(e.time).toLocaleTimeString();
       return `<div class="csv-item" data-id="${e.id}" title="${escapeHtml(e.sql || e.name)}">
         <span class="csv-name">📄 ${escapeHtml(e.name)}</span>
         <span class="csv-rows">${e.rows} 行 · ${time}</span>
+        <button class="copy-btn csv-prompt-btn" data-name="${escapeHtml(e.name)}" title="复制让 Agent 读这个文件的 Prompt">🤖</button>
       </div>`;
     }).join('');
     csvList.querySelectorAll('.csv-item').forEach(el => {
-      el.onclick = () => {
+      el.onclick = (ev) => {
+        // 点复制按钮不触发行点击（下载）
+        if (ev.target.closest('.csv-prompt-btn')) return;
         chrome.runtime.sendMessage({ type: 'CSV_DOWNLOAD', id: Number(el.dataset.id) }, (r) => {
           if (chrome.runtime.lastError || !r || !r.ok) {
             console.warn('重新下载失败:', r && r.msg);
@@ -162,10 +167,61 @@ function refreshCsvList() {
         });
       };
     });
+    csvList.querySelectorAll('.csv-prompt-btn').forEach(btn => {
+      btn.onclick = (ev) => {
+        ev.stopPropagation();
+        const id = Number(btn.closest('.csv-item').dataset.id);
+        chrome.runtime.sendMessage({ type: 'CSV_PROMPT', id }, (r) => {
+          if (chrome.runtime.lastError || !r || !r.ok) {
+            console.warn('prompt 失败:', r && r.msg);
+            return;
+          }
+          navigator.clipboard.writeText(r.prompt).then(() => {
+            btn.textContent = '✓';
+            setTimeout(() => { btn.textContent = '🤖'; }, 1500);
+          }).catch(() => {});
+        });
+      };
+    });
   });
 }
 refreshCsvList();
 
+// 清空 CSV 导出记录
+document.getElementById('btnCsvClear').onclick = () => {
+  chrome.runtime.sendMessage({ type: 'CSV_CLEAR' }, () => refreshCsvList());
+};
+
+// 复制 Agent 使用 Prompt（粘给任意 AI 助手即可用桥接查 Yearning）
+const AGENT_PROMPT = `你的机器上已装好 Terminal Bridge（Yearning SQL 桥接）。请按以下方式查询数据库：
+
+## 查询（Yearning）
+cd ~/.terminal-bridge/proxy && node yr-example.mjs "SELECT ...;" [超时ms]
+
+返回 JSON：{ ok, output(结果 JSON: field 列定义 + data 数据行), error }。
+
+## 说明
+- SQL 会自动注入到用户浏览器里选中的 Yearning 页面并点「查询」，结果自动返回
+- 前置：用户已在 Yearning 页面选择数据库，且插件 popup 已「监听当前 Yearning 页」
+- 多个 Yearning 页面时：用户在 popup 列表点选目标页面（显示 数据源 · 数据库）
+- 报 database-not-selected = 页面未选数据库，请让用户先选
+- 报 timeout = 查询超时或页面未监听，让用户确认 popup 状态
+- 只读查询即可；每次查询的结果会自动出现在插件 popup 的「CSV 导出记录」里供用户下载
+
+## 终端命令（JumpServer/Arthas，同一代理）
+cd ~/.terminal-bridge/proxy && node client-example.mjs "linux 命令" [超时ms]
+多层引号命令加 --b64 参数下发。`;
+
+document.getElementById('btnCopyPrompt').onclick = () => {
+  navigator.clipboard.writeText(AGENT_PROMPT).then(() => {
+    const btn = document.getElementById('btnCopyPrompt');
+    btn.textContent = '✓ 已复制，粘贴给 Agent 即可';
+    setTimeout(() => { btn.textContent = '📋 复制 Agent 使用 Prompt'; }, 2000);
+  }).catch(() => {
+    console.warn('复制失败');
+  });
+};
+
 // ============== 代理控制 ==============
 const proxyDot = document.getElementById('proxyDot');
 const proxyText = document.getElementById('proxyStatusText');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "terminal-bridge-setup",
-  "version": "3.0.0",
+  "version": "3.2.0",
   "description": "一次性安装器：释放终端桥接（JumpServer / Arthas）的本地代理 + Chrome 插件，并注册 native messaging host。让 Agent 能通过浏览器 xterm 终端执行命令并拿回输出。",
   "license": "MIT",
   "author": "encorearon",