1import { execSync, spawnSync } from 'node:child_process';
2import fs from 'node:fs';
3import path from 'node:path';
4import process from 'node:process';
5import { fileURLToPath } from 'node:url';
6 
7import {
8  CdpConnection,
9  findChromeExecutable as findChromeExecutableBase,
10  findExistingChromeDebugPort as findExistingChromeDebugPortBase,
11  getFreePort as getFreePortBase,
12  gracefulKillChrome,
13  killChrome,
14  launchChrome as launchChromeBase,
15  openPageSession,
16  resolveSharedChromeProfileDir,
17  sleep,
18  waitForChromeDebugPort,
19  type PlatformCandidates,
20} from 'baoyu-chrome-cdp';
21 
22export { CdpConnection, gracefulKillChrome, killChrome, openPageSession, sleep, waitForChromeDebugPort };
23export type { PlatformCandidates } from 'baoyu-chrome-cdp';
24 
25const X_SESSION_URLS = ['https://x.com/', 'https://twitter.com/'] as const;
26const REQUIRED_X_SESSION_COOKIES = ['auth_token', 'ct0'] as const;
27 
28interface CookieLike {
29  name?: string;
30  value?: string | null;
31}
32 
33interface NetworkGetCookiesResult {
34  cookies?: CookieLike[];
35}
36 
37export const CHROME_CANDIDATES_BASIC: PlatformCandidates = {
38  darwin: [
39    '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
40    '/Applications/Google Chrome Canary.app/Contents/MacOS/Google Chrome Canary',
41    '/Applications/Chromium.app/Contents/MacOS/Chromium',
42  ],
43  win32: [
44    'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',
45    'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe',
46  ],
47  default: [
48    '/usr/bin/google-chrome',
49    '/usr/bin/chromium',
50    '/usr/bin/chromium-browser',
51  ],
52};
53 
54export const CHROME_CANDIDATES_FULL: PlatformCandidates = {
55  darwin: [
56    '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
57    '/Applications/Google Chrome Canary.app/Contents/MacOS/Google Chrome Canary',
58    '/Applications/Google Chrome Beta.app/Contents/MacOS/Google Chrome Beta',
59    '/Applications/Chromium.app/Contents/MacOS/Chromium',
60    '/Applications/Microsoft Edge.app/Contents/MacOS/Microsoft Edge',
61  ],
62  win32: [
63    'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',
64    'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe',
65    'C:\\Program Files\\Microsoft\\Edge\\Application\\msedge.exe',
66    'C:\\Program Files (x86)\\Microsoft\\Edge\\Application\\msedge.exe',
67  ],
68  default: [
69    '/usr/bin/google-chrome',
70    '/usr/bin/google-chrome-stable',
71    '/usr/bin/chromium',
72    '/usr/bin/chromium-browser',
73    '/snap/bin/chromium',
74    '/usr/bin/microsoft-edge',
75  ],
76};
77 
78export function findChromeExecutable(candidates: PlatformCandidates): string | undefined {
79  return findChromeExecutableBase({
80    candidates,
81    envNames: ['X_BROWSER_CHROME_PATH'],
82  });
83}
84 
85let _wslHome: string | null | undefined;
86function getWslWindowsHome(): string | null {
87  if (_wslHome !== undefined) return _wslHome;
88  if (!process.env.WSL_DISTRO_NAME) { _wslHome = null; return null; }
89  try {
90    const raw = execSync('cmd.exe /C "echo %USERPROFILE%"', { encoding: 'utf-8', timeout: 5000 }).trim().replace(/\r/g, '');
91    _wslHome = execSync(`wslpath -u "${raw}"`, { encoding: 'utf-8', timeout: 5000 }).trim() || null;
92  } catch { _wslHome = null; }
93  return _wslHome;
94}
95 
96export function getDefaultProfileDir(): string {
97  return resolveSharedChromeProfileDir({
98    envNames: ['BAOYU_CHROME_PROFILE_DIR', 'X_BROWSER_PROFILE_DIR'],
99    wslWindowsHome: getWslWindowsHome(),
100  });
101}
102 
103export async function getFreePort(): Promise<number> {
104  return await getFreePortBase('X_BROWSER_DEBUG_PORT');
105}
106 
107export async function findExistingChromeDebugPort(profileDir: string): Promise<number | null> {
108  return await findExistingChromeDebugPortBase({ profileDir });
109}
110 
111const CHROME_LOCK_FILES = ['SingletonLock', 'SingletonSocket', 'SingletonCookie', 'chrome.pid'] as const;
112 
113export function hasChromeLockArtifacts(entries: readonly string[]): boolean {
114  return CHROME_LOCK_FILES.some((name) => entries.includes(name));
115}
116 
117export function shouldRetryChromeLaunch(options: {
118  lockArtifactsPresent: boolean;
119  hasLiveOwner: boolean;
120}): boolean {
121  return options.lockArtifactsPresent && !options.hasLiveOwner;
122}
123 
124export function buildXSessionCookieMap(cookies: readonly CookieLike[]): Record<string, string> {
125  const cookieMap: Record<string, string> = {};
126  for (const cookie of cookies) {
127    const name = cookie.name?.trim();
128    const value = cookie.value?.trim();
129    if (!name || !value) {
130      continue;
131    }
132    cookieMap[name] = value;
133  }
134  return cookieMap;
135}
136 
137export function hasRequiredXSessionCookies(cookieMap: Record<string, string>): boolean {
138  return REQUIRED_X_SESSION_COOKIES.every((name) => Boolean(cookieMap[name]));
139}
140 
141export async function readXSessionCookieMap(
142  cdp: CdpConnection,
143  sessionId?: string,
144): Promise<Record<string, string>> {
145  const { cookies } = await cdp.send<NetworkGetCookiesResult>(
146    'Network.getCookies',
147    { urls: [...X_SESSION_URLS] },
148    {
149      sessionId,
150      timeoutMs: 5_000,
151    },
152  );
153  return buildXSessionCookieMap(cookies ?? []);
154}
155 
156export async function waitForXSessionPersistence(options: {
157  cdp: CdpConnection;
158  sessionId?: string;
159  timeoutMs?: number;
160  pollIntervalMs?: number;
161}): Promise<boolean> {
162  const timeoutMs = options.timeoutMs ?? 15_000;
163  const pollIntervalMs = options.pollIntervalMs ?? 1_000;
164  const start = Date.now();
165 
166  while (Date.now() - start < timeoutMs) {
167    const cookieMap = await readXSessionCookieMap(options.cdp, options.sessionId).catch(() => ({}));
168    if (hasRequiredXSessionCookies(cookieMap)) {
169      return true;
170    }
171    await sleep(pollIntervalMs);
172  }
173 
174  return false;
175}
176 
177function cleanStaleLockFiles(profileDir: string): void {
178  for (const name of CHROME_LOCK_FILES) {
179    try { fs.unlinkSync(path.join(profileDir, name)); } catch {}
180  }
181}
182 
183function hasLiveChromeOwner(profileDir: string): boolean {
184  if (process.platform === 'win32') return false;
185  try {
186    const result = spawnSync('ps', ['aux'], {
187      encoding: 'utf8',
188      timeout: 5000,
189    });
190    if (result.status !== 0 || !result.stdout) return false;
191    return result.stdout.split('\n').some((line) => line.includes(`--user-data-dir=${profileDir}`));
192  } catch {
193    return false;
194  }
195}
196 
197async function listProfileDirEntries(profileDir: string): Promise<string[]> {
198  try {
199    return await fs.promises.readdir(profileDir);
200  } catch {
201    return [];
202  }
203}
204 
205async function launchChromeOnce(
206  url: string,
207  profileDir: string,
208  chromePath: string,
209): Promise<{ chrome: Awaited<ReturnType<typeof launchChromeBase>>; port: number }> {
210  const port = await getFreePort();
211  const chrome = await launchChromeBase({
212    chromePath,
213    profileDir,
214    port,
215    url,
216    extraArgs: ['--start-maximized'],
217  });
218 
219  try {
220    await waitForChromeDebugPort(port, 30_000, { includeLastError: true });
221    return { chrome, port };
222  } catch (error) {
223    killChrome(chrome);
224    throw error;
225  }
226}
227 
228export async function launchChrome(
229  url: string,
230  profileDir: string,
231  candidates: PlatformCandidates,
232  chromePathOverride?: string,
233): Promise<{ chrome: Awaited<ReturnType<typeof launchChromeBase>>; port: number }> {
234  const chromePath = chromePathOverride?.trim() || findChromeExecutable(candidates);
235  if (!chromePath) throw new Error('Chrome not found. Set X_BROWSER_CHROME_PATH env var.');
236 
237  try {
238    return await launchChromeOnce(url, profileDir, chromePath);
239  } catch (error) {
240    const entries = await listProfileDirEntries(profileDir);
241    const shouldRetry = shouldRetryChromeLaunch({
242      lockArtifactsPresent: hasChromeLockArtifacts(entries),
243      hasLiveOwner: hasLiveChromeOwner(profileDir),
244    });
245    if (!shouldRetry) throw error;
246 
247    cleanStaleLockFiles(profileDir);
248    return await launchChromeOnce(url, profileDir, chromePath);
249  }
250}
251 
252export function getScriptDir(): string {
253  return path.dirname(fileURLToPath(import.meta.url));
254}
255 
256function runBunScript(scriptPath: string, args: string[]): boolean {
257  const result = spawnSync('npx', ['-y', 'bun', scriptPath, ...args], { stdio: 'inherit' });
258  return result.status === 0;
259}
260 
261export function copyImageToClipboard(imagePath: string): boolean {
262  const copyScript = path.join(getScriptDir(), 'copy-to-clipboard.ts');
263  return runBunScript(copyScript, ['image', imagePath]);
264}
265 
266export function copyHtmlToClipboard(htmlPath: string): boolean {
267  const copyScript = path.join(getScriptDir(), 'copy-to-clipboard.ts');
268  return runBunScript(copyScript, ['html', '--file', htmlPath]);
269}
270 
271export function pasteFromClipboard(targetApp?: string, retries = 3, delayMs = 500): boolean {
272  const pasteScript = path.join(getScriptDir(), 'paste-from-clipboard.ts');
273  const args = ['--retries', String(retries), '--delay', String(delayMs)];
274  if (targetApp) args.push('--app', targetApp);
275  return runBunScript(pasteScript, args);
276}
277