terminal-bridge-setup 3.0.0 → 3.1.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);
@@ -688,15 +693,27 @@ function resultJsonToCsv(jsonText) {
   return "\uFEFF" + [headers.join(","), ...rows].join("\r\n") + "\r\n";
 }
 
+// 文件名：表名_库_数据源_日期。表名从 SQL 提取（from/into/update 后的词），
+// 库和数据源从目标 tab 的 meta（tapTabMeta）取。
+function buildCsvFileName(sql, tabId) {
+  const sqlText = sql || "";
+  const tableMatch = sqlText.match(/\b(?:from|into|update|join)\s+[`"]?(\w+)[`"]?/i);
+  const table = (tableMatch ? tableMatch[1] : "query").slice(0, 40);
+  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 `${table}_${database}_${source}_${date}_${time}.csv`;
+}
+
 async function handleYrExportCsv(frame) {
   const csv = resultJsonToCsv(frame.payload || "");
   if (!csv) {
     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 = buildCsvFileName(frame.sql, frame.tabId);
   const dataUrl = "data:text/csv;charset=utf-8," + encodeURIComponent(csv);
   await addCsvExport({
     id: Date.now(),

package/files/extension/popup.html

@@ -200,10 +200,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,8 +144,10 @@ 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)}">
@@ -166,6 +168,41 @@ function refreshCsvList() {
 }
 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.1.0",
   "description": "一次性安装器：释放终端桥接（JumpServer / Arthas）的本地代理 + Chrome 插件，并注册 native messaging host。让 Agent 能通过浏览器 xterm 终端执行命令并拿回输出。",
   "license": "MIT",
   "author": "encorearon",