2015-01-30 14:52:18 +08:00
|
|
|
// Package redirect is middleware for redirecting certain requests
|
2015-01-14 03:43:45 +08:00
|
|
|
// to other locations.
|
2015-01-30 13:05:54 +08:00
|
|
|
package redirect
|
2015-01-19 14:11:21 +08:00
|
|
|
|
2015-01-30 13:05:54 +08:00
|
|
|
import (
|
2015-06-10 13:44:40 +08:00
|
|
|
"fmt"
|
|
|
|
|
"html"
|
2015-01-30 13:05:54 +08:00
|
|
|
"net/http"
|
2015-07-12 22:01:32 +08:00
|
|
|
"net/url"
|
|
|
|
|
"regexp"
|
2015-01-30 13:05:54 +08:00
|
|
|
|
|
|
|
|
"github.com/mholt/caddy/middleware"
|
|
|
|
|
)
|
2015-01-19 14:11:21 +08:00
|
|
|
|
2015-04-12 07:06:09 +08:00
|
|
|
// Redirect is middleware to respond with HTTP redirects
|
|
|
|
|
type Redirect struct {
|
|
|
|
|
Next middleware.Handler
|
|
|
|
|
Rules []Rule
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// ServeHTTP implements the middleware.Handler interface.
|
|
|
|
|
func (rd Redirect) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
|
|
|
for _, rule := range rd.Rules {
|
2015-04-13 08:13:58 +08:00
|
|
|
if rule.From == "/" {
|
|
|
|
|
// Catchall redirect preserves path (TODO: Standardize/formalize this behavior)
|
2015-07-12 22:01:32 +08:00
|
|
|
toURL, err := url.ParseRequestURI(rule.To)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 500, err
|
|
|
|
|
}
|
|
|
|
|
newPath := toURL.Host + toURL.Path + r.URL.Path
|
|
|
|
|
rmSlashs := regexp.MustCompile("//+")
|
|
|
|
|
newPath = rmSlashs.ReplaceAllString(newPath, "/")
|
|
|
|
|
newPath = toURL.Scheme + "://" + newPath
|
|
|
|
|
parameters := toURL.Query()
|
|
|
|
|
for k, v := range r.URL.Query() {
|
|
|
|
|
parameters.Set(k, v[0])
|
|
|
|
|
}
|
|
|
|
|
if len(parameters) > 0 {
|
|
|
|
|
newPath = newPath + "?" + parameters.Encode()
|
|
|
|
|
}
|
2015-06-10 13:44:40 +08:00
|
|
|
if rule.Meta {
|
|
|
|
|
fmt.Fprintf(w, metaRedir, html.EscapeString(newPath))
|
|
|
|
|
} else {
|
|
|
|
|
http.Redirect(w, r, newPath, rule.Code)
|
|
|
|
|
}
|
2015-04-13 08:13:58 +08:00
|
|
|
return 0, nil
|
|
|
|
|
}
|
2015-04-13 07:40:59 +08:00
|
|
|
if r.URL.Path == rule.From {
|
2015-06-10 13:44:40 +08:00
|
|
|
if rule.Meta {
|
|
|
|
|
fmt.Fprintf(w, metaRedir, html.EscapeString(rule.To))
|
|
|
|
|
} else {
|
|
|
|
|
http.Redirect(w, r, rule.To, rule.Code)
|
|
|
|
|
}
|
2015-04-12 07:06:09 +08:00
|
|
|
return 0, nil
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return rd.Next.ServeHTTP(w, r)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Rule describes an HTTP redirect rule.
|
|
|
|
|
type Rule struct {
|
|
|
|
|
From, To string
|
|
|
|
|
Code int
|
2015-06-10 13:44:40 +08:00
|
|
|
Meta bool
|
2015-01-14 03:43:45 +08:00
|
|
|
}
|
2015-06-10 13:44:40 +08:00
|
|
|
|
|
|
|
|
var metaRedir = `<html>
|
|
|
|
|
<head>
|
|
|
|
|
<meta http-equiv="refresh" content="0;URL='%s'">
|
|
|
|
|
</head>
|
|
|
|
|
<body>redirecting...</body>
|
|
|
|
|
</html>`
|
