45 lines
2.3 KiB
JavaScript
45 lines
2.3 KiB
JavaScript
|
// 浅色模式截图验证（切换 data-theme 后截图）
|
||
|
|
import { spawn } from 'node:child_process';
|
||
|
|
import { mkdtempSync, writeFileSync } from 'node:fs';
|
||
|
|
import { tmpdir } from 'node:os';
|
||
|
|
import { join } from 'node:path';
|
||
|
|
|
||
|
|
const CHROME = 'C:/Users/QunLIn/.agent-browser/browsers/chrome-154.0.8037.92/chrome.exe';
|
||
|
|
const PORT = 9406;
|
||
|
|
const URL_ = 'http://127.0.0.1:1313/comment.html';
|
||
|
|
const OUT = process.argv[2] || 'light.png';
|
||
|
|
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
|
||
|
|
const userDir = mkdtempSync(join(tmpdir(), 'cdp-light-'));
|
||
|
|
|
||
|
|
const chrome = spawn(CHROME, [
|
||
|
|
'--headless=new', `--remote-debugging-port=${PORT}`, `--user-data-dir=${userDir}`,
|
||
|
|
'--no-first-run', '--disable-gpu', '--window-size=1280,1400', 'about:blank',
|
||
|
|
], { stdio: 'ignore' });
|
||
|
|
|
||
|
|
try {
|
||
|
|
for (let i = 0; i < 40; i++) {
|
||
|
|
try { const r = await fetch(`http://127.0.0.1:${PORT}/json/version`); if (r.ok) break; } catch { /* */ }
|
||
|
|
await sleep(300);
|
||
|
|
}
|
||
|
|
const tab = await (await fetch(`http://127.0.0.1:${PORT}/json/new?${encodeURIComponent(URL_)}`, { method: 'PUT' })).json();
|
||
|
|
const ws = new WebSocket(tab.webSocketDebuggerUrl);
|
||
|
|
let id = 0; const pending = new Map();
|
||
|
|
ws.addEventListener('message', (ev) => { const m = JSON.parse(ev.data); if (m.id && pending.has(m.id)) { pending.get(m.id)(m); pending.delete(m.id); } });
|
||
|
|
await new Promise((res) => ws.addEventListener('open', res));
|
||
|
|
const send = (method, params = {}) => new Promise((res) => { const mid = ++id; pending.set(mid, res); ws.send(JSON.stringify({ id: mid, method, params })); });
|
||
|
|
const ev = async (expr) => {
|
||
|
|
const r = await send('Runtime.evaluate', { expression: expr, returnByValue: true, awaitPromise: true });
|
||
|
|
return r.result?.result?.value ?? r.result?.value ?? null;
|
||
|
|
};
|
||
|
|
await sleep(9000);
|
||
|
|
await ev(`document.documentElement.setAttribute('data-theme','light')`);
|
||
|
|
await sleep(800);
|
||
|
|
const s = await send('Page.captureScreenshot', { format: 'png' });
|
||
|
|
writeFileSync(OUT, Buffer.from(s.result.data, 'base64'));
|
||
|
|
console.log('btn:', await ev(`JSON.stringify((() => { const b = document.querySelector('.atk-send-btn'); const s = getComputedStyle(b); return { text: b.textContent, color: s.color, bg: s.backgroundColor, opacity: getComputedStyle(b.closest('.atk-main-editor')).opacity }; })())`));
|
||
|
|
ws.close();
|
||
|
|
} catch (e) {
|
||
|
|
console.error('ERROR', e.message);
|
||
|
|
} finally {
|
||
|
|
chrome.kill();
|
||
|
|
}
|