add curried forms

Author: eush77

README.md

@@ -82,11 +82,15 @@ All the relevant parts of [Selectors Level 3][spec]:
 
 ## API
 
-#### `select(ast, selector)`
+### `select(ast, selector)`
+
+*Curried form: `select(ast)(selector)`*
 
 Applies `selector` to `ast`, returns array of matching nodes.
 
-#### `select.one(ast, selector)`
+### `select.one(ast, selector)`
+
+*Curried form: `select.one(ast)(selector)`*
 
 Returns a single node matching `selector`.
 

index.js

@@ -7,6 +7,10 @@ var debug = require('debug')('unist-util-select');
 
 
 var select = function select (ast, selector) {
+  if (arguments.length == 1) {
+    return select.bind(this, ast);
+  }
+
   debug('Selector: %j', selector);
   selector = parseSelector(selector);
   debug('AST: %s',
@@ -16,6 +20,10 @@ var select = function select (ast, selector) {
 
 
 select.one = function selectOne (ast, selector) {
+  if (arguments.length == 1) {
+    return selectOne.bind(this, ast);
+  }
+
   var nodes = select(ast, selector);
 
   if (!nodes.length) {

test/curried.js

@@ -0,0 +1,13 @@
+'use strict';
+
+var select = require('..'),
+    ast = require('./lib/ast')();
+
+var test = require('tape');
+
+
+test('curried forms', function (t) {
+  t.deepEqual(select(ast)('paragraph'), select(ast, 'paragraph'));
+  t.equal(select.one(ast)('table'), select.one(ast, 'table'));
+  t.end();
+});

test/select-one.js

@@ -2,8 +2,7 @@
 
 var select = require('..'),
     select1 = select.one,
-    ast = require('./lib/ast')(),
-    path = require('./lib/path');
+    ast = require('./lib/ast')();
 
 var test = require('tape');