849 lines
32 KiB
JavaScript
849 lines
32 KiB
JavaScript
"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 page_exports = {};
|
|
__export(page_exports, {
|
|
InitScript: () => InitScript,
|
|
Page: () => Page,
|
|
PageBinding: () => PageBinding,
|
|
Worker: () => Worker
|
|
});
|
|
module.exports = __toCommonJS(page_exports);
|
|
var accessibility = __toESM(require("./accessibility"));
|
|
var import_browserContext = require("./browserContext");
|
|
var import_console = require("./console");
|
|
var import_errors = require("./errors");
|
|
var import_fileChooser = require("./fileChooser");
|
|
var frames = __toESM(require("./frames"));
|
|
var import_helper = require("./helper");
|
|
var input = __toESM(require("./input"));
|
|
var import_instrumentation = require("./instrumentation");
|
|
var js = __toESM(require("./javascript"));
|
|
var import_screenshotter = require("./screenshotter");
|
|
var import_utils = require("../utils");
|
|
var import_utils2 = require("../utils");
|
|
var import_comparators = require("./utils/comparators");
|
|
var import_debugLogger = require("./utils/debugLogger");
|
|
var import_selectorParser = require("../utils/isomorphic/selectorParser");
|
|
var import_manualPromise = require("../utils/isomorphic/manualPromise");
|
|
var import_utilityScriptSerializers = require("../utils/isomorphic/utilityScriptSerializers");
|
|
var import_callLog = require("./callLog");
|
|
var rawBindingsControllerSource = __toESM(require("../generated/bindingsControllerSource"));
|
|
class Page extends import_instrumentation.SdkObject {
|
|
constructor(delegate, browserContext) {
|
|
super(browserContext, "page");
|
|
this._closedState = "open";
|
|
this._closedPromise = new import_manualPromise.ManualPromise();
|
|
this._initializedPromise = new import_manualPromise.ManualPromise();
|
|
this._eventsToEmitAfterInitialized = [];
|
|
this._crashed = false;
|
|
this.openScope = new import_utils.LongStandingScope();
|
|
this._emulatedMedia = {};
|
|
this._fileChooserInterceptedBy = /* @__PURE__ */ new Set();
|
|
this._pageBindings = /* @__PURE__ */ new Map();
|
|
this.initScripts = [];
|
|
this._workers = /* @__PURE__ */ new Map();
|
|
this.requestInterceptors = [];
|
|
this.video = null;
|
|
this._locatorHandlers = /* @__PURE__ */ new Map();
|
|
this._lastLocatorHandlerUid = 0;
|
|
this._locatorHandlerRunningCounter = 0;
|
|
// Aiming at 25 fps by default - each frame is 40ms, but we give some slack with 35ms.
|
|
// When throttling for tracing, 200ms between frames, except for 10 frames around the action.
|
|
this._frameThrottler = new FrameThrottler(10, 35, 200);
|
|
this.lastSnapshotFrameIds = [];
|
|
this.attribution.page = this;
|
|
this.delegate = delegate;
|
|
this.browserContext = browserContext;
|
|
this.accessibility = new accessibility.Accessibility(delegate.getAccessibilityTree.bind(delegate));
|
|
this.keyboard = new input.Keyboard(delegate.rawKeyboard, this);
|
|
this.mouse = new input.Mouse(delegate.rawMouse, this);
|
|
this.touchscreen = new input.Touchscreen(delegate.rawTouchscreen, this);
|
|
this.screenshotter = new import_screenshotter.Screenshotter(this);
|
|
this.frameManager = new frames.FrameManager(this);
|
|
if (delegate.pdf)
|
|
this.pdf = delegate.pdf.bind(delegate);
|
|
this.coverage = delegate.coverage ? delegate.coverage() : null;
|
|
this.isStorageStatePage = browserContext.isCreatingStorageStatePage();
|
|
}
|
|
static {
|
|
this.Events = {
|
|
Close: "close",
|
|
Crash: "crash",
|
|
Download: "download",
|
|
EmulatedSizeChanged: "emulatedsizechanged",
|
|
FileChooser: "filechooser",
|
|
FrameAttached: "frameattached",
|
|
FrameDetached: "framedetached",
|
|
InternalFrameNavigatedToNewDocument: "internalframenavigatedtonewdocument",
|
|
LocatorHandlerTriggered: "locatorhandlertriggered",
|
|
ScreencastFrame: "screencastframe",
|
|
Video: "video",
|
|
WebSocket: "websocket",
|
|
Worker: "worker"
|
|
};
|
|
}
|
|
async reportAsNew(opener, error = void 0, contextEvent = import_browserContext.BrowserContext.Events.Page) {
|
|
if (opener) {
|
|
const openerPageOrError = await opener.waitForInitializedOrError();
|
|
if (openerPageOrError instanceof Page && !openerPageOrError.isClosed())
|
|
this._opener = openerPageOrError;
|
|
}
|
|
this._markInitialized(error, contextEvent);
|
|
}
|
|
_markInitialized(error = void 0, contextEvent = import_browserContext.BrowserContext.Events.Page) {
|
|
if (error) {
|
|
if (this.browserContext.isClosingOrClosed())
|
|
return;
|
|
this.frameManager.createDummyMainFrameIfNeeded();
|
|
}
|
|
this._initialized = error || this;
|
|
this.emitOnContext(contextEvent, this);
|
|
for (const { event, args } of this._eventsToEmitAfterInitialized)
|
|
this.browserContext.emit(event, ...args);
|
|
this._eventsToEmitAfterInitialized = [];
|
|
if (this.isClosed())
|
|
this.emit(Page.Events.Close);
|
|
else
|
|
this.instrumentation.onPageOpen(this);
|
|
this._initializedPromise.resolve(this._initialized);
|
|
}
|
|
initializedOrUndefined() {
|
|
return this._initialized ? this : void 0;
|
|
}
|
|
waitForInitializedOrError() {
|
|
return this._initializedPromise;
|
|
}
|
|
emitOnContext(event, ...args) {
|
|
if (this.isStorageStatePage)
|
|
return;
|
|
this.browserContext.emit(event, ...args);
|
|
}
|
|
emitOnContextOnceInitialized(event, ...args) {
|
|
if (this.isStorageStatePage)
|
|
return;
|
|
if (this._initialized)
|
|
this.browserContext.emit(event, ...args);
|
|
else
|
|
this._eventsToEmitAfterInitialized.push({ event, args });
|
|
}
|
|
async resetForReuse(progress) {
|
|
await this.mainFrame().gotoImpl(progress, "about:blank", {});
|
|
this._emulatedSize = void 0;
|
|
this._emulatedMedia = {};
|
|
this._extraHTTPHeaders = void 0;
|
|
await Promise.all([
|
|
this.delegate.updateEmulatedViewportSize(),
|
|
this.delegate.updateEmulateMedia(),
|
|
this.delegate.updateExtraHTTPHeaders()
|
|
]);
|
|
await this.delegate.resetForReuse(progress);
|
|
}
|
|
_didClose() {
|
|
this.frameManager.dispose();
|
|
this._frameThrottler.dispose();
|
|
(0, import_utils.assert)(this._closedState !== "closed", "Page closed twice");
|
|
this._closedState = "closed";
|
|
this.emit(Page.Events.Close);
|
|
this._closedPromise.resolve();
|
|
this.instrumentation.onPageClose(this);
|
|
this.openScope.close(new import_errors.TargetClosedError());
|
|
}
|
|
_didCrash() {
|
|
this.frameManager.dispose();
|
|
this._frameThrottler.dispose();
|
|
this.emit(Page.Events.Crash);
|
|
this._crashed = true;
|
|
this.instrumentation.onPageClose(this);
|
|
this.openScope.close(new Error("Page crashed"));
|
|
}
|
|
async _onFileChooserOpened(handle) {
|
|
let multiple;
|
|
try {
|
|
multiple = await handle.evaluate((element) => !!element.multiple);
|
|
} catch (e) {
|
|
return;
|
|
}
|
|
if (!this.listenerCount(Page.Events.FileChooser)) {
|
|
handle.dispose();
|
|
return;
|
|
}
|
|
const fileChooser = new import_fileChooser.FileChooser(this, handle, multiple);
|
|
this.emit(Page.Events.FileChooser, fileChooser);
|
|
}
|
|
opener() {
|
|
return this._opener;
|
|
}
|
|
mainFrame() {
|
|
return this.frameManager.mainFrame();
|
|
}
|
|
frames() {
|
|
return this.frameManager.frames();
|
|
}
|
|
async exposeBinding(progress, name, needsHandle, playwrightBinding) {
|
|
if (this._pageBindings.has(name))
|
|
throw new Error(`Function "${name}" has been already registered`);
|
|
if (this.browserContext._pageBindings.has(name))
|
|
throw new Error(`Function "${name}" has been already registered in the browser context`);
|
|
await progress.race(this.browserContext.exposePlaywrightBindingIfNeeded());
|
|
const binding = new PageBinding(name, playwrightBinding, needsHandle);
|
|
this._pageBindings.set(name, binding);
|
|
try {
|
|
await progress.race(this.delegate.addInitScript(binding.initScript));
|
|
await progress.race(this.safeNonStallingEvaluateInAllFrames(binding.initScript.source, "main"));
|
|
return binding;
|
|
} catch (error) {
|
|
this._pageBindings.delete(name);
|
|
throw error;
|
|
}
|
|
}
|
|
async removeExposedBindings(bindings) {
|
|
bindings = bindings.filter((binding) => this._pageBindings.get(binding.name) === binding);
|
|
for (const binding of bindings)
|
|
this._pageBindings.delete(binding.name);
|
|
await this.delegate.removeInitScripts(bindings.map((binding) => binding.initScript));
|
|
const cleanup = bindings.map((binding) => `{ ${binding.cleanupScript} };
|
|
`).join("");
|
|
await this.safeNonStallingEvaluateInAllFrames(cleanup, "main");
|
|
}
|
|
async setExtraHTTPHeaders(progress, headers) {
|
|
const oldHeaders = this._extraHTTPHeaders;
|
|
try {
|
|
this._extraHTTPHeaders = headers;
|
|
await progress.race(this.delegate.updateExtraHTTPHeaders());
|
|
} catch (error) {
|
|
this._extraHTTPHeaders = oldHeaders;
|
|
this.delegate.updateExtraHTTPHeaders().catch(() => {
|
|
});
|
|
throw error;
|
|
}
|
|
}
|
|
extraHTTPHeaders() {
|
|
return this._extraHTTPHeaders;
|
|
}
|
|
async onBindingCalled(payload, context) {
|
|
if (this._closedState === "closed")
|
|
return;
|
|
await PageBinding.dispatch(this, payload, context);
|
|
}
|
|
addConsoleMessage(type, args, location, text) {
|
|
const message = new import_console.ConsoleMessage(this, type, text, args, location);
|
|
const intercepted = this.frameManager.interceptConsoleMessage(message);
|
|
if (intercepted) {
|
|
args.forEach((arg) => arg.dispose());
|
|
return;
|
|
}
|
|
this.emitOnContextOnceInitialized(import_browserContext.BrowserContext.Events.Console, message);
|
|
}
|
|
async reload(progress, options) {
|
|
return this.mainFrame().raceNavigationAction(progress, async () => {
|
|
const [response] = await Promise.all([
|
|
// Reload must be a new document, and should not be confused with a stray pushState.
|
|
this.mainFrame()._waitForNavigation(progress, true, options),
|
|
progress.race(this.delegate.reload())
|
|
]);
|
|
return response;
|
|
});
|
|
}
|
|
async goBack(progress, options) {
|
|
return this.mainFrame().raceNavigationAction(progress, async () => {
|
|
let error;
|
|
const waitPromise = this.mainFrame()._waitForNavigation(progress, false, options).catch((e) => {
|
|
error = e;
|
|
return null;
|
|
});
|
|
const result = await progress.race(this.delegate.goBack());
|
|
if (!result) {
|
|
waitPromise.catch(() => {
|
|
});
|
|
return null;
|
|
}
|
|
const response = await waitPromise;
|
|
if (error)
|
|
throw error;
|
|
return response;
|
|
});
|
|
}
|
|
async goForward(progress, options) {
|
|
return this.mainFrame().raceNavigationAction(progress, async () => {
|
|
let error;
|
|
const waitPromise = this.mainFrame()._waitForNavigation(progress, false, options).catch((e) => {
|
|
error = e;
|
|
return null;
|
|
});
|
|
const result = await progress.race(this.delegate.goForward());
|
|
if (!result) {
|
|
waitPromise.catch(() => {
|
|
});
|
|
return null;
|
|
}
|
|
const response = await waitPromise;
|
|
if (error)
|
|
throw error;
|
|
return response;
|
|
});
|
|
}
|
|
requestGC() {
|
|
return this.delegate.requestGC();
|
|
}
|
|
registerLocatorHandler(selector, noWaitAfter) {
|
|
const uid = ++this._lastLocatorHandlerUid;
|
|
this._locatorHandlers.set(uid, { selector, noWaitAfter });
|
|
return uid;
|
|
}
|
|
resolveLocatorHandler(uid, remove) {
|
|
const handler = this._locatorHandlers.get(uid);
|
|
if (remove)
|
|
this._locatorHandlers.delete(uid);
|
|
if (handler) {
|
|
handler.resolved?.resolve();
|
|
handler.resolved = void 0;
|
|
}
|
|
}
|
|
unregisterLocatorHandler(uid) {
|
|
this._locatorHandlers.delete(uid);
|
|
}
|
|
async performActionPreChecks(progress) {
|
|
await this._performWaitForNavigationCheck(progress);
|
|
await this._performLocatorHandlersCheckpoint(progress);
|
|
await this._performWaitForNavigationCheck(progress);
|
|
}
|
|
async _performWaitForNavigationCheck(progress) {
|
|
const mainFrame = this.frameManager.mainFrame();
|
|
if (!mainFrame || !mainFrame.pendingDocument())
|
|
return;
|
|
const url = mainFrame.pendingDocument()?.request?.url();
|
|
const toUrl = url ? `" ${(0, import_utils.trimStringWithEllipsis)(url, 200)}"` : "";
|
|
progress.log(` waiting for${toUrl} navigation to finish...`);
|
|
await import_helper.helper.waitForEvent(progress, mainFrame, frames.Frame.Events.InternalNavigation, (e) => {
|
|
if (!e.isPublic)
|
|
return false;
|
|
if (!e.error)
|
|
progress.log(` navigated to "${(0, import_utils.trimStringWithEllipsis)(mainFrame.url(), 200)}"`);
|
|
return true;
|
|
}).promise;
|
|
}
|
|
async _performLocatorHandlersCheckpoint(progress) {
|
|
if (this._locatorHandlerRunningCounter)
|
|
return;
|
|
for (const [uid, handler] of this._locatorHandlers) {
|
|
if (!handler.resolved) {
|
|
if (await this.mainFrame().isVisibleInternal(progress, handler.selector, { strict: true })) {
|
|
handler.resolved = new import_manualPromise.ManualPromise();
|
|
this.emit(Page.Events.LocatorHandlerTriggered, uid);
|
|
}
|
|
}
|
|
if (handler.resolved) {
|
|
++this._locatorHandlerRunningCounter;
|
|
progress.log(` found ${(0, import_utils2.asLocator)(this.browserContext._browser.sdkLanguage(), handler.selector)}, intercepting action to run the handler`);
|
|
const promise = handler.resolved.then(async () => {
|
|
if (!handler.noWaitAfter) {
|
|
progress.log(` locator handler has finished, waiting for ${(0, import_utils2.asLocator)(this.browserContext._browser.sdkLanguage(), handler.selector)} to be hidden`);
|
|
await this.mainFrame().waitForSelector(progress, handler.selector, false, { state: "hidden" });
|
|
} else {
|
|
progress.log(` locator handler has finished`);
|
|
}
|
|
});
|
|
await progress.race(this.openScope.race(promise)).finally(() => --this._locatorHandlerRunningCounter);
|
|
progress.log(` interception handler has finished, continuing`);
|
|
}
|
|
}
|
|
}
|
|
async emulateMedia(progress, options) {
|
|
const oldEmulatedMedia = { ...this._emulatedMedia };
|
|
if (options.media !== void 0)
|
|
this._emulatedMedia.media = options.media;
|
|
if (options.colorScheme !== void 0)
|
|
this._emulatedMedia.colorScheme = options.colorScheme;
|
|
if (options.reducedMotion !== void 0)
|
|
this._emulatedMedia.reducedMotion = options.reducedMotion;
|
|
if (options.forcedColors !== void 0)
|
|
this._emulatedMedia.forcedColors = options.forcedColors;
|
|
if (options.contrast !== void 0)
|
|
this._emulatedMedia.contrast = options.contrast;
|
|
try {
|
|
await progress.race(this.delegate.updateEmulateMedia());
|
|
} catch (error) {
|
|
this._emulatedMedia = oldEmulatedMedia;
|
|
this.delegate.updateEmulateMedia().catch(() => {
|
|
});
|
|
throw error;
|
|
}
|
|
}
|
|
emulatedMedia() {
|
|
const contextOptions = this.browserContext._options;
|
|
return {
|
|
media: this._emulatedMedia.media || "no-override",
|
|
colorScheme: this._emulatedMedia.colorScheme !== void 0 ? this._emulatedMedia.colorScheme : contextOptions.colorScheme ?? "light",
|
|
reducedMotion: this._emulatedMedia.reducedMotion !== void 0 ? this._emulatedMedia.reducedMotion : contextOptions.reducedMotion ?? "no-preference",
|
|
forcedColors: this._emulatedMedia.forcedColors !== void 0 ? this._emulatedMedia.forcedColors : contextOptions.forcedColors ?? "none",
|
|
contrast: this._emulatedMedia.contrast !== void 0 ? this._emulatedMedia.contrast : contextOptions.contrast ?? "no-preference"
|
|
};
|
|
}
|
|
async setViewportSize(progress, viewportSize) {
|
|
const oldEmulatedSize = this._emulatedSize;
|
|
try {
|
|
this._setEmulatedSize({ viewport: { ...viewportSize }, screen: { ...viewportSize } });
|
|
await progress.race(this.delegate.updateEmulatedViewportSize());
|
|
} catch (error) {
|
|
this._emulatedSize = oldEmulatedSize;
|
|
this.delegate.updateEmulatedViewportSize().catch(() => {
|
|
});
|
|
throw error;
|
|
}
|
|
}
|
|
setEmulatedSizeFromWindowOpen(emulatedSize) {
|
|
this._setEmulatedSize(emulatedSize);
|
|
}
|
|
_setEmulatedSize(emulatedSize) {
|
|
this._emulatedSize = emulatedSize;
|
|
this.emit(Page.Events.EmulatedSizeChanged);
|
|
}
|
|
emulatedSize() {
|
|
if (this._emulatedSize)
|
|
return this._emulatedSize;
|
|
const contextOptions = this.browserContext._options;
|
|
return contextOptions.viewport ? { viewport: contextOptions.viewport, screen: contextOptions.screen || contextOptions.viewport } : void 0;
|
|
}
|
|
async bringToFront() {
|
|
await this.delegate.bringToFront();
|
|
}
|
|
async addInitScript(progress, source) {
|
|
const initScript = new InitScript(source);
|
|
this.initScripts.push(initScript);
|
|
try {
|
|
await progress.race(this.delegate.addInitScript(initScript));
|
|
} catch (error) {
|
|
this.removeInitScripts([initScript]).catch(() => {
|
|
});
|
|
throw error;
|
|
}
|
|
return initScript;
|
|
}
|
|
async removeInitScripts(initScripts) {
|
|
const set = new Set(initScripts);
|
|
this.initScripts = this.initScripts.filter((script) => !set.has(script));
|
|
await this.delegate.removeInitScripts(initScripts);
|
|
}
|
|
needsRequestInterception() {
|
|
return this.requestInterceptors.length > 0 || this.browserContext.requestInterceptors.length > 0;
|
|
}
|
|
async addRequestInterceptor(progress, handler, prepend) {
|
|
if (prepend)
|
|
this.requestInterceptors.unshift(handler);
|
|
else
|
|
this.requestInterceptors.push(handler);
|
|
await this.delegate.updateRequestInterception();
|
|
}
|
|
async removeRequestInterceptor(handler) {
|
|
const index = this.requestInterceptors.indexOf(handler);
|
|
if (index === -1)
|
|
return;
|
|
this.requestInterceptors.splice(index, 1);
|
|
await this.browserContext.notifyRoutesInFlightAboutRemovedHandler(handler);
|
|
await this.delegate.updateRequestInterception();
|
|
}
|
|
async expectScreenshot(progress, options) {
|
|
const locator = options.locator;
|
|
const rafrafScreenshot = locator ? async (timeout) => {
|
|
return await locator.frame.rafrafTimeoutScreenshotElementWithProgress(progress, locator.selector, timeout, options || {});
|
|
} : async (timeout) => {
|
|
await this.performActionPreChecks(progress);
|
|
await this.mainFrame().rafrafTimeout(progress, timeout);
|
|
return await this.screenshotter.screenshotPage(progress, options || {});
|
|
};
|
|
const comparator = (0, import_comparators.getComparator)("image/png");
|
|
if (!options.expected && options.isNot)
|
|
return { errorMessage: '"not" matcher requires expected result' };
|
|
try {
|
|
const format = (0, import_screenshotter.validateScreenshotOptions)(options || {});
|
|
if (format !== "png")
|
|
throw new Error("Only PNG screenshots are supported");
|
|
} catch (error) {
|
|
return { errorMessage: error.message };
|
|
}
|
|
let intermediateResult;
|
|
const areEqualScreenshots = (actual, expected, previous) => {
|
|
const comparatorResult = actual && expected ? comparator(actual, expected, options) : void 0;
|
|
if (comparatorResult !== void 0 && !!comparatorResult === !!options.isNot)
|
|
return true;
|
|
if (comparatorResult)
|
|
intermediateResult = { errorMessage: comparatorResult.errorMessage, diff: comparatorResult.diff, actual, previous };
|
|
return false;
|
|
};
|
|
try {
|
|
let actual;
|
|
let previous;
|
|
const pollIntervals = [0, 100, 250, 500];
|
|
progress.log(`${(0, import_utils.renderTitleForCall)(progress.metadata)}${options.timeout ? ` with timeout ${options.timeout}ms` : ""}`);
|
|
if (options.expected)
|
|
progress.log(` verifying given screenshot expectation`);
|
|
else
|
|
progress.log(` generating new stable screenshot expectation`);
|
|
let isFirstIteration = true;
|
|
while (true) {
|
|
if (this.isClosed())
|
|
throw new Error("The page has closed");
|
|
const screenshotTimeout = pollIntervals.shift() ?? 1e3;
|
|
if (screenshotTimeout)
|
|
progress.log(`waiting ${screenshotTimeout}ms before taking screenshot`);
|
|
previous = actual;
|
|
actual = await rafrafScreenshot(screenshotTimeout).catch((e) => {
|
|
if (this.mainFrame().isNonRetriableError(e))
|
|
throw e;
|
|
progress.log(`failed to take screenshot - ` + e.message);
|
|
return void 0;
|
|
});
|
|
if (!actual)
|
|
continue;
|
|
const expectation = options.expected && isFirstIteration ? options.expected : previous;
|
|
if (areEqualScreenshots(actual, expectation, previous))
|
|
break;
|
|
if (intermediateResult)
|
|
progress.log(intermediateResult.errorMessage);
|
|
isFirstIteration = false;
|
|
}
|
|
if (!isFirstIteration)
|
|
progress.log(`captured a stable screenshot`);
|
|
if (!options.expected)
|
|
return { actual };
|
|
if (isFirstIteration) {
|
|
progress.log(`screenshot matched expectation`);
|
|
return {};
|
|
}
|
|
if (areEqualScreenshots(actual, options.expected, void 0)) {
|
|
progress.log(`screenshot matched expectation`);
|
|
return {};
|
|
}
|
|
throw new Error(intermediateResult.errorMessage);
|
|
} catch (e) {
|
|
if (js.isJavaScriptErrorInEvaluate(e) || (0, import_selectorParser.isInvalidSelectorError)(e))
|
|
throw e;
|
|
let errorMessage = e.message;
|
|
if (e instanceof import_errors.TimeoutError && intermediateResult?.previous)
|
|
errorMessage = `Failed to take two consecutive stable screenshots.`;
|
|
return {
|
|
log: (0, import_callLog.compressCallLog)(e.message ? [...progress.metadata.log, e.message] : progress.metadata.log),
|
|
...intermediateResult,
|
|
errorMessage,
|
|
timedOut: e instanceof import_errors.TimeoutError
|
|
};
|
|
}
|
|
}
|
|
async screenshot(progress, options) {
|
|
return await this.screenshotter.screenshotPage(progress, options);
|
|
}
|
|
async close(options = {}) {
|
|
if (this._closedState === "closed")
|
|
return;
|
|
if (options.reason)
|
|
this.closeReason = options.reason;
|
|
const runBeforeUnload = !!options.runBeforeUnload;
|
|
if (this._closedState !== "closing") {
|
|
this._closedState = "closing";
|
|
await this.delegate.closePage(runBeforeUnload).catch((e) => import_debugLogger.debugLogger.log("error", e));
|
|
}
|
|
if (!runBeforeUnload)
|
|
await this._closedPromise;
|
|
}
|
|
isClosed() {
|
|
return this._closedState === "closed";
|
|
}
|
|
hasCrashed() {
|
|
return this._crashed;
|
|
}
|
|
isClosedOrClosingOrCrashed() {
|
|
return this._closedState !== "open" || this._crashed;
|
|
}
|
|
addWorker(workerId, worker) {
|
|
this._workers.set(workerId, worker);
|
|
this.emit(Page.Events.Worker, worker);
|
|
}
|
|
removeWorker(workerId) {
|
|
const worker = this._workers.get(workerId);
|
|
if (!worker)
|
|
return;
|
|
worker.didClose();
|
|
this._workers.delete(workerId);
|
|
}
|
|
clearWorkers() {
|
|
for (const [workerId, worker] of this._workers) {
|
|
worker.didClose();
|
|
this._workers.delete(workerId);
|
|
}
|
|
}
|
|
async setFileChooserInterceptedBy(enabled, by) {
|
|
const wasIntercepted = this.fileChooserIntercepted();
|
|
if (enabled)
|
|
this._fileChooserInterceptedBy.add(by);
|
|
else
|
|
this._fileChooserInterceptedBy.delete(by);
|
|
if (wasIntercepted !== this.fileChooserIntercepted())
|
|
await this.delegate.updateFileChooserInterception();
|
|
}
|
|
fileChooserIntercepted() {
|
|
return this._fileChooserInterceptedBy.size > 0;
|
|
}
|
|
frameNavigatedToNewDocument(frame) {
|
|
this.emit(Page.Events.InternalFrameNavigatedToNewDocument, frame);
|
|
const origin = frame.origin();
|
|
if (origin)
|
|
this.browserContext.addVisitedOrigin(origin);
|
|
}
|
|
allInitScripts() {
|
|
const bindings = [...this.browserContext._pageBindings.values(), ...this._pageBindings.values()].map((binding) => binding.initScript);
|
|
if (this.browserContext.bindingsInitScript)
|
|
bindings.unshift(this.browserContext.bindingsInitScript);
|
|
return [...bindings, ...this.browserContext.initScripts, ...this.initScripts];
|
|
}
|
|
getBinding(name) {
|
|
return this._pageBindings.get(name) || this.browserContext._pageBindings.get(name);
|
|
}
|
|
setScreencastOptions(options) {
|
|
this.delegate.setScreencastOptions(options).catch((e) => import_debugLogger.debugLogger.log("error", e));
|
|
this._frameThrottler.setThrottlingEnabled(!!options);
|
|
}
|
|
throttleScreencastFrameAck(ack) {
|
|
this._frameThrottler.ack(ack);
|
|
}
|
|
temporarilyDisableTracingScreencastThrottling() {
|
|
this._frameThrottler.recharge();
|
|
}
|
|
async safeNonStallingEvaluateInAllFrames(expression, world, options = {}) {
|
|
await Promise.all(this.frames().map(async (frame) => {
|
|
try {
|
|
await frame.nonStallingEvaluateInExistingContext(expression, world);
|
|
} catch (e) {
|
|
if (options.throwOnJSErrors && js.isJavaScriptErrorInEvaluate(e))
|
|
throw e;
|
|
}
|
|
}));
|
|
}
|
|
async hideHighlight() {
|
|
await Promise.all(this.frames().map((frame) => frame.hideHighlight().catch(() => {
|
|
})));
|
|
}
|
|
async snapshotForAI(progress) {
|
|
this.lastSnapshotFrameIds = [];
|
|
const snapshot = await snapshotFrameForAI(progress, this.mainFrame(), 0, this.lastSnapshotFrameIds);
|
|
return snapshot.join("\n");
|
|
}
|
|
}
|
|
class Worker extends import_instrumentation.SdkObject {
|
|
constructor(parent, url) {
|
|
super(parent, "worker");
|
|
this.existingExecutionContext = null;
|
|
this.openScope = new import_utils.LongStandingScope();
|
|
this.url = url;
|
|
this._executionContextCallback = () => {
|
|
};
|
|
this._executionContextPromise = new Promise((x) => this._executionContextCallback = x);
|
|
}
|
|
static {
|
|
this.Events = {
|
|
Close: "close"
|
|
};
|
|
}
|
|
createExecutionContext(delegate) {
|
|
this.existingExecutionContext = new js.ExecutionContext(this, delegate, "worker");
|
|
this._executionContextCallback(this.existingExecutionContext);
|
|
return this.existingExecutionContext;
|
|
}
|
|
didClose() {
|
|
if (this.existingExecutionContext)
|
|
this.existingExecutionContext.contextDestroyed("Worker was closed");
|
|
this.emit(Worker.Events.Close, this);
|
|
this.openScope.close(new Error("Worker closed"));
|
|
}
|
|
async evaluateExpression(expression, isFunction, arg) {
|
|
return js.evaluateExpression(await this._executionContextPromise, expression, { returnByValue: true, isFunction }, arg);
|
|
}
|
|
async evaluateExpressionHandle(expression, isFunction, arg) {
|
|
return js.evaluateExpression(await this._executionContextPromise, expression, { returnByValue: false, isFunction }, arg);
|
|
}
|
|
}
|
|
class PageBinding {
|
|
static {
|
|
this.kController = "__playwright__binding__controller__";
|
|
}
|
|
static {
|
|
this.kBindingName = "__playwright__binding__";
|
|
}
|
|
static createInitScript() {
|
|
return new InitScript(`
|
|
(() => {
|
|
const module = {};
|
|
${rawBindingsControllerSource.source}
|
|
const property = '${PageBinding.kController}';
|
|
if (!globalThis[property])
|
|
globalThis[property] = new (module.exports.BindingsController())(globalThis, '${PageBinding.kBindingName}');
|
|
})();
|
|
`);
|
|
}
|
|
constructor(name, playwrightFunction, needsHandle) {
|
|
this.name = name;
|
|
this.playwrightFunction = playwrightFunction;
|
|
this.initScript = new InitScript(`globalThis['${PageBinding.kController}'].addBinding(${JSON.stringify(name)}, ${needsHandle})`);
|
|
this.needsHandle = needsHandle;
|
|
this.cleanupScript = `globalThis['${PageBinding.kController}'].removeBinding(${JSON.stringify(name)})`;
|
|
}
|
|
static async dispatch(page, payload, context) {
|
|
const { name, seq, serializedArgs } = JSON.parse(payload);
|
|
try {
|
|
(0, import_utils.assert)(context.world);
|
|
const binding = page.getBinding(name);
|
|
if (!binding)
|
|
throw new Error(`Function "${name}" is not exposed`);
|
|
let result;
|
|
if (binding.needsHandle) {
|
|
const handle = await context.evaluateExpressionHandle(`arg => globalThis['${PageBinding.kController}'].takeBindingHandle(arg)`, { isFunction: true }, { name, seq }).catch((e) => null);
|
|
result = await binding.playwrightFunction({ frame: context.frame, page, context: page.browserContext }, handle);
|
|
} else {
|
|
if (!Array.isArray(serializedArgs))
|
|
throw new Error(`serializedArgs is not an array. This can happen when Array.prototype.toJSON is defined incorrectly`);
|
|
const args = serializedArgs.map((a) => (0, import_utilityScriptSerializers.parseEvaluationResultValue)(a));
|
|
result = await binding.playwrightFunction({ frame: context.frame, page, context: page.browserContext }, ...args);
|
|
}
|
|
context.evaluateExpressionHandle(`arg => globalThis['${PageBinding.kController}'].deliverBindingResult(arg)`, { isFunction: true }, { name, seq, result }).catch((e) => import_debugLogger.debugLogger.log("error", e));
|
|
} catch (error) {
|
|
context.evaluateExpressionHandle(`arg => globalThis['${PageBinding.kController}'].deliverBindingResult(arg)`, { isFunction: true }, { name, seq, error }).catch((e) => import_debugLogger.debugLogger.log("error", e));
|
|
}
|
|
}
|
|
}
|
|
class InitScript {
|
|
constructor(source) {
|
|
this.source = `(() => {
|
|
${source}
|
|
})();`;
|
|
}
|
|
}
|
|
class FrameThrottler {
|
|
constructor(nonThrottledFrames, defaultInterval, throttlingInterval) {
|
|
this._acks = [];
|
|
this._throttlingEnabled = false;
|
|
this._nonThrottledFrames = nonThrottledFrames;
|
|
this._budget = nonThrottledFrames;
|
|
this._defaultInterval = defaultInterval;
|
|
this._throttlingInterval = throttlingInterval;
|
|
this._tick();
|
|
}
|
|
dispose() {
|
|
if (this._timeoutId) {
|
|
clearTimeout(this._timeoutId);
|
|
this._timeoutId = void 0;
|
|
}
|
|
}
|
|
setThrottlingEnabled(enabled) {
|
|
this._throttlingEnabled = enabled;
|
|
}
|
|
recharge() {
|
|
for (const ack of this._acks)
|
|
ack();
|
|
this._acks = [];
|
|
this._budget = this._nonThrottledFrames;
|
|
if (this._timeoutId) {
|
|
clearTimeout(this._timeoutId);
|
|
this._tick();
|
|
}
|
|
}
|
|
ack(ack) {
|
|
if (!this._timeoutId) {
|
|
ack();
|
|
return;
|
|
}
|
|
this._acks.push(ack);
|
|
}
|
|
_tick() {
|
|
const ack = this._acks.shift();
|
|
if (ack) {
|
|
--this._budget;
|
|
ack();
|
|
}
|
|
if (this._throttlingEnabled && this._budget <= 0) {
|
|
this._timeoutId = setTimeout(() => this._tick(), this._throttlingInterval);
|
|
} else {
|
|
this._timeoutId = setTimeout(() => this._tick(), this._defaultInterval);
|
|
}
|
|
}
|
|
}
|
|
async function snapshotFrameForAI(progress, frame, frameOrdinal, frameIds) {
|
|
const snapshot = await frame.retryWithProgressAndTimeouts(progress, [1e3, 2e3, 4e3, 8e3], async (continuePolling) => {
|
|
try {
|
|
const context = await progress.race(frame._utilityContext());
|
|
const injectedScript = await progress.race(context.injectedScript());
|
|
const snapshotOrRetry = await progress.race(injectedScript.evaluate((injected, refPrefix) => {
|
|
const node = injected.document.body;
|
|
if (!node)
|
|
return true;
|
|
return injected.ariaSnapshot(node, { mode: "ai", refPrefix });
|
|
}, frameOrdinal ? "f" + frameOrdinal : ""));
|
|
if (snapshotOrRetry === true)
|
|
return continuePolling;
|
|
return snapshotOrRetry;
|
|
} catch (e) {
|
|
if (frame.isNonRetriableError(e))
|
|
throw e;
|
|
return continuePolling;
|
|
}
|
|
});
|
|
const lines = snapshot.split("\n");
|
|
const result = [];
|
|
for (const line of lines) {
|
|
const match = line.match(/^(\s*)- iframe (?:\[active\] )?\[ref=([^\]]*)\]/);
|
|
if (!match) {
|
|
result.push(line);
|
|
continue;
|
|
}
|
|
const leadingSpace = match[1];
|
|
const ref = match[2];
|
|
const frameSelector = `aria-ref=${ref} >> internal:control=enter-frame`;
|
|
const frameBodySelector = `${frameSelector} >> body`;
|
|
const child = await progress.race(frame.selectors.resolveFrameForSelector(frameBodySelector, { strict: true }));
|
|
if (!child) {
|
|
result.push(line);
|
|
continue;
|
|
}
|
|
const frameOrdinal2 = frameIds.length + 1;
|
|
frameIds.push(child.frame._id);
|
|
try {
|
|
const childSnapshot = await snapshotFrameForAI(progress, child.frame, frameOrdinal2, frameIds);
|
|
result.push(line + ":", ...childSnapshot.map((l) => leadingSpace + " " + l));
|
|
} catch {
|
|
result.push(line);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
// Annotate the CommonJS export names for ESM import in node:
|
|
0 && (module.exports = {
|
|
InitScript,
|
|
Page,
|
|
PageBinding,
|
|
Worker
|
|
});
|