1

server/services/puppeteer_runner.js

@@ -1,4 +1,5 @@
 import dotenv from 'dotenv';
+import fs from 'node:fs';
 import path from 'node:path';
 import puppeteer from 'puppeteer';
 
@@ -30,17 +31,64 @@ function get_extension_id_from_targets(targets) {
   return null;
 }
 
+async function wait_for_extension_id(browser, timeout_ms) {
+  const existing = get_extension_id_from_targets(browser.targets());
+  if (existing) {
+    return existing;
+  }
+
+  const target = await browser
+    .waitForTarget((t) => {
+      const url = t.url();
+      return typeof url === 'string' && url.startsWith('chrome-extension://');
+    }, { timeout: timeout_ms })
+    .catch(() => null);
+
+  if (!target) {
+    return null;
+  }
+
+  return get_extension_id_from_targets([target]);
+}
+
+function get_chrome_executable_path() {
+  // 优先环境变量，方便你后续切换版本
+  const from_env = process.env.CHROME_EXECUTABLE_PATH;
+  if (from_env) {
+    return path.resolve(from_env);
+  }
+
+  // 默认使用项目根目录的 chrome-win/chrome.exe
+  // 当前进程 cwd 通常是 server/，所以回到上一级
+  return path.resolve(process.cwd(), '../chrome-win/chrome.exe');
+}
+
 export async function get_or_create_browser() {
   if (browser_singleton) {
     return browser_singleton;
   }
 
-  const extension_path = path.resolve(get_crx_src_path());
+  const chrome_executable_path = get_chrome_executable_path();
+  if (!fs.existsSync(chrome_executable_path)) {
+    throw new Error(`Chrome 不存在: ${chrome_executable_path}`);
+  }
+
+  const raw_extension_path = path.resolve(get_crx_src_path());
+  const manifest_path = path.resolve(raw_extension_path, 'manifest.json');
+  if (!fs.existsSync(manifest_path)) {
+    throw new Error(`扩展 manifest.json 不存在: ${manifest_path}`);
+  }
+
+  const extension_path = raw_extension_path.replace(/\\/g, '/');
   const headless = String(process.env.PUPPETEER_HEADLESS || 'false') === 'true';
+  const user_data_dir = path.resolve(process.cwd(), 'puppeteer_profile');
 
   browser_singleton = await puppeteer.launch({
+    executablePath: chrome_executable_path,
     headless,
     args: [
+      `--user-data-dir=${user_data_dir}`,
+      '--enable-extensions',
       `--disable-extensions-except=${extension_path}`,
       `--load-extension=${extension_path}`,
       '--no-default-browser-check',
@@ -58,11 +106,19 @@ export async function invoke_extension_action(action_name, action_payload) {
   const page = await browser.newPage();
   await page.goto('about:blank');
 
-  const targets = await browser.targets();
-  const extension_id = get_extension_id_from_targets(targets);
+  // 尝试先打开 chrome://extensions 触发扩展初始化（某些环境下扩展 target 不会立刻出现）
+  try {
+    await page.goto('chrome://extensions/', { waitUntil: 'domcontentloaded' });
+  } catch (err) {
+    // ignore
+  }
+
+  const extension_id = await wait_for_extension_id(browser, 15000);
   if (!extension_id) {
     await page.close();
-    throw new Error('未找到扩展 extension_id（请确认 CRX_SRC_PATH 指向 src 且成功加载）');
+    throw new Error(
+      '未找到扩展 extension_id：Chrome 未加载扩展（常见原因：MV2 被禁用/企业策略未生效/CRX_SRC_PATH 不正确/使用了 headless）'
+    );
   }
 
   const bridge_url = `chrome-extension://${extension_id}/bridge/bridge.html`;