mirror of
https://github.com/umputun/reproxy.git
synced 2024-11-24 08:12:31 +02:00
114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
package proxy
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
|
|
log "github.com/go-pkgz/lgr"
|
|
R "github.com/go-pkgz/rest"
|
|
"github.com/gorilla/handlers"
|
|
)
|
|
|
|
func headersHandler(headers []string) func(next http.Handler) http.Handler {
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if len(headers) == 0 {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
for _, h := range headers {
|
|
elems := strings.Split(h, ":")
|
|
if len(elems) != 2 {
|
|
continue
|
|
}
|
|
w.Header().Set(strings.TrimSpace(elems[0]), strings.TrimSpace(elems[1]))
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
func maxReqSizeHandler(maxSize int64) func(next http.Handler) http.Handler {
|
|
if maxSize <= 0 {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
log.Printf("[DEBUG] request size limited to %d", maxSize)
|
|
return func(next http.Handler) http.Handler {
|
|
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// check ContentLength
|
|
if r.ContentLength > maxSize {
|
|
w.WriteHeader(http.StatusRequestEntityTooLarge)
|
|
return
|
|
}
|
|
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxSize)
|
|
next.ServeHTTP(w, r)
|
|
}
|
|
return http.HandlerFunc(fn)
|
|
}
|
|
}
|
|
|
|
func accessLogHandler(wr io.Writer) func(next http.Handler) http.Handler {
|
|
return func(next http.Handler) http.Handler {
|
|
return handlers.CombinedLoggingHandler(wr, next)
|
|
}
|
|
}
|
|
|
|
func stdoutLogHandler(enable bool, lh func(next http.Handler) http.Handler) func(next http.Handler) http.Handler {
|
|
|
|
if enable {
|
|
log.Printf("[DEBUG] stdout logging enabled")
|
|
return func(next http.Handler) http.Handler {
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
|
// don't log to stdout GET ~/(.*)/ping$ requests
|
|
if r.Method == "GET" && strings.HasSuffix(r.URL.Path, "/ping") {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
lh(next).ServeHTTP(w, r)
|
|
}
|
|
return http.HandlerFunc(fn)
|
|
}
|
|
}
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
func gzipHandler(enabled bool) func(next http.Handler) http.Handler {
|
|
if enabled {
|
|
log.Printf("[DEBUG] gzip enabled")
|
|
return handlers.CompressHandler
|
|
}
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
func signatureHandler(enabled bool, version string) func(next http.Handler) http.Handler {
|
|
if enabled {
|
|
log.Printf("[DEBUG] signature headers enabled")
|
|
return R.AppInfo("reproxy", "umputun", version)
|
|
}
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|