caddy/caddyhttp/rewrite/setup.go

108 lines
2.0 KiB
Go
Raw Normal View History

2016-06-06 11:51:56 +08:00
package rewrite
import (
"net/http"
"strings"
"github.com/mholt/caddy"
"github.com/mholt/caddy/caddyhttp/httpserver"
)
func init() {
caddy.RegisterPlugin("rewrite", caddy.Plugin{
2016-06-06 11:51:56 +08:00
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)
2016-06-06 11:51:56 +08:00
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) ([]httpserver.HandlerConfig, error) {
var rules []httpserver.HandlerConfig
2016-06-06 11:51:56 +08:00
for c.Next() {
var rule Rule
var err error
var base = "/"
var pattern, to string
var ext []string
args := c.RemainingArgs()
var matcher httpserver.RequestMatcher
2016-06-06 11:51:56 +08:00
switch len(args) {
case 1:
base = args[0]
fallthrough
case 0:
// Integrate request matcher for 'if' conditions.
2016-07-21 22:42:38 +08:00
matcher, err = httpserver.SetupIfMatcher(c)
if err != nil {
return nil, err
}
2016-07-21 22:42:38 +08:00
2016-06-06 11:51:56 +08:00
for c.NextBlock() {
2016-07-21 22:42:38 +08:00
if httpserver.IfMatcherKeyword(c) {
continue
}
2016-06-06 11:51:56 +08:00
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
default:
return nil, c.ArgErr()
}
}
// ensure to is specified
if to == "" {
2016-06-06 11:51:56 +08:00
return nil, c.ArgErr()
}
if rule, err = NewComplexRule(base, pattern, to, ext, matcher); err != nil {
2016-06-06 11:51:56 +08:00
return nil, err
}
rules = append(rules, rule)
2016-06-06 11:51:56 +08:00
// the only unhandled case is 2 and above
2016-06-06 11:51:56 +08:00
default:
rule = NewSimpleRule(args[0], strings.Join(args[1:], " "))
rules = append(rules, rule)
2016-06-06 11:51:56 +08:00
}
}
return rules, nil
2016-06-06 11:51:56 +08:00
}