| import { SelectorType, AttributeAction } from "./types"; |
| const attribValChars = ["\\", '"']; |
| const pseudoValChars = [...attribValChars, "(", ")"]; |
| const charsToEscapeInAttributeValue = new Set(attribValChars.map((c) => c.charCodeAt(0))); |
| const charsToEscapeInPseudoValue = new Set(pseudoValChars.map((c) => c.charCodeAt(0))); |
| const charsToEscapeInName = new Set([ |
| ...pseudoValChars, |
| "~", |
| "^", |
| "$", |
| "*", |
| "+", |
| "!", |
| "|", |
| ":", |
| "[", |
| "]", |
| " ", |
| ".", |
| ].map((c) => c.charCodeAt(0))); |
| /** |
| * Turns `selector` back into a string. |
| * |
| * @param selector Selector to stringify. |
| */ |
| export function stringify(selector) { |
| return selector |
| .map((token) => token.map(stringifyToken).join("")) |
| .join(", "); |
| } |
| function stringifyToken(token, index, arr) { |
| switch (token.type) { |
| // Simple types |
| case SelectorType.Child: |
| return index === 0 ? "> " : " > "; |
| case SelectorType.Parent: |
| return index === 0 ? "< " : " < "; |
| case SelectorType.Sibling: |
| return index === 0 ? "~ " : " ~ "; |
| case SelectorType.Adjacent: |
| return index === 0 ? "+ " : " + "; |
| case SelectorType.Descendant: |
| return " "; |
| case SelectorType.ColumnCombinator: |
| return index === 0 ? "|| " : " || "; |
| case SelectorType.Universal: |
| // Return an empty string if the selector isn't needed. |
| return token.namespace === "*" && |
| index + 1 < arr.length && |
| "name" in arr[index + 1] |
| ? "" |
| : `${getNamespace(token.namespace)}*`; |
| case SelectorType.Tag: |
| return getNamespacedName(token); |
| case SelectorType.PseudoElement: |
| return `::${escapeName(token.name, charsToEscapeInName)}${token.data === null |
| ? "" |
| : `(${escapeName(token.data, charsToEscapeInPseudoValue)})`}`; |
| case SelectorType.Pseudo: |
| return `:${escapeName(token.name, charsToEscapeInName)}${token.data === null |
| ? "" |
| : `(${typeof token.data === "string" |
| ? escapeName(token.data, charsToEscapeInPseudoValue) |
| : stringify(token.data)})`}`; |
| case SelectorType.Attribute: { |
| if (token.name === "id" && |
| token.action === AttributeAction.Equals && |
| token.ignoreCase === "quirks" && |
| !token.namespace) { |
| return `#${escapeName(token.value, charsToEscapeInName)}`; |
| } |
| if (token.name === "class" && |
| token.action === AttributeAction.Element && |
| token.ignoreCase === "quirks" && |
| !token.namespace) { |
| return `.${escapeName(token.value, charsToEscapeInName)}`; |
| } |
| const name = getNamespacedName(token); |
| if (token.action === AttributeAction.Exists) { |
| return `[${name}]`; |
| } |
| return `[${name}${getActionValue(token.action)}="${escapeName(token.value, charsToEscapeInAttributeValue)}"${token.ignoreCase === null ? "" : token.ignoreCase ? " i" : " s"}]`; |
| } |
| } |
| } |
| function getActionValue(action) { |
| switch (action) { |
| case AttributeAction.Equals: |
| return ""; |
| case AttributeAction.Element: |
| return "~"; |
| case AttributeAction.Start: |
| return "^"; |
| case AttributeAction.End: |
| return "$"; |
| case AttributeAction.Any: |
| return "*"; |
| case AttributeAction.Not: |
| return "!"; |
| case AttributeAction.Hyphen: |
| return "|"; |
| case AttributeAction.Exists: |
| throw new Error("Shouldn't be here"); |
| } |
| } |
| function getNamespacedName(token) { |
| return `${getNamespace(token.namespace)}${escapeName(token.name, charsToEscapeInName)}`; |
| } |
| function getNamespace(namespace) { |
| return namespace !== null |
| ? `${namespace === "*" |
| ? "*" |
| : escapeName(namespace, charsToEscapeInName)}|` |
| : ""; |
| } |
| function escapeName(str, charsToEscape) { |
| let lastIdx = 0; |
| let ret = ""; |
| for (let i = 0; i < str.length; i++) { |
| if (charsToEscape.has(str.charCodeAt(i))) { |
| ret += `${str.slice(lastIdx, i)}\\${str.charAt(i)}`; |
| lastIdx = i + 1; |
| } |
| } |
| return ret.length > 0 ? ret + str.slice(lastIdx) : str; |
| } |