mirror of
https://github.com/coredns/coredns.git
synced 2025-10-28 08:44:17 -04:00
29 lines
524 B
Go
29 lines
524 B
Go
|
package whoami
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/miekg/coredns/core/dnsserver"
|
||
|
|
"github.com/miekg/coredns/middleware"
|
||
|
|
|
||
|
|
"github.com/mholt/caddy"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
caddy.RegisterPlugin("whoami", caddy.Plugin{
|
||
|
|
ServerType: "dns",
|
||
|
|
Action: setupWhoami,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func setupWhoami(c *caddy.Controller) error {
|
||
|
|
c.Next() // 'whoami'
|
||
|
|
if c.NextArg() {
|
||
|
|
return middleware.Error("whoami", c.ArgErr())
|
||
|
|
}
|
||
|
|
|
||
|
|
dnsserver.GetConfig(c).AddMiddleware(func(next dnsserver.Handler) dnsserver.Handler {
|
||
|
|
return Whoami{Next: next}
|
||
|
|
})
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|