mirror of
https://github.com/yusing/godoxy.git
synced 2025-06-01 09:32:35 +02:00
4a8bd48ad5
Some checks are pending
Docker Image CI (socket-proxy) / build (push) Waiting to run
* refactor: simplify io code and make utils module independent * fix(docker): agent and socket-proxy docker event flushing with modified reverse proxy handler * refactor: remove unused code * refactor: remove the use of logging module in most code * refactor: streamline domain mismatch check in certState function * tweak: use ecdsa p-256 for autocert * fix(tests): update health check tests for invalid host and add case for port in host * feat(acme): custom acme directory * refactor: code refactor and improved context and error handling * tweak: optimize memory usage under load * fix(oidc): restore old user matching behavior * docs: add ChatGPT assistant to README --------- Co-authored-by: yusing <yusing@6uo.me>
80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
gphttp "github.com/yusing/go-proxy/internal/net/gphttp"
|
|
"github.com/yusing/go-proxy/internal/net/gphttp/httpheaders"
|
|
"github.com/yusing/go-proxy/internal/net/gphttp/middleware/errorpage"
|
|
)
|
|
|
|
type customErrorPage struct{}
|
|
|
|
var CustomErrorPage = NewMiddleware[customErrorPage]()
|
|
|
|
const StaticFilePathPrefix = "/$gperrorpage/"
|
|
|
|
// before implements RequestModifier.
|
|
func (customErrorPage) before(w http.ResponseWriter, r *http.Request) (proceed bool) {
|
|
return !ServeStaticErrorPageFile(w, r)
|
|
}
|
|
|
|
// modifyResponse implements ResponseModifier.
|
|
func (customErrorPage) modifyResponse(resp *http.Response) error {
|
|
// only handles non-success status code and html/plain content type
|
|
contentType := gphttp.GetContentType(resp.Header)
|
|
if !gphttp.IsSuccess(resp.StatusCode) && (contentType.IsHTML() || contentType.IsPlainText()) {
|
|
errorPage, ok := errorpage.GetErrorPageByStatus(resp.StatusCode)
|
|
if ok {
|
|
log.Debug().Msgf("error page for status %d loaded", resp.StatusCode)
|
|
_, _ = io.Copy(io.Discard, resp.Body) // drain the original body
|
|
resp.Body.Close()
|
|
resp.Body = io.NopCloser(bytes.NewReader(errorPage))
|
|
resp.ContentLength = int64(len(errorPage))
|
|
resp.Header.Set(httpheaders.HeaderContentLength, strconv.Itoa(len(errorPage)))
|
|
resp.Header.Set(httpheaders.HeaderContentType, "text/html; charset=utf-8")
|
|
} else {
|
|
log.Error().Msgf("unable to load error page for status %d", resp.StatusCode)
|
|
}
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ServeStaticErrorPageFile(w http.ResponseWriter, r *http.Request) (served bool) {
|
|
path := r.URL.Path
|
|
if path != "" && path[0] != '/' {
|
|
path = "/" + path
|
|
}
|
|
if strings.HasPrefix(path, StaticFilePathPrefix) {
|
|
filename := path[len(StaticFilePathPrefix):]
|
|
file, ok := errorpage.GetStaticFile(filename)
|
|
if !ok {
|
|
log.Error().Msg("unable to load resource " + filename)
|
|
return false
|
|
}
|
|
ext := filepath.Ext(filename)
|
|
switch ext {
|
|
case ".html":
|
|
w.Header().Set(httpheaders.HeaderContentType, "text/html; charset=utf-8")
|
|
case ".js":
|
|
w.Header().Set(httpheaders.HeaderContentType, "application/javascript; charset=utf-8")
|
|
case ".css":
|
|
w.Header().Set(httpheaders.HeaderContentType, "text/css; charset=utf-8")
|
|
default:
|
|
log.Error().Msgf("unexpected file type %q for %s", ext, filename)
|
|
}
|
|
if _, err := w.Write(file); err != nil {
|
|
log.Err(err).Msg("unable to write resource " + filename)
|
|
http.Error(w, "Error page failure", http.StatusInternalServerError)
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|