Allow optional redirect host rewriting.

Author: samccone

README.md

@@ -325,6 +325,7 @@ proxyServer.listen(8015);
  *  **secure**: true/false, if you want to verify the SSL Certs
  *  **xfwd**: true/false, adds x-forward headers
  *  **toProxy**: passes the absolute URL as the `path` (useful for proxying to proxies)
+ *  **hostRewrite**: rewrites the location hostname on (301/302/307/308) redirects.
 
 If you are using the `proxyServer.listen` method, the following options are also applicable:
 

lib/http-proxy.js

@@ -41,6 +41,7 @@ module.exports.createProxyServer =
    *    prependPath: <true/false, Default: true - specify whether you want to prepend the target's path to the proxy path>
    *    localAddress : <Local interface string to bind for outgoing connections>
    *    changeOrigin: <true/false, Default: false - changes the origin of the host header to the target URL>
+   *    hostRewrite: rewrites the location hostname on (301/302/307/308) redirects, Default: null.
    *  }
    *
    *  NOTE: `options.ws` and `options.ssl` are optional.

lib/http-proxy/passes/web-incoming.js

@@ -144,7 +144,7 @@ web_o = Object.keys(web_o).map(function(pass) {
     proxyReq.on('response', function(proxyRes) {
       if(server) { server.emit('proxyRes', proxyRes, req, res); }
       for(var i=0; i < web_o.length; i++) {
-       if(web_o[i](req, res, proxyRes)) { break; }
+       if(web_o[i](req, res, proxyRes, options)) { break; }
       }
 
       // Allow us to listen when the proxy has completed

lib/http-proxy/passes/web-outgoing.js

@@ -1,3 +1,4 @@
+var url    = require('url')
 var passes = exports;
 
 /*!
@@ -43,6 +44,13 @@ var passes = exports;
     }
   },
 
+  function setRedirectHostRewrite(req, res, proxyRes, options) {
+    if (options.hostRewrite && /^30(1|2|7|8)$/.test(proxyRes.statusCode)) {
+      var u = url.parse(proxyRes.headers['location']);
+      u.host = options.hostRewrite;
+      proxyRes.headers['location'] = u.format();
+    }
+  },
   /**
    * Copy headers from proxyResponse to response
    * set each header in response object.

test/lib-http-proxy-passes-web-outgoing-test.js

@@ -12,7 +12,7 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
         }
       }, {}, proxyRes);
 
-      expect(proxyRes.headers.connection).to.eql('close'); 
+      expect(proxyRes.headers.connection).to.eql('close');
     });
 
     it('set the right connection with 1.0 - req.connection', function() {
@@ -24,7 +24,7 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
         }
       }, {}, proxyRes);
 
-      expect(proxyRes.headers.connection).to.eql('hey'); 
+      expect(proxyRes.headers.connection).to.eql('hey');
     });
 
     it('set the right connection - req.connection', function() {
@@ -36,7 +36,7 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
         }
       }, {}, proxyRes);
 
-      expect(proxyRes.headers.connection).to.eql('hola'); 
+      expect(proxyRes.headers.connection).to.eql('hola');
     });
 
     it('set the right connection - `keep-alive`', function() {
@@ -48,7 +48,7 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
         }
       }, {}, proxyRes);
 
-      expect(proxyRes.headers.connection).to.eql('keep-alive'); 
+      expect(proxyRes.headers.connection).to.eql('keep-alive');
     });
 
   });
@@ -101,4 +101,4 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
   });
 
 });
- 
+