Add universal selector

Author: eush77

README.md

@@ -28,8 +28,8 @@ select(ast, 'paragraph emphasis > text')
 - [x] Sibling selectors: `paragraph ~ text`
 - [x] Adjacent sibling selectors: `paragraph + text`
 - [x] Group selectors: `paragraph, text`
+- [x] Universal selector: `*`
 - [ ] Attribute selectors: `text[value*="substr"]`
-- [ ] Universal selectors: `*`
 
 ## API
 

lib/select.js

@@ -46,22 +46,24 @@ select.rule = function (selector, ast) {
   return result;
 
   function walk (node, parent) {
-    if (node.type == selector.tagName) {
+    if (selector.tagName == '*' || node.type == selector.tagName) {
       if (!selector.rule) {
         append(result, [node]);
       }
       else if (!selector.rule.nestingOperator ||
                selector.rule.nestingOperator == '>') {
-        if (!node.children) return;
-        node.children.forEach(function (childNode) {
-          append(result, select.rule(selector.rule, childNode));
-        });
+        if (node.children) {
+          node.children.forEach(function (childNode) {
+            append(result, select.rule(selector.rule, childNode));
+          });
+        }
       }
       else {
-        if (!parent) return;
-        append(result, select.rule(selector.rule, {
-          children: parent.children.slice(parent.children.indexOf(node) + 1)
-        }));
+        if (parent) {
+          append(result, select.rule(selector.rule, {
+            children: parent.children.slice(parent.children.indexOf(node) + 1)
+          }));
+        }
       }
     }
 

test/test.js

@@ -83,3 +83,19 @@ test('grouping', function (t) {
   ]);
   t.end();
 });
+
+
+test('universal selector', function (t) {
+  t.equal(select(ast, '*').length, totalNodes(ast));
+  t.deepEqual(select(ast, '* ~ heading'), select(ast, 'heading ~ heading'));
+  t.true(select(ast, 'list > *').every(function (listItem) {
+    return listItem.type == 'listItem';
+  }));
+  t.end();
+
+  function totalNodes (ast) {
+    return 1 + (ast.children || []).map(totalNodes).reduce(function (a, b) {
+      return a + b;
+    }, 0);
+  }
+});