1import { execSync, spawnSync } from 'node:child_process';
2import path from 'node:path';
3import process from 'node:process';
4import { fileURLToPath } from 'node:url';
5 
6import {
7  CdpConnection,
8  findChromeExecutable as findChromeExecutableBase,
9  findExistingChromeDebugPort as findExistingChromeDebugPortBase,
10  getFreePort as getFreePortBase,
11  launchChrome as launchChromeBase,
12  resolveSharedChromeProfileDir,
13  sleep,
14  waitForChromeDebugPort,
15  type PlatformCandidates,
16} from 'baoyu-chrome-cdp';
17 
18export { CdpConnection, sleep, waitForChromeDebugPort };
19 
20export const CHROME_CANDIDATES: PlatformCandidates = {
21  darwin: [
22    '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
23    '/Applications/Google Chrome Canary.app/Contents/MacOS/Google Chrome Canary',
24    '/Applications/Chromium.app/Contents/MacOS/Chromium',
25  ],
26  win32: [
27    'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',
28    'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe',
29  ],
30  default: [
31    '/usr/bin/google-chrome',
32    '/usr/bin/chromium',
33    '/usr/bin/chromium-browser',
34  ],
35};
36 
37let wslHome: string | null | undefined;
38function getWslWindowsHome(): string | null {
39  if (wslHome !== undefined) return wslHome;
40  if (!process.env.WSL_DISTRO_NAME) {
41    wslHome = null;
42    return null;
43  }
44  try {
45    const raw = execSync('cmd.exe /C "echo %USERPROFILE%"', {
46      encoding: 'utf-8',
47      timeout: 5_000,
48    }).trim().replace(/\r/g, '');
49    wslHome = execSync(`wslpath -u "${raw}"`, {
50      encoding: 'utf-8',
51      timeout: 5_000,
52    }).trim() || null;
53  } catch {
54    wslHome = null;
55  }
56  return wslHome;
57}
58 
59export function findChromeExecutable(chromePathOverride?: string): string | undefined {
60  if (chromePathOverride?.trim()) return chromePathOverride.trim();
61  return findChromeExecutableBase({
62    candidates: CHROME_CANDIDATES,
63    envNames: ['WEIBO_BROWSER_CHROME_PATH'],
64  });
65}
66 
67export async function findExistingChromeDebugPort(profileDir: string): Promise<number | null> {
68  return await findExistingChromeDebugPortBase({ profileDir });
69}
70 
71export function killChromeByProfile(profileDir: string): void {
72  try {
73    const result = spawnSync('ps', ['aux'], { encoding: 'utf-8', timeout: 5_000 });
74    if (result.status !== 0 || !result.stdout) return;
75    for (const line of result.stdout.split('\n')) {
76      if (!line.includes(profileDir) || !line.includes('--remote-debugging-port=')) continue;
77      const pid = line.trim().split(/\s+/)[1];
78      if (pid) {
79        try {
80          process.kill(Number(pid), 'SIGTERM');
81        } catch {}
82      }
83    }
84  } catch {}
85}
86 
87export function getDefaultProfileDir(): string {
88  return resolveSharedChromeProfileDir({
89    envNames: ['BAOYU_CHROME_PROFILE_DIR', 'WEIBO_BROWSER_PROFILE_DIR'],
90    wslWindowsHome: getWslWindowsHome(),
91  });
92}
93 
94export async function getFreePort(): Promise<number> {
95  return await getFreePortBase('WEIBO_BROWSER_DEBUG_PORT');
96}
97 
98export async function launchChrome(url: string, profileDir: string, chromePathOverride?: string): Promise<number> {
99  const chromePath = findChromeExecutable(chromePathOverride);
100  if (!chromePath) throw new Error('Chrome not found. Set WEIBO_BROWSER_CHROME_PATH env var.');
101 
102  const port = await getFreePort();
103  console.log(`[weibo-cdp] Launching Chrome (profile: ${profileDir})`);
104  await launchChromeBase({
105    chromePath,
106    profileDir,
107    port,
108    url,
109    extraArgs: ['--disable-blink-features=AutomationControlled', '--start-maximized'],
110  });
111  return port;
112}
113 
114export function getScriptDir(): string {
115  return path.dirname(fileURLToPath(import.meta.url));
116}
117 
118function runBunScript(scriptPath: string, args: string[]): boolean {
119  const result = spawnSync('npx', ['-y', 'bun', scriptPath, ...args], { stdio: 'inherit' });
120  return result.status === 0;
121}
122 
123export function copyImageToClipboard(imagePath: string): boolean {
124  const copyScript = path.join(getScriptDir(), 'copy-to-clipboard.ts');
125  return runBunScript(copyScript, ['image', imagePath]);
126}
127 
128export function copyHtmlToClipboard(htmlPath: string): boolean {
129  const copyScript = path.join(getScriptDir(), 'copy-to-clipboard.ts');
130  return runBunScript(copyScript, ['html', '--file', htmlPath]);
131}
132 
133export function pasteFromClipboard(targetApp?: string, retries = 3, delayMs = 500): boolean {
134  const pasteScript = path.join(getScriptDir(), 'paste-from-clipboard.ts');
135  const args = ['--retries', String(retries), '--delay', String(delayMs)];
136  if (targetApp) args.push('--app', targetApp);
137  return runBunScript(pasteScript, args);
138}
139