update deps

README.md

@@ -80,6 +80,26 @@ Type: `String`
 
 [estemplate](https://github.com/RReverser/estemplate) string to be used for generating AST.
 
+#### queryMapper
+
+Custom `queryMapper` RxJS mapper or function that produces such a mapper (when passed `options`).
+
+#### replaces
+
+Custom `replaces` ie. `Rx.Observable.fromArray(...)` or a function that produces it.
+
+#### equery
+
+Custom `equery`, by default `require('grasp-equery')`
+
+#### traverse
+
+Custom `traverse` replacer, by default `require('estraverse').replace`
+
+#### estemplate
+
+Custom `estemplate`, by default `require('estemplate')`
+
 ## License
 
 [MIT License](http://en.wikipedia.org/wiki/MIT_License)

index.js

@@ -2,17 +2,48 @@
 
 var Rx = require('rx');
 var equery = require('grasp-equery');
-var Map = require('es6-map');
+var Map = Map || require('es6-map');
 var traverse = require('estraverse').replace;
 var estemplate = require('estemplate');
 
+function defaultMapper(options) {
+	return function (files) {
+		return files.flatMap(function (file) {
+			return options.replaces
+				.flatMap(function (replace) {
+					var handler = replace.handler;
+
+					return Rx.Observable.fromArray(options.equery.queryParsed(replace.selector, file.program)).map(function (node) {
+						var named = node._named || {};
+						delete node._named;
+						return [node, handler(node, named)];
+					});
+				})
+				.filter(function (replace) { return replace[1] !== undefined })
+				.toArray()
+				.map(function (replaces) {
+					file.program = options.traverse(file.program, {
+						leave: Map.prototype.get.bind(new Map(replaces))
+					});
+
+					return file;
+				});
+		});
+	}
+}
+
 module.exports = function (options) {
-	var replaces = Rx.Observable.fromArray(Object.keys(options)).map(function (selector) {
+	options = options || {}
+	options.equery = options.equery || equery
+	options.traverse = options.traverse || traverse
+	options.estemplate = options.estemplate || estemplate
+
+	var defaultReplaces = Rx.Observable.fromArray(Object.keys(options)).map(function (selector) {
 		var handler = options[selector];
 
 		if (typeof handler === 'string') {
 			var canBeExprStmt = handler.slice(-1) === ';';
-			var tmpl = estemplate.compile(handler, {tolerant: true});
+			var tmpl = options.estemplate.compile(handler, {tolerant: true});
 
 			handler = function (node, named) {
 				var ast = tmpl(named);
@@ -32,32 +63,17 @@ module.exports = function (options) {
 		}
 
 		return {
-			selector: equery.parse(selector),
+			selector: options.equery.parse(selector),
 			handler: handler
 		};
 	});
 
-	return function (files) {
-		return files.flatMap(function (file) {
-			return replaces
-				.flatMap(function (replace) {
-					var handler = replace.handler;
 
-					return Rx.Observable.fromArray(equery.queryParsed(replace.selector, file.program)).map(function (node) {
-						var named = node._named || {};
-						delete node._named;
-						return [node, handler(node, named)];
-					});
-				})
-				.filter(function (replace) { return replace[1] !== undefined })
-				.toArray()
-				.map(function (replaces) {
-					file.program = traverse(file.program, {
-						leave: Map.prototype.get.bind(new Map(replaces))
-					});
+	var replaces = options.replaces || defaultReplaces
+	replaces  = typeof replaces === 'function' ? replaces(options) : replaces
+	options.replaces = replaces
 
-					return file;
-				});
-		});
-	}
+	var qMapper = options.queryMapper || defaultMapper
+	qMapper = typeof qMapper === 'function' ? qMapper(options) : qMapper
+	return qMapper
 };

package.json

@@ -21,16 +21,16 @@
     "test": "mocha"
   },
   "dependencies": {
-    "es6-map": "^0.1.0",
-    "estemplate": "^0.1.1",
-    "estraverse": "^1.5.1",
-    "grasp-equery": "^0.2.1",
-    "rx": "^2.3.11"
+    "es6-map": "^0.1.4",
+    "estemplate": "^0.5.1",
+    "estraverse": "^4.2.0",
+    "grasp-equery": "^0.3.1",
+    "rx": "^4.1.0"
   },
   "devDependencies": {
     "chai": "*",
-    "escodegen": "^1.4.1",
-    "esprima": "^1.2.2",
+    "escodegen": "^1.8.1",
+    "esprima": "^3.1.0",
     "mocha": "*"
   }
 }

test/test.js

@@ -34,7 +34,12 @@ it('function handler', function (done) {
 	})(Rx.Observable.fromArray(input))
 	.pluck('program')
 	.map(generate)
-	.zip(expected, assert.equal)
+	// .zip(expected, assert.equal)
+	.do(function (file) {
+		assert.equal(file, expected[0]);
+		// assert.equal(file.program.type, 'Program');
+	})
+
 	.subscribe(function () {}, done, done);
 });
 
@@ -54,6 +59,11 @@ it('template handler', function (done) {
 	})(Rx.Observable.fromArray(input))
 	.pluck('program')
 	.map(generate)
-	.zip(expected, assert.equal)
+	// .zip(expected, assert.equal)
+	.do(function (file) {
+		assert.equal(file, expected[0]);
+		// assert.equal(file.program.type, 'Program');
+	})
+
 	.subscribe(function () {}, done, done);
 });