✨ v3 (feature): new redirection methods

client_test.go

@@ -1078,9 +1078,9 @@ func Test_Client_Agent_MaxRedirectsCount(t *testing.T) {
 
 	app.Get("/", func(c Ctx) error {
 		if c.Request().URI().QueryArgs().Has("foo") {
-			return c.Redirect("/foo")
+			return c.Redirect().To("/foo")
 		}
-		return c.Redirect("/")
+		return c.Redirect().To("/")
 	})
 	app.Get("/foo", func(c Ctx) error {
 		return c.SendString("redirect")

ctx.go

@@ -52,6 +52,9 @@ type DefaultCtx struct {
 	matched             bool                 // Non use route matched
 	viewBindMap         *dictpool.Dict       // Default view map to bind template engine
 	bind                *Bind                // Default bind reference
+	redirect            *Redirect            // Default redirect reference
+	flashMessages       map[string]string    // flash messages sent by redirection cookie
+	oldInput            map[string]string    // old input data sent by redirection cookie
 }
 
 // TLSHandler object
@@ -896,16 +899,16 @@ func (c *DefaultCtx) Range(size int) (rangeData Range, err error) {
 	return
 }
 
-// Redirect to the URL derived from the specified path, with specified status.
+// Redirect returns the Redirect reference.
+// Use Redirect().Status() to set custom redirection status code.
 // If status is not specified, status defaults to 302 Found.
-func (c *DefaultCtx) Redirect(location string, status ...int) error {
-	c.setCanonical(HeaderLocation, location)
-	if len(status) > 0 {
-		c.Status(status[0])
-	} else {
-		c.Status(StatusFound)
+// You can use Redirect().To(), Redirect().Route() and Redirect().Back() for redirection.
+func (c *DefaultCtx) Redirect() *Redirect {
+	if c.redirect == nil {
+		c.redirect = newRedirect(c)
 	}
-	return nil
+
+	return c.redirect
 }
 
 // Add vars to default view var map binding to template engine.
@@ -956,45 +959,6 @@ func (c *DefaultCtx) GetRouteURL(routeName string, params Map) (string, error) {
 	return c.getLocationFromRoute(c.App().GetRoute(routeName), params)
 }
 
-// RedirectToRoute to the Route registered in the app with appropriate parameters
-// If status is not specified, status defaults to 302 Found.
-// If you want to send queries to route, you must add "queries" key typed as map[string]string to params.
-func (c *DefaultCtx) RedirectToRoute(routeName string, params Map, status ...int) error {
-	location, err := c.getLocationFromRoute(c.App().GetRoute(routeName), params)
-	if err != nil {
-		return err
-	}
-
-	// Check queries
-	if queries, ok := params["queries"].(map[string]string); ok {
-		queryText := bytebufferpool.Get()
-		defer bytebufferpool.Put(queryText)
-
-		i := 1
-		for k, v := range queries {
-			_, _ = queryText.WriteString(k + "=" + v)
-
-			if i != len(queries) {
-				_, _ = queryText.WriteString("&")
-			}
-			i++
-		}
-
-		return c.Redirect(location+"?"+queryText.String(), status...)
-	}
-	return c.Redirect(location, status...)
-}
-
-// RedirectBack to the URL to referer
-// If status is not specified, status defaults to 302 Found.
-func (c *DefaultCtx) RedirectBack(fallback string, status ...int) error {
-	location := c.Get(HeaderReferer)
-	if location == "" {
-		location = fallback
-	}
-	return c.Redirect(location, status...)
-}
-
 // Render a template with data and sends a text/html response.
 // We support the following engines: https://github.com/gofiber/template
 func (c *DefaultCtx) Render(name string, bind Map, layouts ...string) error {

ctx_interface.go

@@ -229,9 +229,11 @@ type Ctx interface {
 	// Range returns a struct containing the type and a slice of ranges.
 	Range(size int) (rangeData Range, err error)
 
-	// Redirect to the URL derived from the specified path, with specified status.
+	// Redirect returns the Redirect reference.
+	// Use Redirect().Status() to set custom redirection status code.
 	// If status is not specified, status defaults to 302 Found.
-	Redirect(location string, status ...int) error
+	// You can use Redirect().To(), Redirect().Route() and Redirect().Back() for redirection.
+	Redirect() *Redirect
 
 	// Add vars to default view var map binding to template engine.
 	// Variables are read by the Render method and may be overwritten.
@@ -240,15 +242,6 @@ type Ctx interface {
 	// GetRouteURL generates URLs to named routes, with parameters. URLs are relative, for example: "/user/1831"
 	GetRouteURL(routeName string, params Map) (string, error)
 
-	// RedirectToRoute to the Route registered in the app with appropriate parameters
-	// If status is not specified, status defaults to 302 Found.
-	// If you want to send queries to route, you must add "queries" key typed as map[string]string to params.
-	RedirectToRoute(routeName string, params Map, status ...int) error
-
-	// RedirectBack to the URL to referer
-	// If status is not specified, status defaults to 302 Found.
-	RedirectBack(fallback string, status ...int) error
-
 	// Render a template with data and sends a text/html response.
 	// We support the following engines: https://github.com/gofiber/template
 	Render(name string, bind Map, layouts ...string) error
@@ -445,6 +438,9 @@ func (c *DefaultCtx) release() {
 	c.route = nil
 	c.fasthttp = nil
 	c.bind = nil
+	c.redirect = nil
+	c.flashMessages = nil
+	c.oldInput = nil
 	if c.viewBindMap != nil {
 		dictpool.ReleaseDict(c.viewBindMap)
 	}

ctx_test.go

@@ -16,7 +16,6 @@ import (
 	"io"
 	"mime/multipart"
 	"net/http/httptest"
-	"net/url"
 	"os"
 	"path/filepath"
 	"strconv"
@@ -2254,137 +2253,6 @@ func Test_Ctx_Next_Error(t *testing.T) {
 	require.Equal(t, "Works", resp.Header.Get("X-Next-Result"))
 }
 
-// go test -run Test_Ctx_Redirect
-func Test_Ctx_Redirect(t *testing.T) {
-	t.Parallel()
-	app := New()
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.Redirect("http://default.com")
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "http://default.com", string(c.Response().Header.Peek(HeaderLocation)))
-
-	c.Redirect("http://example.com", 301)
-	require.Equal(t, 301, c.Response().StatusCode())
-	require.Equal(t, "http://example.com", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectToRouteWithParams
-func Test_Ctx_RedirectToRouteWithParams(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/user/:name", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectToRoute("user", Map{
-		"name": "fiber",
-	})
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/user/fiber", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectToRouteWithParams
-func Test_Ctx_RedirectToRouteWithQueries(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/user/:name", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectToRoute("user", Map{
-		"name":    "fiber",
-		"queries": map[string]string{"data[0][name]": "john", "data[0][age]": "10", "test": "doe"},
-	})
-	require.Equal(t, 302, c.Response().StatusCode())
-	// analysis of query parameters with url parsing, since a map pass is always randomly ordered
-	location, err := url.Parse(string(c.Response().Header.Peek(HeaderLocation)))
-	require.NoError(t, err, "url.Parse(location)")
-	require.Equal(t, "/user/fiber", location.Path)
-	require.Equal(t, url.Values{"data[0][name]": []string{"john"}, "data[0][age]": []string{"10"}, "test": []string{"doe"}}, location.Query())
-}
-
-// go test -run Test_Ctx_RedirectToRouteWithOptionalParams
-func Test_Ctx_RedirectToRouteWithOptionalParams(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/user/:name?", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectToRoute("user", Map{
-		"name": "fiber",
-	})
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/user/fiber", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectToRouteWithOptionalParamsWithoutValue
-func Test_Ctx_RedirectToRouteWithOptionalParamsWithoutValue(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/user/:name?", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectToRoute("user", Map{})
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/user/", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectToRouteWithGreedyParameters
-func Test_Ctx_RedirectToRouteWithGreedyParameters(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/user/+", func(c Ctx) error {
-		return c.JSON(c.Params("+"))
-	}).Name("user")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectToRoute("user", Map{
-		"+": "test/routes",
-	})
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/user/test/routes", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectBack
-func Test_Ctx_RedirectBack(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/", func(c Ctx) error {
-		return c.JSON("Home")
-	}).Name("home")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.RedirectBack("/")
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-// go test -run Test_Ctx_RedirectBackWithReferer
-func Test_Ctx_RedirectBackWithReferer(t *testing.T) {
-	t.Parallel()
-	app := New()
-	app.Get("/", func(c Ctx) error {
-		return c.JSON("Home")
-	}).Name("home")
-	app.Get("/back", func(c Ctx) error {
-		return c.JSON("Back")
-	}).Name("back")
-	c := app.NewCtx(&fasthttp.RequestCtx{})
-
-	c.Request().Header.Set(HeaderReferer, "/back")
-	c.RedirectBack("/")
-	require.Equal(t, 302, c.Response().StatusCode())
-	require.Equal(t, "/back", c.Get(HeaderReferer))
-	require.Equal(t, "/back", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
 // go test -run Test_Ctx_Render
 func Test_Ctx_Render(t *testing.T) {
 	t.Parallel()
@@ -2597,53 +2465,6 @@ func Benchmark_Ctx_RenderWithLocalsAndBindVars(b *testing.B) {
 	require.Equal(b, "<h1>Hello, World! Test</h1>", string(c.Response().Body()))
 }
 
-func Benchmark_Ctx_RedirectToRoute(b *testing.B) {
-	app := New()
-	app.Get("/user/:name", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-
-	c := app.NewCtx(&fasthttp.RequestCtx{}).(*DefaultCtx)
-
-	b.ReportAllocs()
-	b.ResetTimer()
-
-	for n := 0; n < b.N; n++ {
-		c.RedirectToRoute("user", Map{
-			"name": "fiber",
-		})
-	}
-
-	require.Equal(b, 302, c.Response().StatusCode())
-	require.Equal(b, "/user/fiber", string(c.Response().Header.Peek(HeaderLocation)))
-}
-
-func Benchmark_Ctx_RedirectToRouteWithQueries(b *testing.B) {
-	app := New()
-	app.Get("/user/:name", func(c Ctx) error {
-		return c.JSON(c.Params("name"))
-	}).Name("user")
-
-	c := app.NewCtx(&fasthttp.RequestCtx{}).(*DefaultCtx)
-
-	b.ReportAllocs()
-	b.ResetTimer()
-
-	for n := 0; n < b.N; n++ {
-		c.RedirectToRoute("user", Map{
-			"name":    "fiber",
-			"queries": map[string]string{"a": "a", "b": "b"},
-		})
-	}
-
-	require.Equal(b, 302, c.Response().StatusCode())
-	// analysis of query parameters with url parsing, since a map pass is always randomly ordered
-	location, err := url.Parse(string(c.Response().Header.Peek(HeaderLocation)))
-	require.NoError(b, err, "url.Parse(location)")
-	require.Equal(b, "/user/fiber", location.Path)
-	require.Equal(b, url.Values{"a": []string{"a"}, "b": []string{"b"}}, location.Query())
-}
-
 func Benchmark_Ctx_RenderLocals(b *testing.B) {
 	engine := &testTemplateEngine{}
 	err := engine.Load()

middleware/expvar/expvar.go

@@ -29,6 +29,6 @@ func New(config ...Config) fiber.Handler {
 			return nil
 		}
 
-		return c.Redirect("/debug/vars", 302)
+		return c.Redirect().To("/debug/vars")
 	}
 }

middleware/pprof/pprof.go

@@ -72,7 +72,7 @@ func New(config ...Config) fiber.Handler {
 				path = "/debug/pprof/"
 			}
 
-			return c.Redirect(path, fiber.StatusFound)
+			return c.Redirect().To(path)
 		}
 		return nil
 	}

middleware/redirect/redirect.go

@@ -61,7 +61,7 @@ func New(config ...Config) fiber.Handler {
 		for k, v := range cfg.rulesRegex {
 			replacer := captureTokens(k, c.Path())
 			if replacer != nil {
-				return c.Redirect(replacer.Replace(v), cfg.StatusCode)
+				return c.Redirect().Status(cfg.StatusCode).To(replacer.Replace(v))
 			}
 		}
 		return c.Next()