mirror of
https://github.com/jwhited/wgsd.git
synced 2025-01-19 06:19:33 +08:00
implement config parsing
This commit is contained in:
parent
48c4ad579d
commit
9963322fb6
24
setup.go
24
setup.go
@ -6,6 +6,7 @@ import (
|
|||||||
"github.com/caddyserver/caddy"
|
"github.com/caddyserver/caddy"
|
||||||
"github.com/coredns/coredns/core/dnsserver"
|
"github.com/coredns/coredns/core/dnsserver"
|
||||||
"github.com/coredns/coredns/plugin"
|
"github.com/coredns/coredns/plugin"
|
||||||
|
"github.com/miekg/dns"
|
||||||
"golang.zx2c4.com/wireguard/wgctrl"
|
"golang.zx2c4.com/wireguard/wgctrl"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -14,19 +15,38 @@ func init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func setup(c *caddy.Controller) error {
|
func setup(c *caddy.Controller) error {
|
||||||
|
c.Next() // Ignore "wgsd" and give us the next token.
|
||||||
|
|
||||||
|
// return an error if there is no zone specified
|
||||||
|
if !c.NextArg() {
|
||||||
|
return plugin.Error("wgsd", c.ArgErr())
|
||||||
|
}
|
||||||
|
zone := dns.Fqdn(c.Val())
|
||||||
|
|
||||||
|
// return an error if there is no device name specified
|
||||||
|
if !c.NextArg() {
|
||||||
|
return plugin.Error("wgsd", c.ArgErr())
|
||||||
|
}
|
||||||
|
device := c.Val()
|
||||||
|
|
||||||
|
// return an error if there are more tokens on this line
|
||||||
|
if c.NextArg() {
|
||||||
|
return plugin.Error("wgsd", c.ArgErr())
|
||||||
|
}
|
||||||
|
|
||||||
client, err := wgctrl.New()
|
client, err := wgctrl.New()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("wgsd: error constructing wgctrl client: %v",
|
return fmt.Errorf("wgsd: error constructing wgctrl client: %v",
|
||||||
err)
|
err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: parse zone and wireguard device name from config
|
|
||||||
|
|
||||||
// Add the Plugin to CoreDNS, so Servers can use it in their plugin chain.
|
// Add the Plugin to CoreDNS, so Servers can use it in their plugin chain.
|
||||||
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
|
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
|
||||||
return &WGSD{
|
return &WGSD{
|
||||||
Next: next,
|
Next: next,
|
||||||
client: client,
|
client: client,
|
||||||
|
zone: zone,
|
||||||
|
device: device,
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user