diff options
author | 2017-09-14 09:36:06 +0100 | |
---|---|---|
committer | 2017-09-14 09:36:06 +0100 | |
commit | d8714e64e400ef873c2adc4d929a07d7890727b9 (patch) | |
tree | c9fa4c157e6af12eb1517654f8d23ca5d5619513 /plugin/autopath/setup.go | |
parent | b984aa45595dc95253b91191afe7d3ee29e71b48 (diff) | |
download | coredns-d8714e64e400ef873c2adc4d929a07d7890727b9.tar.gz coredns-d8714e64e400ef873c2adc4d929a07d7890727b9.tar.zst coredns-d8714e64e400ef873c2adc4d929a07d7890727b9.zip |
Remove the word middleware (#1067)
* Rename middleware to plugin
first pass; mostly used 'sed', few spots where I manually changed
text.
This still builds a coredns binary.
* fmt error
* Rename AddMiddleware to AddPlugin
* Readd AddMiddleware to remain backwards compat
Diffstat (limited to 'plugin/autopath/setup.go')
-rw-r--r-- | plugin/autopath/setup.go | 93 |
1 files changed, 93 insertions, 0 deletions
diff --git a/plugin/autopath/setup.go b/plugin/autopath/setup.go new file mode 100644 index 000000000..c83912a63 --- /dev/null +++ b/plugin/autopath/setup.go @@ -0,0 +1,93 @@ +package autopath + +import ( + "fmt" + + "github.com/coredns/coredns/core/dnsserver" + "github.com/coredns/coredns/plugin" + "github.com/coredns/coredns/plugin/erratic" + "github.com/coredns/coredns/plugin/kubernetes" + + "github.com/mholt/caddy" + "github.com/miekg/dns" +) + +func init() { + caddy.RegisterPlugin("autopath", caddy.Plugin{ + ServerType: "dns", + Action: setup, + }) + +} + +func setup(c *caddy.Controller) error { + ap, mw, err := autoPathParse(c) + if err != nil { + return plugin.Error("autopath", err) + } + + // Do this in OnStartup, so all plugin has been initialized. + c.OnStartup(func() error { + m := dnsserver.GetConfig(c).Handler(mw) + if m == nil { + return nil + } + if x, ok := m.(*kubernetes.Kubernetes); ok { + ap.searchFunc = x.AutoPath + } + if x, ok := m.(*erratic.Erratic); ok { + ap.searchFunc = x.AutoPath + } + return nil + }) + + dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler { + ap.Next = next + return ap + }) + + return nil +} + +// allowedMiddleware has a list of plugin that can be used by autopath. +var allowedMiddleware = map[string]bool{ + "@kubernetes": true, + "@erratic": true, +} + +func autoPathParse(c *caddy.Controller) (*AutoPath, string, error) { + ap := &AutoPath{} + mw := "" + + for c.Next() { + zoneAndresolv := c.RemainingArgs() + if len(zoneAndresolv) < 1 { + return ap, "", fmt.Errorf("no resolv-conf specified") + } + resolv := zoneAndresolv[len(zoneAndresolv)-1] + if resolv[0] == '@' { + _, ok := allowedMiddleware[resolv] + if ok { + mw = resolv[1:] + } + } else { + // assume file on disk + rc, err := dns.ClientConfigFromFile(resolv) + if err != nil { + return ap, "", fmt.Errorf("failed to parse %q: %v", resolv, err) + } + ap.search = rc.Search + plugin.Zones(ap.search).Normalize() + ap.search = append(ap.search, "") // sentinal value as demanded. + } + ap.Zones = zoneAndresolv[:len(zoneAndresolv)-1] + if len(ap.Zones) == 0 { + ap.Zones = make([]string, len(c.ServerBlockKeys)) + copy(ap.Zones, c.ServerBlockKeys) + } + for i, str := range ap.Zones { + ap.Zones[i] = plugin.Host(str).Normalize() + } + } + return ap, mw, nil +} |