v0.0.7 add modifyHash plugin

Author: zxcabs

lib/ProxyServer.js

@@ -147,7 +147,8 @@ ProxyServer.prototype.__addHost = function addHost(raw) {
 
 //plugins
 ProxyServer.plugins = {
-	replaceParamsInHash: require('./plugins/replaceParamsInHash.js')
+	replaceParamsInHash: require('./plugins/replaceParamsInHash.js'),
+	modifyHash: require('./plugins/modifyHash')
 };
 
 module.exports = ProxyServer;

lib/plugins/modifyHash.js

@@ -0,0 +1,81 @@
+/**
+ * User: jo
+ * Date: 21.05.13
+ * Time: 18:15
+ *
+ */
+
+var url = require('url'),
+	async = require('async'),
+	tools = require('../tools'),
+	getHash = tools.getHash,
+	getEndHadersIndex = tools.getEndHeaderIndex;
+
+module.exports = function makeModifyHash(modFuncArr) {
+	return function modifyHash(obj, next) {
+		var raw = obj.raw,
+			index = getEndHadersIndex(raw),
+			head = parseHead(raw.slice(0, index));
+
+		head._raw = raw;
+		head._hash = obj.hash;
+		head.body = raw.slice(index + 4);
+
+		async.applyEachSeries(modFuncArr, head, function () {
+			var newHead = makeHead(head),
+				hash = tools.getHash()
+							.update(newHead)
+							.update(head.body)
+							.end();
+
+			obj.hash = hash;
+			next();
+		});
+	};
+};
+
+
+function parseHead(buff) {
+	var lns = buff.toString('utf8').split(/\r\n/),
+		first = lns[0].split(' '),
+		version,
+		urlObj = url.parse(first[1], true),
+		headers = {};
+
+	for (var i = 1, l = lns.length; l > i; i += 1) {
+		var ln = lns[i].match(/^([\w-]+): (.*)$/);
+		if (!ln) continue;
+
+		headers[ln[1]] = ln[2];
+	}
+
+	if (first[2]) {
+		version = first[2].split('/')[1];
+	}
+
+
+
+	return {
+		method: first[0],
+		uri: {
+			pathname: urlObj.pathname,
+			query: urlObj.query
+		},
+		version: version,
+		headers: headers
+	};
+}
+
+function makeHead(obj) {
+	var headers = obj.headers,
+		arr = [
+			obj.method.toUpperCase() + ' ' + url.format(obj.uri) + (obj.version ? ' HTTP/' + obj.version: '')
+		];
+
+	for (var name in headers) {
+		arr.push(name + ': ' + headers[name]);
+	}
+
+	arr.push('\r\n');
+	return new Buffer(arr.join('\r\n'));
+}

package.json

@@ -1,7 +1,7 @@
 {
 	"name": "node-http-proxy-cache",
 	"description": "Http caching proxy for the development clients to different websites and api",
-	"version": "0.0.6",
+	"version": "0.0.7",
 	"author": "Evgeny Reznichenko <[email protected]>",
 	"dependencies": {
 		"async": "0.2.8"

test/plugins.js

@@ -7,6 +7,7 @@
  */
 
 var should = require('should'),
+	url = require('url'),
 	Proxy = require('../lib/ProxyServer.js'),
 	tools = require('../lib/tools.js');
 
@@ -46,4 +47,84 @@ describe('plugins', function () {
 		});
 	});
 
+	/**
+	 * Test for modifyHash
+	 */
+	describe('#modifyHash', function () {
+		var modifyHash = plugins.modifyHash;
+
+		it('should call mod function and delegate parsed params', function (done) {
+			var raw = new Buffer('POST /index?bar=1&foo=2 HTTP/1.1\r\nHost: localhost\r\n\r\nsome data'),
+				hash = tools.getHash().update(raw).end(),
+				is1 = false,
+				is2 = false,
+				urlObj = url.parse('/index?bar=1&foo=2', true),
+				uri = {
+					pathname: urlObj.pathname,
+					query: urlObj.query
+				};
+
+			var obj = { raw: raw, hash: hash };
+
+			function modfn1 (req, next) {
+				is1 = true;
+				req.should.have.property('_raw', raw);
+				req.should.have.property('_hash', hash);
+				req.should.have.property('method', 'POST');
+				req.should.have.property('uri').and.obj.should.eql(uri);
+				req.should.have.property('version', '1.1');
+				req.should.have.property('headers').and.obj.should.eql({ 'Host': 'localhost' });
+				req.should.have.property('body');
+				next();
+			}
+
+			function modfn2 (req, next) {
+				is2 = true;
+				req.should.have.property('_raw', raw);
+				req.should.have.property('_hash', hash);
+				req.should.have.property('method', 'POST');
+				req.should.have.property('uri').and.obj.should.eql(uri);
+				req.should.have.property('version', '1.1');
+				req.should.have.property('headers').and.obj.should.eql({ 'Host': 'localhost' });
+				req.should.have.property('body');
+				next();
+			}
+
+			var plugin = modifyHash([modfn1, modfn2]);
+
+			plugin(obj, function (err) {
+				is1.should.be.ok;
+				is2.should.be.ok;
+
+				done();
+			});
+		});
+
+		it('should calculate new hash sum', function (done) {
+			var raw = new Buffer('GET /index?bar=1&foo=2 HTTP/1.1\r\nHost: localhost\r\n\r\n'),
+				hash = tools.getHash().update(raw).end(),
+				expectHash = tools.getHash()
+								.update(new Buffer('GET /index?bar=1&foo=4 HTTP/1.1\r\nHost: localhost1\r\n\r\n'))
+								.end();
+
+			var obj = { raw: raw, hash: hash };
+
+			function mod(req, next) {
+				req.uri.query.foo = 4;
+				req.headers.Host = 'localhost1';
+				next();
+			}
+
+			var plugin = modifyHash([mod]);
+
+			plugin(obj, function (err) {
+				should.not.exist(err);
+				obj.raw.should.eql(raw);
+				obj.hash.should.eql(expectHash);
+
+				done();
+			});
+		});
+	});
+
 });