57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/go-chi/chi"
|
||
|
|
)
|
||
|
|
|
||
|
|
// StripSlashes is a middleware that will match request paths with a trailing
|
||
|
|
// slash, strip it from the path and continue routing through the mux, if a route
|
||
|
|
// matches, then it will serve the handler.
|
||
|
|
func StripSlashes(next http.Handler) http.Handler {
|
||
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
var path string
|
||
|
|
rctx := chi.RouteContext(r.Context())
|
||
|
|
if rctx.RoutePath != "" {
|
||
|
|
path = rctx.RoutePath
|
||
|
|
} else {
|
||
|
|
path = r.URL.Path
|
||
|
|
}
|
||
|
|
if len(path) > 1 && path[len(path)-1] == '/' {
|
||
|
|
rctx.RoutePath = path[:len(path)-1]
|
||
|
|
}
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
}
|
||
|
|
return http.HandlerFunc(fn)
|
||
|
|
}
|
||
|
|
|
||
|
|
// RedirectSlashes is a middleware that will match request paths with a trailing
|
||
|
|
// slash and redirect to the same path, less the trailing slash.
|
||
|
|
//
|
||
|
|
// NOTE: RedirectSlashes middleware is *incompatible* with http.FileServer,
|
||
|
|
// see https://github.com/go-chi/chi/issues/343
|
||
|
|
func RedirectSlashes(next http.Handler) http.Handler {
|
||
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
var path string
|
||
|
|
rctx := chi.RouteContext(r.Context())
|
||
|
|
if rctx.RoutePath != "" {
|
||
|
|
path = rctx.RoutePath
|
||
|
|
} else {
|
||
|
|
path = r.URL.Path
|
||
|
|
}
|
||
|
|
if len(path) > 1 && path[len(path)-1] == '/' {
|
||
|
|
if r.URL.RawQuery != "" {
|
||
|
|
path = fmt.Sprintf("%s?%s", path[:len(path)-1], r.URL.RawQuery)
|
||
|
|
} else {
|
||
|
|
path = path[:len(path)-1]
|
||
|
|
}
|
||
|
|
http.Redirect(w, r, path, 301)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
}
|
||
|
|
return http.HandlerFunc(fn)
|
||
|
|
}
|