Use css-selector-parser

Author: eush77

index.js

@@ -1,29 +1,10 @@
 'use strict';
 
-
-function select (ast, selector) {
-  var result = [];
-
-  (function walk (node) {
-    if (node.type == selector[0]) {
-      if (selector.length == 1) {
-        result.push(node);
-      }
-      else if (node.children) {
-        node.children.forEach(function (child) {
-          [].push.apply(result, select(child, selector.slice(1)));
-        });
-      }
-    }
-    if (node.children) {
-      node.children.forEach(walk);
-    }
-  }(ast));
-
-  return result;
-}
+var parseSelector = require('./lib/selector')(),
+    select = require('./lib/select');
 
 
 module.exports = function (ast, selector) {
-  return select(ast, selector.split(/\s+/g));
+  selector = parseSelector(selector);
+  return selector ? select[selector.type](selector, ast) : [];
 };

lib/select.js

@@ -0,0 +1,35 @@
+'use strict';
+
+
+var select = exports;
+
+
+select.ruleSet = function (selector, ast) {
+  return select.rule(selector.rule, ast);
+};
+
+
+select.rule = function (selector, ast) {
+  var result = [];
+
+  (function walk (node) {
+    if (node.type == selector.tagName) {
+      if (!selector.rule) {
+        return result.push(node);
+      }
+      if (!node.children) {
+        return;
+      }
+
+      node.children.forEach(function (childNode) {
+        [].push.apply(result, select.rule(selector.rule, childNode));
+      });
+    }
+
+    if (node.children) {
+      node.children.forEach(walk);
+    }
+  }(ast));
+
+  return result;
+};

lib/selector.js

@@ -0,0 +1,10 @@
+'use strict';
+
+var Parser = require('css-selector-parser').CssSelectorParser;
+
+
+module.exports = function SelectorParser () {
+  var parser = new Parser;
+  parser.registerNestingOperators('>');
+  return parser.parse.bind(parser);
+};

package.json

@@ -30,7 +30,9 @@
     "visit",
     "walk"
   ],
-  "dependencies": {},
+  "dependencies": {
+    "css-selector-parser": "^1.1.0"
+  },
   "devDependencies": {
     "tape": "^4.2.0"
   }

test/test.js

@@ -6,6 +6,13 @@ var select = require('..'),
 var test = require('tape');
 
 
+test('edge cases', function (t) {
+  t.deepEqual(select(ast, ''), []);
+  t.deepEqual(select(ast, '\t '), []);
+  t.end();
+});
+
+
 test('type selector', function (t) {
   t.equal(select(ast, 'root').length, 1);
   t.equal(select(ast, 'root')[0], ast);