1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
package handlers
import (
"auth-proxy/logging"
"net/http"
"net/http/httputil"
"net/url"
)
// NewProxy creates a new reverse proxy handler using httputil.ReverseProxy.
func NewProxy(targetURL string) http.Handler {
target, err := url.Parse(targetURL)
if err != nil {
panic("Invalid target URL: " + err.Error())
}
proxy := httputil.NewSingleHostReverseProxy(target)
// The default director is mostly fine, but we want to log and ensure the Host header is correct.
director := proxy.Director
proxy.Director = func(req *http.Request) {
director(req) // Run the default director
req.Host = target.Host // Set the Host header to the target's host
// Log the proxying action
ip := logging.GetClientIP(req)
logging.AppLog.Info("Proxying request", "url", req.URL.String(), "ip", ip)
}
// Custom error handler for logging
proxy.ErrorHandler = func(w http.ResponseWriter, r *http.Request, err error) {
ip := logging.GetClientIP(r)
logging.AppLog.Error("Proxy error", "error", err, "ip", ip)
http.Error(w, "Proxy Error", http.StatusBadGateway)
}
return proxy
}
|
