101 lines
3.5 KiB
TypeScript
101 lines
3.5 KiB
TypeScript
import type { Command } from "commander";
|
|
import { resolveBrowserControlUrl } from "../browser/client.js";
|
|
import {
|
|
browserConsoleMessages,
|
|
browserPdfSave,
|
|
browserResponseBody,
|
|
} from "../browser/client-actions.js";
|
|
import { danger } from "../globals.js";
|
|
import { defaultRuntime } from "../runtime.js";
|
|
import type { BrowserParentOpts } from "./browser-cli-shared.js";
|
|
import { runCommandWithRuntime } from "./cli-utils.js";
|
|
|
|
function runBrowserObserve(action: () => Promise<void>) {
|
|
return runCommandWithRuntime(defaultRuntime, action, (err) => {
|
|
defaultRuntime.error(danger(String(err)));
|
|
defaultRuntime.exit(1);
|
|
});
|
|
}
|
|
|
|
export function registerBrowserActionObserveCommands(
|
|
browser: Command,
|
|
parentOpts: (cmd: Command) => BrowserParentOpts,
|
|
) {
|
|
browser
|
|
.command("console")
|
|
.description("Get recent console messages")
|
|
.option("--level <level>", "Filter by level (error, warn, info)")
|
|
.option("--target-id <id>", "CDP target id (or unique prefix)")
|
|
.action(async (opts, cmd) => {
|
|
const parent = parentOpts(cmd);
|
|
const baseUrl = resolveBrowserControlUrl(parent?.url);
|
|
const profile = parent?.browserProfile;
|
|
await runBrowserObserve(async () => {
|
|
const result = await browserConsoleMessages(baseUrl, {
|
|
level: opts.level?.trim() || undefined,
|
|
targetId: opts.targetId?.trim() || undefined,
|
|
profile,
|
|
});
|
|
if (parent?.json) {
|
|
defaultRuntime.log(JSON.stringify(result, null, 2));
|
|
return;
|
|
}
|
|
defaultRuntime.log(JSON.stringify(result.messages, null, 2));
|
|
});
|
|
});
|
|
|
|
browser
|
|
.command("pdf")
|
|
.description("Save page as PDF")
|
|
.option("--target-id <id>", "CDP target id (or unique prefix)")
|
|
.action(async (opts, cmd) => {
|
|
const parent = parentOpts(cmd);
|
|
const baseUrl = resolveBrowserControlUrl(parent?.url);
|
|
const profile = parent?.browserProfile;
|
|
await runBrowserObserve(async () => {
|
|
const result = await browserPdfSave(baseUrl, {
|
|
targetId: opts.targetId?.trim() || undefined,
|
|
profile,
|
|
});
|
|
if (parent?.json) {
|
|
defaultRuntime.log(JSON.stringify(result, null, 2));
|
|
return;
|
|
}
|
|
defaultRuntime.log(`PDF: ${result.path}`);
|
|
});
|
|
});
|
|
|
|
browser
|
|
.command("responsebody")
|
|
.description("Wait for a network response and return its body")
|
|
.argument("<url>", "URL (exact, substring, or glob like **/api)")
|
|
.option("--target-id <id>", "CDP target id (or unique prefix)")
|
|
.option(
|
|
"--timeout-ms <ms>",
|
|
"How long to wait for the response (default: 20000)",
|
|
(v: string) => Number(v),
|
|
)
|
|
.option("--max-chars <n>", "Max body chars to return (default: 200000)", (v: string) =>
|
|
Number(v),
|
|
)
|
|
.action(async (url: string, opts, cmd) => {
|
|
const parent = parentOpts(cmd);
|
|
const baseUrl = resolveBrowserControlUrl(parent?.url);
|
|
const profile = parent?.browserProfile;
|
|
await runBrowserObserve(async () => {
|
|
const result = await browserResponseBody(baseUrl, {
|
|
url,
|
|
targetId: opts.targetId?.trim() || undefined,
|
|
timeoutMs: Number.isFinite(opts.timeoutMs) ? opts.timeoutMs : undefined,
|
|
maxChars: Number.isFinite(opts.maxChars) ? opts.maxChars : undefined,
|
|
profile,
|
|
});
|
|
if (parent?.json) {
|
|
defaultRuntime.log(JSON.stringify(result, null, 2));
|
|
return;
|
|
}
|
|
defaultRuntime.log(result.response.body);
|
|
});
|
|
});
|
|
}
|