From 0a9a19305c7edef8e299f29cb23214e1b6f15a8d Mon Sep 17 00:00:00 2001
From: Matthew Holt <Matthew.Holt+git@gmail.com>
Date: Sun, 29 Mar 2015 21:48:53 -0600
Subject: [PATCH] Made catch-all redirects possible

---
 middleware/redirect/redirect.go | 54 +++++++++++++++++----------------
 1 file changed, 28 insertions(+), 26 deletions(-)

diff --git a/middleware/redirect/redirect.go b/middleware/redirect/redirect.go
index 59f72cab..6cdbcfa6 100644
--- a/middleware/redirect/redirect.go
+++ b/middleware/redirect/redirect.go
@@ -4,47 +4,49 @@ package redirect
 
 import (
 	"net/http"
+	"strings"
 
 	"github.com/mholt/caddy/middleware"
 )
 
 // New creates a new redirect middleware.
 func New(c middleware.Controller) (middleware.Middleware, error) {
-	var redirects []redirect
+	var redirects []Redirect
 
 	for c.Next() {
-		var rule redirect
+		var rule Redirect
+		args := c.RemainingArgs()
 
-		// From
-		if !c.NextArg() {
-			return nil, c.ArgErr()
-		}
-		rule.From = c.Val()
-
-		// To
-		if !c.NextArg() {
-			return nil, c.ArgErr()
-		}
-		rule.To = c.Val()
-
-		// Status Code
-		if !c.NextArg() {
-			return nil, c.ArgErr()
-		}
-
-		if code, ok := httpRedirs[c.Val()]; !ok {
-			return nil, c.Err("Invalid redirect code '" + c.Val() + "'")
+		if len(args) == 1 {
+			// Only 'To' specified
+			rule.From = "/"
+			rule.To = c.Val()
+			rule.Code = 307 // TODO: Consider 301 instead?
+			redirects = append(redirects, rule)
+		} else if len(args) == 3 {
+			// From, To, and Code specified
+			rule.From = args[0]
+			rule.To = args[1]
+			if code, ok := httpRedirs[args[2]]; !ok {
+				return nil, c.Err("Invalid redirect code '" + c.Val() + "'")
+			} else {
+				rule.Code = code
+			}
+			redirects = append(redirects, rule)
 		} else {
-			rule.Code = code
+			return nil, c.ArgErr()
 		}
-
-		redirects = append(redirects, rule)
 	}
 
 	return func(next middleware.HandlerFunc) middleware.HandlerFunc {
 		return func(w http.ResponseWriter, r *http.Request) (int, error) {
 			for _, rule := range redirects {
-				if r.URL.Path == rule.From {
+				if middleware.Path(r.URL.Path).Matches(rule.From) {
+					if rule.From == "/" {
+						// Catchall redirect preserves path (TODO: This should be made more consistent...)
+						http.Redirect(w, r, strings.TrimSuffix(rule.To, "/")+r.URL.Path, rule.Code)
+						return 0, nil
+					}
 					http.Redirect(w, r, rule.To, rule.Code)
 					return 0, nil
 				}
@@ -55,7 +57,7 @@ func New(c middleware.Controller) (middleware.Middleware, error) {
 }
 
 // redirect describes an HTTP redirect rule.
-type redirect struct {
+type Redirect struct {
 	From string
 	To   string
 	Code int