happy-dom 9.9.2 → 9.10.1

package/src/query-selector/SelectorParser.ts

@@ -0,0 +1,148 @@
+import SelectorItem from './SelectorItem';
+import SelectorCombinatorEnum from './SelectorCombinatorEnum';
+import DOMException from '../exception/DOMException';
+
+/**
+ * Selector RegExp.
+ *
+ * Group 1: All (e.g. "*")
+ * Group 2: Tag name (e.g. "div")
+ * Group 3: ID (e.g. "#id")
+ * Group 4: Class (e.g. ".class")
+ * Group 5: Attribute name when no value (e.g. "attr1")
+ * Group 6: Attribute name when there is a value (e.g. "attr1")
+ * Group 7: Attribute operator (e.g. "~")
+ * Group 8: Attribute value (e.g. "value1")
+ * Group 9: Pseudo (e.g. "nth-child")
+ * Group 10: Arguments of pseudo (e.g. "2n + 1")
+ * Group 11: Combinator.
+ */
+const SELECTOR_REGEXP =
+	/(\*)|([a-zA-Z0-9-]+)|#((?:[a-zA-Z0-9-_]|\\.)+)|\.((?:[a-zA-Z0-9-_]|\\.)+)|\[([a-zA-Z0-9-_]+)\]|\[([a-zA-Z0-9-_]+)([~|^$*]{0,1}) *= *["']{0,1}([^"']+)["']{0,1}\]|:([a-zA-Z-:]+)|\(([^)]+)\)|([ ,+>]*)/g;
+
+/**
+ * Escaped Character RegExp.
+ */
+const CLASS_ESCAPED_CHARACTER_REGEXP = /\\/g;
+
+/**
+ * Simple Selector RegExp.
+ *
+ * Group 1: Tag name (e.g. "div")
+ * Group 2: Class (e.g. ".classA.classB")
+ * Group 3: ID (e.g. "#id")
+ */
+const SIMPLE_SELECTOR_REGEXP = /(^[a-zA-Z0-9-]+$)|(^\.[a-zA-Z0-9-_.]+$)|(^#[a-zA-Z0-9-_]+$)/;
+
+/**
+ * Utility for parsing a selection string.
+ */
+export default class SelectorParser {
+	/**
+	 * Parses a selector string and returns an instance of SelectorItem.
+	 *
+	 * @param selector Selector.
+	 * @returns Selector itme.
+	 */
+	public static getSelectorItem(selector: string): SelectorItem {
+		return this.getSelectorGroups(selector)[0][0];
+	}
+
+	/**
+	 * Parses a selector string and returns groups with SelectorItem instances.
+	 *
+	 * @param selector Selector.
+	 * @returns Selector groups.
+	 */
+	public static getSelectorGroups(selector: string): Array<Array<SelectorItem>> {
+		const simpleMatch = selector.match(SIMPLE_SELECTOR_REGEXP);
+
+		if (simpleMatch) {
+			if (simpleMatch[1]) {
+				return [[new SelectorItem({ tagName: selector.toUpperCase() })]];
+			} else if (simpleMatch[2]) {
+				return [[new SelectorItem({ classNames: selector.replace('.', '').split('.') })]];
+			} else if (simpleMatch[3]) {
+				return [[new SelectorItem({ id: selector.replace('#', '') })]];
+			}
+		}
+
+		const regexp = new RegExp(SELECTOR_REGEXP);
+		let currentSelectorItem: SelectorItem = new SelectorItem({
+			combinator: SelectorCombinatorEnum.descendant
+		});
+		let currentGroup: SelectorItem[] = [currentSelectorItem];
+		const groups: Array<Array<SelectorItem>> = [currentGroup];
+		let isValid = false;
+		let match;
+
+		while ((match = regexp.exec(selector))) {
+			if (match[0]) {
+				isValid = true;
+
+				if (match[1]) {
+					currentSelectorItem.all = '*';
+				} else if (match[2]) {
+					currentSelectorItem.tagName = match[2].toUpperCase();
+				} else if (match[3]) {
+					currentSelectorItem.id = match[3].replace(CLASS_ESCAPED_CHARACTER_REGEXP, '');
+				} else if (match[4]) {
+					currentSelectorItem.classNames = currentSelectorItem.classNames || [];
+					currentSelectorItem.classNames.push(match[4].replace(CLASS_ESCAPED_CHARACTER_REGEXP, ''));
+				} else if (match[5]) {
+					currentSelectorItem.attributes = currentSelectorItem.attributes || [];
+					currentSelectorItem.attributes.push({
+						name: match[5].toLowerCase(),
+						operator: null,
+						value: null
+					});
+				} else if (match[6] && match[8]) {
+					currentSelectorItem.attributes = currentSelectorItem.attributes || [];
+					currentSelectorItem.attributes.push({
+						name: match[6].toLowerCase(),
+						operator: match[7] || null,
+						value: match[8]
+					});
+				} else if (match[9]) {
+					currentSelectorItem.pseudoClass = match[9].toLowerCase();
+				} else if (match[10]) {
+					currentSelectorItem.pseudoArguments = match[10];
+				} else if (match[11]) {
+					switch (match[11].trim()) {
+						case ',':
+							currentSelectorItem = new SelectorItem({
+								combinator: SelectorCombinatorEnum.descendant
+							});
+							currentGroup = [currentSelectorItem];
+							groups.push(currentGroup);
+							break;
+						case '>':
+							currentSelectorItem = new SelectorItem({ combinator: SelectorCombinatorEnum.child });
+							currentGroup.push(currentSelectorItem);
+							break;
+						case '+':
+							currentSelectorItem = new SelectorItem({
+								combinator: SelectorCombinatorEnum.adjacentSibling
+							});
+							currentGroup.push(currentSelectorItem);
+							break;
+						case '':
+							currentSelectorItem = new SelectorItem({
+								combinator: SelectorCombinatorEnum.descendant
+							});
+							currentGroup.push(currentSelectorItem);
+							break;
+					}
+				}
+			} else {
+				break;
+			}
+		}
+
+		if (!isValid) {
+			throw new DOMException(`Invalid selector: "${selector}"`);
+		}
+
+		return groups;
+	}
+}