Commit 6f46dbe

mo khan <mo@mokhan.ca>
2025-03-27 15:26:30
refactor: use proxy request to add forwarded headers
1 parent f7b7b77
Changed files (1)
pkg
pkg/prxy/prxy.go
@@ -18,17 +18,16 @@ func New(routes map[string]string) http.Handler {
 	}
 
 	return &httputil.ReverseProxy{
-		Director: func(r *http.Request) {
-			host, _, err := net.SplitHostPort(r.Host)
+		Rewrite: func(r *httputil.ProxyRequest) {
+			host, _, err := net.SplitHostPort(r.In.Host)
 			if err != nil {
-				fmt.Printf("%v\n", err)
+				log.Println(err)
 				return
 			}
 
 			destination := mapped[host]
-			r.URL.Scheme = destination.Scheme
-			r.Host = destination.Host
-			r.URL.Host = destination.Host
+			r.SetXForwarded()
+			r.SetURL(destination)
 		},
 		Transport:     http.DefaultTransport,
 		FlushInterval: -1,