1 files changed, 172 insertions, 0 deletions
diff --git a/plugin/auto/setup.go b/plugin/auto/setup.go
new file mode 100644
index 000000000..75966f8a0
--- /dev/null
+++ b/plugin/auto/setup.go
@@ -0,0 +1,172 @@
+package auto
+
+import (
+	"log"
+	"os"
+	"path"
+	"regexp"
+	"strconv"
+	"time"
+
+	"github.com/coredns/coredns/core/dnsserver"
+	"github.com/coredns/coredns/plugin"
+	"github.com/coredns/coredns/plugin/file"
+	"github.com/coredns/coredns/plugin/metrics"
+	"github.com/coredns/coredns/plugin/pkg/dnsutil"
+	"github.com/coredns/coredns/plugin/proxy"
+
+	"github.com/mholt/caddy"
+)
+
+func init() {
+	caddy.RegisterPlugin("auto", caddy.Plugin{
+		ServerType: "dns",
+		Action:     setup,
+	})
+}
+
+func setup(c *caddy.Controller) error {
+	a, err := autoParse(c)
+	if err != nil {
+		return plugin.Error("auto", err)
+	}
+
+	c.OnStartup(func() error {
+		m := dnsserver.GetConfig(c).Handler("prometheus")
+		if m == nil {
+			return nil
+		}
+		(&a).metrics = m.(*metrics.Metrics)
+		return nil
+	})
+
+	walkChan := make(chan bool)
+
+	c.OnStartup(func() error {
+		err := a.Walk()
+		if err != nil {
+			return err
+		}
+
+		go func() {
+			ticker := time.NewTicker(a.loader.duration)
+			for {
+				select {
+				case <-walkChan:
+					return
+				case <-ticker.C:
+					a.Walk()
+				}
+			}
+		}()
+		return nil
+	})
+
+	c.OnShutdown(func() error {
+		close(walkChan)
+		return nil
+	})
+
+	dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
+		a.Next = next
+		return a
+	})
+
+	return nil
+}
+
+func autoParse(c *caddy.Controller) (Auto, error) {
+	var a = Auto{
+		loader: loader{template: "${1}", re: regexp.MustCompile(`db\.(.*)`), duration: 60 * time.Second},
+		Zones:  &Zones{},
+	}
+
+	config := dnsserver.GetConfig(c)
+
+	for c.Next() {
+		// auto [ZONES...]
+		a.Zones.origins = make([]string, len(c.ServerBlockKeys))
+		copy(a.Zones.origins, c.ServerBlockKeys)
+
+		args := c.RemainingArgs()
+		if len(args) > 0 {
+			a.Zones.origins = args
+		}
+		for i := range a.Zones.origins {
+			a.Zones.origins[i] = plugin.Host(a.Zones.origins[i]).Normalize()
+		}
+
+		for c.NextBlock() {
+			switch c.Val() {
+			case "directory": // directory DIR [REGEXP [TEMPLATE] [DURATION]]
+				if !c.NextArg() {
+					return a, c.ArgErr()
+				}
+				a.loader.directory = c.Val()
+				if !path.IsAbs(a.loader.directory) && config.Root != "" {
+					a.loader.directory = path.Join(config.Root, a.loader.directory)
+				}
+				_, err := os.Stat(a.loader.directory)
+				if err != nil {
+					if os.IsNotExist(err) {
+						log.Printf("[WARNING] Directory does not exist: %s", a.loader.directory)
+					} else {
+						return a, c.Errf("Unable to access root path '%s': %v", a.loader.directory, err)
+					}
+				}
+
+				// regexp
+				if c.NextArg() {
+					a.loader.re, err = regexp.Compile(c.Val())
+					if err != nil {
+						return a, err
+					}
+					if a.loader.re.NumSubexp() == 0 {
+						return a, c.Errf("Need at least one sub expression")
+					}
+				}
+
+				// template
+				if c.NextArg() {
+					a.loader.template = rewriteToExpand(c.Val())
+				}
+
+				// duration
+				if c.NextArg() {
+					i, err := strconv.Atoi(c.Val())
+					if err != nil {
+						return a, err
+					}
+					if i < 1 {
+						i = 1
+					}
+					a.loader.duration = time.Duration(i) * time.Second
+				}
+
+			case "no_reload":
+				a.loader.noReload = true
+
+			case "upstream":
+				args := c.RemainingArgs()
+				if len(args) == 0 {
+					return a, c.ArgErr()
+				}
+				ups, err := dnsutil.ParseHostPortOrFile(args...)
+				if err != nil {
+					return a, err
+				}
+				a.loader.proxy = proxy.NewLookup(ups)
+
+			default:
+				t, _, e := file.TransferParse(c, false)
+				if e != nil {
+					return a, e
+				}
+				if t != nil {
+					a.loader.transferTo = append(a.loader.transferTo, t...)
+				}
+			}
+		}
+	}
+	return a, nil
+}