feat: Support parsing column combinators

fb55 · Dec 25, 2021 · 8030f67 · 8030f67
1 parent a812a1c
commit 8030f67
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 1 deletion.
diff --git a/src/__fixtures__/tests.ts b/src/__fixtures__/tests.ts
@@ -1016,4 +1016,25 @@ export const tests: [
         ],
         "case-sensitive attribute selector",
     ],
+    [
+        "foo || bar",
+        [
+            [
+                {
+                    name: "foo",
+                    namespace: null,
+                    type: SelectorType.Tag,
+                },
+                {
+                    type: SelectorType.ColumnCombinator,
+                },
+                {
+                    name: "bar",
+                    namespace: null,
+                    type: SelectorType.Tag,
+                },
+            ],
+        ],
+        "column combinator",
+    ],
 ];
diff --git a/src/parse.ts b/src/parse.ts
@@ -137,6 +137,7 @@ export function isTraversal(selector: Selector): selector is Traversal {
         case SelectorType.Descendant:
         case SelectorType.Parent:
         case SelectorType.Sibling:
+        case SelectorType.ColumnCombinator:
             return true;
         default:
             return false;
@@ -616,6 +617,14 @@ function parseSelector(
                     name = "*";
                 } else if (firstChar === CharCode.Pipe) {
                     name = "";
+
+                    if (
+                        selector.charCodeAt(selectorIndex + 1) === CharCode.Pipe
+                    ) {
+                        addTraversal(SelectorType.ColumnCombinator);
+                        stripWhitespace(2);
+                        break;
+                    }
                 } else if (reName.test(selector.slice(selectorIndex))) {
                     name = getName(0);
                 } else {

diff --git a/src/stringify.ts b/src/stringify.ts
@@ -32,6 +32,8 @@ function stringifyToken(token: Selector): string {
             return " + ";
         case SelectorType.Descendant:
             return " ";
+        case SelectorType.ColumnCombinator:
+            return " || ";
         case SelectorType.Universal:
             return `${getNamespace(token.namespace)}*`;
 

diff --git a/src/types.ts b/src/types.ts
@@ -42,6 +42,7 @@ export enum SelectorType {
     Descendant = "descendant",
     Parent = "parent",
     Sibling = "sibling",
+    ColumnCombinator = "column-combinator",
 }
 
 export interface AttributeSelector {
@@ -97,4 +98,5 @@ export type TraversalType =
     | SelectorType.Child
     | SelectorType.Descendant
     | SelectorType.Parent
-    | SelectorType.Sibling;
+    | SelectorType.Sibling
+    | SelectorType.ColumnCombinator;