@hangox/mg-cli 1.3.3 → 1.4.0

package/bin/mg-native-host.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+import '../dist/native-host.js'

package/dist/cli.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 
 // src/cli/index.ts
-import { Command as Command14 } from "commander";
+import { Command as Command15 } from "commander";
 
 // src/cli/version-check.ts
 import { existsSync as existsSync2, mkdirSync, readFileSync as readFileSync2, writeFileSync } from "fs";
@@ -1694,7 +1694,7 @@ var MGClient = class {
         const currentVersion = getVersion();
         if (!isVersionMatch(currentVersion, serverInfo.version)) {
           console.warn(`\u26A0\uFE0F \u7248\u672C\u4E0D\u5339\u914D: CLI ${currentVersion} vs Server ${serverInfo.version}`);
-          console.warn("\u63D0\u793A: \u5982\u9700\u5BF9\u9F50\u7248\u672C\uFF0C\u8BF7\u624B\u52A8\u8FD0\u884C `npx -y @hangox/mg-cli@latest server restart`");
+          console.warn("\u63D0\u793A: \u5982\u9700\u5BF9\u9F50\u7248\u672C\uFF0C\u8BF7\u624B\u52A8\u8FD0\u884C `mg-cli server restart`");
         }
         try {
           await this.tryConnect(serverInfo.port);
@@ -2890,7 +2890,7 @@ async function handleOpenPage(options) {
         console.error("\u9519\u8BEF: \u6CA1\u6709 Chrome \u6269\u5C55\u8FDE\u63A5\u5230 Server\u3002\u8BF7\u786E\u4FDD\u6D4F\u89C8\u5668\u5DF2\u6253\u5F00\u5E76\u5B89\u88C5\u4E86 MG Plugin");
         console.error("\n\u{1F4A1} \u89E3\u51B3\u65B9\u6CD5:");
         console.error("  1. \u67E5\u770B\u53EF\u7528\u7684\u6269\u5C55\u5B9E\u4F8B:");
-        console.error("     npx -y @hangox/mg-cli@latest list_extensions");
+        console.error("     mg-cli list_extensions");
         console.error("");
         console.error("  2. \u5982\u679C\u6CA1\u6709\u6269\u5C55\u8FDE\u63A5\uFF0C\u8BF7\u68C0\u67E5:");
         console.error("     \u2022 \u6D4F\u89C8\u5668\u662F\u5426\u5DF2\u6253\u5F00");
@@ -2926,7 +2926,7 @@ async function handleOpenPage(options) {
     if (errorMsg.includes("NO_EXTENSION_CONNECTED") || errorMsg.includes("\u672A\u627E\u5230")) {
       console.error("\n\u{1F4A1} \u89E3\u51B3\u65B9\u6CD5:");
       console.error("  1. \u67E5\u770B\u53EF\u7528\u7684\u6269\u5C55\u5B9E\u4F8B:");
-      console.error("     npx -y @hangox/mg-cli@latest list_extensions");
+      console.error("     mg-cli list_extensions");
       console.error("");
       console.error("  2. \u5982\u679C\u6CA1\u6709\u6269\u5C55\u8FDE\u63A5\uFF0C\u8BF7\u68C0\u67E5:");
       console.error("     \u2022 \u6D4F\u89C8\u5668\u662F\u5426\u5DF2\u6253\u5F00");
@@ -2981,7 +2981,7 @@ async function handleNavigateToNode(options) {
         console.error("\u9519\u8BEF: \u6CA1\u6709 Chrome \u6269\u5C55\u8FDE\u63A5\u5230 Server\u3002\u8BF7\u786E\u4FDD\u6D4F\u89C8\u5668\u5DF2\u6253\u5F00\u5E76\u5B89\u88C5\u4E86 MG Plugin");
         console.error("\n\u{1F4A1} \u89E3\u51B3\u65B9\u6CD5:");
         console.error("  1. \u67E5\u770B\u53EF\u7528\u7684\u6269\u5C55\u5B9E\u4F8B:");
-        console.error("     npx -y @hangox/mg-cli@latest list_extensions");
+        console.error("     mg-cli list_extensions");
         console.error("");
         console.error("  2. \u5982\u679C\u6CA1\u6709\u6269\u5C55\u8FDE\u63A5\uFF0C\u8BF7\u68C0\u67E5:");
         console.error("     \u2022 \u6D4F\u89C8\u5668\u662F\u5426\u5DF2\u6253\u5F00");
@@ -3023,7 +3023,7 @@ async function handleNavigateToNode(options) {
     if (errorMsg.includes("NO_EXTENSION_CONNECTED") || errorMsg.includes("\u672A\u627E\u5230")) {
       console.error("\n\u{1F4A1} \u89E3\u51B3\u65B9\u6CD5:");
       console.error("  1. \u67E5\u770B\u53EF\u7528\u7684\u6269\u5C55\u5B9E\u4F8B:");
-      console.error("     npx -y @hangox/mg-cli@latest list_extensions");
+      console.error("     mg-cli list_extensions");
       console.error("");
       console.error("  2. \u5982\u679C\u6CA1\u6709\u6269\u5C55\u8FDE\u63A5\uFF0C\u8BF7\u68C0\u67E5:");
       console.error("     \u2022 \u6D4F\u89C8\u5668\u662F\u5426\u5DF2\u6253\u5F00");
@@ -3408,8 +3408,180 @@ async function handleVisualize(options) {
   }
 }
 
+// src/cli/commands/register.ts
+import { Command as Command14 } from "commander";
+
+// src/scripts/register-utils.ts
+import fs from "fs";
+import path from "path";
+import os from "os";
+import { execSync as execSync2 } from "child_process";
+import { fileURLToPath as fileURLToPath3 } from "url";
+var HOST_NAME = "com.hangox.mgplugin";
+var EXTENSION_ID = "ddhihanlpcdneicohnglnaliefnkaeja";
+var DESCRIPTION = "MasterGo Plugin Native Messaging Host";
+function getBrowserPaths(browser) {
+  const home = os.homedir();
+  const platform2 = os.platform();
+  const browserDirs = {
+    chrome: {
+      darwin: path.join(home, "Library/Application Support/Google/Chrome/NativeMessagingHosts"),
+      linux: path.join(home, ".config/google-chrome/NativeMessagingHosts"),
+      win32: path.join(
+        process.env.APPDATA || path.join(home, "AppData/Roaming"),
+        "Google/Chrome/NativeMessagingHosts"
+      )
+    },
+    chromium: {
+      darwin: path.join(home, "Library/Application Support/Chromium/NativeMessagingHosts"),
+      linux: path.join(home, ".config/chromium/NativeMessagingHosts"),
+      win32: path.join(
+        process.env.APPDATA || path.join(home, "AppData/Roaming"),
+        "Chromium/NativeMessagingHosts"
+      )
+    },
+    edge: {
+      darwin: path.join(
+        home,
+        "Library/Application Support/Microsoft Edge/NativeMessagingHosts"
+      ),
+      linux: path.join(home, ".config/microsoft-edge/NativeMessagingHosts"),
+      win32: path.join(
+        process.env.APPDATA || path.join(home, "AppData/Roaming"),
+        "Microsoft/Edge/NativeMessagingHosts"
+      )
+    }
+  };
+  const registryKeys = {
+    chrome: `HKCU\\Software\\Google\\Chrome\\NativeMessagingHosts\\${HOST_NAME}`,
+    chromium: `HKCU\\Software\\Chromium\\NativeMessagingHosts\\${HOST_NAME}`,
+    edge: `HKCU\\Software\\Microsoft\\Edge\\NativeMessagingHosts\\${HOST_NAME}`
+  };
+  const manifestDir = browserDirs[browser][platform2];
+  if (!manifestDir) {
+    throw new Error(`\u4E0D\u652F\u6301\u7684\u5E73\u53F0: ${platform2}`);
+  }
+  return {
+    manifestDir,
+    registryKey: platform2 === "win32" ? registryKeys[browser] : void 0
+  };
+}
+function getNativeHostPath() {
+  const currentFile = fileURLToPath3(import.meta.url);
+  const distDir = path.dirname(currentFile);
+  return path.join(distDir, "native-host.js");
+}
+function createManifest() {
+  return {
+    name: HOST_NAME,
+    description: DESCRIPTION,
+    path: getNativeHostPath(),
+    type: "stdio",
+    allowed_origins: [`chrome-extension://${EXTENSION_ID}/`]
+  };
+}
+function register(browser = "chrome") {
+  const config = getBrowserPaths(browser);
+  const manifestPath = path.join(config.manifestDir, `${HOST_NAME}.json`);
+  const manifest = createManifest();
+  console.log("\u6B63\u5728\u6CE8\u518C Native Messaging Host...");
+  console.log(`  Host Name: ${HOST_NAME}`);
+  console.log(`  Extension ID: ${EXTENSION_ID}`);
+  console.log(`  Browser: ${browser}`);
+  console.log(`  Native Host: ${manifest.path}`);
+  console.log(`  Manifest: ${manifestPath}`);
+  fs.mkdirSync(config.manifestDir, { recursive: true });
+  fs.writeFileSync(manifestPath, JSON.stringify(manifest, null, 2));
+  if (os.platform() !== "win32") {
+    try {
+      fs.chmodSync(manifest.path, "755");
+    } catch {
+      console.warn("  \u26A0\uFE0F \u65E0\u6CD5\u8BBE\u7F6E\u6267\u884C\u6743\u9650\uFF0C\u53EF\u80FD\u9700\u8981\u624B\u52A8\u8BBE\u7F6E");
+    }
+  }
+  if (config.registryKey) {
+    try {
+      execSync2(`reg add "${config.registryKey}" /ve /t REG_SZ /d "${manifestPath}" /f`, {
+        stdio: "pipe"
+      });
+      console.log("  \u2713 Windows \u6CE8\u518C\u8868\u5DF2\u66F4\u65B0");
+    } catch {
+      console.warn("  \u26A0\uFE0F \u6CE8\u518C\u8868\u5199\u5165\u5931\u8D25\uFF0C\u53EF\u80FD\u9700\u8981\u7BA1\u7406\u5458\u6743\u9650");
+    }
+  }
+  console.log("\u2713 Native Messaging Host \u6CE8\u518C\u6210\u529F");
+}
+function unregister(browser = "chrome") {
+  const config = getBrowserPaths(browser);
+  const manifestPath = path.join(config.manifestDir, `${HOST_NAME}.json`);
+  if (fs.existsSync(manifestPath)) {
+    fs.unlinkSync(manifestPath);
+    console.log(`  \u5DF2\u5220\u9664: ${manifestPath}`);
+  }
+  if (config.registryKey) {
+    try {
+      execSync2(`reg delete "${config.registryKey}" /f`, { stdio: "pipe" });
+      console.log("  \u2713 Windows \u6CE8\u518C\u8868\u9879\u5DF2\u5220\u9664");
+    } catch {
+    }
+  }
+}
+function checkRegistration(browser = "chrome") {
+  const config = getBrowserPaths(browser);
+  const manifestPath = path.join(config.manifestDir, `${HOST_NAME}.json`);
+  if (!fs.existsSync(manifestPath)) {
+    return { registered: false };
+  }
+  try {
+    const manifest = JSON.parse(fs.readFileSync(manifestPath, "utf-8"));
+    return {
+      registered: true,
+      manifestPath,
+      manifest
+    };
+  } catch {
+    return { registered: false, manifestPath };
+  }
+}
+
+// src/cli/commands/register.ts
+function createRegisterCommand() {
+  const cmd = new Command14("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;
+    }
+    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}`);
+        }
+      }
+      return;
+    }
+    if (options.unregister) {
+      unregister(browser);
+      console.log("\u2713 \u5DF2\u5378\u8F7D Native Host");
+      return;
+    }
+    register(browser);
+  });
+  return cmd;
+}
+
 // src/cli/index.ts
-var program = new Command14();
+var program = new Command15();
 program.name("mg-cli").description("MasterGo CLI \u5DE5\u5177 - \u7528\u4E8E Claude Code \u4E0E MasterGo \u901A\u4FE1").version(getVersion());
 program.addCommand(createServerCommand());
 program.addCommand(createGetNodeByIdCommand());
@@ -3424,6 +3596,7 @@ program.addCommand(createListExtensionsCommand());
 program.addCommand(createOpenPageCommand());
 program.addCommand(createNavigateToNodeCommand());
 program.addCommand(createVisualizeCommand());
+program.addCommand(createRegisterCommand());
 checkForUpdates();
 program.parse();
 //# sourceMappingURL=cli.js.map