123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.FrameSelectors = void 0;
- var _selectorParser = require("../utils/isomorphic/selectorParser");
- var _locatorGenerators = require("../utils/isomorphic/locatorGenerators");
- class FrameSelectors {
- constructor(frame) {
- this.frame = void 0;
- this.frame = frame;
- }
- _parseSelector(selector, options) {
- const strict = typeof (options === null || options === void 0 ? void 0 : options.strict) === 'boolean' ? options.strict : !!this.frame._page.context()._options.strictSelectors;
- return this.frame._page.context().selectors().parseSelector(selector, strict);
- }
- async query(selector, options, scope) {
- const resolved = await this.resolveInjectedForSelector(selector, options, scope);
-
- if (!resolved) return null;
- const handle = await resolved.injected.evaluateHandle((injected, {
- info,
- scope
- }) => {
- return injected.querySelector(info.parsed, scope || document, info.strict);
- }, {
- info: resolved.info,
- scope: resolved.scope
- });
- const elementHandle = handle.asElement();
- if (!elementHandle) {
- handle.dispose();
- return null;
- }
- return adoptIfNeeded(elementHandle, await resolved.frame._mainContext());
- }
- async queryArrayInMainWorld(selector, scope) {
- const resolved = await this.resolveInjectedForSelector(selector, {
- mainWorld: true
- }, scope);
-
- if (!resolved) throw new Error(`Failed to find frame for selector "${selector}"`);
- return await resolved.injected.evaluateHandle((injected, {
- info,
- scope
- }) => {
- return injected.querySelectorAll(info.parsed, scope || document);
- }, {
- info: resolved.info,
- scope: resolved.scope
- });
- }
- async queryCount(selector) {
- const resolved = await this.resolveInjectedForSelector(selector);
-
- if (!resolved) throw new Error(`Failed to find frame for selector "${selector}"`);
- return await resolved.injected.evaluate((injected, {
- info
- }) => {
- return injected.querySelectorAll(info.parsed, document).length;
- }, {
- info: resolved.info
- });
- }
- async queryAll(selector, scope) {
- const resolved = await this.resolveInjectedForSelector(selector, {}, scope);
-
- if (!resolved) return [];
- const arrayHandle = await resolved.injected.evaluateHandle((injected, {
- info,
- scope
- }) => {
- return injected.querySelectorAll(info.parsed, scope || document);
- }, {
- info: resolved.info,
- scope: resolved.scope
- });
- const properties = await arrayHandle.getProperties();
- arrayHandle.dispose();
-
-
- const targetContext = await resolved.frame._mainContext();
- const result = [];
- for (const property of properties.values()) {
- const elementHandle = property.asElement();
- if (elementHandle) result.push(adoptIfNeeded(elementHandle, targetContext));else property.dispose();
- }
- return Promise.all(result);
- }
- async resolveFrameForSelector(selector, options = {}, scope) {
- let frame = this.frame;
- const frameChunks = (0, _selectorParser.splitSelectorByFrame)(selector);
- for (const chunk of frameChunks) {
- (0, _selectorParser.visitAllSelectorParts)(chunk, (part, nested) => {
- if (nested && part.name === 'internal:control' && part.body === 'enter-frame') {
- const locator = (0, _locatorGenerators.asLocator)(this.frame._page.attribution.playwright.options.sdkLanguage, selector);
- throw new _selectorParser.InvalidSelectorError(`Frame locators are not allowed inside composite locators, while querying "${locator}"`);
- }
- });
- }
- for (let i = 0; i < frameChunks.length - 1; ++i) {
- const info = this._parseSelector(frameChunks[i], options);
- const context = await frame._context(info.world);
- const injectedScript = await context.injectedScript();
- const handle = await injectedScript.evaluateHandle((injected, {
- info,
- scope,
- selectorString
- }) => {
- const element = injected.querySelector(info.parsed, scope || document, info.strict);
- if (element && element.nodeName !== 'IFRAME' && element.nodeName !== 'FRAME') throw injected.createStacklessError(`Selector "${selectorString}" resolved to ${injected.previewNode(element)}, <iframe> was expected`);
- return element;
- }, {
- info,
- scope: i === 0 ? scope : undefined,
- selectorString: (0, _selectorParser.stringifySelector)(info.parsed)
- });
- const element = handle.asElement();
- if (!element) return null;
- const maybeFrame = await frame._page._delegate.getContentFrame(element);
- element.dispose();
- if (!maybeFrame) return null;
- frame = maybeFrame;
- }
-
- if (frame !== this.frame) scope = undefined;
- return {
- frame,
- info: frame.selectors._parseSelector(frameChunks[frameChunks.length - 1], options),
- scope
- };
- }
- async resolveInjectedForSelector(selector, options, scope) {
- const resolved = await this.resolveFrameForSelector(selector, options, scope);
-
- if (!resolved) return;
- const context = await resolved.frame._context(options !== null && options !== void 0 && options.mainWorld ? 'main' : resolved.info.world);
- const injected = await context.injectedScript();
- return {
- injected,
- info: resolved.info,
- frame: resolved.frame,
- scope: resolved.scope
- };
- }
- }
- exports.FrameSelectors = FrameSelectors;
- async function adoptIfNeeded(handle, context) {
- if (handle._context === context) return handle;
- const adopted = handle._page._delegate.adoptElementHandle(handle, context);
- handle.dispose();
- return adopted;
- }
|