1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- package middleware
-
- import (
- "fmt"
- "net/http"
-
- "github.com/go-chi/chi/v5"
- )
-
- // 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 != nil && rctx.RoutePath != "" {
- path = rctx.RoutePath
- } else {
- path = r.URL.Path
- }
- if len(path) > 1 && path[len(path)-1] == '/' {
- newPath := path[:len(path)-1]
- if rctx == nil {
- r.URL.Path = newPath
- } else {
- rctx.RoutePath = newPath
- }
- }
- 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 != nil && 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]
- }
- redirectURL := fmt.Sprintf("//%s%s", r.Host, path)
- http.Redirect(w, r, redirectURL, 301)
- return
- }
- next.ServeHTTP(w, r)
- }
- return http.HandlerFunc(fn)
- }
|