From 49360a363a4b4c4da38b2a7c94dce6d4cdf914d9 Mon Sep 17 00:00:00 2001
From: Baokun Lee <nototon@gmail.com>
Date: Sun, 4 Aug 2019 13:31:18 +0800
Subject: [PATCH] custom proxy director

---
 proxy/router.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/proxy/router.go b/proxy/router.go
index e628fe7..abd9b08 100644
--- a/proxy/router.go
+++ b/proxy/router.go
@@ -32,7 +32,14 @@ func NewRouter(srv *Server, ops *RouterOps) *Router {
 	}
 	if ops != nil {
 		if remote, err := url.Parse(ops.Proxy); err == nil {
-			rt.proxy = httputil.NewSingleHostReverseProxy(remote)
+			proxy := httputil.NewSingleHostReverseProxy(remote)
+			director := proxy.Director
+			proxy.Director = func(r *http.Request) {
+				director(r)
+				r.Host = remote.Host
+			}
+			rt.proxy = proxy
+
 			rt.proxy.Transport = &http.Transport{
 				TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 			}