421 lines
16 KiB
JavaScript
421 lines
16 KiB
JavaScript
import { SelectorType, AttributeAction, } from "./types";
|
|
const reName = /^[^\\#]?(?:\\(?:[\da-f]{1,6}\s?|.)|[\w\-\u00b0-\uFFFF])+/;
|
|
const reEscape = /\\([\da-f]{1,6}\s?|(\s)|.)/gi;
|
|
const actionTypes = new Map([
|
|
[126 /* Tilde */, AttributeAction.Element],
|
|
[94 /* Circumflex */, AttributeAction.Start],
|
|
[36 /* Dollar */, AttributeAction.End],
|
|
[42 /* Asterisk */, AttributeAction.Any],
|
|
[33 /* ExclamationMark */, AttributeAction.Not],
|
|
[124 /* Pipe */, AttributeAction.Hyphen],
|
|
]);
|
|
// Pseudos, whose data property is parsed as well.
|
|
const unpackPseudos = new Set([
|
|
"has",
|
|
"not",
|
|
"matches",
|
|
"is",
|
|
"where",
|
|
"host",
|
|
"host-context",
|
|
]);
|
|
/**
|
|
* Checks whether a specific selector is a traversal.
|
|
* This is useful eg. in swapping the order of elements that
|
|
* are not traversals.
|
|
*
|
|
* @param selector Selector to check.
|
|
*/
|
|
export function isTraversal(selector) {
|
|
switch (selector.type) {
|
|
case SelectorType.Adjacent:
|
|
case SelectorType.Child:
|
|
case SelectorType.Descendant:
|
|
case SelectorType.Parent:
|
|
case SelectorType.Sibling:
|
|
case SelectorType.ColumnCombinator:
|
|
return true;
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
const stripQuotesFromPseudos = new Set(["contains", "icontains"]);
|
|
// Unescape function taken from https://github.com/jquery/sizzle/blob/master/src/sizzle.js#L152
|
|
function funescape(_, escaped, escapedWhitespace) {
|
|
const high = parseInt(escaped, 16) - 0x10000;
|
|
// NaN means non-codepoint
|
|
return high !== high || escapedWhitespace
|
|
? escaped
|
|
: high < 0
|
|
? // BMP codepoint
|
|
String.fromCharCode(high + 0x10000)
|
|
: // Supplemental Plane codepoint (surrogate pair)
|
|
String.fromCharCode((high >> 10) | 0xd800, (high & 0x3ff) | 0xdc00);
|
|
}
|
|
function unescapeCSS(str) {
|
|
return str.replace(reEscape, funescape);
|
|
}
|
|
function isQuote(c) {
|
|
return c === 39 /* SingleQuote */ || c === 34 /* DoubleQuote */;
|
|
}
|
|
function isWhitespace(c) {
|
|
return (c === 32 /* Space */ ||
|
|
c === 9 /* Tab */ ||
|
|
c === 10 /* NewLine */ ||
|
|
c === 12 /* FormFeed */ ||
|
|
c === 13 /* CarriageReturn */);
|
|
}
|
|
/**
|
|
* Parses `selector`, optionally with the passed `options`.
|
|
*
|
|
* @param selector Selector to parse.
|
|
* @param options Options for parsing.
|
|
* @returns Returns a two-dimensional array.
|
|
* The first dimension represents selectors separated by commas (eg. `sub1, sub2`),
|
|
* the second contains the relevant tokens for that selector.
|
|
*/
|
|
export function parse(selector) {
|
|
const subselects = [];
|
|
const endIndex = parseSelector(subselects, `${selector}`, 0);
|
|
if (endIndex < selector.length) {
|
|
throw new Error(`Unmatched selector: ${selector.slice(endIndex)}`);
|
|
}
|
|
return subselects;
|
|
}
|
|
function parseSelector(subselects, selector, selectorIndex) {
|
|
let tokens = [];
|
|
function getName(offset) {
|
|
const match = selector.slice(selectorIndex + offset).match(reName);
|
|
if (!match) {
|
|
throw new Error(`Expected name, found ${selector.slice(selectorIndex)}`);
|
|
}
|
|
const [name] = match;
|
|
selectorIndex += offset + name.length;
|
|
return unescapeCSS(name);
|
|
}
|
|
function stripWhitespace(offset) {
|
|
selectorIndex += offset;
|
|
while (selectorIndex < selector.length &&
|
|
isWhitespace(selector.charCodeAt(selectorIndex))) {
|
|
selectorIndex++;
|
|
}
|
|
}
|
|
function readValueWithParenthesis() {
|
|
selectorIndex += 1;
|
|
const start = selectorIndex;
|
|
let counter = 1;
|
|
for (; counter > 0 && selectorIndex < selector.length; selectorIndex++) {
|
|
if (selector.charCodeAt(selectorIndex) ===
|
|
40 /* LeftParenthesis */ &&
|
|
!isEscaped(selectorIndex)) {
|
|
counter++;
|
|
}
|
|
else if (selector.charCodeAt(selectorIndex) ===
|
|
41 /* RightParenthesis */ &&
|
|
!isEscaped(selectorIndex)) {
|
|
counter--;
|
|
}
|
|
}
|
|
if (counter) {
|
|
throw new Error("Parenthesis not matched");
|
|
}
|
|
return unescapeCSS(selector.slice(start, selectorIndex - 1));
|
|
}
|
|
function isEscaped(pos) {
|
|
let slashCount = 0;
|
|
while (selector.charCodeAt(--pos) === 92 /* BackSlash */)
|
|
slashCount++;
|
|
return (slashCount & 1) === 1;
|
|
}
|
|
function ensureNotTraversal() {
|
|
if (tokens.length > 0 && isTraversal(tokens[tokens.length - 1])) {
|
|
throw new Error("Did not expect successive traversals.");
|
|
}
|
|
}
|
|
function addTraversal(type) {
|
|
if (tokens.length > 0 &&
|
|
tokens[tokens.length - 1].type === SelectorType.Descendant) {
|
|
tokens[tokens.length - 1].type = type;
|
|
return;
|
|
}
|
|
ensureNotTraversal();
|
|
tokens.push({ type });
|
|
}
|
|
function addSpecialAttribute(name, action) {
|
|
tokens.push({
|
|
type: SelectorType.Attribute,
|
|
name,
|
|
action,
|
|
value: getName(1),
|
|
namespace: null,
|
|
ignoreCase: "quirks",
|
|
});
|
|
}
|
|
/**
|
|
* We have finished parsing the current part of the selector.
|
|
*
|
|
* Remove descendant tokens at the end if they exist,
|
|
* and return the last index, so that parsing can be
|
|
* picked up from here.
|
|
*/
|
|
function finalizeSubselector() {
|
|
if (tokens.length &&
|
|
tokens[tokens.length - 1].type === SelectorType.Descendant) {
|
|
tokens.pop();
|
|
}
|
|
if (tokens.length === 0) {
|
|
throw new Error("Empty sub-selector");
|
|
}
|
|
subselects.push(tokens);
|
|
}
|
|
stripWhitespace(0);
|
|
if (selector.length === selectorIndex) {
|
|
return selectorIndex;
|
|
}
|
|
loop: while (selectorIndex < selector.length) {
|
|
const firstChar = selector.charCodeAt(selectorIndex);
|
|
switch (firstChar) {
|
|
// Whitespace
|
|
case 32 /* Space */:
|
|
case 9 /* Tab */:
|
|
case 10 /* NewLine */:
|
|
case 12 /* FormFeed */:
|
|
case 13 /* CarriageReturn */: {
|
|
if (tokens.length === 0 ||
|
|
tokens[0].type !== SelectorType.Descendant) {
|
|
ensureNotTraversal();
|
|
tokens.push({ type: SelectorType.Descendant });
|
|
}
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
// Traversals
|
|
case 62 /* GreaterThan */: {
|
|
addTraversal(SelectorType.Child);
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
case 60 /* LessThan */: {
|
|
addTraversal(SelectorType.Parent);
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
case 126 /* Tilde */: {
|
|
addTraversal(SelectorType.Sibling);
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
case 43 /* Plus */: {
|
|
addTraversal(SelectorType.Adjacent);
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
// Special attribute selectors: .class, #id
|
|
case 46 /* Period */: {
|
|
addSpecialAttribute("class", AttributeAction.Element);
|
|
break;
|
|
}
|
|
case 35 /* Hash */: {
|
|
addSpecialAttribute("id", AttributeAction.Equals);
|
|
break;
|
|
}
|
|
case 91 /* LeftSquareBracket */: {
|
|
stripWhitespace(1);
|
|
// Determine attribute name and namespace
|
|
let name;
|
|
let namespace = null;
|
|
if (selector.charCodeAt(selectorIndex) === 124 /* Pipe */) {
|
|
// Equivalent to no namespace
|
|
name = getName(1);
|
|
}
|
|
else if (selector.startsWith("*|", selectorIndex)) {
|
|
namespace = "*";
|
|
name = getName(2);
|
|
}
|
|
else {
|
|
name = getName(0);
|
|
if (selector.charCodeAt(selectorIndex) === 124 /* Pipe */ &&
|
|
selector.charCodeAt(selectorIndex + 1) !==
|
|
61 /* Equal */) {
|
|
namespace = name;
|
|
name = getName(1);
|
|
}
|
|
}
|
|
stripWhitespace(0);
|
|
// Determine comparison operation
|
|
let action = AttributeAction.Exists;
|
|
const possibleAction = actionTypes.get(selector.charCodeAt(selectorIndex));
|
|
if (possibleAction) {
|
|
action = possibleAction;
|
|
if (selector.charCodeAt(selectorIndex + 1) !==
|
|
61 /* Equal */) {
|
|
throw new Error("Expected `=`");
|
|
}
|
|
stripWhitespace(2);
|
|
}
|
|
else if (selector.charCodeAt(selectorIndex) === 61 /* Equal */) {
|
|
action = AttributeAction.Equals;
|
|
stripWhitespace(1);
|
|
}
|
|
// Determine value
|
|
let value = "";
|
|
let ignoreCase = null;
|
|
if (action !== "exists") {
|
|
if (isQuote(selector.charCodeAt(selectorIndex))) {
|
|
const quote = selector.charCodeAt(selectorIndex);
|
|
let sectionEnd = selectorIndex + 1;
|
|
while (sectionEnd < selector.length &&
|
|
(selector.charCodeAt(sectionEnd) !== quote ||
|
|
isEscaped(sectionEnd))) {
|
|
sectionEnd += 1;
|
|
}
|
|
if (selector.charCodeAt(sectionEnd) !== quote) {
|
|
throw new Error("Attribute value didn't end");
|
|
}
|
|
value = unescapeCSS(selector.slice(selectorIndex + 1, sectionEnd));
|
|
selectorIndex = sectionEnd + 1;
|
|
}
|
|
else {
|
|
const valueStart = selectorIndex;
|
|
while (selectorIndex < selector.length &&
|
|
((!isWhitespace(selector.charCodeAt(selectorIndex)) &&
|
|
selector.charCodeAt(selectorIndex) !==
|
|
93 /* RightSquareBracket */) ||
|
|
isEscaped(selectorIndex))) {
|
|
selectorIndex += 1;
|
|
}
|
|
value = unescapeCSS(selector.slice(valueStart, selectorIndex));
|
|
}
|
|
stripWhitespace(0);
|
|
// See if we have a force ignore flag
|
|
const forceIgnore = selector.charCodeAt(selectorIndex) | 0x20;
|
|
// If the forceIgnore flag is set (either `i` or `s`), use that value
|
|
if (forceIgnore === 115 /* LowerS */) {
|
|
ignoreCase = false;
|
|
stripWhitespace(1);
|
|
}
|
|
else if (forceIgnore === 105 /* LowerI */) {
|
|
ignoreCase = true;
|
|
stripWhitespace(1);
|
|
}
|
|
}
|
|
if (selector.charCodeAt(selectorIndex) !==
|
|
93 /* RightSquareBracket */) {
|
|
throw new Error("Attribute selector didn't terminate");
|
|
}
|
|
selectorIndex += 1;
|
|
const attributeSelector = {
|
|
type: SelectorType.Attribute,
|
|
name,
|
|
action,
|
|
value,
|
|
namespace,
|
|
ignoreCase,
|
|
};
|
|
tokens.push(attributeSelector);
|
|
break;
|
|
}
|
|
case 58 /* Colon */: {
|
|
if (selector.charCodeAt(selectorIndex + 1) === 58 /* Colon */) {
|
|
tokens.push({
|
|
type: SelectorType.PseudoElement,
|
|
name: getName(2).toLowerCase(),
|
|
data: selector.charCodeAt(selectorIndex) ===
|
|
40 /* LeftParenthesis */
|
|
? readValueWithParenthesis()
|
|
: null,
|
|
});
|
|
continue;
|
|
}
|
|
const name = getName(1).toLowerCase();
|
|
let data = null;
|
|
if (selector.charCodeAt(selectorIndex) ===
|
|
40 /* LeftParenthesis */) {
|
|
if (unpackPseudos.has(name)) {
|
|
if (isQuote(selector.charCodeAt(selectorIndex + 1))) {
|
|
throw new Error(`Pseudo-selector ${name} cannot be quoted`);
|
|
}
|
|
data = [];
|
|
selectorIndex = parseSelector(data, selector, selectorIndex + 1);
|
|
if (selector.charCodeAt(selectorIndex) !==
|
|
41 /* RightParenthesis */) {
|
|
throw new Error(`Missing closing parenthesis in :${name} (${selector})`);
|
|
}
|
|
selectorIndex += 1;
|
|
}
|
|
else {
|
|
data = readValueWithParenthesis();
|
|
if (stripQuotesFromPseudos.has(name)) {
|
|
const quot = data.charCodeAt(0);
|
|
if (quot === data.charCodeAt(data.length - 1) &&
|
|
isQuote(quot)) {
|
|
data = data.slice(1, -1);
|
|
}
|
|
}
|
|
data = unescapeCSS(data);
|
|
}
|
|
}
|
|
tokens.push({ type: SelectorType.Pseudo, name, data });
|
|
break;
|
|
}
|
|
case 44 /* Comma */: {
|
|
finalizeSubselector();
|
|
tokens = [];
|
|
stripWhitespace(1);
|
|
break;
|
|
}
|
|
default: {
|
|
if (selector.startsWith("/*", selectorIndex)) {
|
|
const endIndex = selector.indexOf("*/", selectorIndex + 2);
|
|
if (endIndex < 0) {
|
|
throw new Error("Comment was not terminated");
|
|
}
|
|
selectorIndex = endIndex + 2;
|
|
// Remove leading whitespace
|
|
if (tokens.length === 0) {
|
|
stripWhitespace(0);
|
|
}
|
|
break;
|
|
}
|
|
let namespace = null;
|
|
let name;
|
|
if (firstChar === 42 /* Asterisk */) {
|
|
selectorIndex += 1;
|
|
name = "*";
|
|
}
|
|
else if (firstChar === 124 /* Pipe */) {
|
|
name = "";
|
|
if (selector.charCodeAt(selectorIndex + 1) === 124 /* Pipe */) {
|
|
addTraversal(SelectorType.ColumnCombinator);
|
|
stripWhitespace(2);
|
|
break;
|
|
}
|
|
}
|
|
else if (reName.test(selector.slice(selectorIndex))) {
|
|
name = getName(0);
|
|
}
|
|
else {
|
|
break loop;
|
|
}
|
|
if (selector.charCodeAt(selectorIndex) === 124 /* Pipe */ &&
|
|
selector.charCodeAt(selectorIndex + 1) !== 124 /* Pipe */) {
|
|
namespace = name;
|
|
if (selector.charCodeAt(selectorIndex + 1) ===
|
|
42 /* Asterisk */) {
|
|
name = "*";
|
|
selectorIndex += 2;
|
|
}
|
|
else {
|
|
name = getName(1);
|
|
}
|
|
}
|
|
tokens.push(name === "*"
|
|
? { type: SelectorType.Universal, namespace }
|
|
: { type: SelectorType.Tag, name, namespace });
|
|
}
|
|
}
|
|
}
|
|
finalizeSubselector();
|
|
return selectorIndex;
|
|
}
|