Add basic nesting

Author: eush77

index.js

@@ -1,17 +1,29 @@
 'use strict';
 
 
-module.exports = function (ast, selector) {
+function select (ast, selector) {
   var result = [];
 
   (function walk (node) {
-    if (node.type == selector) {
-      result.push(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;
+}
+
+
+module.exports = function (ast, selector) {
+  return select(ast, selector.split(/\s+/g));
 };

test/test.js

@@ -12,5 +12,18 @@ test('type selector', function (t) {
   t.equal(select(ast, 'text').length, 39);
   t.equal(select(ast, 'text')[1], ast.children[1].children[0]);
   t.equal(select(ast, 'tableRow').length, 2);
+  t.equal(select(ast, 'heading').length, 5);
+  t.end();
+});
+
+
+test('nesting', function (t) {
+  t.deepEqual(select(ast, 'root heading'), select(ast, 'heading'));
+  t.deepEqual(select(ast, 'paragraph emphasis'), [
+    ast.children[2].children[0].children[1],
+    ast.children[3].children[1],
+    ast.children[4].children[1].children[1].children[1]
+      .children[0].children[0].children[1]
+  ]);
   t.end();
 });