1import fs from "node:fs";
2import os from "node:os";
3import path from "node:path";
4import process from "node:process";
5import { spawnSync } from "node:child_process";
6 
7export interface ResolveSharedChromeProfileDirOptions {
8  envNames?: string[];
9  appDataDirName?: string;
10  profileDirName?: string;
11}
12 
13export interface FindExistingChromeDebugPortOptions {
14  profileDir: string;
15  timeoutMs?: number;
16}
17 
18interface ChromeVersionResponse {
19  webSocketDebuggerUrl?: string;
20}
21 
22const CHROME_LOCK_FILE_NAMES = ["SingletonLock", "SingletonSocket", "SingletonCookie", "chrome.pid"] as const;
23 
24function resolveDataBaseDir(): string {
25  if (process.platform === "darwin") {
26    return path.join(os.homedir(), "Library", "Application Support");
27  }
28  if (process.platform === "win32") {
29    return process.env.APPDATA ?? path.join(os.homedir(), "AppData", "Roaming");
30  }
31  return process.env.XDG_DATA_HOME ?? path.join(os.homedir(), ".local", "share");
32}
33 
34export function resolveSharedChromeProfileDir(
35  options: ResolveSharedChromeProfileDirOptions = {},
36): string {
37  for (const envName of options.envNames ?? []) {
38    const override = process.env[envName]?.trim();
39    if (override) {
40      return path.resolve(override);
41    }
42  }
43 
44  const appDataDirName = options.appDataDirName ?? "baoyu-skills";
45  const profileDirName = options.profileDirName ?? "chrome-profile";
46  return path.join(resolveDataBaseDir(), appDataDirName, profileDirName);
47}
48 
49export function resolveChromeProfileDir(profileDir?: string): string {
50  if (profileDir?.trim()) {
51    return path.resolve(profileDir.trim());
52  }
53 
54  return resolveSharedChromeProfileDir({
55    envNames: ["BAOYU_CHROME_PROFILE_DIR"],
56    appDataDirName: "baoyu-skills",
57    profileDirName: "chrome-profile",
58  });
59}
60 
61export function ensureChromeProfileDir(profileDir: string): string {
62  fs.mkdirSync(profileDir, { recursive: true });
63  return profileDir;
64}
65 
66export function hasChromeLockArtifacts(entries: readonly string[]): boolean {
67  return CHROME_LOCK_FILE_NAMES.some((name) => entries.includes(name));
68}
69 
70export function shouldRetryChromeLaunchRecovery(options: {
71  hasLockArtifacts: boolean;
72  hasLiveOwner: boolean;
73}): boolean {
74  return options.hasLockArtifacts && !options.hasLiveOwner;
75}
76 
77export function findChromeProcessUsingProfile(profileDir: string): boolean {
78  if (process.platform === "win32") {
79    return false;
80  }
81 
82  try {
83    const result = spawnSync("ps", ["aux"], {
84      encoding: "utf8",
85      timeout: 5_000,
86    });
87    if (result.status !== 0 || !result.stdout) {
88      return false;
89    }
90 
91    return result.stdout
92      .split("\n")
93      .some((line) => line.includes(`--user-data-dir=${profileDir}`));
94  } catch {
95    return false;
96  }
97}
98 
99export function cleanChromeLockArtifacts(profileDir: string): void {
100  for (const name of CHROME_LOCK_FILE_NAMES) {
101    try {
102      fs.unlinkSync(path.join(profileDir, name));
103    } catch {
104      // Ignore missing files and continue cleaning the remaining artifacts.
105    }
106  }
107}
108 
109export async function listChromeProfileEntries(profileDir: string): Promise<string[]> {
110  try {
111    return await fs.promises.readdir(profileDir);
112  } catch {
113    return [];
114  }
115}
116 
117async function fetchWithTimeout(url: string, timeoutMs = 3_000): Promise<Response> {
118  const controller = new AbortController();
119  const timer = setTimeout(() => controller.abort(), timeoutMs);
120  try {
121    return await fetch(url, {
122      redirect: "follow",
123      signal: controller.signal,
124    });
125  } finally {
126    clearTimeout(timer);
127  }
128}
129 
130async function fetchJson<T>(url: string, timeoutMs = 3_000): Promise<T> {
131  const response = await fetchWithTimeout(url, timeoutMs);
132  if (!response.ok) {
133    throw new Error(`Request failed: ${response.status} ${response.statusText}`);
134  }
135  return (await response.json()) as T;
136}
137 
138async function isDebugPortReady(port: number, timeoutMs = 3_000): Promise<boolean> {
139  try {
140    const version = await fetchJson<ChromeVersionResponse>(`http://127.0.0.1:${port}/json/version`, timeoutMs);
141    return Boolean(version.webSocketDebuggerUrl);
142  } catch {
143    return false;
144  }
145}
146 
147function parseDevToolsActivePort(filePath: string): { port: number; wsPath: string } | null {
148  try {
149    const content = fs.readFileSync(filePath, "utf8");
150    const lines = content.split(/\r?\n/);
151    const port = Number.parseInt(lines[0]?.trim() ?? "", 10);
152    const wsPath = lines[1]?.trim() ?? "";
153    if (port > 0 && wsPath) {
154      return { port, wsPath };
155    }
156  } catch {
157    // Ignore and fall back to process inspection.
158  }
159 
160  return null;
161}
162 
163export async function findExistingChromeDebugPort(
164  options: FindExistingChromeDebugPortOptions,
165): Promise<number | null> {
166  const timeoutMs = options.timeoutMs ?? 3_000;
167  const activePort = parseDevToolsActivePort(path.join(options.profileDir, "DevToolsActivePort"));
168  if (activePort && await isDebugPortReady(activePort.port, timeoutMs)) {
169    return activePort.port;
170  }
171 
172  if (process.platform === "win32") {
173    return null;
174  }
175 
176  try {
177    const result = spawnSync("ps", ["aux"], {
178      encoding: "utf8",
179      timeout: 5_000,
180    });
181    if (result.status !== 0 || !result.stdout) {
182      return null;
183    }
184 
185    const lines = result.stdout
186      .split("\n")
187      .filter((line) => line.includes(options.profileDir) && line.includes("--remote-debugging-port="));
188 
189    for (const line of lines) {
190      const match = line.match(/--remote-debugging-port=(\d+)/);
191      const port = Number.parseInt(match?.[1] ?? "", 10);
192      if (port > 0 && await isDebugPortReady(port, timeoutMs)) {
193        return port;
194      }
195    }
196  } catch {
197    // Ignore and report no reusable debugger.
198  }
199 
200  return null;
201}
202