♻️ Improved redirecting middleware configuration

internal/middleware/redirect.go

@@ -5,13 +5,16 @@ import (
 	"net/url"
 )
 
-type RedirectConfig struct {
+type (
-	Skipper Skipper
+	RedirectConfig struct {
-	URL     *url.URL
+		Skipper Skipper
-	Code    int
+		Code    int
-}
+	}
 
-func Redirect(config RedirectConfig) Interceptor {
+	redirectLogic func(u *url.URL) (url string, ok bool)
+)
+
+func Redirect(config RedirectConfig, redirect redirectLogic) Interceptor {
 	if config.Skipper == nil {
 		config.Skipper = DefaultSkipper
 	}
@@ -27,6 +30,20 @@ func Redirect(config RedirectConfig) Interceptor {
 			return
 		}
 
-		http.RedirectHandler(config.URL.String(), config.Code).ServeHTTP(w, r)
+		u := &url.URL{
+			Scheme: "http",
+			Host:   r.Host,
+			Path:   r.RequestURI,
+		}
+
+		if r.TLS != nil {
+			u.Scheme += "s"
+		}
+
+		if target, ok := redirect(u); ok {
+			http.RedirectHandler(target, config.Code).ServeHTTP(w, r)
+		} else {
+			next(w, r)
+		}
 	}
 }