5a7466b67d
📁 Files changed: 13 📝 Lines changed: 595 • PLAN.md • TODO.md • browser.d.ts • browser.d.ts.map • browser.js • browser.js.map • mcp.js • mcp.js.map • types.d.ts • types.d.ts.map • browser.ts • mcp.ts • types.ts
281 lines
9.8 KiB
JavaScript
281 lines
9.8 KiB
JavaScript
import { resolve } from 'node:path';
|
|
import { webkit } from 'playwright';
|
|
import * as image from './image.js';
|
|
export class ClaudeBrowser {
|
|
browser = null;
|
|
context = null;
|
|
page = null;
|
|
options;
|
|
consoleMessages = [];
|
|
constructor(options = {}) {
|
|
this.options = {
|
|
headless: options.headless ?? true,
|
|
width: options.width ?? 1280,
|
|
height: options.height ?? 800,
|
|
};
|
|
}
|
|
async launch() {
|
|
this.browser = await webkit.launch({ headless: this.options.headless });
|
|
this.context = await this.browser.newContext({
|
|
viewport: {
|
|
width: this.options.width,
|
|
height: this.options.height,
|
|
},
|
|
});
|
|
this.page = await this.context.newPage();
|
|
this.setupConsoleListener(this.page);
|
|
}
|
|
setupConsoleListener(page) {
|
|
page.on('console', (msg) => {
|
|
const location = msg.location();
|
|
this.consoleMessages.push({
|
|
level: msg.type(),
|
|
text: msg.text(),
|
|
timestamp: Date.now(),
|
|
location: location.url ? `${location.url}:${location.lineNumber}` : undefined,
|
|
});
|
|
});
|
|
}
|
|
async close() {
|
|
if (this.browser) {
|
|
await this.browser.close();
|
|
this.browser = null;
|
|
this.context = null;
|
|
this.page = null;
|
|
}
|
|
}
|
|
ensurePage() {
|
|
if (!this.page) {
|
|
throw new Error('Browser not launched. Call launch() first.');
|
|
}
|
|
return this.page;
|
|
}
|
|
/** Get the current page instance (for advanced usage) */
|
|
getPage() {
|
|
return this.page;
|
|
}
|
|
/** Get the browser context (for advanced usage like cookies) */
|
|
getContext() {
|
|
return this.context;
|
|
}
|
|
async goto(url) {
|
|
const page = this.ensurePage();
|
|
await page.goto(url, { waitUntil: 'networkidle' });
|
|
return { url: page.url(), title: await page.title() };
|
|
}
|
|
async click(selector) {
|
|
const page = this.ensurePage();
|
|
await page.click(selector);
|
|
await page.waitForLoadState('networkidle').catch(() => { });
|
|
return { url: page.url() };
|
|
}
|
|
async type(selector, text) {
|
|
const page = this.ensurePage();
|
|
await page.fill(selector, text);
|
|
}
|
|
async query(selector) {
|
|
const page = this.ensurePage();
|
|
return page.$$eval(selector, (nodes) => nodes.map((el) => {
|
|
const attrs = {};
|
|
for (const attr of el.attributes) {
|
|
attrs[attr.name] = attr.value;
|
|
}
|
|
return {
|
|
tag: el.tagName.toLowerCase(),
|
|
text: el.textContent?.trim().slice(0, 200) || '',
|
|
attributes: attrs,
|
|
};
|
|
}));
|
|
}
|
|
async screenshot(path, fullPage = false) {
|
|
const page = this.ensurePage();
|
|
const resolvedPath = resolve(path || 'screenshot.png');
|
|
const buffer = await page.screenshot({ path: resolvedPath, fullPage });
|
|
return { path: resolvedPath, buffer };
|
|
}
|
|
async getUrl() {
|
|
const page = this.ensurePage();
|
|
return { url: page.url(), title: await page.title() };
|
|
}
|
|
async getHtml(full = false) {
|
|
const page = this.ensurePage();
|
|
const html = await page.content();
|
|
return full ? html : html.slice(0, 10000);
|
|
}
|
|
async back() {
|
|
const page = this.ensurePage();
|
|
await page.goBack();
|
|
return { url: page.url() };
|
|
}
|
|
async forward() {
|
|
const page = this.ensurePage();
|
|
await page.goForward();
|
|
return { url: page.url() };
|
|
}
|
|
async reload() {
|
|
const page = this.ensurePage();
|
|
await page.reload();
|
|
return { url: page.url() };
|
|
}
|
|
async wait(ms = 1000) {
|
|
const page = this.ensurePage();
|
|
await page.waitForTimeout(ms);
|
|
}
|
|
async newPage() {
|
|
if (!this.context) {
|
|
throw new Error('Browser not launched. Call launch() first.');
|
|
}
|
|
this.page = await this.context.newPage();
|
|
this.setupConsoleListener(this.page);
|
|
}
|
|
async eval(script) {
|
|
const page = this.ensurePage();
|
|
return page.evaluate(script);
|
|
}
|
|
getConsole(level, clear = false) {
|
|
let messages = this.consoleMessages;
|
|
if (level && level !== 'all') {
|
|
messages = messages.filter((m) => m.level === level);
|
|
}
|
|
if (clear) {
|
|
this.consoleMessages = [];
|
|
}
|
|
return messages;
|
|
}
|
|
clearConsole() {
|
|
this.consoleMessages = [];
|
|
}
|
|
async executeCommand(cmd) {
|
|
try {
|
|
switch (cmd.cmd) {
|
|
case 'goto': {
|
|
const result = await this.goto(cmd.url);
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'click': {
|
|
const result = await this.click(cmd.selector);
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'type': {
|
|
await this.type(cmd.selector, cmd.text);
|
|
return { ok: true };
|
|
}
|
|
case 'query': {
|
|
const elements = await this.query(cmd.selector);
|
|
return { ok: true, count: elements.length, elements };
|
|
}
|
|
case 'screenshot': {
|
|
const result = await this.screenshot(cmd.path, cmd.fullPage);
|
|
return { ok: true, path: result.path };
|
|
}
|
|
case 'url': {
|
|
const result = await this.getUrl();
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'html': {
|
|
const html = await this.getHtml(cmd.full);
|
|
return { ok: true, html };
|
|
}
|
|
case 'back': {
|
|
const result = await this.back();
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'forward': {
|
|
const result = await this.forward();
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'reload': {
|
|
const result = await this.reload();
|
|
return { ok: true, ...result };
|
|
}
|
|
case 'wait': {
|
|
await this.wait(cmd.ms);
|
|
return { ok: true };
|
|
}
|
|
case 'newpage': {
|
|
await this.newPage();
|
|
return { ok: true };
|
|
}
|
|
case 'close': {
|
|
await this.close();
|
|
return { ok: true };
|
|
}
|
|
case 'eval': {
|
|
const result = await this.eval(cmd.script);
|
|
return { ok: true, result };
|
|
}
|
|
case 'console': {
|
|
const messages = this.getConsole(cmd.level, cmd.clear);
|
|
return { ok: true, count: messages.length, messages };
|
|
}
|
|
case 'favicon': {
|
|
const result = await image.createFavicon(cmd.input, cmd.outputDir);
|
|
return { ok: true, files: result.files, outputDir: result.outputDir };
|
|
}
|
|
case 'convert': {
|
|
const result = await image.convert(cmd.input, cmd.output, cmd.format);
|
|
return {
|
|
ok: true,
|
|
path: result.path,
|
|
width: result.width,
|
|
height: result.height,
|
|
format: result.format,
|
|
size: result.size,
|
|
};
|
|
}
|
|
case 'resize': {
|
|
const result = await image.resize(cmd.input, cmd.output, cmd.width, cmd.height, cmd.fit);
|
|
return {
|
|
ok: true,
|
|
path: result.path,
|
|
width: result.width,
|
|
height: result.height,
|
|
format: result.format,
|
|
size: result.size,
|
|
};
|
|
}
|
|
case 'crop': {
|
|
const result = await image.crop(cmd.input, cmd.output, cmd.left, cmd.top, cmd.width, cmd.height);
|
|
return {
|
|
ok: true,
|
|
path: result.path,
|
|
width: result.width,
|
|
height: result.height,
|
|
format: result.format,
|
|
size: result.size,
|
|
};
|
|
}
|
|
case 'compress': {
|
|
const result = await image.compress(cmd.input, cmd.output, cmd.quality);
|
|
return {
|
|
ok: true,
|
|
path: result.path,
|
|
width: result.width,
|
|
height: result.height,
|
|
format: result.format,
|
|
size: result.size,
|
|
};
|
|
}
|
|
case 'thumbnail': {
|
|
const result = await image.thumbnail(cmd.input, cmd.output, cmd.size);
|
|
return {
|
|
ok: true,
|
|
path: result.path,
|
|
width: result.width,
|
|
height: result.height,
|
|
format: result.format,
|
|
size: result.size,
|
|
};
|
|
}
|
|
default: {
|
|
const _exhaustive = cmd;
|
|
return { ok: false, error: `Unknown command: ${_exhaustive.cmd}` };
|
|
}
|
|
}
|
|
}
|
|
catch (err) {
|
|
return { ok: false, error: err.message };
|
|
}
|
|
}
|
|
}
|
|
//# sourceMappingURL=browser.js.map
|