mirror of
https://github.com/caddyserver/caddy.git
synced 2024-11-25 17:56:34 +08:00
a798e0c951
- Server types no longer need to store their own contexts; they are stored on the caddy.Instance, which means each context will be properly GC'ed when the instance is stopped. Server types should use type assertions to convert from caddy.Context to their concrete context type when they need to use it. - Pass the entire context into httpserver.GetConfig instead of only the Key field. - caddy.NewTestController now requires a server type string so it can create a controller with the proper concrete context associated with that server type. Tests still need more attention so that we can test the proper creation of startup functions, etc.
121 lines
2.4 KiB
Go
121 lines
2.4 KiB
Go
package rewrite
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/mholt/caddy"
|
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
|
)
|
|
|
|
func init() {
|
|
caddy.RegisterPlugin("rewrite", caddy.Plugin{
|
|
ServerType: "http",
|
|
Action: setup,
|
|
})
|
|
}
|
|
|
|
// setup configures a new Rewrite middleware instance.
|
|
func setup(c *caddy.Controller) error {
|
|
rewrites, err := rewriteParse(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cfg := httpserver.GetConfig(c)
|
|
|
|
cfg.AddMiddleware(func(next httpserver.Handler) httpserver.Handler {
|
|
return Rewrite{
|
|
Next: next,
|
|
FileSys: http.Dir(cfg.Root),
|
|
Rules: rewrites,
|
|
}
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func rewriteParse(c *caddy.Controller) ([]Rule, error) {
|
|
var simpleRules []Rule
|
|
var regexpRules []Rule
|
|
|
|
for c.Next() {
|
|
var rule Rule
|
|
var err error
|
|
var base = "/"
|
|
var pattern, to string
|
|
var status int
|
|
var ext []string
|
|
|
|
args := c.RemainingArgs()
|
|
|
|
var ifs []If
|
|
|
|
switch len(args) {
|
|
case 1:
|
|
base = args[0]
|
|
fallthrough
|
|
case 0:
|
|
for c.NextBlock() {
|
|
switch c.Val() {
|
|
case "r", "regexp":
|
|
if !c.NextArg() {
|
|
return nil, c.ArgErr()
|
|
}
|
|
pattern = c.Val()
|
|
case "to":
|
|
args1 := c.RemainingArgs()
|
|
if len(args1) == 0 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
to = strings.Join(args1, " ")
|
|
case "ext":
|
|
args1 := c.RemainingArgs()
|
|
if len(args1) == 0 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
ext = args1
|
|
case "if":
|
|
args1 := c.RemainingArgs()
|
|
if len(args1) != 3 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
ifCond, err := NewIf(args1[0], args1[1], args1[2])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ifs = append(ifs, ifCond)
|
|
case "status":
|
|
if !c.NextArg() {
|
|
return nil, c.ArgErr()
|
|
}
|
|
status, _ = strconv.Atoi(c.Val())
|
|
if status < 200 || (status > 299 && status < 400) || status > 499 {
|
|
return nil, c.Err("status must be 2xx or 4xx")
|
|
}
|
|
default:
|
|
return nil, c.ArgErr()
|
|
}
|
|
}
|
|
// ensure to or status is specified
|
|
if to == "" && status == 0 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
if rule, err = NewComplexRule(base, pattern, to, status, ext, ifs); err != nil {
|
|
return nil, err
|
|
}
|
|
regexpRules = append(regexpRules, rule)
|
|
|
|
// the only unhandled case is 2 and above
|
|
default:
|
|
rule = NewSimpleRule(args[0], strings.Join(args[1:], " "))
|
|
simpleRules = append(simpleRules, rule)
|
|
}
|
|
|
|
}
|
|
|
|
// put simple rules in front to avoid regexp computation for them
|
|
return append(simpleRules, regexpRules...), nil
|
|
}
|