This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
2021-05-02 22:49:16 +00:00
|
|
|
package web
|
|
|
|
|
|
|
|
import (
|
2021-05-04 16:14:25 +00:00
|
|
|
"net/http"
|
2021-05-02 22:49:16 +00:00
|
|
|
"net/http/httputil"
|
|
|
|
"net/url"
|
|
|
|
)
|
|
|
|
|
|
|
|
func (ws *WebServer) configureProxy() {
|
|
|
|
// Reverse proxy to the application server
|
|
|
|
u, _ := url.Parse("http://localhost:8000")
|
|
|
|
rp := httputil.NewSingleHostReverseProxy(u)
|
2021-05-04 16:14:25 +00:00
|
|
|
rp.ErrorHandler = ws.proxyErrorHandler
|
2021-05-04 16:45:28 +00:00
|
|
|
rp.ModifyResponse = ws.proxyModifyResponse
|
2021-05-02 22:49:16 +00:00
|
|
|
ws.m.PathPrefix("/").Handler(rp)
|
|
|
|
}
|
2021-05-04 16:14:25 +00:00
|
|
|
|
|
|
|
func (ws *WebServer) proxyErrorHandler(rw http.ResponseWriter, req *http.Request, err error) {
|
|
|
|
ws.log.WithError(err).Warning("proxy error")
|
|
|
|
rw.WriteHeader(http.StatusBadGateway)
|
|
|
|
}
|
2021-05-04 16:45:28 +00:00
|
|
|
|
|
|
|
func (ws *WebServer) proxyModifyResponse(r *http.Response) error {
|
|
|
|
r.Header.Set("X-authentik-from", "authentik")
|
|
|
|
return nil
|
|
|
|
}
|