Skip to content

Commit

Permalink
chore: rename internal query handler
Browse files Browse the repository at this point in the history
  • Loading branch information
jrandolf committed Sep 15, 2022
1 parent a205f7b commit 0bcde40
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 15 deletions.
10 changes: 5 additions & 5 deletions src/common/AriaQueryHandler.ts
Expand Up @@ -20,7 +20,7 @@ import {CDPSession} from './Connection.js';
import {ElementHandle} from './ElementHandle.js';
import {Frame} from './Frame.js';
import {MAIN_WORLD, PUPPETEER_WORLD} from './IsolatedWorld.js';
import {InternalQueryHandler} from './QueryHandler.js';
import {PuppeteerQueryHandler} from './QueryHandler.js';

async function queryAXTree(
client: CDPSession,
Expand Down Expand Up @@ -95,7 +95,7 @@ const queryOneId = async (element: ElementHandle<Node>, selector: string) => {
return res[0].backendDOMNodeId;
};

const queryOne: InternalQueryHandler['queryOne'] = async (
const queryOne: PuppeteerQueryHandler['queryOne'] = async (
element,
selector
) => {
Expand All @@ -108,7 +108,7 @@ const queryOne: InternalQueryHandler['queryOne'] = async (
)) as ElementHandle<Node>;
};

const waitFor: InternalQueryHandler['waitFor'] = async (
const waitFor: PuppeteerQueryHandler['waitFor'] = async (
elementOrFrame,
selector,
options
Expand Down Expand Up @@ -158,7 +158,7 @@ const waitFor: InternalQueryHandler['waitFor'] = async (
return result.frame.worlds[MAIN_WORLD].transferHandle(result);
};

const queryAll: InternalQueryHandler['queryAll'] = async (
const queryAll: PuppeteerQueryHandler['queryAll'] = async (
element,
selector
) => {
Expand All @@ -178,7 +178,7 @@ const queryAll: InternalQueryHandler['queryAll'] = async (
/**
* @internal
*/
export const ariaHandler: InternalQueryHandler = {
export const ariaHandler: PuppeteerQueryHandler = {
queryOne,
waitFor,
queryAll,
Expand Down
20 changes: 10 additions & 10 deletions src/common/QueryHandler.ts
Expand Up @@ -40,7 +40,7 @@ export interface CustomQueryHandler {
/**
* @internal
*/
export interface InternalQueryHandler {
export interface PuppeteerQueryHandler {
/**
* Queries for a single node given a selector and {@link ElementHandle}.
*
Expand Down Expand Up @@ -71,10 +71,10 @@ export interface InternalQueryHandler {
) => Promise<ElementHandle<Node> | null>;
}

function internalizeCustomQueryHandler(
function createPuppeteerQueryHandler(
handler: CustomQueryHandler
): InternalQueryHandler {
const internalHandler: InternalQueryHandler = {};
): PuppeteerQueryHandler {
const internalHandler: PuppeteerQueryHandler = {};

if (handler.queryOne) {
const queryOne = handler.queryOne;
Expand Down Expand Up @@ -138,7 +138,7 @@ function internalizeCustomQueryHandler(
return internalHandler;
}

const defaultHandler = internalizeCustomQueryHandler({
const defaultHandler = createPuppeteerQueryHandler({
queryOne: (element, selector) => {
if (!('querySelector' in element)) {
throw new Error(
Expand All @@ -165,7 +165,7 @@ const defaultHandler = internalizeCustomQueryHandler({
},
});

const pierceHandler = internalizeCustomQueryHandler({
const pierceHandler = createPuppeteerQueryHandler({
queryOne: (element, selector) => {
let found: Node | null = null;
const search = (root: Node) => {
Expand Down Expand Up @@ -215,7 +215,7 @@ const pierceHandler = internalizeCustomQueryHandler({
},
});

const xpathHandler = internalizeCustomQueryHandler({
const xpathHandler = createPuppeteerQueryHandler({
queryOne: (element, selector) => {
const doc = element.ownerDocument || document;
const result = doc.evaluate(
Expand Down Expand Up @@ -245,7 +245,7 @@ const xpathHandler = internalizeCustomQueryHandler({
});

interface RegisteredQueryHandler {
handler: InternalQueryHandler;
handler: PuppeteerQueryHandler;
transformSelector?: (selector: string) => string;
}

Expand Down Expand Up @@ -294,7 +294,7 @@ export function registerCustomQueryHandler(
throw new Error(`Custom query handler names may only contain [a-zA-Z]`);
}

QUERY_HANDLERS.set(name, {handler: internalizeCustomQueryHandler(handler)});
QUERY_HANDLERS.set(name, {handler: createPuppeteerQueryHandler(handler)});
}

/**
Expand Down Expand Up @@ -331,7 +331,7 @@ const CUSTOM_QUERY_SEPARATORS = ['=', '/'];
*/
export function getQueryHandlerAndSelector(selector: string): {
updatedSelector: string;
queryHandler: InternalQueryHandler;
queryHandler: PuppeteerQueryHandler;
} {
for (const handlerMap of [QUERY_HANDLERS, INTERNAL_QUERY_HANDLERS]) {
for (const [
Expand Down

0 comments on commit 0bcde40

Please sign in to comment.