92 lines
3.0 KiB
JavaScript
92 lines
3.0 KiB
JavaScript
const fs = require('fs')
|
|
const path = require('path')
|
|
|
|
const [url, selector, output, width = '360', height = '800', action] = process.argv.slice(2)
|
|
|
|
if (!url || !selector || !output) {
|
|
throw new Error('Usage: node scripts/capture-chrome-page.js <url> <selector> <output> [width] [height] [sms-tab]')
|
|
}
|
|
|
|
const sleep = (milliseconds) => new Promise((resolve) => setTimeout(resolve, milliseconds))
|
|
|
|
const connect = async () => {
|
|
const pages = await (await fetch('http://127.0.0.1:9222/json/list')).json()
|
|
const page = pages.find((item) => item.type === 'page' && item.url.includes('localhost:5173'))
|
|
if (!page) throw new Error('Chrome debugging has no localhost:5173 page')
|
|
|
|
const socket = new WebSocket(page.webSocketDebuggerUrl)
|
|
await new Promise((resolve, reject) => {
|
|
socket.addEventListener('open', resolve, { once: true })
|
|
socket.addEventListener('error', reject, { once: true })
|
|
})
|
|
|
|
let id = 0
|
|
const pending = new Map()
|
|
socket.addEventListener('message', (event) => {
|
|
const message = JSON.parse(event.data)
|
|
const request = pending.get(message.id)
|
|
if (!request) return
|
|
pending.delete(message.id)
|
|
if (message.error) request.reject(new Error(message.error.message))
|
|
else request.resolve(message.result)
|
|
})
|
|
|
|
const send = (method, params = {}) => new Promise((resolve, reject) => {
|
|
id += 1
|
|
pending.set(id, { resolve, reject })
|
|
socket.send(JSON.stringify({ id, method, params }))
|
|
})
|
|
|
|
return { socket, send }
|
|
}
|
|
|
|
const capture = async () => {
|
|
const { socket, send } = await connect()
|
|
const viewportWidth = Number(width)
|
|
const viewportHeight = Number(height)
|
|
|
|
try {
|
|
await send('Page.enable')
|
|
await send('Runtime.enable')
|
|
await send('Emulation.setDeviceMetricsOverride', {
|
|
width: viewportWidth,
|
|
height: viewportHeight,
|
|
deviceScaleFactor: 1,
|
|
mobile: true,
|
|
screenWidth: viewportWidth,
|
|
screenHeight: viewportHeight
|
|
})
|
|
await send('Page.navigate', { url })
|
|
|
|
for (let attempt = 0; attempt < 30; attempt += 1) {
|
|
const result = await send('Runtime.evaluate', {
|
|
expression: `Boolean(document.querySelector(${JSON.stringify(selector)}))`,
|
|
returnByValue: true
|
|
})
|
|
if (result.result?.value) break
|
|
if (attempt === 29) throw new Error(`Expected page selector did not render: ${selector}`)
|
|
await sleep(100)
|
|
}
|
|
|
|
if (action === 'sms-tab') {
|
|
await send('Runtime.evaluate', {
|
|
expression: "document.querySelectorAll('.login-tab')[1]?.click()"
|
|
})
|
|
}
|
|
|
|
await sleep(300)
|
|
const screenshot = await send('Page.captureScreenshot', { format: 'png', fromSurface: true, captureBeyondViewport: false })
|
|
fs.mkdirSync(path.dirname(output), { recursive: true })
|
|
fs.writeFileSync(output, Buffer.from(screenshot.data, 'base64'))
|
|
process.stdout.write(`CAPTURED ${output}\n`)
|
|
} finally {
|
|
try { await send('Emulation.clearDeviceMetricsOverride') } catch (_) {}
|
|
socket.close()
|
|
}
|
|
}
|
|
|
|
capture().catch((error) => {
|
|
process.stderr.write(`${error.stack || error.message}\n`)
|
|
process.exit(1)
|
|
})
|