aboutsummaryrefslogtreecommitdiff
path: root/plugin/autopath/setup.go
diff options
context:
space:
mode:
authorGravatar Miek Gieben <miek@miek.nl> 2017-09-14 09:36:06 +0100
committerGravatar GitHub <noreply@github.com> 2017-09-14 09:36:06 +0100
commitd8714e64e400ef873c2adc4d929a07d7890727b9 (patch)
treec9fa4c157e6af12eb1517654f8d23ca5d5619513 /plugin/autopath/setup.go
parentb984aa45595dc95253b91191afe7d3ee29e71b48 (diff)
downloadcoredns-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.go93
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
+}