@srgay/cursor-extension 1.0.0

package/src/mcp-followup/install-cursor-mcp-followup.mjs

@@ -0,0 +1,93 @@
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { dirname, join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const port = Number(process.env.CURSOR_DEBUG_PORT || "9222");
+const sourcePath = resolve(__dirname, "cursor-mcp-followup.js");
+
+if (typeof WebSocket !== "function") {
+  throw new Error("This script needs Node.js with global WebSocket support. Use Node.js 20 or newer.");
+}
+
+async function json(url) {
+  const response = await fetch(url);
+  if (!response.ok) {
+    throw new Error(`${url} returned ${response.status}`);
+  }
+  return response.json();
+}
+
+function pickWorkbenchTarget(targets) {
+  return targets.find((target) => {
+    return target.type === "page" && typeof target.url === "string" && target.url.includes("workbench.html");
+  });
+}
+
+function connect(wsUrl) {
+  const ws = new WebSocket(wsUrl);
+  let id = 0;
+
+  const opened = new Promise((resolveOpen, rejectOpen) => {
+    ws.addEventListener("open", resolveOpen, { once: true });
+    ws.addEventListener("error", rejectOpen, { once: true });
+  });
+
+  function call(method, params = {}) {
+    return new Promise((resolveCall, rejectCall) => {
+      const msgId = ++id;
+      const timer = setTimeout(() => rejectCall(new Error(`CDP call timed out: ${method}`)), 5000);
+
+      function onMessage(event) {
+        const data = JSON.parse(event.data);
+        if (data.id !== msgId) return;
+
+        clearTimeout(timer);
+        ws.removeEventListener("message", onMessage);
+        if (data.error) {
+          rejectCall(new Error(JSON.stringify(data.error)));
+        } else {
+          resolveCall(data.result);
+        }
+      }
+
+      ws.addEventListener("message", onMessage);
+      ws.send(JSON.stringify({ id: msgId, method, params }));
+    });
+  }
+
+  return { ws, opened, call };
+}
+
+const targets = await json(`http://127.0.0.1:${port}/json/list`);
+const target = pickWorkbenchTarget(targets);
+
+if (!target) {
+  throw new Error(
+    `Cursor workbench target was not found on port ${port}. Start Cursor with: open -na /Applications/Cursor.app --args --remote-debugging-port=${port}`
+  );
+}
+
+// 面板运行在浏览器环境（无 fs），改用 WS run_command 读取常用提示词；
+// 这里在 Node 侧把占位符替换成 ui_settings.json 的真实绝对路径（shell=False 下 ~ 不展开）。
+const settingsPath = join(homedir(), ".config", "mcp-feedback-enhanced", "ui_settings.json");
+const source = (await readFile(sourcePath, "utf8")).replace("__MCP_SETTINGS_PATH__", settingsPath);
+const client = connect(target.webSocketDebuggerUrl);
+
+await client.opened;
+await client.call("Runtime.enable");
+
+const result = await client.call("Runtime.evaluate", {
+  expression: source,
+  returnByValue: true,
+  awaitPromise: true,
+});
+
+client.ws.close();
+
+if (result.exceptionDetails) {
+  throw new Error(result.exceptionDetails.text || "Injection failed");
+}
+
+console.log(JSON.stringify(result.result?.value || result.result || {}, null, 2));

package/src/shared/cursor-workbench-paths.mjs

@@ -0,0 +1,46 @@
+import { existsSync } from "node:fs";
+import { homedir } from "node:os";
+import { resolve } from "node:path";
+
+const relativeWorkbenchPath = "resources/app/out/vs/code/electron-sandbox/workbench";
+
+function windowsLocalAppData() {
+  return process.env.LOCALAPPDATA || resolve(homedir(), "AppData/Local");
+}
+
+function defaultWorkbenchDir() {
+  if (process.platform === "darwin") {
+    return "/Applications/Cursor.app/Contents/Resources/app/out/vs/code/electron-sandbox/workbench";
+  }
+
+  if (process.platform === "win32") {
+    const candidates = [
+      resolve(windowsLocalAppData(), "Programs/Cursor", relativeWorkbenchPath),
+      resolve("C:/Program Files/cursor", relativeWorkbenchPath),
+      resolve("C:/Program Files/Cursor", relativeWorkbenchPath),
+    ];
+
+    return candidates.find((path) => existsSync(resolve(path, "workbench.html"))) || candidates[0];
+  }
+
+  throw new Error(
+    "Unsupported platform. Set CURSOR_WORKBENCH_DIR to Cursor's workbench directory."
+  );
+}
+
+export function resolveWorkbenchPaths() {
+  const workbenchDir = process.env.CURSOR_WORKBENCH_DIR || defaultWorkbenchDir();
+  const appRoot = resolve(workbenchDir, "../../../../..");
+
+  return {
+    appRoot,
+    workbenchDir,
+    workbenchHtml: resolve(workbenchDir, "workbench.html"),
+    guardTarget: resolve(workbenchDir, "cursor-max-mode-guard.js"),
+    followupTarget: resolve(workbenchDir, "cursor-mcp-followup.js"),
+    imeFixTarget: resolve(workbenchDir, "cursor-ime-enter-fix.js"),
+    backupHtml: resolve(workbenchDir, "workbench.html.cursor-max-guard.bak"),
+    productJson: resolve(appRoot, "product.json"),
+    backupProductJson: resolve(appRoot, "product.json.cursor-max-guard.bak"),
+  };
+}

package/src/shared/patch-cursor-workbench.mjs

@@ -0,0 +1,177 @@
+import { createHash } from "node:crypto";
+import { copyFile, readFile, stat, writeFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { dirname, join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+import { resolveWorkbenchPaths } from "./cursor-workbench-paths.mjs";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+
+const guardSource = resolve(__dirname, "../max-mode-guard/cursor-max-mode-guard.js");
+const followupSource = resolve(__dirname, "../mcp-followup/cursor-mcp-followup.js");
+const imeFixSource = resolve(__dirname, "../ime-enter-fix/cursor-ime-enter-fix.js");
+// 面板在浏览器环境无 fs，靠 WS run_command 读取 ui_settings.json；patch 时把占位符替换成真实绝对路径。
+const settingsPath = join(homedir(), ".config", "mcp-feedback-enhanced", "ui_settings.json");
+const {
+  workbenchDir,
+  workbenchHtml,
+  guardTarget,
+  followupTarget,
+  imeFixTarget,
+  backupHtml,
+  productJson,
+  backupProductJson,
+} = resolveWorkbenchPaths();
+
+const markerStart = "<!-- cursor-max-mode-guard:start -->";
+const markerEnd = "<!-- cursor-max-mode-guard:end -->";
+const injection = `${markerStart}
+\t<script src="./cursor-max-mode-guard.js" defer></script>
+\t${markerEnd}`;
+
+const followupMarkerStart = "<!-- cursor-mcp-followup:start -->";
+const followupMarkerEnd = "<!-- cursor-mcp-followup:end -->";
+const followupInjection = `${followupMarkerStart}
+\t<script src="./cursor-mcp-followup.js" defer></script>
+\t${followupMarkerEnd}`;
+
+const imeMarkerStart = "<!-- cursor-ime-enter-fix:start -->";
+const imeMarkerEnd = "<!-- cursor-ime-enter-fix:end -->";
+const imeInjection = `${imeMarkerStart}
+\t<script src="./cursor-ime-enter-fix.js" defer></script>
+\t${imeMarkerEnd}`;
+
+function explainPermissionError(error) {
+  if (error && error.code === "EPERM" && process.platform === "win32") {
+    return [
+      "Permission denied while patching Cursor.",
+      "",
+      `Target: ${workbenchDir}`,
+      "",
+      "Cursor is installed under a protected Windows directory, such as C:\\Program Files.",
+      "Run PowerShell as Administrator and execute npm run patch again.",
+      "",
+      "Alternative:",
+      '$env:CURSOR_WORKBENCH_DIR="C:\\Path\\To\\Cursor\\resources\\app\\out\\vs\\code\\electron-sandbox\\workbench"',
+      "npm run patch",
+    ].join("\n");
+  }
+
+  return null;
+}
+
+async function exists(path) {
+  try {
+    await stat(path);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function checksumBase64NoPadding(content) {
+  return createHash("sha256").update(content).digest("base64").replace(/=+$/, "");
+}
+
+async function updateProductChecksum() {
+  if (!(await exists(productJson))) {
+    return { updated: false, reason: "product.json not found" };
+  }
+
+  const productText = await readFile(productJson, "utf8");
+  if (!(await exists(backupProductJson))) {
+    await writeFile(backupProductJson, productText);
+  }
+
+  const product = JSON.parse(productText);
+  product.checksums ||= {};
+
+  const checksumKey = "vs/code/electron-sandbox/workbench/workbench.html";
+  const htmlBuffer = await readFile(workbenchHtml);
+  const checksum = checksumBase64NoPadding(htmlBuffer);
+  const previous = product.checksums[checksumKey];
+
+  product.checksums[checksumKey] = checksum;
+  await writeFile(productJson, `${JSON.stringify(product, null, 2)}\n`);
+
+  return {
+    updated: previous !== checksum,
+    key: checksumKey,
+    previous,
+    checksum,
+    productJson,
+    backupProductJson,
+  };
+}
+
+async function main() {
+  const html = await readFile(workbenchHtml, "utf8");
+
+  if (!(await exists(backupHtml))) {
+    await writeFile(backupHtml, html);
+  }
+
+  await copyFile(guardSource, guardTarget);
+
+  // MCP 面板：不是单纯拷贝，需把 __MCP_SETTINGS_PATH__ 占位符替换成真实路径后再写入。
+  const followupCode = (await readFile(followupSource, "utf8")).replace("__MCP_SETTINGS_PATH__", settingsPath);
+  await writeFile(followupTarget, followupCode);
+
+  // 输入法回车修复：无占位符，直接拷贝。
+  await copyFile(imeFixSource, imeFixTarget);
+
+  const needle = '<script src="./workbench.js" type="module"></script>';
+
+  let nextHtml = html;
+  if (!nextHtml.includes(markerStart)) {
+    if (!nextHtml.includes(needle)) {
+      throw new Error(`Cannot find workbench startup script in ${workbenchHtml}`);
+    }
+    nextHtml = nextHtml.replace(needle, `${needle}\n\t${injection}`);
+  }
+  if (!nextHtml.includes(followupMarkerStart)) {
+    if (!nextHtml.includes(needle)) {
+      throw new Error(`Cannot find workbench startup script in ${workbenchHtml}`);
+    }
+    nextHtml = nextHtml.replace(needle, `${needle}\n\t${followupInjection}`);
+  }
+  if (!nextHtml.includes(imeMarkerStart)) {
+    if (!nextHtml.includes(needle)) {
+      throw new Error(`Cannot find workbench startup script in ${workbenchHtml}`);
+    }
+    nextHtml = nextHtml.replace(needle, `${needle}\n\t${imeInjection}`);
+  }
+
+  if (nextHtml !== html) {
+    await writeFile(workbenchHtml, nextHtml);
+  }
+
+  const productChecksum = await updateProductChecksum();
+
+  console.log(
+    JSON.stringify(
+      {
+        patched: true,
+        platform: process.platform,
+        workbenchDir,
+        workbenchHtml,
+        backupHtml,
+        guardTarget,
+        followupTarget,
+        imeFixTarget,
+        settingsPath,
+        productChecksum,
+        alreadyPatched: html.includes(markerStart),
+        followupAlreadyPatched: html.includes(followupMarkerStart),
+        imeFixAlreadyPatched: html.includes(imeMarkerStart),
+      },
+      null,
+      2
+    )
+  );
+}
+
+main().catch((error) => {
+  console.error(explainPermissionError(error) || (error && error.stack) || error);
+  process.exitCode = 1;
+});

package/src/shared/unpatch-cursor-workbench.mjs

@@ -0,0 +1,96 @@
+import { copyFile, readFile, rm, stat, writeFile } from "node:fs/promises";
+import { resolveWorkbenchPaths } from "./cursor-workbench-paths.mjs";
+
+const {
+  workbenchDir,
+  workbenchHtml,
+  guardTarget,
+  followupTarget,
+  imeFixTarget,
+  backupHtml,
+  productJson,
+  backupProductJson,
+} = resolveWorkbenchPaths();
+
+const markerStart = "<!-- cursor-max-mode-guard:start -->";
+const markerEnd = "<!-- cursor-max-mode-guard:end -->";
+const followupMarkerStart = "<!-- cursor-mcp-followup:start -->";
+const followupMarkerEnd = "<!-- cursor-mcp-followup:end -->";
+const imeMarkerStart = "<!-- cursor-ime-enter-fix:start -->";
+const imeMarkerEnd = "<!-- cursor-ime-enter-fix:end -->";
+
+function explainPermissionError(error) {
+  if (error && error.code === "EPERM" && process.platform === "win32") {
+    return [
+      "Permission denied while restoring Cursor.",
+      "",
+      `Target: ${workbenchDir}`,
+      "",
+      "Cursor is installed under a protected Windows directory, such as C:\\Program Files.",
+      "Run PowerShell as Administrator and execute npm run unpatch again.",
+      "",
+      "Alternative:",
+      '$env:CURSOR_WORKBENCH_DIR="C:\\Path\\To\\Cursor\\resources\\app\\out\\vs\\code\\electron-sandbox\\workbench"',
+      "npm run unpatch",
+    ].join("\n");
+  }
+
+  return null;
+}
+
+async function exists(path) {
+  try {
+    await stat(path);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function main() {
+  if (await exists(backupHtml)) {
+    await copyFile(backupHtml, workbenchHtml);
+  } else {
+    const html = await readFile(workbenchHtml, "utf8");
+    const guardPattern = new RegExp(`\\n?\\t?${markerStart}[\\s\\S]*?${markerEnd}`, "m");
+    const followupPattern = new RegExp(`\\n?\\t?${followupMarkerStart}[\\s\\S]*?${followupMarkerEnd}`, "m");
+    const imePattern = new RegExp(`\\n?\\t?${imeMarkerStart}[\\s\\S]*?${imeMarkerEnd}`, "m");
+    await writeFile(
+      workbenchHtml,
+      html.replace(guardPattern, "").replace(followupPattern, "").replace(imePattern, "")
+    );
+  }
+
+  await rm(guardTarget, { force: true });
+  await rm(followupTarget, { force: true });
+  await rm(imeFixTarget, { force: true });
+
+  const backupProductJsonFound = await exists(backupProductJson);
+  if (backupProductJsonFound) {
+    await copyFile(backupProductJson, productJson);
+  }
+
+  console.log(
+    JSON.stringify(
+      {
+        unpatched: true,
+        platform: process.platform,
+        workbenchDir,
+        workbenchHtml,
+        backupHtmlFound: await exists(backupHtml),
+        guardRemoved: guardTarget,
+        followupRemoved: followupTarget,
+        imeFixRemoved: imeFixTarget,
+        productJson,
+        backupProductJsonFound,
+      },
+      null,
+      2
+    )
+  );
+}
+
+main().catch((error) => {
+  console.error(explainPermissionError(error) || (error && error.stack) || error);
+  process.exitCode = 1;
+});