@fangyb/ahchat-bridge 0.1.2 → 0.1.4

package/dist/cli.js

@@ -18,15 +18,225 @@ import {
 
 // src/cli.ts
 import cac from "cac";
-var logger = createModuleLogger("bridge");
+
+// src/protocol.ts
+import { execSync } from "child_process";
+import fs from "fs";
+import os from "os";
+import path from "path";
+import { fileURLToPath } from "url";
+var logger = createModuleLogger("bridge.protocol");
+var __filename = fileURLToPath(import.meta.url);
+var __dirname = path.dirname(__filename);
+function getBridgeExePath() {
+  const pkgDir = path.resolve(__dirname, "..");
+  return path.join(pkgDir, "dist", "cli.js");
+}
+function registerProtocolHandler() {
+  const platform = os.platform();
+  if (platform === "win32") {
+    registerWindows();
+  } else if (platform === "darwin") {
+    registerMacOS();
+  } else {
+    registerLinux();
+  }
+  logger.info("Protocol handler registered", { platform });
+}
+function registerWindows() {
+  const exe = getBridgeExePath();
+  const nodeExe = process.execPath;
+  const ahchatDir = path.join(os.homedir(), ".ahchat");
+  const urlFilePath = path.join(ahchatDir, ".bridge-launch-url");
+  fs.mkdirSync(ahchatDir, { recursive: true });
+  const psScriptPath = path.join(ahchatDir, "launch-bridge.ps1");
+  const psContent = `param([string]$url)
+if (-not $url) {
+  if (Test-Path '${urlFilePath}') {
+    $url = Get-Content '${urlFilePath}' -Raw
+  }
+}
+if (-not $url) {
+  Write-Error "No URL provided"
+  exit 1
+}
+& '${nodeExe}' '${exe}' launch --url $url
+`;
+  fs.writeFileSync(psScriptPath, psContent);
+  const handler = `powershell -ExecutionPolicy Bypass -File "${psScriptPath}" -url "%1"`;
+  const regCommands = [
+    `REG ADD "HKCU\\Software\\Classes\\ahchat" /ve /d "URL:ahchat" /f`,
+    `REG ADD "HKCU\\Software\\Classes\\ahchat" /v "URL Protocol" /d "" /f`,
+    `REG ADD "HKCU\\Software\\Classes\\ahchat\\DefaultIcon" /ve /d "${nodeExe}" /f`,
+    `REG ADD "HKCU\\Software\\Classes\\ahchat\\shell\\open\\command" /ve /t REG_SZ /d "${handler}" /f`
+  ];
+  for (const cmd of regCommands) {
+    try {
+      execSync(cmd, { stdio: "pipe" });
+    } catch (e) {
+      logger.error("Failed to register Windows protocol handler", { error: e, cmd });
+      throw new Error(`Failed to register protocol handler: ${cmd}`);
+    }
+  }
+  logger.info("Windows protocol handler registered", { psScriptPath });
+}
+function registerMacOS() {
+  const appDir = path.join(os.homedir(), "Applications", "AHChatBridge.app");
+  const contentsDir = path.join(appDir, "Contents");
+  const macosDir = path.join(contentsDir, "MacOS");
+  const resourcesDir = path.join(contentsDir, "Resources");
+  fs.mkdirSync(macosDir, { recursive: true });
+  fs.mkdirSync(resourcesDir, { recursive: true });
+  const infoPlist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>CFBundleName</key>
+  <string>AHChatBridge</string>
+  <key>CFBundleDisplayName</key>
+  <string>AHChat Bridge</string>
+  <key>CFBundleIdentifier</key>
+  <string>com.fangyb.ahchat-bridge</string>
+  <key>CFBundleVersion</key>
+  <string>0.1.0</string>
+  <key>CFBundlePackageType</key>
+  <string>APPL</string>
+  <key>CFBundleExecutable</key>
+  <string>launch.sh</string>
+  <key>CFBundleURLTypes</key>
+  <array>
+    <dict>
+      <key>CFBundleURLName</key>
+      <string>AHChat Bridge</string>
+      <key>CFBundleURLSchemes</key>
+      <array>
+        <string>ahchat</string>
+      </array>
+    </dict>
+  </array>
+</dict>
+</plist>`;
+  const launchScript = `#!/bin/bash
+URL="$1"
+exec "${process.execPath}" "${getBridgeExePath()}" launch --url "$URL"`;
+  fs.writeFileSync(path.join(contentsDir, "Info.plist"), infoPlist);
+  fs.writeFileSync(path.join(macosDir, "launch.sh"), launchScript);
+  fs.chmodSync(path.join(macosDir, "launch.sh"), 493);
+  logger.info("macOS protocol handler registered", { appDir });
+}
+function registerLinux() {
+  const desktopFile = `[Desktop Entry]
+Name=AHChat Bridge
+Exec=${process.execPath} ${getBridgeExePath()} launch --url %u
+Type=Application
+NoDisplay=true
+MimeType=x-scheme-handler/ahchat;`;
+  const desktopPath = path.join(
+    os.homedir(),
+    ".local",
+    "share",
+    "applications",
+    "ahchat-bridge.desktop"
+  );
+  fs.mkdirSync(path.dirname(desktopPath), { recursive: true });
+  fs.writeFileSync(desktopPath, desktopFile);
+  try {
+    execSync("update-desktop-database ~/.local/share/applications/", { stdio: "pipe" });
+  } catch {
+  }
+  logger.info("Linux protocol handler registered", { desktopPath });
+}
+function unregisterProtocolHandler() {
+  const platform = os.platform();
+  if (platform === "win32") {
+    try {
+      execSync('REG DELETE "HKCU\\Software\\Classes\\ahchat" /f', { stdio: "pipe" });
+      const psScriptPath = path.join(os.homedir(), ".ahchat", "launch-bridge.ps1");
+      const urlFilePath = path.join(os.homedir(), ".ahchat", ".bridge-launch-url");
+      try {
+        fs.unlinkSync(psScriptPath);
+      } catch {
+      }
+      try {
+        fs.unlinkSync(urlFilePath);
+      } catch {
+      }
+      logger.info("Windows protocol handler unregistered");
+    } catch (e) {
+      logger.warn("Failed to unregister Windows protocol handler", { error: e });
+    }
+  } else if (platform === "darwin") {
+    const appDir = path.join(os.homedir(), "Applications", "AHChatBridge.app");
+    try {
+      fs.rmSync(appDir, { recursive: true, force: true });
+      logger.info("macOS protocol handler unregistered");
+    } catch (e) {
+      logger.warn("Failed to unregister macOS protocol handler", { error: e });
+    }
+  } else {
+    const desktopPath = path.join(
+      os.homedir(),
+      ".local",
+      "share",
+      "applications",
+      "ahchat-bridge.desktop"
+    );
+    try {
+      fs.unlinkSync(desktopPath);
+      logger.info("Linux protocol handler unregistered");
+    } catch (e) {
+      logger.warn("Failed to unregister Linux protocol handler", { error: e });
+    }
+  }
+}
+function isProtocolRegistered() {
+  const platform = os.platform();
+  if (platform === "win32") {
+    try {
+      execSync('REG QUERY "HKCU\\Software\\Classes\\ahchat" /ve', { stdio: "pipe" });
+      return true;
+    } catch {
+      return false;
+    }
+  } else if (platform === "darwin") {
+    const appDir = path.join(os.homedir(), "Applications", "AHChatBridge.app");
+    return fs.existsSync(path.join(appDir, "Contents", "Info.plist"));
+  } else {
+    const desktopPath = path.join(
+      os.homedir(),
+      ".local",
+      "share",
+      "applications",
+      "ahchat-bridge.desktop"
+    );
+    return fs.existsSync(desktopPath);
+  }
+}
+
+// src/cli.ts
+var logger2 = createModuleLogger("bridge");
 var cli = cac("ahchat-bridge");
-cli.command("[command]", "Start the bridge (default: run)").option("--server-url <url>", "WebSocket URL of the AHChat server").option("--token <token>", "Auth token for server registration").option("--data-dir <dir>", "Data directory (default: ~/.ahchat)").option("--log-level <level>", "Log level (default: INFO)").action((args) => {
-  void runBridge({
-    serverUrl: args["server-url"],
-    token: args.token,
-    dataDir: args["data-dir"],
-    logLevel: args["log-level"]
-  });
+cli.command("install", "Register ahchat:// protocol handler (one-time setup)").action(() => {
+  registerProtocolHandler();
+  console.log("ahchat:// protocol handler registered successfully.");
+  console.log("You can now launch the bridge from your browser with one click.");
+});
+cli.command("uninstall", "Remove ahchat:// protocol handler").action(() => {
+  unregisterProtocolHandler();
+  console.log("ahchat:// protocol handler removed.");
+});
+cli.command("status", "Check if protocol handler is registered").action(() => {
+  const registered = isProtocolRegistered();
+  console.log(registered ? "ahchat:// protocol is registered." : "ahchat:// protocol is NOT registered.");
+  console.log('Run "npx @fangyb/ahchat-bridge install" to register it.');
+});
+cli.command("launch", "Launch bridge from ahchat:// URL (called by OS)").option("--url <url>", "ahchat:// URL with server and token params").action((args) => {
+  const url = args.url;
+  if (!url) {
+    console.error("Error: --url is required");
+    process.exit(1);
+  }
+  void runBridgeFromUrl(url);
 });
 cli.command("run", "Start the bridge and connect to server").option("--server-url <url>", "WebSocket URL of the AHChat server").option("--token <token>", "Auth token for server registration").option("--data-dir <dir>", "Data directory (default: ~/.ahchat)").option("--log-level <level>", "Log level (default: INFO)").action((args) => {
   void runBridge({
@@ -42,6 +252,35 @@ cli.command("version", "Show bridge version").action(() => {
 cli.help();
 cli.version("0.1.0");
 cli.parse();
+function parseAhchatUrl(url) {
+  try {
+    if (!url.startsWith("ahchat://")) return null;
+    const afterProtocol = url.slice("ahchat://".length);
+    const tildeIndex = afterProtocol.indexOf("~");
+    if (tildeIndex < 0) return null;
+    const rest = afterProtocol.slice(tildeIndex + 1);
+    const lastTilde = rest.lastIndexOf("~");
+    if (lastTilde < 0) return null;
+    const serverUrl = decodeURIComponent(rest.slice(0, lastTilde));
+    const token = decodeURIComponent(rest.slice(lastTilde + 1));
+    if (!serverUrl || !token) return null;
+    return { serverUrl, token };
+  } catch {
+    return null;
+  }
+}
+async function runBridgeFromUrl(url) {
+  const parsed = parseAhchatUrl(url);
+  if (!parsed) {
+    console.error("Invalid ahchat:// URL:", url);
+    console.error("Expected format: ahchat://bridge?server=ws://host:port/ws/bridge&token=xxx");
+    process.exit(1);
+  }
+  await runBridge({
+    serverUrl: parsed.serverUrl,
+    token: parsed.token
+  });
+}
 async function runBridge(args) {
   let config = loadBridgeConfig();
   if (args.serverUrl) {
@@ -60,7 +299,7 @@ async function runBridge(args) {
   }
   ensureDir(config.dataDir);
   acquireLock(config.dataDir);
-  logger.info("Bridge starting", {
+  logger2.info("Bridge starting", {
     bridgeId: config.bridgeId,
     serverUrl: config.serverUrl,
     serverApiUrl: config.serverApiUrl
@@ -101,21 +340,21 @@ async function runBridge(args) {
       switch (msg.type) {
         case "bridge:list_models_request": {
           const { requestId } = msg.payload;
-          logger.info("list_models request received", { requestId });
+          logger2.info("list_models request received", { requestId });
           try {
             const models = await listModels();
             connector?.send({
               type: "bridge:list_models_response",
               payload: { requestId, models }
             });
-            logger.info("list_models response sent", { requestId, count: models.length });
+            logger2.info("list_models response sent", { requestId, count: models.length });
           } catch (e) {
             const err = e instanceof Error ? e.message : String(e);
             connector?.send({
               type: "bridge:list_models_response",
               payload: { requestId, error: err }
             });
-            logger.error("list_models failed", { requestId, error: e });
+            logger2.error("list_models failed", { requestId, error: e });
           }
           break;
         }
@@ -123,7 +362,7 @@ async function runBridge(args) {
           await agentManager.terminate(msg.payload.agentId);
           break;
         case "agent:terminate_scope":
-          logger.info("agent:terminate_scope received", {
+          logger2.info("agent:terminate_scope received", {
             agentId: msg.payload.agentId,
             scope: msg.payload.scope
           });
@@ -140,7 +379,7 @@ async function runBridge(args) {
           const p = msg.payload;
           const answerText = formatAnswerForSDK(p);
           const ok = askQuestionRegistry.resolve(p.questionId, answerText);
-          logger.info("user:answer_question handled", {
+          logger2.info("user:answer_question handled", {
             questionId: p.questionId,
             agentId: p.agentId,
             resolved: ok,
@@ -161,7 +400,7 @@ async function runBridge(args) {
     });
   }, config.queryConfig.statusReportIntervalMs);
   const shutdown = async (signal) => {
-    logger.info("Shutdown signal received", { signal });
+    logger2.info("Shutdown signal received", { signal });
     if (statusInterval) {
       clearInterval(statusInterval);
       statusInterval = null;
@@ -169,7 +408,7 @@ async function runBridge(args) {
     wsMetrics.stop();
     connector?.close();
     await agentManager.shutdownAll();
-    logger.info("Bridge stopped");
+    logger2.info("Bridge stopped");
     process.exit(0);
   };
   process.on("SIGINT", () => void shutdown("SIGINT"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fangyb/ahchat-bridge",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "AHChat Bridge CLI — connect your local Claude Code agents to an AHChat server",
   "type": "module",
   "main": "dist/index.js",