74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package setup
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/miekg/coredns/middleware"
|
|
"github.com/miekg/coredns/middleware/file"
|
|
)
|
|
|
|
// File sets up the file middleware.
|
|
func File(c *Controller) (middleware.Middleware, error) {
|
|
zones, err := fileParse(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Set start function is transfer is specified
|
|
|
|
return func(next middleware.Handler) middleware.Handler {
|
|
return file.File{Next: next, Zones: zones}
|
|
}, nil
|
|
|
|
}
|
|
|
|
func fileParse(c *Controller) (file.Zones, error) {
|
|
z := make(map[string]*file.Zone)
|
|
names := []string{}
|
|
for c.Next() {
|
|
if c.Val() == "file" {
|
|
// file db.file [origin]
|
|
if !c.NextArg() {
|
|
return file.Zones{}, c.ArgErr()
|
|
}
|
|
fileName := c.Val()
|
|
|
|
origin := c.ServerBlockHosts[c.ServerBlockHostIndex]
|
|
if c.NextArg() {
|
|
origin = c.Val()
|
|
}
|
|
// normalize this origin
|
|
origin = middleware.Host(origin).Standard()
|
|
|
|
reader, err := os.Open(fileName)
|
|
if err != nil {
|
|
return file.Zones{}, err
|
|
}
|
|
zone, err := file.Parse(reader, origin, fileName)
|
|
if err == nil {
|
|
z[origin] = zone
|
|
}
|
|
names = append(names, origin)
|
|
if c.NextBlock() {
|
|
what := c.Val()
|
|
if !c.NextArg() {
|
|
return file.Zones{}, c.ArgErr()
|
|
}
|
|
value := c.Val()
|
|
var err error
|
|
switch what {
|
|
case "transfer":
|
|
if value == "out" {
|
|
z[origin].Transfer.Out = true
|
|
}
|
|
if value == "in" {
|
|
z[origin].Transfer.In = true
|
|
}
|
|
}
|
|
if err != nil {
|
|
return file.Zones{}, err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return file.Zones{Z: z, Names: names}, nil
|
|
}
|