Add support for setting headers on the proxy (#7)

JSainsburyPLC · Nov 25, 2019 · 9f6d40d · 9f6d40d
1 parent 31291f5
commit 9f6d40d
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -48,6 +48,9 @@ See `examples/config.json`
     "path_pattern": "/test-ui/(.*)",
     "to": "/$1",
   }],
+  "proxy_pass_headers": { // additional proxy headers. Optional
+    "Referer": "https://www.test.example.com"
+  }
 }
 ```
 

diff --git a/domain/config.go b/domain/config.go
@@ -17,12 +17,13 @@ type Config struct {
 }
 
 type Route struct {
-	Type        string       `json:"type"`
-	PathPattern *PathPattern `json:"path_pattern"`
-	Backend     *Backend     `json:"backend"`
-	Mock        *Mock        `json:"mock"`
-	Rewrite     []Rewrite    `json:"rewrite"`
-	Redirect    *Redirect    `json:"redirect"`
+	Type             string            `json:"type"`
+	PathPattern      *PathPattern      `json:"path_pattern"`
+	Backend          *Backend          `json:"backend"`
+	Mock             *Mock             `json:"mock"`
+	Rewrite          []Rewrite         `json:"rewrite"`
+	Redirect         *Redirect         `json:"redirect"`
+	ProxyPassHeaders map[string]string `json:"proxy_pass_headers"`
 }
 
 type Rewrite struct {

diff --git a/examples/config.json b/examples/config.json
@@ -33,7 +33,10 @@
       "rewrite": [{
         "path_pattern": "/test-ui/(.*)",
         "to": "/$1"
-      }]
+      }],
+      "proxy_pass_headers": {
+        "Referer": "https://www.test1.example.co.uk/"
+      }
     },
     {
       "type": "redirect",

diff --git a/proxy/proxy.go b/proxy/proxy.go
@@ -84,6 +84,11 @@ func director(defaultBackend *url.URL, logger *log.Logger) func(req *http.Reques
 				break
 			}
 		}
+
+		// set any proxy pass headers from config
+		for name, value := range route.ProxyPassHeaders {
+			req.Header.Set(name, value)
+		}
 	}
 }
 

diff --git a/proxy/proxy_test.go b/proxy/proxy_test.go
@@ -238,6 +238,7 @@ func otherProxyMock(status int, responseBody string) *apitest.Mock {
 
 func userProxyMock(status int, responseBody string) *apitest.Mock {
 	return apitest.NewMock().Get("http://localhost:3001/test-ui/users/info").
+		Header("Referer", "https://www.test.example.com").
 		RespondWith().
 		Status(status).
 		Body(responseBody).
@@ -267,6 +268,9 @@ func config() domain.Config {
 				Type:        "proxy",
 				PathPattern: &domain.PathPattern{Regexp: regexp.MustCompile("^/test-ui/users/.*")},
 				Backend:     &domain.Backend{URL: mockProxyUrlUserUi},
+				ProxyPassHeaders: map[string]string{
+					"Referer": "https://www.test.example.com",
+				},
 			},
 			{
 				Type:        "proxy",
-Original file line number
+Diff line change
@@ Expand Up @@
     				break
     			}
     		}
+    		// set any proxy pass headers from config
+    		for name, value := range route.ProxyPassHeaders {
+    			req.Header.Set(name, value)
+    		}
     	}
     }
@@ Expand Down @@