blob: fc91d288762494ce5bd184876da5ffe5698fb95e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
package rewrite
import (
"net/http"
"net/url"
"strings"
"bitbucket.org/lightcodelabs/caddy2"
"bitbucket.org/lightcodelabs/caddy2/modules/caddyhttp"
)
func init() {
caddy2.RegisterModule(caddy2.Module{
Name: "http.middleware.rewrite",
New: func() interface{} { return new(Rewrite) },
})
}
// Rewrite is a middleware which can rewrite HTTP requests.
type Rewrite struct {
Method string `json:"method,omitempty"`
URI string `json:"uri,omitempty"`
Rehandle bool `json:"rehandle,omitempty"`
}
func (rewr Rewrite) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddyhttp.Handler) error {
repl := r.Context().Value(caddy2.ReplacerCtxKey).(caddy2.Replacer)
var rehandleNeeded bool
if rewr.Method != "" {
method := r.Method
r.Method = strings.ToUpper(repl.ReplaceAll(rewr.Method, ""))
if r.Method != method {
rehandleNeeded = true
}
}
if rewr.URI != "" {
oldURI := r.RequestURI
newURI := repl.ReplaceAll(rewr.URI, "")
u, err := url.Parse(newURI)
if err != nil {
return caddyhttp.Error(http.StatusInternalServerError, err)
}
r.RequestURI = newURI
r.URL.Path = u.Path
if u.RawQuery != "" {
r.URL.RawQuery = u.RawQuery
}
if u.Fragment != "" {
r.URL.Fragment = u.Fragment
}
if newURI != oldURI {
rehandleNeeded = true
}
}
if rehandleNeeded && rewr.Rehandle {
return caddyhttp.ErrRehandle
}
return next.ServeHTTP(w, r)
}
// Interface guard
var _ caddyhttp.MiddlewareHandler = (*Rewrite)(nil)
|
