cookieDomainRewrite option

Author: Jeremy Judeaux

README.md

@@ -337,6 +337,7 @@ proxyServer.listen(8015);
 *  **hostRewrite**: rewrites the location hostname on (301/302/307/308) redirects.
 *  **autoRewrite**: rewrites the location host/port on (301/302/307/308) redirects based on requested host/port. Default: false.
 *  **protocolRewrite**: rewrites the location protocol on (301/302/307/308) redirects to 'http' or 'https'. Default: null.
+*  **cookieDomainRewrite**: rewrites domain of `set-cookie` headers. `""` to remove the domain. Can provide an object to map domains one by one. Default: `false`.
 *  **headers**: object with extra headers to be added to target requests.
 
 **NOTE:**

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

@@ -80,12 +80,44 @@ var redirectRegex = /^30(1|2|7|8)$/;
    *
    * @api private
    */
-  function writeHeaders(req, res, proxyRes) {
+  function writeHeaders(req, res, proxyRes, options) {
+    var rewriteCookieDomainConfig = options ? options.cookieDomainRewrite : undefined;
+    if(typeof rewriteCookieDomainConfig === "string") { //also test for ""
+      rewriteCookieDomainConfig = {"*": rewriteCookieDomainConfig};
+    }
     Object.keys(proxyRes.headers).forEach(function(key) {
-      if(proxyRes.headers[key] != undefined){
-        res.setHeader(String(key).trim(), proxyRes.headers[key]);
+      var header = proxyRes.headers[key];
+      if(header != undefined){
+        if(rewriteCookieDomainConfig && key.toLowerCase() === 'set-cookie') {
+          header = rewriteCookieDomain(header);
+        }
+        res.setHeader(String(key).trim(), header);
       }
     });
+
+    function rewriteCookieDomain(header) {
+      if(Array.isArray(header)) {
+        return header.map(rewriteCookieDomain);
+      }
+      return header.replace(/(;\s*domain=)([^;]+)/, function(match, prefix, previousDomain) {
+        var newDomain;
+        if(previousDomain in rewriteCookieDomainConfig) {
+          newDomain = rewriteCookieDomainConfig[previousDomain];
+        } else if ("*" in rewriteCookieDomainConfig) {
+          newDomain = rewriteCookieDomainConfig["*"];
+        } else {
+          //no match, return previous domain
+          return match;
+        }
+        if(newDomain) {
+          //replace domain
+          return prefix + newDomain;
+        } else {
+          //remove domain
+          return "";
+        }
+      });
+    }
   },
 
   /**

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

@@ -206,24 +206,72 @@ describe('lib/http-proxy/passes/web-outgoing.js', function () {
   });
 
   describe('#writeHeaders', function() {
-    var proxyRes = {
-      headers: {
-        hey: 'hello',
-        how: 'are you?'
-      }
-    };
+    beforeEach(function() {
+      this.proxyRes = {
+        headers: {
+          hey: 'hello',
+          how: 'are you?',
+          'set-cookie': 'hello; domain=my.domain; path=/'
+        }
+      };
+      this.res = {
+        setHeader: function(k, v) {
+          this.headers[k] = v;
+        },
+        headers: {}
+      };
+    });
 
-    var res = {
-      setHeader: function(k, v) {
-        this.headers[k] = v;
-      },
-      headers: {}
-    };
+    it("writes headers", function() {
+      httpProxy.writeHeaders({}, this.res, this.proxyRes);
 
-    httpProxy.writeHeaders({}, res, proxyRes);
+      expect(this.res.headers.hey).to.eql('hello');
+      expect(this.res.headers.how).to.eql('are you?');
+    });
 
-    expect(res.headers.hey).to.eql('hello');
-    expect(res.headers.how).to.eql('are you?');
+    it("rewrites domain", function() {
+      var options = {
+        cookieDomainRewrite: "my.new.domain"
+      };
+
+      httpProxy.writeHeaders({}, this.res, this.proxyRes, options);
+
+      expect(this.res.headers['set-cookie']).to.eql('hello; domain=my.new.domain; path=/');
+    });
+    
+    it("removes domain", function() {
+      var options = {
+        cookieDomainRewrite: ""
+      };
+
+      httpProxy.writeHeaders({}, this.res, this.proxyRes, options);
+
+      expect(this.res.headers['set-cookie']).to.eql('hello; path=/');
+    });
+
+    it("rewrites headers with advanced configuration", function() {
+      var options = {
+        cookieDomainRewrite: {
+          '*': '',
+          'my.old.domain': 'my.new.domain',
+          'my.special.domain': 'my.special.domain'
+        }
+      };
+      this.proxyRes.headers['set-cookie'] = [
+        'hello-on-my.domain; domain=my.domain; path=/',
+        'hello-on-my.old.domain; domain=my.old.domain; path=/',
+        'hello-on-my.special.domain; domain=my.special.domain; path=/'
+      ];
+
+      httpProxy.writeHeaders({}, this.res, this.proxyRes, options);
+
+      expect(this.res.headers['set-cookie'])
+        .to.contain('hello-on-my.domain; path=/');
+      expect(this.res.headers['set-cookie'])
+        .to.contain('hello-on-my.old.domain; domain=my.new.domain; path=/');
+      expect(this.res.headers['set-cookie'])
+        .to.contain('hello-on-my.special.domain; domain=my.special.domain; path=/');
+    });
   });