设计完成30%

2026-07-14 07:01:32 +08:00
parent 355f47976a
commit 1015e60ff7
95 changed files with 1991 additions and 349 deletions
@@ -0,0 +1,91 @@
+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)
+})