import * as PropertySymbol from '../PropertySymbol.js'; import NodeList from '../nodes/node/NodeList.js'; import NodeTypeEnum from '../nodes/node/NodeTypeEnum.js'; import SelectorCombinatorEnum from './SelectorCombinatorEnum.js'; import SelectorParser from './SelectorParser.js'; /** * Invalid Selector RegExp. */ const INVALID_SELECTOR_REGEXP = /^[.#\[]?\d|[.#]$/; /** * Utility for query selection in an HTML element. * * @class QuerySelector */ export default class QuerySelector { /** * Finds elements based on a query selector. * * @param node Node to search in. * @param selector Selector. * @returns HTML elements. */ static querySelectorAll(node, selector) { if (selector === '') { throw new Error(`Failed to execute 'querySelectorAll' on '${node.constructor.name}': The provided selector is empty.`); } if (selector === null || selector === undefined) { return new NodeList(); } if (INVALID_SELECTOR_REGEXP.test(selector)) { throw new Error(`Failed to execute 'querySelectorAll' on '${node.constructor.name}': '${selector}' is not a valid selector.`); } const groups = SelectorParser.getSelectorGroups(selector); let matches = []; for (const items of groups) { matches = matches.concat(node[PropertySymbol.nodeType] === NodeTypeEnum.elementNode ? this.findAll(node, [node], items) : this.findAll(null, node[PropertySymbol.children], items)); } const nodeList = new NodeList(); const matchesMap = {}; for (let i = 0, max = matches.length; i < max; i++) { matchesMap[matches[i].documentPosition] = matches[i].element; } const keys = Object.keys(matchesMap).sort(); for (let i = 0, max = keys.length; i < max; i++) { nodeList.push(matchesMap[keys[i]]); } return nodeList; } /** * Finds an element based on a query selector. * * @param node Node to search in. * @param selector Selector. * @returns HTML element. */ static querySelector(node, selector) { if (selector === '') { throw new Error(`Failed to execute 'querySelector' on '${node.constructor.name}': The provided selector is empty.`); } if (selector === null || selector === undefined) { return null; } if (INVALID_SELECTOR_REGEXP.test(selector)) { throw new Error(`Failed to execute 'querySelector' on '${node.constructor.name}': '${selector}' is not a valid selector.`); } for (const items of SelectorParser.getSelectorGroups(selector)) { const match = node[PropertySymbol.nodeType] === NodeTypeEnum.elementNode ? this.findFirst(node, [node], items) : this.findFirst(null, node[PropertySymbol.children], items); if (match) { return match; } } return null; } /** * Checks if an element matches a selector and returns priority weight. * * @param element Element to match. * @param selector Selector to match with. * @returns Result. */ static match(element, selector) { if (!selector) { return null; } if (selector === '*') { return { priorityWeight: 1 }; } if (INVALID_SELECTOR_REGEXP.test(selector)) { throw new Error(`Failed to execute 'match' on '${element.constructor.name}': '${selector}' is not a valid selector.`); } for (const items of SelectorParser.getSelectorGroups(selector)) { const result = this.matchSelector(element, element, items.reverse()); if (result) { return result; } } return null; } /** * Checks if a node matches a selector. * * @param targetElement Target element. * @param currentElement Current element. * @param selectorItems Selector items. * @param [priorityWeight] Priority weight. * @returns Result. */ static matchSelector(targetElement, currentElement, selectorItems, priorityWeight = 0) { const selectorItem = selectorItems[0]; const result = selectorItem.match(currentElement); if (result) { if (selectorItems.length === 1) { return { priorityWeight: priorityWeight + result.priorityWeight }; } switch (selectorItem.combinator) { case SelectorCombinatorEnum.adjacentSibling: if (currentElement.previousElementSibling) { const match = this.matchSelector(targetElement, currentElement.previousElementSibling, selectorItems.slice(1), priorityWeight + result.priorityWeight); if (match) { return match; } } break; case SelectorCombinatorEnum.child: case SelectorCombinatorEnum.descendant: if (currentElement.parentElement) { const match = this.matchSelector(targetElement, currentElement.parentElement, selectorItems.slice(1), priorityWeight + result.priorityWeight); if (match) { return match; } } break; } } if (selectorItem.combinator === SelectorCombinatorEnum.descendant && targetElement !== currentElement && currentElement.parentElement) { return this.matchSelector(targetElement, currentElement.parentElement, selectorItems, priorityWeight); } return null; } /** * Finds elements based on a query selector for a part of a list of selectors separated with comma. * * @param rootElement Root element. * @param children Child elements. * @param selectorItems Selector items. * @param [documentPosition] Document position of the element. * @returns Document position and element map. */ static findAll(rootElement, children, selectorItems, documentPosition) { const selectorItem = selectorItems[0]; const nextSelectorItem = selectorItems[1]; let matched = []; for (let i = 0, max = children.length; i < max; i++) { const child = children[i]; const position = (documentPosition ? documentPosition + '>' : '') + String.fromCharCode(i); if (selectorItem.match(child)) { if (!nextSelectorItem) { if (rootElement !== child) { matched.push({ documentPosition: position, element: child }); } } else { switch (nextSelectorItem.combinator) { case SelectorCombinatorEnum.adjacentSibling: if (child.nextElementSibling) { matched = matched.concat(this.findAll(rootElement, [child.nextElementSibling], selectorItems.slice(1), position)); } break; case SelectorCombinatorEnum.descendant: case SelectorCombinatorEnum.child: matched = matched.concat(this.findAll(rootElement, child[PropertySymbol.children], selectorItems.slice(1), position)); break; } } } if (selectorItem.combinator === SelectorCombinatorEnum.descendant && child[PropertySymbol.children].length) { matched = matched.concat(this.findAll(rootElement, child[PropertySymbol.children], selectorItems, position)); } } return matched; } /** * Finds an element based on a query selector for a part of a list of selectors separated with comma. * * @param rootElement Root element. * @param children Child elements. * @param selectorItems Selector items. * @returns HTML element. */ static findFirst(rootElement, children, selectorItems) { const selectorItem = selectorItems[0]; const nextSelectorItem = selectorItems[1]; for (const child of children) { if (selectorItem.match(child)) { if (!nextSelectorItem) { if (rootElement !== child) { return child; } } else { switch (nextSelectorItem.combinator) { case SelectorCombinatorEnum.adjacentSibling: if (child.nextElementSibling) { const match = this.findFirst(rootElement, [child.nextElementSibling], selectorItems.slice(1)); if (match) { return match; } } break; case SelectorCombinatorEnum.descendant: case SelectorCombinatorEnum.child: const match = this.findFirst(rootElement, child[PropertySymbol.children], selectorItems.slice(1)); if (match) { return match; } break; } } } if (selectorItem.combinator === SelectorCombinatorEnum.descendant && child[PropertySymbol.children].length) { const match = this.findFirst(rootElement, child[PropertySymbol.children], selectorItems); if (match) { return match; } } } return null; } } //# sourceMappingURL=QuerySelector.js.map