yiyan-browser-agent 1.1.3 → 1.2.0

package/dist/cli.d.ts

@@ -4,12 +4,13 @@ import { C as CliOutput } from './types-BhQ78DYf.js';
 declare function printHelp(): void;
 /** 解析后的 CLI 参数 */
 interface ParsedArgs {
-    command: 'ask' | 'status' | 'reset' | 'login' | 'help' | null;
+    command: 'ask' | 'status' | 'reset' | 'login' | 'help' | 'config' | null;
     question?: string;
     timeout?: number;
     retry?: number;
     headful?: boolean;
     verbose?: boolean;
+    chromePath?: string;
 }
 /**
  * 解析 CLI 参数

package/dist/cli.js

@@ -2,7 +2,7 @@
 
 // src/agent.ts
 import path3 from "path";
-import os2 from "os";
+import os3 from "os";
 
 // src/types.ts
 var YiyanAgentError = class extends Error {
@@ -23,17 +23,68 @@ var YIYAN_CHAT_URL = "https://yiyan.baidu.com/";
 
 // src/profile.ts
 import path from "path";
+import os from "os";
 import fs from "fs";
 import fsp from "fs/promises";
-function getChromeExecutablePath(platform) {
+var CONFIG_FILE_PATH = path.join(os.homedir(), ".yiyan-browser-agent", "config.json");
+function readConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE_PATH)) {
+      const content = fs.readFileSync(CONFIG_FILE_PATH, "utf-8");
+      return JSON.parse(content);
+    }
+  } catch {
+  }
+  return {};
+}
+function saveConfig(config) {
+  const configDir = path.dirname(CONFIG_FILE_PATH);
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+  }
+  fs.writeFileSync(CONFIG_FILE_PATH, JSON.stringify(config, null, 2));
+}
+var LINUX_CHROME_PATHS = [
+  "/usr/bin/google-chrome-stable",
+  "/usr/bin/google-chrome",
+  "/usr/bin/chromium-browser",
+  "/usr/bin/chromium",
+  "/snap/bin/google-chrome"
+  // Ubuntu snap 安装
+];
+function detectLinuxChromePath() {
+  for (const chromePath of LINUX_CHROME_PATHS) {
+    if (fs.existsSync(chromePath)) {
+      return chromePath;
+    }
+  }
+  return null;
+}
+function getChromeExecutablePath(platform, userPath) {
+  if (userPath && fs.existsSync(userPath)) {
+    return userPath;
+  }
   switch (platform) {
     case "win32":
       const programFiles = process.env.PROGRAMFILES || "C:\\Program Files";
-      return path.join(programFiles, "Google", "Chrome", "Application", "chrome.exe");
+      const winPath = path.join(programFiles, "Google", "Chrome", "Application", "chrome.exe");
+      if (fs.existsSync(winPath)) {
+        return winPath;
+      }
+      const programFilesX86 = process.env["PROGRAMFILES(X86)"] || "C:\\Program Files (x86)";
+      const winPathX86 = path.join(programFilesX86, "Google", "Chrome", "Application", "chrome.exe");
+      if (fs.existsSync(winPathX86)) {
+        return winPathX86;
+      }
+      return null;
     case "darwin":
-      return "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome";
+      const macPath = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome";
+      if (fs.existsSync(macPath)) {
+        return macPath;
+      }
+      return null;
     case "linux":
-      return "/usr/bin/google-chrome";
+      return detectLinuxChromePath();
     default:
       return null;
   }
@@ -55,10 +106,10 @@ import { chromium } from "playwright-core";
 import http from "http";
 import { spawn, execSync } from "child_process";
 import path2 from "path";
-import os from "os";
+import os2 from "os";
 import fs2 from "fs";
 var CDP_PORT = 19222;
-var DEBUG_DIR = path2.join(os.homedir(), ".yiyan-browser-agent", "debug");
+var DEBUG_DIR = path2.join(os2.homedir(), ".yiyan-browser-agent", "debug");
 function log(verbose, msg) {
   if (verbose) {
     process.stderr.write(`[yiyan-agent] ${msg}
@@ -148,6 +199,12 @@ function waitForCDP(port, timeout = 15e3) {
 }
 async function launchBrowser(options) {
   const { chromePath, profilePath, headless, timeout = 3e4, verbose = false } = options;
+  if (!chromePath || !fs2.existsSync(chromePath)) {
+    throw new YiyanAgentError(
+      "BROWSER_LAUNCH",
+      `Chrome not found at: ${chromePath || "unknown path"}. Please install Chrome or use --chrome-path to specify a custom path.`
+    );
+  }
   try {
     log(verbose, "\u6E05\u7406\u6B8B\u7559 Chrome \u8FDB\u7A0B...");
     killProcessOnPort(CDP_PORT);
@@ -159,10 +216,18 @@ async function launchBrowser(options) {
       "--no-default-browser-check",
       "--disable-blink-features=AutomationControlled"
     ];
+    if (os2.platform() === "linux") {
+      chromeArgs.push(
+        "--no-sandbox",
+        "--disable-setuid-sandbox",
+        "--disable-dev-shm-usage"
+      );
+    }
     if (headless) {
       chromeArgs.push("--headless=new");
     }
     log(verbose, `\u542F\u52A8 Chrome: ${chromePath} ${headless ? "(headless)" : "(headed)"}`);
+    log(verbose, `Chrome \u53C2\u6570: ${chromeArgs.join(" ")}`);
     const chromeProcess = spawn(chromePath, chromeArgs, {
       detached: true,
       stdio: "ignore"
@@ -720,13 +785,14 @@ async function extractReply(page, question, verbose = false) {
 
 // src/agent.ts
 var DEFAULT_PROFILE_BASE_DIR = path3.join(
-  os2.homedir(),
+  os3.homedir(),
   ".yiyan-browser-agent"
 );
 var YiyanAgent = class {
   options;
   profileDir;
   verbose;
+  savedChromePath;
   constructor(options) {
     this.options = {
       ...DEFAULT_OPTIONS,
@@ -734,6 +800,29 @@ var YiyanAgent = class {
     };
     this.profileDir = this.options.profileDir || DEFAULT_PROFILE_BASE_DIR;
     this.verbose = options?.verbose ?? false;
+    const config = readConfig();
+    this.savedChromePath = config.chromePath || null;
+  }
+  /**
+   * 保存 Chrome 路径到配置文件
+   */
+  saveChromePath(chromePath) {
+    saveConfig({ chromePath });
+    this.savedChromePath = chromePath;
+  }
+  /**
+   * 获取有效的 Chrome 路径
+   * 优先级：用户指定 > 配置文件保存 > 自动探测
+   */
+  getEffectiveChromePath(userPath) {
+    const platform = os3.platform();
+    if (userPath) {
+      return userPath;
+    }
+    if (this.savedChromePath) {
+      return this.savedChromePath;
+    }
+    return getChromeExecutablePath(platform);
   }
   /**
    * 发送问题并获取答案
@@ -762,12 +851,12 @@ var YiyanAgent = class {
    * 执行单次问答
    */
   async executeAsk(question, headful) {
-    const platform = os2.platform();
-    const chromePath = this.options.chromePath || getChromeExecutablePath(platform);
+    const platform = os3.platform();
+    const chromePath = this.getEffectiveChromePath(this.options.chromePath);
     if (!chromePath) {
       throw new YiyanAgentError(
         "BROWSER_LAUNCH",
-        `Unsupported platform: ${platform}`
+        `Chrome not found on ${platform}. Please install Chrome or use --chrome-path to specify a custom path.`
       );
     }
     const profilePath = path3.join(this.profileDir, "chrome-profile");
@@ -804,12 +893,12 @@ var YiyanAgent = class {
    * 登录成功后，后续的 ask 调用将自动使用登录状态
    */
   async login() {
-    const platform = os2.platform();
-    const chromePath = this.options.chromePath || getChromeExecutablePath(platform);
+    const platform = os3.platform();
+    const chromePath = this.getEffectiveChromePath(this.options.chromePath);
     if (!chromePath) {
       throw new YiyanAgentError(
         "BROWSER_LAUNCH",
-        `Unsupported platform: ${platform}`
+        `Chrome not found on ${platform}. Please install Chrome or use --chrome-path to specify a custom path.`
       );
     }
     const profilePath = path3.join(this.profileDir, "chrome-profile");
@@ -867,36 +956,49 @@ function printHelp() {
 yiyan-agent - \u6587\u5FC3\u4E00\u8A00\u6D4F\u89C8\u5668\u4EE3\u7406 CLI
 
 \u7528\u6CD5:
-  yiyan-agent login                    \u9996\u6B21\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u4F1A\u6253\u5F00\u6D4F\u89C8\u5668\u7A97\u53E3\uFF09
-  yiyan-agent ask "\u95EE\u9898" [--timeout ms] [--retry n] [--headful] [--verbose]
-  yiyan-agent status                   \u68C0\u67E5\u767B\u5F55\u72B6\u6001
-  yiyan-agent reset                    \u6E05\u9664\u4FDD\u5B58\u7684 profile
+  yiyan-agent login [--chrome-path <path>]    \u9996\u6B21\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u4F1A\u6253\u5F00\u6D4F\u89C8\u5668\u7A97\u53E3\uFF09
+  yiyan-agent ask "\u95EE\u9898" [--timeout ms] [--retry n] [--headful] [--verbose] [--chrome-path <path>]
+  yiyan-agent status                          \u68C0\u67E5\u767B\u5F55\u72B6\u6001
+  yiyan-agent reset                           \u6E05\u9664\u4FDD\u5B58\u7684 profile
+  yiyan-agent config                          \u663E\u793A\u5F53\u524D\u914D\u7F6E
 
 \u547D\u4EE4:
   login      \u6253\u5F00\u6D4F\u89C8\u5668\u624B\u52A8\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u9996\u6B21\u4F7F\u7528\u5FC5\u987B\u5148\u767B\u5F55\uFF09
   ask        \u53D1\u9001\u95EE\u9898\u5E76\u83B7\u53D6\u7B54\u6848\uFF08\u9ED8\u8BA4\u65E0\u5934\u6A21\u5F0F\uFF0C\u4E0D\u5F39\u7A97\uFF1B\u767B\u5F55\u540E\u76F4\u63A5\u4F7F\u7528\uFF09
   status     \u68C0\u67E5\u767B\u5F55\u72B6\u6001
   reset      \u6E05\u9664\u4FDD\u5B58\u7684 profile
+  config     \u663E\u793A\u5F53\u524D\u914D\u7F6E
 
 \u9009\u9879:
-  --timeout <ms>    \u8D85\u65F6\u65F6\u95F4\uFF08\u6BEB\u79D2\uFF09\uFF0C\u9ED8\u8BA4 120000
-  --retry <n>       \u91CD\u8BD5\u6B21\u6570\uFF0C\u9ED8\u8BA4 3
-  --headful         \u4F7F\u7528\u6709\u5934\u6D4F\u89C8\u5668\uFF08\u53EF\u89C1\u7A97\u53E3\uFF0C\u7528\u4E8E\u624B\u52A8\u8FC7\u9A8C\u8BC1\u7801\uFF09
-  --verbose         \u663E\u793A\u8BE6\u7EC6\u65E5\u5FD7\uFF08\u8C03\u8BD5\u7528\uFF09
-  --help            \u663E\u793A\u5E2E\u52A9\u4FE1\u606F
+  --timeout <ms>      \u8D85\u65F6\u65F6\u95F4\uFF08\u6BEB\u79D2\uFF09\uFF0C\u9ED8\u8BA4 120000
+  --retry <n>         \u91CD\u8BD5\u6B21\u6570\uFF0C\u9ED8\u8BA4 3
+  --headful           \u4F7F\u7528\u6709\u5934\u6D4F\u89C8\u5668\uFF08\u53EF\u89C1\u7A97\u53E3\uFF0C\u7528\u4E8E\u624B\u52A8\u8FC7\u9A8C\u8BC1\u7801\uFF09
+  --verbose           \u663E\u793A\u8BE6\u7EC6\u65E5\u5FD7\uFF08\u8C03\u8BD5\u7528\uFF09
+  --chrome-path <path> \u6307\u5B9A Chrome \u53EF\u6267\u884C\u6587\u4EF6\u8DEF\u5F84\uFF08\u4F1A\u4FDD\u5B58\u914D\u7F6E\uFF0C\u540E\u7EED\u81EA\u52A8\u4F7F\u7528\uFF09
+  --help              \u663E\u793A\u5E2E\u52A9\u4FE1\u606F
 
 \u793A\u4F8B:
   yiyan-agent login                           # \u9996\u6B21\u4F7F\u7528\uFF1A\u767B\u5F55\u6587\u5FC3\u4E00\u8A00
+  yiyan-agent login --chrome-path "/usr/bin/google-chrome-stable"  # Linux \u6307\u5B9A Chrome \u8DEF\u5F84
   yiyan-agent ask "\u4EC0\u4E48\u662F TypeScript\uFF1F"         # \u63D0\u95EE\uFF08\u9759\u9ED8\u6A21\u5F0F\uFF09
   yiyan-agent ask "\u89E3\u91CA Promise" --verbose    # \u663E\u793A\u8BE6\u7EC6\u65E5\u5FD7
   yiyan-agent ask "30+30=" --headful         # \u6709\u5934\u6A21\u5F0F\uFF08\u53EF\u624B\u52A8\u8FC7\u9A8C\u8BC1\u7801\uFF09
   yiyan-agent status
   yiyan-agent reset
+  yiyan-agent config                          # \u663E\u793A\u4FDD\u5B58\u7684 Chrome \u8DEF\u5F84\u7B49\u914D\u7F6E
 
 \u6D41\u7A0B:
   1. \u5148\u8FD0\u884C yiyan-agent login \u767B\u5F55\uFF08\u53EA\u9700\u4E00\u6B21\uFF0C\u767B\u5F55\u72B6\u6001\u4F1A\u4FDD\u5B58\uFF09
   2. \u4E4B\u540E\u76F4\u63A5 yiyan-agent ask "\u95EE\u9898" \u5373\u53EF\uFF0C\u65E0\u9700\u518D\u767B\u5F55
   3. \u5982\u679C\u9047\u5230\u9A8C\u8BC1\u7801\uFF0C\u52A0 --headful \u5207\u6362\u6709\u5934\u6A21\u5F0F\u624B\u52A8\u5904\u7406
+
+Linux \u63D0\u793A:
+  \u5982\u679C login \u6CA1\u6709\u53CD\u5E94\uFF0C\u8BF7\u5C1D\u8BD5\u6307\u5B9A Chrome \u8DEF\u5F84\uFF1A
+    yiyan-agent login --chrome-path "/usr/bin/google-chrome-stable"
+  \u5E38\u89C1\u8DEF\u5F84\uFF1A
+    /usr/bin/google-chrome-stable  (Debian/Ubuntu apt \u5B89\u88C5)
+    /usr/bin/chromium-browser      (\u67D0\u4E9B Ubuntu \u7248\u672C)
+    /snap/bin/google-chrome        (Ubuntu snap \u5B89\u88C5)
 `);
 }
 function parseCliArgs(args) {
@@ -925,6 +1027,9 @@ function parseCliArgs(args) {
     if (arg === "login") {
       result.command = "login";
     }
+    if (arg === "config") {
+      result.command = "config";
+    }
     if (arg === "--timeout") {
       if (i + 1 < args.length) {
         result.timeout = parseInt(args[i + 1], 10);
@@ -943,6 +1048,12 @@ function parseCliArgs(args) {
     if (arg === "--verbose") {
       result.verbose = true;
     }
+    if (arg === "--chrome-path") {
+      if (i + 1 < args.length) {
+        result.chromePath = args[i + 1];
+        i++;
+      }
+    }
   }
   return result;
 }
@@ -965,11 +1076,33 @@ async function runCli(args) {
   if (parsed.verbose) {
     options.verbose = true;
   }
+  if (parsed.chromePath) {
+    options.chromePath = parsed.chromePath;
+  }
   const agent = new YiyanAgent(options);
+  if (parsed.command === "config") {
+    const status = agent.status();
+    const configInfo = {
+      ...status,
+      savedChromePath: parsed.chromePath || void 0
+    };
+    console.log(formatCliOutput({
+      success: true,
+      question: "config",
+      answer: JSON.stringify(configInfo, null, 2),
+      duration: 0
+    }));
+    return;
+  }
   if (parsed.command === "login") {
-    console.log("Opening browser for login... Please login to Yiyan (\u6587\u5FC3\u4E00\u8A00) in the browser window.");
+    if (parsed.verbose) {
+      console.log("Opening browser for login... Please login to Yiyan (\u6587\u5FC3\u4E00\u8A00) in the browser window.");
+    }
     try {
       await agent.login();
+      if (parsed.chromePath) {
+        agent.saveChromePath(parsed.chromePath);
+      }
       console.log(formatCliOutput({
         success: true,
         question: "login",