@srgay/cursor-extension 1.0.1 → 1.0.3

package/README.md

@@ -3,7 +3,7 @@
 一组本机 Cursor workbench 增强脚本，包含三部分：
 
 1. **MAX Mode 守护**：保持 `MAX Mode` 关闭，并在彩色 `MAX` 标记出现且聊天框有内容时阻止发送。
-2. **MCP Follow-up 面板**：在 Cursor 聊天框上方注入一个 MCP 反馈面板，连接 `mcp-feedback-enhanced` 的 WebSocket，让你直接在 Cursor 里回复 `interactive_feedback`，支持端口自动扫描/自定义，并能按当前项目自动匹配对应端口（多项目、多端口同时在线时也不误连）。
+2. **MCP Follow-up 面板**：在 Cursor 聊天框上方注入一个 MCP 反馈面板，连接 `mcp-feedback-enhanced` 的 WebSocket，让你直接在 Cursor 里回复 `interactive_feedback`，支持端口自动扫描/自定义，并能按当前窗口自动匹配对应端口（多窗口、多端口同时在线时也不误连）。
 3. **输入法回车修复**：修复用中文输入法（拼音/注音等）组字时，按回车上屏候选词会被 Cursor 误当成「提交」的问题（含自带的 AskQuestion「Other」输入框）。
 
 ## 安装与使用（npx）
@@ -115,10 +115,10 @@ npm run inject
 
 - 注入到聊天框上方，样式贴合 Cursor 原生输入框。
 - 自动扫描端口 `8765–8769`，识别正在监听的 MCP 服务（连得上或返回 `4004 无 session` 即视为活跃）。
-- 端口下拉：动态填充扫描到的活跃端口，末尾提供「自定义端口…」可手动输入任意端口。
-- 下拉展开时每项显示「状态 · 项目名」（如 `Port 8766 · waiting · my-project`），收起时只显示「Port X · 状态」以贴合宽度，避免与右侧项目名重复。
-- 放大镜按钮可随时重新扫描。
-- **自动匹配当前项目端口**：从 Cursor 原生 `window.vscode` 取当前窗口的工作区路径，优先连接 `project_directory` 与之相等的 MCP 端口；多个端口同时在线（如别的项目窗口或常驻实例）时也能选对本项目，绝不误连。本项目端口尚未出现时持续重扫等待，出现后自动连上；取不到工作区（多根工作区/空窗口）时回退为原有的端口选择逻辑。
+- 端口下拉（自定义浮层菜单）：**点开下拉即扫描一次**——先显示「扫描中…」，扫完用最新结果填充并展示可用端口；点选**一次即生效**（自定义菜单不依赖原生 `<select>`，不打断、不闪），末尾提供「自定义端口…」可手动输入任意端口。点空白处或按 `Esc` 关闭菜单。
+- 菜单每项显示「状态 · 项目名」（如 `Port 8766 · waiting · my-project`）便于区分；触发按钮收起时只显示「Port X · 状态」以贴合宽度，避免与右侧项目名重复。
+- 放大镜按钮可随时重新扫描（全量扫描并自动连上本窗口端口）。
+- **自动匹配当前窗口端口**：每个 Cursor 窗口会起一个独立的 mcp-feedback 实例。面板从 Cursor 原生 `window.vscode` 取当前窗口工作区路径，再借 `run_command` 读取每个端口所在实例的 `WORKSPACE_FOLDER_PATHS`（窗口真实工作区，由 Cursor 注入、**不受 AI 传入的 `project_directory` 影响**），据此连接「本窗口」对应的端口；读不到时退用 `CURSOR_WORKSPACE_LABEL`（项目名）做次级匹配。多端口同时在线（别的窗口/常驻实例）也能选对、绝不误连；本窗口端口尚未出现时降频重扫等待，出现后自动连上；取不到工作区（多根工作区/空窗口）时回退原有逻辑。
 - 持续保持为 WebSocket「最后连接」（每 3 秒重连 + 输入框聚焦时重连），确保始终能收到当前会话并成功提交反馈。
 - 底部「常用提示词」下拉：一键把 `ui_settings.json` 里的提示词追加到输入框；旁边刷新按钮可重新拉取配置。
 - 齿轮按钮展开**配置抽屉**：
@@ -147,9 +147,9 @@ npm run inject:followup
 ### 端口扫描与自定义
 
 - 面板安装时自动扫描一次 `8765–8769`。
-- **按项目自动选端口**：扫描后优先选中 `project_directory` 等于当前窗口工作区路径的端口；若当前选中端口已知属于别的项目则不连接，避免误连到其它项目或常驻 feedback 实例。
-- 本项目端口还没出现时，每 3 秒重扫一次去发现它；连上本项目端口后改为定期重连以保持「最后连接」。
-- 点击放大镜按钮可手动重扫。
+- **按窗口自动选端口**：扫描时借 `run_command` 读取每个端口实例的 `WORKSPACE_FOLDER_PATHS`，优先选中与当前窗口工作区相等的端口（次级用 `CURSOR_WORKSPACE_LABEL` 项目名）；已知属于别的窗口的端口直接不连。这样即使「在 A 窗口让 AI 改 B 项目代码」（会话 `project_directory` 变成 B），也仍连回 A 窗口自己的端口。
+- **扫描时机**：只在「面板启动时」和「每次点开端口下拉时」各扫一次，不做后台轮询扫描；连上本窗口端口后只发心跳维持 WebSocket「最后连接」。本窗口端口暂未出现时，点开下拉重扫即可发现并连上。
+- 点击放大镜按钮可手动重扫（全量扫描并自动连上本窗口端口）。
 - 在端口下拉里选择「自定义端口…」，就地输入端口号回车即可连接；输入框失焦或按 `Esc` 会回退到第一个可用端口。
 
 ### 查看状态
@@ -177,7 +177,7 @@ window.__cursorMcpFollowup.uninstall()    // 卸载面板
 脚本顶部 `config` 可调整：
 
 - `scanStart` / `scanCount`：扫描起始端口与数量（默认从 `8765` 起、共 `5` 个，即 `8765–8769`）。
-- `probeTimeoutMs`：单端口探测超时（默认 `1200` ms）。
+- `probeTimeoutMs`：单端口探测超时（默认 `2500` ms；含一次 `run_command` 读取窗口工作区的往返）。
 - `reconnectMs`：自动重连间隔（默认 `3000` ms）。
 
 ### 排障

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@srgay/cursor-extension",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "本机 Cursor workbench 增强：MAX Mode 守护、MCP Follow-up 面板、输入法回车修复（支持随 Cursor 启动持久加载，或 CDP 临时注入）。",
   "type": "module",
   "bin": {

package/src/mcp-followup/cursor-mcp-followup.js

@@ -12,7 +12,8 @@
     reconnectMs: 3000,
     scanStart: 8765,
     scanCount: 5,
-    probeTimeoutMs: 1200,
+    probeTimeoutMs: 2500,
+    // 按需扫描：仅在「启动时」和「展开端口下拉框时」各扫一次，不做后台/持续扫描。
     customValue: "__custom__",
     lang: "zh-CN",
     // 发送场景重连成功（onopen）后，若服务端未及时推送会话状态，则等待此毫秒数后兜底直接发送。
@@ -40,8 +41,10 @@
     scanning: false,
     foundPorts: [],
     portProjects: {},
-    // 端口 → 完整 project_directory（扫描/会话时记录），用于把面板锁定到当前窗口项目对应的端口。
-    portPaths: {},
+    // 端口 → 窗口真实工作区路径 / 项目名（取自 server 进程的 WORKSPACE_FOLDER_PATHS / CURSOR_WORKSPACE_LABEL，
+    // 由 Cursor 注入、不受 AI 传入的 project_directory 影响），用于把面板锁定到「本窗口」对应的端口。
+    portWorkspaces: {},
+    portLabels: {},
     portStatuses: {},
     selectedStatus: null,
     expanded: false,
@@ -63,6 +66,9 @@
     // 输入法组字（拼音/注音等）进行中：此间回车用于上屏候选词，不应触发发送。
     composing: false,
     timers: [],
+    // 自定义端口下拉的文档级监听（点空白处 / Esc 关闭菜单），uninstall 时移除。
+    onDocPointerDown: null,
+    onDocKeydown: null,
   };
 
   const SVG_NS = "http://www.w3.org/2000/svg";
@@ -176,6 +182,45 @@
       #${config.panelId} .cmf-port:focus { background-color: var(--vscode-list-hoverBackground, rgba(228, 228, 228, .07)); }
       #${config.panelId} .cmf-prompts { max-width: 170px; flex: 0 1 auto; }
 
+      /* 自定义端口下拉：触发按钮 + 浮层菜单（替代原生 <select>，以支持「点开→扫描→填充→展示」而不打断点选）。 */
+      #${config.panelId} .cmf-portwrap { position: relative; display: inline-flex; flex: 0 0 auto; }
+      #${config.panelId} .cmf-portbtn {
+        appearance: none; -webkit-appearance: none;
+        height: 22px; padding: 0 20px 0 8px;
+        border-radius: 4px;
+        border: 1px solid transparent;
+        background-color: transparent;
+        background-image: url("${chevron}");
+        background-repeat: no-repeat;
+        background-position: right 5px center;
+        color: var(--vscode-descriptionForeground, rgba(228, 228, 228, .7));
+        font-family: inherit; font-size: 11.5px; cursor: pointer; outline: none;
+        white-space: nowrap;
+        transition: background-color .12s ease;
+      }
+      #${config.panelId} .cmf-portbtn:hover { background-color: var(--vscode-list-hoverBackground, rgba(228, 228, 228, .07)); }
+      #${config.panelId} .cmf-portbtn.open { background-color: var(--vscode-list-hoverBackground, rgba(228, 228, 228, .1)); }
+      #${config.panelId} .cmf-portmenu {
+        position: absolute; top: calc(100% + 4px); right: 0; z-index: 1000;
+        min-width: 100%; max-width: 300px; max-height: 240px; overflow-y: auto;
+        padding: 4px;
+        border: 1px solid var(--vscode-input-border, rgba(228, 228, 228, .18));
+        border-radius: 8px;
+        background: var(--vscode-menu-background, var(--vscode-editorWidget-background, #252526));
+        box-shadow: 0 6px 20px rgba(0, 0, 0, .32);
+      }
+      #${config.panelId} .cmf-opt {
+        display: block; white-space: nowrap;
+        padding: 4px 8px; border-radius: 5px;
+        font-size: 11.5px; line-height: 1.5;
+        color: var(--vscode-foreground, rgba(228, 228, 228, .9));
+        cursor: pointer;
+      }
+      #${config.panelId} .cmf-opt:hover { background: var(--vscode-list-hoverBackground, rgba(228, 228, 228, .1)); }
+      #${config.panelId} .cmf-opt.sel { color: var(--vscode-foreground, #fff); background: var(--vscode-list-activeSelectionBackground, rgba(129, 161, 193, .22)); }
+      #${config.panelId} .cmf-opt-custom { margin-top: 3px; padding-top: 6px; border-top: 1px solid var(--vscode-input-border, rgba(228, 228, 228, .12)); color: var(--vscode-descriptionForeground, rgba(228, 228, 228, .7)); }
+      #${config.panelId} .cmf-menuhint { display: flex; align-items: center; gap: 7px; padding: 6px 8px; font-size: 11.5px; color: var(--vscode-descriptionForeground, rgba(228, 228, 228, .65)); }
+
       #${config.panelId} .cmf-scan {
         flex: 0 0 auto;
         width: 22px; height: 22px;
@@ -346,6 +391,18 @@
     refreshOptionLabels();
   }
 
+  // 确保隐藏 select（数据载体）里存在某端口的 option。自定义浮层用 foundPorts 渲染，
+  // 但隐藏 select 只在全量扫描时重填；按需扫描新发现的端口必须补进来，否则给原生
+  // <select> 设一个不存在的 value 会被静默置空 → 连空端口 offline。
+  function ensurePortOption(value) {
+    if (!els.portSelect) return;
+    const v = String(value);
+    const opts = Array.from(els.portSelect.options);
+    if (opts.some((o) => o.value === v)) return;
+    const custom = opts.find((o) => o.value === config.customValue);
+    els.portSelect.insertBefore(h("option", { value: v }), custom || null);
+  }
+
   function makeScanIcon() {
     const svg = svgEl("svg", { viewBox: "0 0 24 24", fill: "none", "aria-hidden": "true" });
     svg.appendChild(svgEl("circle", { cx: "11", cy: "11", r: "6", stroke: "currentColor", "stroke-width": "2" }));
@@ -433,10 +490,35 @@
   }
 
   function buildPanel() {
-    const portSelect = h("select", { class: "cmf-port", "aria-label": "选择 MCP 前端端口" });
+    // 隐藏的原生 <select> 仅作「数据载体」：继续承载当前选中 value 与 options 列表，
+    // 让 fillPortOptions / connectSelectedPort 等既有读写逻辑（els.portSelect.value）保持不变；
+    // 用户实际看到/交互的是下面的自定义触发按钮 + 浮层菜单（portBtn / portMenu）。
+    const portSelect = h("select", { class: "cmf-port", "aria-hidden": "true", tabindex: "-1" });
+    portSelect.style.display = "none";
     els.portSelect = portSelect;
     fillPortOptions(defaultPorts(), String(config.scanStart));
 
+    const portBtnText = h("span", { class: "cmf-portbtn-text" }, "Port " + config.scanStart);
+    const portBtn = h(
+      "button",
+      {
+        class: "cmf-portbtn",
+        type: "button",
+        "aria-haspopup": "listbox",
+        "aria-expanded": "false",
+        "aria-label": "选择 MCP 前端端口",
+        title: "选择 MCP 前端端口",
+      },
+      portBtnText
+    );
+    const portMenu = h("div", { class: "cmf-portmenu", role: "listbox" });
+    portMenu.style.display = "none";
+    const portWrap = h("div", { class: "cmf-portwrap" }, portBtn, portMenu, portSelect);
+    els.portBtn = portBtn;
+    els.portBtnText = portBtnText;
+    els.portMenu = portMenu;
+    els.portWrap = portWrap;
+
     const customInput = h("input", {
       class: "cmf-custom",
       type: "text",
@@ -476,7 +558,7 @@
       status,
       h("span", { class: "cmf-spacer" }),
       scanBtn,
-      portSelect,
+      portWrap,
       customInput,
       projectName,
       gear
@@ -537,6 +619,7 @@
     // 状态文本已从面板移除；游离占位元素让现有状态写入成为无害空操作。
     els.hint = document.createElement("span");
 
+    updatePortButton();
     wireEvents();
   }
 
@@ -620,13 +703,41 @@
     }
   }
 
-  // 从一轮扫描结果里挑出 project_directory 与当前窗口工作区相等的端口（按项目锁定端口的核心优先级）。
+  // 在 server 进程里读「窗口真实工作区」的命令：WORKSPACE_FOLDER_PATHS / CURSOR_WORKSPACE_LABEL 由 Cursor 注入，
+  // 不受 AI 传入的 project_directory 影响。用单个 print 表达式（不含 ';'）+ chr(10) 分隔（不含 '|'），
+  // 规避服务端 _safe_parse_command 的危险子串过滤。
+  function buildWorkspaceProbeCmd() {
+    const code =
+      "print(__import__('os').environ.get('WORKSPACE_FOLDER_PATHS','')+chr(10)+__import__('os').environ.get('CURSOR_WORKSPACE_LABEL',''))";
+    return config.pyCmd + ' -c "' + code + '"';
+  }
+
+  // WORKSPACE_FOLDER_PATHS 可能含多个路径（多根工作区）；按常见分隔符拆开后看是否包含当前工作区。
+  function matchFolders(folders, ws) {
+    if (!folders || !ws) return false;
+    return String(folders)
+      .split(/[:;,\n]/)
+      .map((s) => normalizePath(s))
+      .filter(Boolean)
+      .includes(ws);
+  }
+
+  // 端口是否属于「当前窗口」：优先用 WORKSPACE_FOLDER_PATHS（完整路径），其次用 CURSOR_WORKSPACE_LABEL（项目名）。
+  function isPortMine(port, ws) {
+    if (!ws) return false;
+    if (matchFolders(state.portWorkspaces[port], ws)) return true;
+    const label = state.portLabels[port];
+    return !!(label && label === basename(ws));
+  }
+
+  // 从一轮扫描结果里挑出「属于当前窗口」的端口：主用 WORKSPACE_FOLDER_PATHS，次用 CURSOR_WORKSPACE_LABEL。
   function pickWorkspacePort(results) {
     const ws = getWorkspacePath();
     if (!ws) return null;
-    const hit = results.find(
-      (item) => item.alive && item.project && normalizePath(item.project) === ws
-    );
+    const wsLabel = basename(ws);
+    let hit = results.find((r) => r.alive && matchFolders(r.workspaceFolders, ws));
+    if (hit) return String(hit.port);
+    hit = results.find((r) => r.alive && r.workspaceLabel && r.workspaceLabel === wsLabel);
     return hit ? String(hit.port) : null;
   }
 
@@ -653,6 +764,7 @@
       const status = isSelected ? state.selectedStatus : state.portStatuses[option.value] || null;
       option.textContent = portText(option.value, status, state.expanded);
     }
+    updatePortButton();
   }
 
   function setOptionLabel(port, status) {
@@ -660,6 +772,91 @@
     refreshOptionLabels();
   }
 
+  // ———————————————————— 自定义端口下拉 ————————————————————
+  // 触发按钮文字 = 当前选中端口（带状态、不带项目名；项目名在右侧 projectName 单独显示）。
+  function updatePortButton() {
+    if (!els.portBtnText) return;
+    const cur = els.portSelect.value;
+    els.portBtnText.textContent =
+      cur === config.customValue ? "自定义端口…" : portText(cur, state.selectedStatus, false);
+  }
+
+  // 渲染浮层菜单：扫描中显示「扫描中…」，否则按最新 foundPorts 列出可用端口（带状态·项目名）+「自定义端口…」。
+  // 关键：菜单仅在「打开瞬间」与「扫描完成时」各渲染一次，渲染后保持稳定 —— 自定义 div 不会像原生 <select>
+  // 那样在用户点击瞬间因重建 DOM 而打断点选，因此点选即时生效、无打断、无闪。
+  function renderPortMenu(opts) {
+    if (!els.portMenu) return;
+    const loading = !!(opts && opts.loading) || state.scanning;
+    clear(els.portMenu);
+    if (loading) {
+      els.portMenu.appendChild(
+        h(
+          "div",
+          { class: "cmf-menuhint" },
+          h("span", { class: "cmf-spinner", "aria-hidden": "true" }),
+          h("span", null, "扫描中…")
+        )
+      );
+      return;
+    }
+    const cur = els.portSelect.value;
+    const ports = state.foundPorts.length ? state.foundPorts.map(String) : defaultPorts();
+    for (const p of ports) {
+      const status = p === cur ? state.selectedStatus : state.portStatuses[p] || null;
+      const item = h("div", { class: "cmf-opt" + (p === cur ? " sel" : ""), role: "option" }, portText(p, status, true));
+      item.addEventListener("click", () => selectPort(p));
+      els.portMenu.appendChild(item);
+    }
+    const customItem = h("div", { class: "cmf-opt cmf-opt-custom", role: "option" }, "自定义端口…");
+    customItem.addEventListener("click", selectCustom);
+    els.portMenu.appendChild(customItem);
+  }
+
+  // 打开菜单：先显示「扫描中…」，按需扫一次，扫完用最新结果填充并展示（满足「先扫描→填充→再展示」）。
+  async function openPortMenu() {
+    if (state.expanded) return;
+    state.expanded = true;
+    els.portMenu.style.display = "";
+    els.portBtn.classList.add("open");
+    els.portBtn.setAttribute("aria-expanded", "true");
+    renderPortMenu({ loading: true });
+    await scanPorts({ refreshOnly: true });
+    if (state.expanded) renderPortMenu();
+  }
+
+  function closePortMenu() {
+    state.expanded = false;
+    if (els.portMenu) els.portMenu.style.display = "none";
+    if (els.portBtn) {
+      els.portBtn.classList.remove("open");
+      els.portBtn.setAttribute("aria-expanded", "false");
+    }
+  }
+
+  function togglePortMenu() {
+    if (state.expanded) closePortMenu();
+    else openPortMenu();
+  }
+
+  // 点选某端口：写回隐藏 select 的 value，关菜单，刷新按钮，再走既有连接逻辑（onPortChange）。
+  function selectPort(port) {
+    const v = String(port);
+    // 关键：浮层从 foundPorts 渲染，可能领先于隐藏 select 的 options；先补齐再赋值，
+    // 否则原生 <select> 对不存在的 value 会静默置空，导致连空端口 → offline。
+    ensurePortOption(v);
+    els.portSelect.value = v;
+    closePortMenu();
+    updatePortButton();
+    onPortChange();
+  }
+
+  function selectCustom() {
+    els.portSelect.value = config.customValue;
+    closePortMenu();
+    updatePortButton();
+    onPortChange();
+  }
+
   function deriveStatusLabel(info) {
     if (!info) return "";
     if (info.feedback_completed === true || info.status === "feedback_submitted" || info.status === "completed") {
@@ -722,7 +919,6 @@
 
     const project = basename(info && info.project_directory);
     if (project) state.portProjects[state.socketPort] = project;
-    if (info && info.project_directory) state.portPaths[state.socketPort] = info.project_directory;
     state.portStatuses[state.socketPort] = deriveStatusLabel(info);
 
     const status = info && info.status;
@@ -1251,6 +1447,10 @@
       let opened = false;
       let project = "";
       let statusLabel = "";
+      let workspaceFolders = "";
+      let workspaceLabel = "";
+      let cmdBuf = null;
+      let needCmd = false;
       let ws = null;
       const finish = (alive) => {
         if (done) return;
@@ -1264,13 +1464,24 @@
             /* noop */
           }
         }
-        resolve({ alive, project, statusLabel });
+        resolve({ alive, project, statusLabel, workspaceFolders, workspaceLabel });
       };
       const timer = window.setTimeout(() => finish(opened), timeoutMs);
       try {
         ws = new WebSocket("ws://127.0.0.1:" + port + "/ws?lang=" + config.lang);
         ws.onopen = () => {
           opened = true;
+          // 连上后用 run_command 读「窗口真实工作区」（不受 AI 传入的 project_directory 影响）。
+          // pyCmd 占位符未被注入替换时跳过，退回靠 session_info 仅识别项目名。
+          if (config.pyCmd.indexOf("__MCP_") !== 0) {
+            cmdBuf = "";
+            needCmd = true;
+            try {
+              ws.send(JSON.stringify({ type: "run_command", command: buildWorkspaceProbeCmd() }));
+            } catch (error) {
+              needCmd = false;
+            }
+          }
         };
         ws.onmessage = (event) => {
           try {
@@ -1279,6 +1490,21 @@
             if (info && info.project_directory) {
               project = info.project_directory;
               statusLabel = deriveStatusLabel(info);
+              // 没发探测命令时（占位符未替换），拿到会话即可结束。
+              if (!needCmd) finish(true);
+            }
+            if (data.type === "command_output") {
+              if (cmdBuf !== null) cmdBuf += data.output || "";
+            }
+            if (data.type === "command_complete" || data.type === "command_error") {
+              const raw = (cmdBuf || "").replace(/\r/g, "");
+              const nl = raw.indexOf("\n");
+              if (nl >= 0) {
+                workspaceFolders = raw.slice(0, nl).trim();
+                workspaceLabel = raw.slice(nl + 1).trim();
+              } else {
+                workspaceFolders = raw.trim();
+              }
               finish(true);
             }
           } catch (error) {
@@ -1293,7 +1519,8 @@
     });
   }
 
-  async function scanPorts() {
+  async function scanPorts(opts) {
+    const refreshOnly = !!(opts && opts.refreshOnly);
     if (state.scanning) return;
     state.scanning = true;
     if (els.scanBtn) {
@@ -1306,44 +1533,53 @@
     els.hint.textContent = "正在扫描端口 " + from + "–" + to + "…";
 
     const ports = defaultPorts();
+    // 跳过探测「当前已连接的健康端口」：probe 会另建连接抢占服务端的「最后连接」，从而把面板自己挤下线。
+    // 对已连端口直接用已知信息标记为活跃，保证下拉扫描/重扫不会断开当前会话。
+    const curPort =
+      state.socket && state.socket.readyState === WebSocket.OPEN ? String(state.socketPort) : null;
     const results = await Promise.all(
-      ports.map((port) =>
-        probePort(port, config.probeTimeoutMs).then((res) => ({
+      ports.map((port) => {
+        if (curPort && String(port) === curPort) {
+          return Promise.resolve({
+            port,
+            alive: true,
+            project: state.currentSession ? state.currentSession.project_directory : "",
+            statusLabel: state.portStatuses[port] || state.selectedStatus || "",
+            workspaceFolders: state.portWorkspaces[port] || "",
+            workspaceLabel: state.portLabels[port] || "",
+          });
+        }
+        return probePort(port, config.probeTimeoutMs).then((res) => ({
           port,
           alive: res.alive,
           project: res.project,
           statusLabel: res.statusLabel,
-        }))
-      )
+          workspaceFolders: res.workspaceFolders,
+          workspaceLabel: res.workspaceLabel,
+        }));
+      })
     );
     const found = [];
     for (const item of results) {
       if (item.alive) {
         found.push(item.port);
-        if (item.project) {
-          state.portProjects[item.port] = basename(item.project);
-          state.portPaths[item.port] = item.project;
-        } else {
-          delete state.portProjects[item.port];
-          delete state.portPaths[item.port];
-        }
+        if (item.project) state.portProjects[item.port] = basename(item.project);
+        else delete state.portProjects[item.port];
+        if (item.workspaceFolders) state.portWorkspaces[item.port] = item.workspaceFolders;
+        else delete state.portWorkspaces[item.port];
+        if (item.workspaceLabel) state.portLabels[item.port] = item.workspaceLabel;
+        else delete state.portLabels[item.port];
         if (item.statusLabel) state.portStatuses[item.port] = item.statusLabel;
         else delete state.portStatuses[item.port];
       } else {
         delete state.portProjects[item.port];
-        delete state.portPaths[item.port];
+        delete state.portWorkspaces[item.port];
+        delete state.portLabels[item.port];
         delete state.portStatuses[item.port];
       }
     }
     state.foundPorts = found;
 
-    const wasCustom = els.portSelect.value === config.customValue;
-    // 优先选中 project_directory 与当前窗口工作区相等的端口；取不到工作区或无匹配端口时维持原有选择
-    //（combo 降级），连接层再由 connectSelectedPort 守卫兜底，避免连到别的项目。
-    const preferred = pickWorkspacePort(results);
-    fillPortOptions(found, preferred || (wasCustom ? null : els.portSelect.value));
-    if (wasCustom && !preferred) els.portSelect.value = config.customValue;
-
     if (els.scanBtn) {
       els.scanBtn.classList.remove("scanning");
       els.scanBtn.disabled = false;
@@ -1354,6 +1590,21 @@
       ? "扫描完成：发现活跃端口 " + found.join("、") + "。"
       : "扫描完成：" + from + "–" + to + " 未发现活跃 MCP 端口。";
 
+    if (refreshOnly) {
+      // 自定义下拉：先把本次扫到的端口补进隐藏 select（数据载体），保证点选时 value 设得上；
+      // 菜单仍展开则用最新结果渲染浮层（渲染后保持稳定，点选即时生效、无打断、无闪）。
+      found.forEach((p) => ensurePortOption(String(p)));
+      if (state.expanded) renderPortMenu();
+      return;
+    }
+
+    const wasCustom = els.portSelect.value === config.customValue;
+    // 优先选中 project_directory 与当前窗口工作区相等的端口；取不到工作区或无匹配端口时维持原有选择
+    //（combo 降级），连接层再由 connectSelectedPort 守卫兜底，避免连到别的项目。
+    const preferred = pickWorkspacePort(results);
+    fillPortOptions(found, preferred || (wasCustom ? null : els.portSelect.value));
+    if (wasCustom && !preferred) els.portSelect.value = config.customValue;
+
     // 探测会短暂抢占连接，扫描结束后立即重连选中端口抢回「最后连接」。
     if (els.portSelect.value !== config.customValue) connectSelectedPort(false);
   }
@@ -1400,10 +1651,11 @@
     const port = els.portSelect.value;
     if (port === config.customValue) return;
 
-    // 守卫：能确定当前窗口项目、且该端口已知属于别的项目时不连接，
-    // 避免误连到其它项目或常驻 feedback 实例（等本项目端口出现后由 refreshTick 重扫接管）。
+    // 守卫：能确定当前窗口、且该端口已知属于别的窗口时不连接，
+    // 避免误连到其它窗口的实例或常驻 feedback 实例（等本窗口端口被识别后由 refreshTick 重扫接管）。
     const ws = getWorkspacePath();
-    if (ws && state.portPaths[port] && normalizePath(state.portPaths[port]) !== ws) {
+    const known = !!(state.portWorkspaces[port] || state.portLabels[port]);
+    if (ws && known && !isPortMine(port, ws)) {
       if (state.socket) {
         state.socket.onclose = null;
         state.socket.onerror = null;
@@ -1411,8 +1663,8 @@
         state.socket.close();
         state.socket = null;
       }
-      setOptionLabel(port, "other project");
-      setVisualState("offline", port + " 属于其它项目，等待本项目的 MCP 会话…");
+      setOptionLabel(port, "other window");
+      setVisualState("offline", port + " 属于其它窗口，等待本窗口的 MCP 会话…");
       return;
     }
 
@@ -1470,6 +1722,10 @@
 
       nextSocket.onerror = () => {
         if (seq !== state.connectSeq) return;
+        // 连接失败：端口很可能已关闭或换了端口（实例重启），清除其归属缓存，
+        // 让 refreshTick 退避重扫去发现新的本窗口端口，而不是一直重连这个死端口。
+        delete state.portWorkspaces[port];
+        delete state.portLabels[port];
         setOptionLabel(port, "offline");
         setVisualState("offline", port + " 连接失败，可能端口未启动或没有 MCP WebUI。");
       };
@@ -1501,17 +1757,22 @@
     if (els.customInput === document.activeElement) return;
     if (els.prompt === document.activeElement && els.prompt.value.trim()) return;
 
-    // 还没连到「本项目」端口时，定期重扫去发现它（scanPorts 会优先选中并连上）；
-    // 已在本项目端口（或取不到工作区时）则维持原有重连，持续保持「最后连接」。
-    const ws = getWorkspacePath();
-    if (ws) {
-      const cur = els.portSelect.value;
-      const onWorkspacePort = !!state.portPaths[cur] && normalizePath(state.portPaths[cur]) === ws;
-      if (!onWorkspacePort) {
-        scanPorts();
-        return;
+    // 不做后台扫描（扫描仅发生在启动时与下拉展开期间）。这里只保持连接：
+    // 已连且健康 → 发 heartbeat 维持「最后连接」；否则重连选中端口（守卫会拦掉别窗口端口；
+    // 端口已关时 onerror 会清归属并置 offline，等待用户展开下拉重新选择本窗口端口）。
+    if (
+      state.socket &&
+      state.socket.readyState === WebSocket.OPEN &&
+      state.socketPort === els.portSelect.value
+    ) {
+      try {
+        state.socket.send(JSON.stringify({ type: "heartbeat", timestamp: Date.now() }));
+      } catch (error) {
+        /* 发送失败则下个周期走重连 */
       }
+      return;
     }
+
     connectSelectedPort(true);
   }
 
@@ -1574,16 +1835,22 @@
   }
 
   function wireEvents() {
-    els.portSelect.addEventListener("change", onPortChange);
-    // 展开下拉前让所有项带上「状态 · 项目名」；收起后选中项去掉项目名（右侧已显示）。
-    els.portSelect.addEventListener("mousedown", () => {
-      state.expanded = true;
-      refreshOptionLabels();
-    });
-    els.portSelect.addEventListener("blur", () => {
-      state.expanded = false;
-      refreshOptionLabels();
+    // 自定义端口下拉：点按钮开/关；打开即「扫描中→填充→展示」；点选即时生效（见 selectPort / selectCustom）。
+    els.portBtn.addEventListener("click", (event) => {
+      event.stopPropagation();
+      togglePortMenu();
     });
+    // 点击按钮与浮层之外的区域、或按 Esc 关闭菜单。
+    state.onDocPointerDown = (event) => {
+      if (!state.expanded) return;
+      if (els.portWrap && els.portWrap.contains(event.target)) return;
+      closePortMenu();
+    };
+    document.addEventListener("mousedown", state.onDocPointerDown, true);
+    state.onDocKeydown = (event) => {
+      if (state.expanded && event.key === "Escape") closePortMenu();
+    };
+    document.addEventListener("keydown", state.onDocKeydown, true);
     els.scanBtn.addEventListener("click", () => {
       scanPorts();
     });
@@ -1655,6 +1922,8 @@
 
   function uninstall() {
     cancelAutoSubmit();
+    if (state.onDocPointerDown) document.removeEventListener("mousedown", state.onDocPointerDown, true);
+    if (state.onDocKeydown) document.removeEventListener("keydown", state.onDocKeydown, true);
     for (const timer of state.timers) {
       window.clearInterval(timer);
     }