feat: reuse a signed-in Chrome profile for browser web search
Add CHROME_USER_DATA_DIR so the browser search fallback can open Chrome against a dedicated, Google-signed-in profile instead of a fresh anonymous session. A returning signed-in profile is what actually avoids Google's /sorry bot-detection page, so this is the reliable way to get browser Google search in plain text turns. Fallback order is now CDP (broadcast Chrome) -> persistent profile (when configured) -> ephemeral headless, all still fail-open to the DDG/Brave/Wikipedia cascade. Document the profile in .env.example and web_search.spec.md. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -7,63 +7,86 @@
|
||||
// - search : Google-search the query, return the top organic results.
|
||||
// - youtube : search YouTube and play the first result.
|
||||
//
|
||||
// When no broadcast Chrome is reachable on CDP (e.g. a plain text conversation
|
||||
// with no active broadcast), `search` falls back to a headless Chromium we
|
||||
// launch ourselves, so browser-based Google search still works at zero API
|
||||
// cost. `youtube` only makes sense on the visible broadcast Chrome, so it never
|
||||
// uses the headless fallback.
|
||||
// Backend selection for `search`:
|
||||
// 1. The broadcast Chrome over CDP (visible on the Go-Live stream).
|
||||
// 2. Else, if CHROME_USER_DATA_DIR is set, a persistent Chrome using that
|
||||
// profile dir. Logging that dedicated profile into Google once lets Google
|
||||
// treat later searches as a returning signed-in user, which avoids the
|
||||
// bot-detection interstitial that blocks a fresh anonymous session.
|
||||
// 3. Else a fresh ephemeral headless Chrome (works only where Google does not
|
||||
// challenge the session, e.g. a non-flagged residential IP).
|
||||
// `youtube` only makes sense on the visible broadcast Chrome, so it never uses
|
||||
// the headless/persistent fallback.
|
||||
import { chromium } from 'playwright';
|
||||
|
||||
const CDP = process.env.CDP_PORT || '9222';
|
||||
// Use 127.0.0.1, not "localhost": in containers localhost can resolve to IPv6
|
||||
// (::1) first while Chrome's CDP listens on IPv4, giving ECONNREFUSED ::1.
|
||||
const CDP_HOST = process.env.CDP_HOST || '127.0.0.1';
|
||||
const USER_DATA_DIR = process.env.CHROME_USER_DATA_DIR || '';
|
||||
const UA =
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ' +
|
||||
'(KHTML, like Gecko) Chrome/148.0.0.0 Safari/537.36';
|
||||
const query = process.argv[2] || '';
|
||||
const mode = (process.argv[3] || 'search').toLowerCase();
|
||||
const out = (o) => { process.stdout.write(JSON.stringify(o)); };
|
||||
|
||||
if (!query) { out({ ok: false, error: 'no query' }); process.exit(1); }
|
||||
|
||||
let browser; // playwright Browser: connected (CDP) or launched (headless)
|
||||
let launched = false; // true when we launched our own headless browser
|
||||
let connected; // CDP Browser (the broadcast Chrome — never kill it)
|
||||
let launchedBrowser; // ephemeral headless Browser we launched
|
||||
let persistent; // persistent BrowserContext we launched
|
||||
let launched = false;
|
||||
let page;
|
||||
|
||||
// Acquire a page from the broadcast Chrome over CDP, or — for search mode only —
|
||||
// from a headless Chromium we launch when no broadcast Chrome is reachable.
|
||||
// Try system Chrome (channel:'chrome') first so no extra Playwright browser
|
||||
// download is needed; fall back to Playwright's bundled chromium.
|
||||
async function tryLaunch(launchFn) {
|
||||
let err;
|
||||
for (const opts of [{ headless: true, channel: 'chrome' }, { headless: true }]) {
|
||||
try {
|
||||
return await launchFn(opts);
|
||||
} catch (e) {
|
||||
err = e;
|
||||
}
|
||||
}
|
||||
throw err;
|
||||
}
|
||||
|
||||
async function acquirePage() {
|
||||
// 1. Broadcast Chrome over CDP.
|
||||
try {
|
||||
browser = await chromium.connectOverCDP(`http://${CDP_HOST}:${CDP}`);
|
||||
const ctx = browser.contexts()[0];
|
||||
connected = await chromium.connectOverCDP(`http://${CDP_HOST}:${CDP}`);
|
||||
const ctx = connected.contexts()[0];
|
||||
page = ctx.pages()[0] || (await ctx.newPage());
|
||||
return;
|
||||
} catch (e) {
|
||||
// YouTube playback must be on the visible broadcast Chrome; no fallback.
|
||||
if (mode === 'youtube') throw e;
|
||||
if (mode === 'youtube') throw e; // youtube needs the visible broadcast Chrome
|
||||
}
|
||||
// Prefer the system Chrome the host already has (channel: 'chrome') so the
|
||||
// fallback needs no extra Playwright browser download; degrade to Playwright's
|
||||
// bundled chromium if no system Chrome is installed.
|
||||
|
||||
// 2. Persistent profile (signed-in) when configured.
|
||||
if (USER_DATA_DIR) {
|
||||
persistent = await tryLaunch((opts) =>
|
||||
chromium.launchPersistentContext(USER_DATA_DIR, { ...opts, locale: 'ko-KR', userAgent: UA }),
|
||||
);
|
||||
launched = true;
|
||||
page = persistent.pages()[0] || (await persistent.newPage());
|
||||
return;
|
||||
}
|
||||
|
||||
// 3. Ephemeral headless.
|
||||
launchedBrowser = await tryLaunch((opts) => chromium.launch(opts));
|
||||
launched = true;
|
||||
let launchErr;
|
||||
for (const opts of [{ headless: true, channel: 'chrome' }, { headless: true }]) {
|
||||
try {
|
||||
browser = await chromium.launch(opts);
|
||||
break;
|
||||
} catch (e) {
|
||||
launchErr = e;
|
||||
browser = undefined;
|
||||
}
|
||||
}
|
||||
if (!browser) throw launchErr;
|
||||
const ctx = await browser.newContext({
|
||||
locale: 'ko-KR',
|
||||
userAgent:
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ' +
|
||||
'(KHTML, like Gecko) Chrome/148.0.0.0 Safari/537.36',
|
||||
});
|
||||
const ctx = await launchedBrowser.newContext({ locale: 'ko-KR', userAgent: UA });
|
||||
page = await ctx.newPage();
|
||||
}
|
||||
|
||||
async function closeAll() {
|
||||
try { await persistent?.close(); } catch { /* ignore */ }
|
||||
try { await launchedBrowser?.close(); } catch { /* ignore */ }
|
||||
try { await connected?.close(); } catch { /* ignore */ }
|
||||
}
|
||||
|
||||
try {
|
||||
await acquirePage();
|
||||
page.setDefaultTimeout(20000);
|
||||
@@ -86,7 +109,7 @@ try {
|
||||
// and fail fast so the Python caller fail-opens to the DDG cascade instead of
|
||||
// treating an empty challenge page as "no results".
|
||||
if (page.url().includes('/sorry/')) {
|
||||
await browser.close();
|
||||
await closeAll();
|
||||
out({ ok: false, error: 'google-bot-challenge', headless: launched });
|
||||
process.exit(1);
|
||||
}
|
||||
@@ -109,9 +132,9 @@ try {
|
||||
});
|
||||
out({ ok: true, mode, query, count: results.length, results, headless: launched });
|
||||
}
|
||||
await browser.close();
|
||||
await closeAll();
|
||||
} catch (e) {
|
||||
try { await browser?.close(); } catch { /* ignore */ }
|
||||
await closeAll();
|
||||
out({ ok: false, error: String(e?.message || e) });
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user