123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244
/// <reference types="./QueryHandler.d.ts" />/** * Copyright 2020 Google Inc. All rights reserved. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * *     http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */import { ariaHandler } from "./AriaQueryHandler.js";function internalizeCustomQueryHandler(handler) {  const internalHandler = {};  if (handler.queryOne) {    const queryOne = handler.queryOne;    internalHandler.queryOne = async (element, selector) => {      const jsHandle = await element.evaluateHandle(queryOne, selector);      const elementHandle = jsHandle.asElement();      if (elementHandle) {        return elementHandle;      }      await jsHandle.dispose();      return null;    };    internalHandler.waitFor = (domWorld, selector, options) => {      return domWorld._waitForSelectorInPage(queryOne, selector, options);    };  }  if (handler.queryAll) {    const queryAll = handler.queryAll;    internalHandler.queryAll = async (element, selector) => {      const jsHandle = await element.evaluateHandle(queryAll, selector);      const properties = await jsHandle.getProperties();      await jsHandle.dispose();      const result = [];      for (const property of properties.values()) {        const elementHandle = property.asElement();        if (elementHandle) {          result.push(elementHandle);        }      }      return result;    };    internalHandler.queryAllArray = async (element, selector) => {      const resultHandle = (await element.evaluateHandle(queryAll, selector));      const arrayHandle = await resultHandle.evaluateHandle((res) => {        return Array.from(res);      });      return arrayHandle;    };  }  return internalHandler;}const defaultHandler = internalizeCustomQueryHandler({  queryOne: (element, selector) => {    if (!("querySelector" in element)) {      throw new Error(        `Could not invoke \`querySelector\` on node of type ${element.nodeName}.`,      );    }    return element.querySelector(selector);  },  queryAll: (element, selector) => {    if (!("querySelectorAll" in element)) {      throw new Error(        `Could not invoke \`querySelectorAll\` on node of type ${element.nodeName}.`,      );    }    return [      ...element.querySelectorAll(selector),    ];  },});const pierceHandler = internalizeCustomQueryHandler({  queryOne: (element, selector) => {    let found = null;    const search = (root) => {      const iter = document.createTreeWalker(root, NodeFilter.SHOW_ELEMENT);      do {        const currentNode = iter.currentNode;        if (currentNode.shadowRoot) {          search(currentNode.shadowRoot);        }        if (currentNode instanceof ShadowRoot) {          continue;        }        if (currentNode !== root && !found && currentNode.matches(selector)) {          found = currentNode;        }      } while (!found && iter.nextNode());    };    if (element instanceof Document) {      element = element.documentElement;    }    search(element);    return found;  },  queryAll: (element, selector) => {    const result = [];    const collect = (root) => {      const iter = document.createTreeWalker(root, NodeFilter.SHOW_ELEMENT);      do {        const currentNode = iter.currentNode;        if (currentNode.shadowRoot) {          collect(currentNode.shadowRoot);        }        if (currentNode instanceof ShadowRoot) {          continue;        }        if (currentNode !== root && currentNode.matches(selector)) {          result.push(currentNode);        }      } while (iter.nextNode());    };    if (element instanceof Document) {      element = element.documentElement;    }    collect(element);    return result;  },});const xpathHandler = internalizeCustomQueryHandler({  queryOne: (element, selector) => {    const doc = element.ownerDocument || document;    const result = doc.evaluate(      selector,      element,      null,      XPathResult.FIRST_ORDERED_NODE_TYPE,    );    return result.singleNodeValue;  },  queryAll: (element, selector) => {    const doc = element.ownerDocument || document;    const iterator = doc.evaluate(      selector,      element,      null,      XPathResult.ORDERED_NODE_ITERATOR_TYPE,    );    const array = [];    let item;    while ((item = iterator.iterateNext())) {      array.push(item);    }    return array;  },});const INTERNAL_QUERY_HANDLERS = new Map([  ["aria", { handler: ariaHandler }],  ["pierce", { handler: pierceHandler }],  ["xpath", { handler: xpathHandler }],]);const QUERY_HANDLERS = new Map();/** * Registers a {@link CustomQueryHandler | custom query handler}. * * @remarks * After registration, the handler can be used everywhere where a selector is * expected by prepending the selection string with `<name>/`. The name is only * allowed to consist of lower- and upper case latin letters. * * @example * * ``` * puppeteer.registerCustomQueryHandler('text', { … }); * const aHandle = await page.$('text/…'); * ``` * * @param name - The name that the custom query handler will be registered * under. * @param queryHandler - The {@link CustomQueryHandler | custom query handler} * to register. * * @public */export function registerCustomQueryHandler(name, handler) {  if (INTERNAL_QUERY_HANDLERS.has(name)) {    throw new Error(`A query handler named "${name}" already exists`);  }  if (QUERY_HANDLERS.has(name)) {    throw new Error(`A custom query handler named "${name}" already exists`);  }  const isValidName = /^[a-zA-Z]+$/.test(name);  if (!isValidName) {    throw new Error(`Custom query handler names may only contain [a-zA-Z]`);  }  QUERY_HANDLERS.set(name, { handler: internalizeCustomQueryHandler(handler) });}/** * @param name - The name of the query handler to unregistered. * * @public */export function unregisterCustomQueryHandler(name) {  QUERY_HANDLERS.delete(name);}/** * @returns a list with the names of all registered custom query handlers. * * @public */export function customQueryHandlerNames() {  return [...QUERY_HANDLERS.keys()];}/** * Clears all registered handlers. * * @public */export function clearCustomQueryHandlers() {  QUERY_HANDLERS.clear();}const CUSTOM_QUERY_SEPARATORS = ["=", "/"];/** * @internal */export function getQueryHandlerAndSelector(selector) {  for (const handlerMap of [QUERY_HANDLERS, INTERNAL_QUERY_HANDLERS]) {    for (      const [name, { handler: queryHandler, transformSelector }] of handlerMap    ) {      for (const separator of CUSTOM_QUERY_SEPARATORS) {        const prefix = `${name}${separator}`;        if (selector.startsWith(prefix)) {          selector = selector.slice(prefix.length);          if (transformSelector) {            selector = transformSelector(selector);          }          return { updatedSelector: selector, queryHandler };        }      }    }  }  return { updatedSelector: selector, queryHandler: defaultHandler };}//# sourceMappingURL=QueryHandler.js.map