6 lines
6.9 KiB
JavaScript
6 lines
6.9 KiB
JavaScript
import crypto from "node:crypto"; import fs from "node:fs"; import http from "node:http"; import os from "node:os"; import path from "node:path"; import { spawn } from "node:child_process";
|
|
const root = path.resolve(path.dirname(new URL(import.meta.url).pathname), ".."); const dist = path.join(root, "web/dist"); const docs = path.join(root, "docs/p6"); const chromePath = ["/usr/bin/google-chrome", "/usr/bin/google-chrome-stable", "/opt/google/chrome/google-chrome"].find(fs.existsSync); const viewport = { width: 1440, height: 900, deviceScaleFactor: 1 }; const wait = ms => new Promise(resolve => setTimeout(resolve, ms)); const hash = value => crypto.createHash("sha256").update(value).digest("hex"); if (!chromePath) throw new Error("Google Chrome unavailable");
|
|
const server = http.createServer((req, res) => { const relative = decodeURIComponent(new URL(req.url ?? "/", "http://127.0.0.1").pathname).replace(/^\/+/, "") || "reverseengineering-python.html"; const file = path.resolve(dist, relative); if (file !== dist && !file.startsWith(`${dist}${path.sep}`)) { res.writeHead(403); return res.end(); } if (!fs.existsSync(file)) { res.writeHead(404); return res.end(); } const type = { ".html": "text/html; charset=utf-8", ".js": "text/javascript; charset=utf-8", ".wasm": "application/wasm" }[path.extname(file)] ?? "application/octet-stream"; res.writeHead(200, { "content-type": type, "cache-control": "no-store" }); fs.createReadStream(file).pipe(res); }); await new Promise((resolve, reject) => { server.once("error", reject); server.listen(0, "127.0.0.1", resolve); }); const origin = `http://127.0.0.1:${server.address().port}`; const profile = fs.mkdtempSync(path.join(os.tmpdir(), "freecad-p6-reverseengineering-python-chrome-")); const chrome = spawn(chromePath, ["--headless=new", "--no-sandbox", "--disable-gpu", "--disable-dev-shm-usage", "--remote-debugging-port=0", "--remote-allow-origins=*", `--user-data-dir=${profile}`, `--window-size=${viewport.width},${viewport.height}`, `${origin}/reverseengineering-python.html`], { stdio: ["ignore", "pipe", "pipe"] }); let output = ""; chrome.stdout.on("data", chunk => { output += chunk; }); chrome.stderr.on("data", chunk => { output += chunk; });
|
|
class Cdp { constructor(url) { this.socket = new WebSocket(url); this.id = 0; this.pending = new Map(); this.listeners = new Map(); this.ready = new Promise((resolve, reject) => { this.socket.addEventListener("open", resolve, { once: true }); this.socket.addEventListener("error", reject, { once: true }); }); this.socket.addEventListener("message", event => { const message = JSON.parse(event.data); if (message.id) { const pending = this.pending.get(message.id); if (pending) { this.pending.delete(message.id); message.error ? pending.reject(new Error(message.error.message)) : pending.resolve(message.result); } } else for (const listener of this.listeners.get(message.method) ?? []) listener(message.params); }); } async send(method, params = {}) { await this.ready; return new Promise((resolve, reject) => { const id = ++this.id; this.pending.set(id, { resolve, reject }); this.socket.send(JSON.stringify({ id, method, params })); }); } on(method, listener) { this.listeners.set(method, [...(this.listeners.get(method) ?? []), listener]); } close() { this.socket.close(); } }
|
|
let cdp; try { let target; for (let i = 0; i < 300 && !target; i++) { const match = output.match(/DevTools listening on ws:\/\/([^\s]+)/); if (match) { const port = match[1].split(":")[1].split("/")[0]; try { target = (await fetch(`http://127.0.0.1:${port}/json/list`).then(response => response.json())).find(item => item.type === "page"); } catch {} } if (!target) await wait(100); } if (!target?.webSocketDebuggerUrl) throw new Error("Chrome CDP target unavailable"); cdp = new Cdp(target.webSocketDebuggerUrl); const requests = []; const exceptions = []; cdp.on("Network.requestWillBeSent", event => requests.push(event.request.url)); cdp.on("Runtime.exceptionThrown", event => exceptions.push(event.exceptionDetails)); await cdp.send("Runtime.enable"); await cdp.send("Network.enable"); await cdp.send("Emulation.setDeviceMetricsOverride", { ...viewport, mobile: false }); const evaluate = async expression => (await cdp.send("Runtime.evaluate", { expression, returnByValue: true, awaitPromise: true })).result?.value; const state = () => evaluate("window.__freecadReverseEngineeringPythonState"); const stage = async wanted => { for (let i = 0; i < 300; i++) { const value = await state(); if (value?.stage === wanted) return value; await wait(100); } throw new Error(`${wanted} timed out`); }; const ready = await stage("ready"); await evaluate("document.querySelector('#run').click()"); const success = await stage("success"); await evaluate("document.querySelector('#error').click()"); const expectedError = await stage("expected-error"); await evaluate("document.querySelector('#reset').click()"); const reset = await stage("reset"); const screenshot = Buffer.from((await cdp.send("Page.captureScreenshot", { format: "png", fromSurface: true })).data, "base64"); fs.writeFileSync(path.join(docs, "reverseengineering-action-03-chrome.png"), screenshot); const externalRequests = requests.filter(url => !url.startsWith(origin)); const version = await new Promise(resolve => { const child = spawn(chromePath, ["--version"], { stdio: ["ignore", "pipe", "ignore"] }); let text = ""; child.stdout.on("data", chunk => { text += chunk; }); child.on("close", () => resolve(text.trim())); }); const result = success?.response?.kind === "result" && success.response.result?.apiCount === 2 && success.response.result?.curveValid === true && expectedError?.response?.kind === "error" && expectedError.response.error?.code === "revision_mismatch" && reset?.stage === "reset" && externalRequests.length === 0 && exceptions.length === 0 ? "pass" : "fail"; const report = { schema: "freecad-web-p6-reverseengineering-action-03-chrome/v1", taskId: "FW-P6-REV-ACT-03-F", module: "ReverseEngineering", action: "python-api", result, chrome: { path: chromePath, version }, mode: "headless-new", origin, viewport, cases: { ready, success, expectedError, reset }, network: { requests, externalRequests, externalRequestCount: externalRequests.length }, exceptions, policy: { browser: "Google Chrome only", nonChromeBrowsersAttempted: false }, evidence: { screenshot: "docs/p6/reverseengineering-action-03-chrome.png", screenshotSha256: hash(screenshot), resultHash: hash(JSON.stringify({ ready, success, expectedError, reset })) } }; fs.writeFileSync(path.join(docs, "reverseengineering-action-03-chrome.json"), `${JSON.stringify(report, null, 2)}\n`); fs.writeFileSync(path.join(root, "docs/freecad-p6-reverseengineering-action-03-f-chrome.log"), `result=${result}\nchrome=${version}\nexternalRequests=${externalRequests.length}\nexceptions=${exceptions.length}\n`); console.log(JSON.stringify({ result, chrome: version, externalRequests: externalRequests.length, exceptions: exceptions.length })); if (result !== "pass") process.exitCode = 1; } finally { cdp?.close(); chrome.kill("SIGTERM"); server.close(); }
|