Something went wrong. Try again.
Monorepo for Aesthetic.Computer aesthetic.computer
Something went wrong. Try again.
3.9 kB · 96 lines
JavaScript
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697#!/usr/bin/env nodeimport puppeteer from "puppeteer-core";import http from "http";
const CDP_HOST = "host.docker.internal";const CDP_PORT = 9222;
function httpGet(url, headers = {}) { return new Promise((resolve, reject) => { const u = new URL(url); const req = http.request({ hostname: u.hostname, port: u.port, path: u.pathname, method: "GET", headers: { Host: "localhost", ...headers } }, (res) => { let data = ""; res.on("data", chunk => data += chunk); res.on("end", () => resolve(data)); }); req.on("error", reject); req.end(); });}
async function getWSEndpoint() { const data = await httpGet(`http://${CDP_HOST}:${CDP_PORT}/json/version`); const json = JSON.parse(data); return json.webSocketDebuggerUrl.replace("ws://localhost", `ws://${CDP_HOST}:${CDP_PORT}`);}
export async function getBrowser() { const wsEndpoint = await getWSEndpoint(); return puppeteer.connect({ browserWSEndpoint: wsEndpoint, defaultViewport: null, });}
export async function listPages() { const browser = await getBrowser(); const pages = await browser.pages(); const info = await Promise.all(pages.map(async (p, i) => ({ index: i, url: p.url(), title: await p.title(), }))); await browser.disconnect(); return info;}
export async function getPage(pattern) { const browser = await getBrowser(); const pages = await browser.pages(); let page = typeof pattern === "number" ? pages[pattern] : pages.find(p => p.url().includes(pattern)); if (!page) { await browser.disconnect(); throw new Error("No page: " + pattern); } return { browser, page };}
export class PageController { constructor(page, browser) { this.page = page; this.browser = browser; } static async connect(pattern) { const { browser, page } = await getPage(pattern); return new PageController(page, browser); } async goto(url, o = {}) { await this.page.goto(url, { waitUntil: "networkidle2", ...o }); return this; } async click(s, o = {}) { await this.page.waitForSelector(s, { timeout: 10000, ...o }); await this.page.click(s); return this; } async type(s, text, o = {}) { await this.page.waitForSelector(s, { timeout: 10000 }); await this.page.click(s); await this.page.type(s, text, o); return this; } async fill(s, v) { await this.page.waitForSelector(s, { timeout: 10000 }); await this.page.$eval(s, (el, val) => { el.value = val; el.dispatchEvent(new Event("input", { bubbles: true })); el.dispatchEvent(new Event("change", { bubbles: true })); }, v); return this; } async getContent() { return this.page.evaluate(() => document.body.innerText); } async getHtml() { return this.page.content(); } async wait(ms) { await new Promise(r => setTimeout(r, ms)); return this; } async waitNav(o = {}) { await this.page.waitForNavigation({ waitUntil: "networkidle2", ...o }); return this; } async waitFor(s, o = {}) { await this.page.waitForSelector(s, { timeout: 10000, ...o }); return this; } async screenshot(path) { await this.page.screenshot({ path, fullPage: true }); return this; } async eval(fn, ...args) { return this.page.evaluate(fn, ...args); } url() { return this.page.url(); } async title() { return this.page.title(); } async close() { await this.browser.disconnect(); }}
if (import.meta.url === `file://${process.argv[1]}`) { const [cmd, arg] = process.argv.slice(2); if (cmd === "list") { const pages = await listPages(); console.log("Tabs:"); pages.forEach(p => console.log(" [" + p.index + "] " + p.title.slice(0,40) + " - " + p.url.slice(0,60))); } else if (cmd === "content") { const ctrl = await PageController.connect(isNaN(arg) ? arg : parseInt(arg) || 0); console.log(await ctrl.getContent()); await ctrl.close(); } else { console.log("Usage: node browser.mjs list | content [pattern]"); }}