mirror of
https://github.com/caddyserver/caddy.git
synced 2024-12-23 22:27:38 -05:00
95 lines
1.6 KiB
Go
95 lines
1.6 KiB
Go
|
package header
|
||
|
|
||
|
import (
|
||
|
"github.com/mholt/caddy"
|
||
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
caddy.RegisterPlugin(caddy.Plugin{
|
||
|
Name: "header",
|
||
|
ServerType: "http",
|
||
|
Action: setup,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// setup configures a new Headers middleware instance.
|
||
|
func setup(c *caddy.Controller) error {
|
||
|
rules, err := headersParse(c)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
httpserver.GetConfig(c.Key).AddMiddleware(func(next httpserver.Handler) httpserver.Handler {
|
||
|
return Headers{Next: next, Rules: rules}
|
||
|
})
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func headersParse(c *caddy.Controller) ([]Rule, error) {
|
||
|
var rules []Rule
|
||
|
|
||
|
for c.NextLine() {
|
||
|
var head Rule
|
||
|
var isNewPattern bool
|
||
|
|
||
|
if !c.NextArg() {
|
||
|
return rules, c.ArgErr()
|
||
|
}
|
||
|
pattern := c.Val()
|
||
|
|
||
|
// See if we already have a definition for this Path pattern...
|
||
|
for _, h := range rules {
|
||
|
if h.Path == pattern {
|
||
|
head = h
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// ...otherwise, this is a new pattern
|
||
|
if head.Path == "" {
|
||
|
head.Path = pattern
|
||
|
isNewPattern = true
|
||
|
}
|
||
|
|
||
|
for c.NextBlock() {
|
||
|
// A block of headers was opened...
|
||
|
|
||
|
h := Header{Name: c.Val()}
|
||
|
|
||
|
if c.NextArg() {
|
||
|
h.Value = c.Val()
|
||
|
}
|
||
|
|
||
|
head.Headers = append(head.Headers, h)
|
||
|
}
|
||
|
if c.NextArg() {
|
||
|
// ... or single header was defined as an argument instead.
|
||
|
|
||
|
h := Header{Name: c.Val()}
|
||
|
|
||
|
h.Value = c.Val()
|
||
|
|
||
|
if c.NextArg() {
|
||
|
h.Value = c.Val()
|
||
|
}
|
||
|
|
||
|
head.Headers = append(head.Headers, h)
|
||
|
}
|
||
|
|
||
|
if isNewPattern {
|
||
|
rules = append(rules, head)
|
||
|
} else {
|
||
|
for i := 0; i < len(rules); i++ {
|
||
|
if rules[i].Path == pattern {
|
||
|
rules[i] = head
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return rules, nil
|
||
|
}
|