@hangox/mg-cli 1.5.0 → 1.5.1

package/dist/cli.js

@@ -4188,6 +4188,7 @@ import { fileURLToPath as fileURLToPath3 } from "url";
 var HOST_NAME = "com.hangox.mgplugin";
 var EXTENSION_ID = "ddhihanlpcdneicohnglnaliefnkaeja";
 var DESCRIPTION = "MasterGo Plugin Native Messaging Host";
+var KNOWN_BROWSERS = ["chrome", "chromium", "edge"];
 function getBrowserPaths(browser) {
   const home = os.homedir();
   const platform2 = os.platform();
@@ -4239,11 +4240,88 @@ function getNativeHostPath() {
   const distDir = path.dirname(currentFile);
   return path.join(distDir, "native-host.js");
 }
-function createManifest() {
+var INSTALL_DIR_NAME = "mg-cli";
+function getInstallDir() {
+  const home = os.homedir();
+  if (os.platform() === "win32") {
+    const appData = process.env.APPDATA || path.join(home, "AppData/Roaming");
+    return path.join(appData, INSTALL_DIR_NAME);
+  }
+  return path.join(home, ".config", INSTALL_DIR_NAME);
+}
+var WRAPPER_FILENAME_UNIX = "native-host-wrapper.sh";
+var WRAPPER_FILENAME_WIN = "native-host-wrapper.cmd";
+function getWrapperPath(installDir) {
+  const filename = os.platform() === "win32" ? WRAPPER_FILENAME_WIN : WRAPPER_FILENAME_UNIX;
+  return path.join(installDir, filename);
+}
+function ensureNativeHostWrapper(sourceNativeHostPath) {
+  const installDir = getInstallDir();
+  fs.mkdirSync(installDir, { recursive: true });
+  const installedNativeHostPath = path.join(installDir, "native-host.js");
+  fs.copyFileSync(sourceNativeHostPath, installedNativeHostPath);
+  const wrapperPath = getWrapperPath(installDir);
+  if (os.platform() === "win32") {
+    const wrapperContent2 = `@echo off\r
+"${process.execPath}" "${installedNativeHostPath}" %*\r
+`;
+    fs.writeFileSync(wrapperPath, wrapperContent2);
+    return wrapperPath;
+  }
+  const wrapperContent = `#!/bin/sh
+exec "${process.execPath}" "${installedNativeHostPath}" "$@"
+`;
+  fs.writeFileSync(wrapperPath, wrapperContent);
+  try {
+    fs.chmodSync(wrapperPath, "755");
+  } catch {
+    console.warn("  \u26A0\uFE0F \u65E0\u6CD5\u8BBE\u7F6E wrapper \u6267\u884C\u6743\u9650\uFF0C\u53EF\u80FD\u9700\u8981\u624B\u52A8\u8BBE\u7F6E");
+  }
+  return wrapperPath;
+}
+function readWrapperNodePath(wrapperPath) {
+  try {
+    const content = fs.readFileSync(wrapperPath, "utf-8");
+    const match = content.match(/"([^"]+)"/);
+    return match?.[1];
+  } catch {
+    return void 0;
+  }
+}
+function cleanupOrphanWrapperIfUnused() {
+  const installDir = getInstallDir();
+  const wrapperPath = getWrapperPath(installDir);
+  if (!fs.existsSync(wrapperPath)) return;
+  const stillReferenced = KNOWN_BROWSERS.some((browser) => {
+    const config = getBrowserPaths(browser);
+    const manifestPath = path.join(config.manifestDir, `${HOST_NAME}.json`);
+    if (!fs.existsSync(manifestPath)) return false;
+    try {
+      const manifest = JSON.parse(fs.readFileSync(manifestPath, "utf-8"));
+      return manifest.path === wrapperPath;
+    } catch {
+      return true;
+    }
+  });
+  if (stillReferenced) return;
+  const installedNativeHostPath = path.join(installDir, "native-host.js");
+  try {
+    fs.unlinkSync(wrapperPath);
+  } catch {
+  }
+  try {
+    if (fs.existsSync(installedNativeHostPath)) {
+      fs.unlinkSync(installedNativeHostPath);
+    }
+  } catch {
+  }
+  console.log(`\u5DF2\u6E05\u7406 wrapper \u5B89\u88C5\u76EE\u5F55: ${installDir}`);
+}
+function createManifest(hostPath) {
   return {
     name: HOST_NAME,
     description: DESCRIPTION,
-    path: getNativeHostPath(),
+    path: hostPath,
     type: "stdio",
     allowed_origins: [`chrome-extension://${EXTENSION_ID}/`]
   };
@@ -4251,7 +4329,9 @@ function createManifest() {
 function register(browser = "chrome") {
   const config = getBrowserPaths(browser);
   const manifestPath = path.join(config.manifestDir, `${HOST_NAME}.json`);
-  const manifest = createManifest();
+  const nativeHostPath = getNativeHostPath();
+  const wrapperPath = ensureNativeHostWrapper(nativeHostPath);
+  const manifest = createManifest(wrapperPath);
   console.log("\u6B63\u5728\u6CE8\u518C Native Messaging Host...");
   console.log(`  Host Name: ${HOST_NAME}`);
   console.log(`  Extension ID: ${EXTENSION_ID}`);
@@ -4293,6 +4373,7 @@ function unregister(browser = "chrome") {
     } catch {
     }
   }
+  cleanupOrphanWrapperIfUnused();
 }
 function checkRegistration(browser = "chrome") {
   const config = getBrowserPaths(browser);
@@ -4300,50 +4381,156 @@ function checkRegistration(browser = "chrome") {
   if (!fs.existsSync(manifestPath)) {
     return { registered: false };
   }
+  let manifest;
   try {
-    const manifest = JSON.parse(fs.readFileSync(manifestPath, "utf-8"));
-    return {
-      registered: true,
-      manifestPath,
-      manifest
-    };
+    manifest = JSON.parse(fs.readFileSync(manifestPath, "utf-8"));
   } catch {
     return { registered: false, manifestPath };
   }
+  const wrapperPath = manifest.path;
+  const wrapperExists = wrapperPath ? fs.existsSync(wrapperPath) : false;
+  if (!wrapperExists) {
+    return { registered: false, manifestPath, manifest, wrapperPath, wrapperExists: false };
+  }
+  return {
+    registered: true,
+    manifestPath,
+    manifest,
+    wrapperPath,
+    wrapperExists: true,
+    wrapperNodePath: wrapperPath ? readWrapperNodePath(wrapperPath) : void 0
+  };
+}
+function registerAll(browsers) {
+  return browsers.map((browser) => {
+    try {
+      register(browser);
+      return { browser, success: true };
+    } catch (error) {
+      return { browser, success: false, error: error instanceof Error ? error.message : String(error) };
+    }
+  });
+}
+function unregisterAll(browsers) {
+  return browsers.map((browser) => {
+    try {
+      unregister(browser);
+      return { browser, success: true };
+    } catch (error) {
+      return { browser, success: false, error: error instanceof Error ? error.message : String(error) };
+    }
+  });
+}
+function checkAll(browsers) {
+  return browsers.map((browser) => ({ browser, ...checkRegistration(browser) }));
 }
 
 // src/cli/commands/register.ts
-function createRegisterCommand() {
-  const cmd = new Command22("register").description("\u6CE8\u518C Native Messaging Host\uFF08\u8BA9 Chrome \u63D2\u4EF6\u80FD\u81EA\u52A8\u542F\u52A8 Server\uFF09").option("--unregister", "\u5378\u8F7D Native Host").option("--check", "\u68C0\u67E5\u6CE8\u518C\u72B6\u6001").option("--browser <type>", "\u6307\u5B9A\u6D4F\u89C8\u5668 (chrome/chromium/edge)", "chrome").action(async (options) => {
-    const browser = options.browser;
-    if (!["chrome", "chromium", "edge"].includes(browser)) {
-      console.error(`\u4E0D\u652F\u6301\u7684\u6D4F\u89C8\u5668: ${browser}`);
-      console.log("\u652F\u6301\u7684\u6D4F\u89C8\u5668: chrome, chromium, edge");
-      process.exitCode = 1;
-      return;
+var DEFAULT_REGISTER_BROWSERS = ["chrome", "chromium"];
+var BROWSER_LABELS = {
+  chrome: "Chrome",
+  chromium: "Chromium",
+  edge: "Edge"
+};
+function resolveTargetBrowsers(options, defaultBrowsers) {
+  const specifiedCount = [options.browser !== void 0, options.browsers !== void 0, options.all === true].filter(
+    Boolean
+  ).length;
+  if (specifiedCount > 1) {
+    return { error: "--browser / --browsers / --all \u4E09\u8005\u4E92\u65A5\uFF0C\u8BF7\u53EA\u4F7F\u7528\u5176\u4E2D\u4E00\u4E2A" };
+  }
+  if (options.browser !== void 0) {
+    if (!KNOWN_BROWSERS.includes(options.browser)) {
+      return { error: `\u4E0D\u652F\u6301\u7684\u6D4F\u89C8\u5668: ${options.browser}\uFF08\u652F\u6301: ${KNOWN_BROWSERS.join(", ")}\uFF09` };
+    }
+    return { browsers: [options.browser] };
+  }
+  if (options.browsers !== void 0) {
+    const list = options.browsers.split(",").map((s) => s.trim()).filter(Boolean);
+    if (list.length === 0) {
+      return { error: "--browsers \u4E0D\u80FD\u4E3A\u7A7A" };
+    }
+    const invalid = list.filter((b) => !KNOWN_BROWSERS.includes(b));
+    if (invalid.length > 0) {
+      return { error: `\u4E0D\u652F\u6301\u7684\u6D4F\u89C8\u5668: ${invalid.join(", ")}\uFF08\u652F\u6301: ${KNOWN_BROWSERS.join(", ")}\uFF09` };
+    }
+    const deduped = Array.from(new Set(list));
+    return { browsers: deduped };
+  }
+  if (options.all) {
+    return { browsers: KNOWN_BROWSERS };
+  }
+  return { browsers: defaultBrowsers };
+}
+function printCheckSummary(results) {
+  const indent = "".padEnd(10);
+  for (const r of results) {
+    const label = BROWSER_LABELS[r.browser].padEnd(10);
+    if (r.registered) {
+      console.log(`${label} \u2713 \u5DF2\u6CE8\u518C  ${r.manifestPath}`);
+      if (r.wrapperPath) {
+        console.log(`${indent}   wrapper: ${r.wrapperPath}`);
+      }
+      if (r.wrapperNodePath) {
+        console.log(`${indent}   node:    ${r.wrapperNodePath}`);
+      }
+    } else if (r.manifest && r.wrapperExists === false) {
+      console.log(`${label} \u2717 \u672A\u6CE8\u518C\uFF08manifest \u5B58\u5728\uFF0C\u4F46\u6307\u5411\u7684 wrapper \u7F3A\u5931: ${r.wrapperPath}\uFF09`);
+    } else {
+      console.log(`${label} \u2717 \u672A\u6CE8\u518C${r.manifestPath ? `\uFF08manifest \u5B58\u5728\u4F46\u89E3\u6790\u5931\u8D25: ${r.manifestPath}\uFF09` : ""}`);
     }
+  }
+}
+function printActionSummary(actionLabel, results) {
+  for (const r of results) {
+    const label = BROWSER_LABELS[r.browser].padEnd(10);
+    if (r.success) {
+      console.log(`${label} \u2713 \u6210\u529F`);
+    } else {
+      console.log(`${label} \u2717 \u5931\u8D25\uFF1A${r.error}`);
+    }
+  }
+  const successCount = results.filter((r) => r.success).length;
+  const failCount = results.length - successCount;
+  console.log("");
+  if (failCount > 0) {
+    console.log(`${failCount} \u9879\u5931\u8D25\uFF0C${successCount} \u9879\u6210\u529F\u3002\u8BF7\u68C0\u67E5\u6743\u9650\u6216\u8FD0\u884C sudo mg-cli register`);
+  } else {
+    console.log(`\u2713 ${actionLabel}\u6210\u529F\uFF08${successCount} \u9879\uFF09`);
+  }
+  return failCount > 0 && successCount === 0;
+}
+function createRegisterCommand() {
+  const cmd = new Command22("register").description("\u6CE8\u518C Native Messaging Host\uFF08\u8BA9 Chrome \u63D2\u4EF6\u80FD\u81EA\u52A8\u542F\u52A8 Server\uFF09\uFF0C\u652F\u6301\u4E00\u6B21\u6CE8\u518C\u5230\u591A\u4E2A\u6D4F\u89C8\u5668\u76EE\u5F55").option("--unregister", "\u5378\u8F7D Native Host").option("--check", "\u68C0\u67E5\u6CE8\u518C\u72B6\u6001").option("--browser <type>", "\u6307\u5B9A\u5355\u4E2A\u6D4F\u89C8\u5668 (chrome/chromium/edge)\uFF0C\u4E0E --browsers/--all \u4E92\u65A5").option("--browsers <list>", "\u9017\u53F7\u5206\u9694\u7684\u591A\u4E2A\u6D4F\u89C8\u5668\uFF0C\u5982 chrome,chromium,edge\uFF0C\u4E0E --browser/--all \u4E92\u65A5").option("--all", "\u6CE8\u518C/\u64CD\u4F5C\u6240\u6709\u5DF2\u77E5\u6D4F\u89C8\u5668 (chrome,chromium,edge)\uFF0C\u4E0E --browser/--browsers \u4E92\u65A5").action((options) => {
     if (options.check) {
-      const status = checkRegistration(browser);
-      if (status.registered) {
-        console.log("\u2713 \u5DF2\u6CE8\u518C");
-        console.log(`  Manifest: ${status.manifestPath}`);
-        if (status.manifest) {
-          console.log(`  Native Host: ${status.manifest.path}`);
-        }
-      } else {
-        console.log("\u2717 \u672A\u6CE8\u518C");
-        if (status.manifestPath) {
-          console.log(`  Manifest \u8DEF\u5F84: ${status.manifestPath}`);
-        }
+      const resolved2 = resolveTargetBrowsers(options, KNOWN_BROWSERS);
+      if ("error" in resolved2) {
+        console.error(resolved2.error);
+        process.exitCode = 1;
+        return;
       }
+      printCheckSummary(checkAll(resolved2.browsers));
       return;
     }
     if (options.unregister) {
-      unregister(browser);
-      console.log("\u2713 \u5DF2\u5378\u8F7D Native Host");
+      const resolved2 = resolveTargetBrowsers(options, KNOWN_BROWSERS);
+      if ("error" in resolved2) {
+        console.error(resolved2.error);
+        process.exitCode = 1;
+        return;
+      }
+      const allFailed2 = printActionSummary("\u5378\u8F7D", unregisterAll(resolved2.browsers));
+      if (allFailed2) process.exitCode = 1;
+      return;
+    }
+    const resolved = resolveTargetBrowsers(options, DEFAULT_REGISTER_BROWSERS);
+    if ("error" in resolved) {
+      console.error(resolved.error);
+      process.exitCode = 1;
       return;
     }
-    register(browser);
+    const allFailed = printActionSummary("\u6CE8\u518C", registerAll(resolved.browsers));
+    if (allFailed) process.exitCode = 1;
   });
   return cmd;
 }