ausboss
46d4ee39d6
The five CDP scripts looked for ComfyUI at http://127.0.0.1:8188/ only; COMFY_URL now overrides it. cdp_nav_debug.mjs says so when no ComfyUI tab is open instead of crashing, and cdp_drag.mjs gives up on a request after 25 s like cdp.mjs rather than waiting forever. The live-testing notes mention COMFY_URL, give the full path of the screenshot pairs, and say that only CDP_RELOAD=1 reloads into a fresh tab.
53 lines
3.5 KiB
JavaScript
53 lines
3.5 KiB
JavaScript
// cdp.mjs <port> <workflow.json|-> <out.png|-> [evalFile] [waitMs]
|
|
// Opens/reuses a ComfyUI tab on the headless Chrome at <port>, optionally
|
|
// loads a workflow JSON into the graph, evaluates a JS snippet file, and
|
|
// screenshots the page. Prints the eval result as JSON.
|
|
import fs from "node:fs";
|
|
const [port, wfPath, outPng, evalFile, waitArg] = process.argv.slice(2);
|
|
const waitMs = Number(waitArg ?? 1500);
|
|
const COMFY = new URL(process.env.COMFY_URL || "http://127.0.0.1:8188/").href;
|
|
const list = await (await fetch(`http://127.0.0.1:${port}/json`)).json();
|
|
let page = list.find((t) => t.type === "page" && t.url.startsWith(COMFY));
|
|
if (page && process.env.CDP_RELOAD === "1") {
|
|
// A fresh tab rather than an in-place navigation: after synthetic mouse
|
|
// input the old tab stops answering on navigate (see cdp_nav_debug.mjs).
|
|
await fetch(`http://127.0.0.1:${port}/json/close/${page.id}`);
|
|
page = null;
|
|
}
|
|
if (!page) page = await (await fetch(`http://127.0.0.1:${port}/json/new?about:blank`, { method: "PUT" })).json();
|
|
const ws = new WebSocket(page.webSocketDebuggerUrl);
|
|
await new Promise((r) => (ws.onopen = r));
|
|
let id = 0; const pending = new Map();
|
|
ws.onmessage = (m) => { const d = JSON.parse(m.data); if (d.id && pending.has(d.id)) { pending.get(d.id)(d); pending.delete(d.id); } else if (d.method === "Page.javascriptDialogOpening") { ws.send(JSON.stringify({ id: ++id, method: "Page.handleJavaScriptDialog", params: { accept: true } })); } };
|
|
const send = (method, params = {}) => new Promise((r, reject) => { const i = ++id; pending.set(i, r); ws.send(JSON.stringify({ id: i, method, params })); setTimeout(() => { if (pending.has(i)) { pending.delete(i); reject(new Error(`CDP timeout: ${method}`)); } }, 25000); });
|
|
const evalJs = async (expr) => {
|
|
const r = await send("Runtime.evaluate", { expression: expr, awaitPromise: true, returnByValue: true });
|
|
if (r.result?.exceptionDetails) throw new Error(JSON.stringify(r.result.exceptionDetails.exception?.description ?? r.result.exceptionDetails));
|
|
return r.result?.result?.value;
|
|
};
|
|
await send("Page.enable"); await send("Runtime.enable");
|
|
await send("Emulation.setDeviceMetricsOverride", { width: 1800, height: 1100, deviceScaleFactor: 1, mobile: false });
|
|
if (process.env.CDP_RELOAD === "1" || !page.url.startsWith(COMFY)) {
|
|
await send("Network.enable"); await send("Network.clearBrowserCache"); await send("Network.setCacheDisabled", { cacheDisabled: true });
|
|
await send("Page.navigate", { url: COMFY });
|
|
for (let i = 0; i < 60; i++) { await new Promise((r) => setTimeout(r, 500)); if (await evalJs("Boolean(window.app?.graph && window.app?.canvas)")) break; }
|
|
await new Promise((r) => setTimeout(r, 1500));
|
|
}
|
|
if (wfPath && wfPath !== "-") {
|
|
const wf = fs.readFileSync(wfPath, "utf-8");
|
|
await evalJs(`(async () => { await window.app.loadGraphData(${wf}, true, true, null, { showMissingNodesDialog: false, showMissingModelsDialog: false }); return true; })()`);
|
|
await new Promise((r) => setTimeout(r, waitMs));
|
|
}
|
|
let result = null;
|
|
if (evalFile && evalFile !== "-") {
|
|
result = await evalJs(fs.readFileSync(evalFile, "utf-8"));
|
|
await new Promise((r) => setTimeout(r, waitMs));
|
|
}
|
|
if (outPng && outPng !== "-") {
|
|
const clip = result && typeof result === "object" && result.__clip ? { ...result.__clip, scale: 1 } : undefined;
|
|
const shot = await send("Page.captureScreenshot", clip ? { format: "png", clip } : { format: "png" });
|
|
fs.writeFileSync(outPng, Buffer.from(shot.result.data, "base64"));
|
|
}
|
|
console.log(JSON.stringify(result, null, 1));
|
|
ws.close();
|