mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 04:42:33 +02:00
74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/yusing/go-proxy/internal/api/v1/error_page"
|
|
gpHTTP "github.com/yusing/go-proxy/internal/net/http"
|
|
)
|
|
|
|
var CustomErrorPage = &Middleware{
|
|
before: func(next http.HandlerFunc, w ResponseWriter, r *Request) {
|
|
if !ServeStaticErrorPageFile(w, r) {
|
|
next(w, r)
|
|
}
|
|
},
|
|
modifyResponse: func(resp *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 := error_page.GetErrorPageByStatus(resp.StatusCode)
|
|
if ok {
|
|
errPageLogger.Debugf("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("Content-Length", fmt.Sprint(len(errorPage)))
|
|
resp.Header.Set("Content-Type", "text/html; charset=utf-8")
|
|
} else {
|
|
errPageLogger.Errorf("unable to load error page for status %d", resp.StatusCode)
|
|
}
|
|
return nil
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func ServeStaticErrorPageFile(w http.ResponseWriter, r *http.Request) bool {
|
|
path := r.URL.Path
|
|
if path != "" && path[0] != '/' {
|
|
path = "/" + path
|
|
}
|
|
if strings.HasPrefix(path, gpHTTP.StaticFilePathPrefix) {
|
|
filename := path[len(gpHTTP.StaticFilePathPrefix):]
|
|
file, ok := error_page.GetStaticFile(filename)
|
|
if !ok {
|
|
errPageLogger.Errorf("unable to load resource %s", filename)
|
|
return false
|
|
} else {
|
|
ext := filepath.Ext(filename)
|
|
switch ext {
|
|
case ".html":
|
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|
case ".js":
|
|
w.Header().Set("Content-Type", "application/javascript; charset=utf-8")
|
|
case ".css":
|
|
w.Header().Set("Content-Type", "text/css; charset=utf-8")
|
|
default:
|
|
errPageLogger.Errorf("unexpected file type %q for %s", ext, filename)
|
|
}
|
|
w.Write(file)
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
var errPageLogger = logrus.WithField("middleware", "error_page")
|