297 lines
11 KiB
Plaintext
297 lines
11 KiB
Plaintext
"use strict";
|
|
var __create = Object.create;
|
|
var __defProp = Object.defineProperty;
|
|
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
var __getProtoOf = Object.getPrototypeOf;
|
|
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
var __export = (target, all) => {
|
|
for (var name in all)
|
|
__defProp(target, name, { get: all[name], enumerable: true });
|
|
};
|
|
var __copyProps = (to, from, except, desc) => {
|
|
if (from && typeof from === "object" || typeof from === "function") {
|
|
for (let key of __getOwnPropNames(from))
|
|
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
}
|
|
return to;
|
|
};
|
|
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
|
|
// If the importer is in node compatibility mode or this is not an ESM
|
|
// file that has been converted to a CommonJS file using a Babel-
|
|
// compatible transform (i.e. "__esModule" has not been set), then set
|
|
// "default" to the CommonJS "module.exports" for node compatibility.
|
|
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
|
|
mod
|
|
));
|
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
var context_exports = {};
|
|
__export(context_exports, {
|
|
Context: () => Context,
|
|
outputDir: () => outputDir,
|
|
outputFile: () => outputFile,
|
|
workspaceFile: () => workspaceFile
|
|
});
|
|
module.exports = __toCommonJS(context_exports);
|
|
var import_fs = __toESM(require("fs"));
|
|
var import_path = __toESM(require("path"));
|
|
var import_utilsBundle = require("../../utilsBundle");
|
|
var import_stringUtils = require("../../utils/isomorphic/stringUtils");
|
|
var import__ = require("../../..");
|
|
var import_tab = require("./tab");
|
|
var import_disposable = require("../../server/utils/disposable");
|
|
var import_eventsHelper = require("../../server/utils/eventsHelper");
|
|
const testDebug = (0, import_utilsBundle.debug)("pw:mcp:test");
|
|
class Context {
|
|
constructor(browserContext, options) {
|
|
this._tabs = [];
|
|
this._routes = [];
|
|
this._disposables = [];
|
|
this.config = options.config;
|
|
this.sessionLog = options.sessionLog;
|
|
this.options = options;
|
|
this._rawBrowserContext = browserContext;
|
|
testDebug("create context");
|
|
}
|
|
async dispose() {
|
|
await (0, import_disposable.disposeAll)(this._disposables);
|
|
for (const tab of this._tabs)
|
|
await tab.dispose();
|
|
this._tabs.length = 0;
|
|
this._currentTab = void 0;
|
|
await this.stopVideoRecording();
|
|
}
|
|
debugger() {
|
|
return this._rawBrowserContext.debugger;
|
|
}
|
|
tabs() {
|
|
return this._tabs;
|
|
}
|
|
currentTab() {
|
|
return this._currentTab;
|
|
}
|
|
currentTabOrDie() {
|
|
if (!this._currentTab)
|
|
throw new Error("No open pages available.");
|
|
return this._currentTab;
|
|
}
|
|
async newTab() {
|
|
const browserContext = await this.ensureBrowserContext();
|
|
const page = await browserContext.newPage();
|
|
this._currentTab = this._tabs.find((t) => t.page === page);
|
|
return this._currentTab;
|
|
}
|
|
async selectTab(index) {
|
|
const tab = this._tabs[index];
|
|
if (!tab)
|
|
throw new Error(`Tab ${index} not found`);
|
|
await tab.page.bringToFront();
|
|
this._currentTab = tab;
|
|
return tab;
|
|
}
|
|
async ensureTab() {
|
|
const browserContext = await this.ensureBrowserContext();
|
|
if (!this._currentTab)
|
|
await browserContext.newPage();
|
|
return this._currentTab;
|
|
}
|
|
async closeTab(index) {
|
|
const tab = index === void 0 ? this._currentTab : this._tabs[index];
|
|
if (!tab)
|
|
throw new Error(`Tab ${index} not found`);
|
|
const url = tab.page.url();
|
|
await tab.page.close();
|
|
return url;
|
|
}
|
|
async workspaceFile(fileName, perCallWorkspaceDir) {
|
|
return await workspaceFile(this.options, fileName, perCallWorkspaceDir);
|
|
}
|
|
async outputFile(template, options) {
|
|
const baseName = template.suggestedFilename || `${template.prefix}-${(template.date ?? /* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}${template.ext ? "." + template.ext : ""}`;
|
|
return await outputFile(this.options, baseName, options);
|
|
}
|
|
async startVideoRecording(fileName, params) {
|
|
if (this._video)
|
|
throw new Error("Video recording has already been started.");
|
|
this._video = { params, fileName, fileNames: [] };
|
|
const browserContext = await this.ensureBrowserContext();
|
|
for (const page of browserContext.pages())
|
|
await this._startPageVideo(page);
|
|
}
|
|
async stopVideoRecording() {
|
|
if (!this._video)
|
|
return [];
|
|
const video = this._video;
|
|
for (const page of this._rawBrowserContext.pages())
|
|
await page.screencast.stop();
|
|
this._video = void 0;
|
|
return [...video.fileNames];
|
|
}
|
|
async _startPageVideo(page) {
|
|
if (!this._video)
|
|
return;
|
|
const suffix = this._video.fileNames.length ? `-${this._video.fileNames.length}` : "";
|
|
let fileName = this._video.fileName;
|
|
if (fileName && suffix) {
|
|
const ext = import_path.default.extname(fileName);
|
|
fileName = import_path.default.basename(fileName, ext) + suffix + ext;
|
|
}
|
|
this._video.fileNames.push(fileName);
|
|
await page.screencast.start({ path: fileName, ...this._video.params });
|
|
}
|
|
_onPageCreated(page) {
|
|
const tab = new import_tab.Tab(this, page, (tab2) => this._onPageClosed(tab2));
|
|
this._tabs.push(tab);
|
|
if (!this._currentTab)
|
|
this._currentTab = tab;
|
|
this._startPageVideo(page).catch(() => {
|
|
});
|
|
}
|
|
_onPageClosed(tab) {
|
|
const index = this._tabs.indexOf(tab);
|
|
if (index === -1)
|
|
return;
|
|
this._tabs.splice(index, 1);
|
|
if (this._currentTab === tab)
|
|
this._currentTab = this._tabs[Math.min(index, this._tabs.length - 1)];
|
|
}
|
|
routes() {
|
|
return this._routes;
|
|
}
|
|
async addRoute(entry) {
|
|
const browserContext = await this.ensureBrowserContext();
|
|
await browserContext.route(entry.pattern, entry.handler);
|
|
this._routes.push(entry);
|
|
}
|
|
async removeRoute(pattern) {
|
|
let removed = 0;
|
|
const browserContext = await this.ensureBrowserContext();
|
|
if (pattern) {
|
|
const toRemove = this._routes.filter((r) => r.pattern === pattern);
|
|
for (const route of toRemove)
|
|
await browserContext.unroute(route.pattern, route.handler);
|
|
this._routes = this._routes.filter((r) => r.pattern !== pattern);
|
|
removed = toRemove.length;
|
|
} else {
|
|
for (const route of this._routes)
|
|
await browserContext.unroute(route.pattern, route.handler);
|
|
removed = this._routes.length;
|
|
this._routes = [];
|
|
}
|
|
return removed;
|
|
}
|
|
isRunningTool() {
|
|
return this._runningToolName !== void 0;
|
|
}
|
|
setRunningTool(name) {
|
|
this._runningToolName = name;
|
|
}
|
|
async _setupRequestInterception(context) {
|
|
if (this.config.network?.allowedOrigins?.length) {
|
|
this._disposables.push(await context.route("**", (route) => route.abort("blockedbyclient")));
|
|
for (const origin of this.config.network.allowedOrigins) {
|
|
const glob = originOrHostGlob(origin);
|
|
this._disposables.push(await context.route(glob, (route) => route.continue()));
|
|
}
|
|
}
|
|
if (this.config.network?.blockedOrigins?.length) {
|
|
for (const origin of this.config.network.blockedOrigins)
|
|
this._disposables.push(await context.route(originOrHostGlob(origin), (route) => route.abort("blockedbyclient")));
|
|
}
|
|
}
|
|
async ensureBrowserContext() {
|
|
if (this._browserContextPromise)
|
|
return this._browserContextPromise;
|
|
this._browserContextPromise = this._initializeBrowserContext();
|
|
return this._browserContextPromise;
|
|
}
|
|
async _initializeBrowserContext() {
|
|
if (this.config.testIdAttribute)
|
|
import__.selectors.setTestIdAttribute(this.config.testIdAttribute);
|
|
const browserContext = this._rawBrowserContext;
|
|
await this._setupRequestInterception(browserContext);
|
|
if (this.config.saveTrace) {
|
|
await browserContext.tracing.start({
|
|
name: "trace-" + Date.now(),
|
|
screenshots: true,
|
|
snapshots: true,
|
|
live: true
|
|
});
|
|
this._disposables.push({
|
|
dispose: async () => {
|
|
await browserContext.tracing.stop();
|
|
}
|
|
});
|
|
}
|
|
for (const initScript of this.config.browser?.initScript || [])
|
|
this._disposables.push(await browserContext.addInitScript({ path: import_path.default.resolve(this.options.cwd, initScript) }));
|
|
for (const page of browserContext.pages())
|
|
this._onPageCreated(page);
|
|
this._disposables.push(import_eventsHelper.eventsHelper.addEventListener(browserContext, "page", (page) => this._onPageCreated(page)));
|
|
return browserContext;
|
|
}
|
|
checkUrlAllowed(url) {
|
|
if (this.config.allowUnrestrictedFileAccess)
|
|
return;
|
|
if (!URL.canParse(url))
|
|
return;
|
|
if (new URL(url).protocol === "file:")
|
|
throw new Error(`Access to "file:" protocol is blocked. Attempted URL: "${url}"`);
|
|
}
|
|
lookupSecret(secretName) {
|
|
if (!this.config.secrets?.[secretName])
|
|
return { value: secretName, code: (0, import_stringUtils.escapeWithQuotes)(secretName, "'") };
|
|
return {
|
|
value: this.config.secrets[secretName],
|
|
code: `process.env['${secretName}']`
|
|
};
|
|
}
|
|
}
|
|
function originOrHostGlob(originOrHost) {
|
|
const wildcardPortMatch = originOrHost.match(/^(https?:\/\/[^/:]+):\*$/);
|
|
if (wildcardPortMatch)
|
|
return `${wildcardPortMatch[1]}:*/**`;
|
|
try {
|
|
const url = new URL(originOrHost);
|
|
if (url.origin !== "null")
|
|
return `${url.origin}/**`;
|
|
} catch {
|
|
}
|
|
return `*://${originOrHost}/**`;
|
|
}
|
|
async function workspaceFile(options, fileName, perCallWorkspaceDir) {
|
|
const workspace = perCallWorkspaceDir ?? options.cwd;
|
|
const resolvedName = import_path.default.resolve(workspace, fileName);
|
|
await checkFile(options, resolvedName, { origin: "llm" });
|
|
return resolvedName;
|
|
}
|
|
function outputDir(options) {
|
|
if (options.config.outputDir)
|
|
return import_path.default.resolve(options.config.outputDir);
|
|
return import_path.default.resolve(options.cwd, options.config.skillMode ? ".playwright-cli" : ".playwright-mcp");
|
|
}
|
|
async function outputFile(options, fileName, flags) {
|
|
const resolvedFile = import_path.default.resolve(outputDir(options), fileName);
|
|
await checkFile(options, resolvedFile, flags);
|
|
await import_fs.default.promises.mkdir(import_path.default.dirname(resolvedFile), { recursive: true });
|
|
(0, import_utilsBundle.debug)("pw:mcp:file")(resolvedFile);
|
|
return resolvedFile;
|
|
}
|
|
async function checkFile(options, resolvedFilename, flags) {
|
|
if (flags.origin === "code" || options.config.allowUnrestrictedFileAccess)
|
|
return;
|
|
const output = outputDir(options);
|
|
const workspace = options.cwd;
|
|
const withinDir = (root) => resolvedFilename === root || resolvedFilename.startsWith(root + import_path.default.sep);
|
|
if (!withinDir(output) && !withinDir(workspace))
|
|
throw new Error(`File access denied: ${resolvedFilename} is outside allowed roots. Allowed roots: ${output}, ${workspace}`);
|
|
}
|
|
// Annotate the CommonJS export names for ESM import in node:
|
|
0 && (module.exports = {
|
|
Context,
|
|
outputDir,
|
|
outputFile,
|
|
workspaceFile
|
|
});
|