github-mirror/caddy
2
0
Fork 0
mirror of https://github.com/caddyserver/caddy.git synced 2024-11-29 12:16:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #952 from abiosoft/condition-patch

Browse Source 
minor condition keyword check refactor
This commit is contained in:
Matt Holt 2016-07-21 12:03:33 -06:00 committed by GitHub
commit 4678471fe0
4 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
caddyhttp/httpserver/condition.go
View File

@ -6,12 +6,13 @@ import (
"regexp"
"strings"
"github.com/mholt/caddy/caddyfile"
"github.com/mholt/caddy"
)
// SetupIfMatcher parses `if` or `if_op` in the current dispenser block.
// It returns a RequestMatcher and an error if any.
func SetupIfMatcher(c caddyfile.Dispenser) (RequestMatcher, error) {
func SetupIfMatcher(controller *caddy.Controller) (RequestMatcher, error) {
var c = controller.Dispenser // copy the dispenser
var matcher IfMatcher
for c.NextBlock() {
switch c.Val() {
@ -193,7 +194,13 @@ func (m IfMatcher) Or(r *http.Request) bool {
return false
}
// IfMatcherKeyword returns if k is a keyword for 'if' config block.
func IfMatcherKeyword(k string) bool {
return k == "if" || k == "if_op"
// IfMatcherKeyword checks if the next value in the dispenser is a keyword for 'if' config block.
// If true, remaining arguments in the dispinser are cleard to keep the dispenser valid for use.
func IfMatcherKeyword(c *caddy.Controller) bool {
if c.Val() == "if" || c.Val() == "if_op" {
// clear remainig args
c.RemainingArgs()
return true
}
return false
}

7
caddyhttp/httpserver/condition_test.go
View File

@ -243,7 +243,7 @@ func TestSetupIfMatcher(t *testing.T) {
for i, test := range tests {
c := caddy.NewTestController("http", test.input)
c.Next()
matcher, err := SetupIfMatcher(c.Dispenser)
matcher, err := SetupIfMatcher(c)
if err == nil && test.shouldErr {
t.Errorf("Test %d didn't error, but it should have", i)
} else if err != nil && !test.shouldErr {
@ -277,8 +277,11 @@ func TestIfMatcherKeyword(t *testing.T) {
{"if_type", false},
{"if_cond", false},
}
for i, test := range tests {
valid := IfMatcherKeyword(test.keyword)
c := caddy.NewTestController("http", test.keyword)
c.Next()
valid := IfMatcherKeyword(c)
if valid != test.expected {
t.Errorf("Test %d: expected %v found %v", i, test.expected, valid)
}

5
caddyhttp/redirect/setup.go
View File

@ -63,7 +63,7 @@ func redirParse(c *caddy.Controller) ([]Rule, error) {
}
for c.Next() {
matcher, err := httpserver.SetupIfMatcher(c.Dispenser)
matcher, err := httpserver.SetupIfMatcher(c)
if err != nil {
return nil, err
}
@ -71,8 +71,7 @@ func redirParse(c *caddy.Controller) ([]Rule, error) {
var hadOptionalBlock bool
for c.NextBlock() {
if httpserver.IfMatcherKeyword(c.Val()) {
c.RemainingArgs()
if httpserver.IfMatcherKeyword(c) {
continue
}

11
caddyhttp/rewrite/setup.go
View File

@ -57,12 +57,15 @@ func rewriteParse(c *caddy.Controller) ([]httpserver.HandlerConfig, error) {
fallthrough
case 0:
// Integrate request matcher for 'if' conditions.
matcher, err = httpserver.SetupIfMatcher(c.Dispenser)
matcher, err = httpserver.SetupIfMatcher(c)
if err != nil {
return nil, err
}
block:
for c.NextBlock() {
if httpserver.IfMatcherKeyword(c) {
continue
}
switch c.Val() {
case "r", "regexp":
if !c.NextArg() {
@ -90,10 +93,6 @@ func rewriteParse(c *caddy.Controller) ([]httpserver.HandlerConfig, error) {
return nil, c.Err("status must be 2xx or 4xx")
}
default:
if httpserver.IfMatcherKeyword(c.Val()) {
c.RemainingArgs()
continue block
}
return nil, c.ArgErr()
}
}