1 files changed, 69 insertions, 0 deletions

diff --git a/dns/cilent.go b/dns/cilent.go
new file mode 100644
index 0000000..63e3734
--- /dev/null
+++ b/dns/cilent.go
@@ -0,0 +1,69 @@
+package dns
+
+import (
+        "code.crute.me/mcrute/go_ddns_manager/bind"
+        "github.com/miekg/dns"
+)
+
+type DNSClient struct {
+        Server string
+}
+
+func (c *DNSClient) AXFR(zone *bind.Zone) (chan *dns.Envelope, error) {
+        k := zone.Keys()[0]
+        t := &dns.Transfer{TsigSecret: k.AsMap()}
+
+        m := &dns.Msg{}
+        m.SetAxfr(zone.Name)
+        k.Sign(m)
+
+        return t.In(m, c.Server)
+}
+
+func (c *DNSClient) Insert(zone *bind.Zone, rrs ...RR) error {
+        k := zone.Keys()[0]
+        dc := &dns.Client{TsigSecret: k.AsMap()}
+
+        m := &dns.Msg{}
+        m.SetUpdate(zone.Name)
+        m.Insert(toRRSet(zone, rrs...))
+        k.Sign(m)
+
+        if _, _, err := dc.Exchange(m, c.Server); err != nil {
+                return err
+        }
+
+        return nil
+}
+
+func (c *DNSClient) Remove(zone *bind.Zone, rrs ...RR) error {
+        k := zone.Keys()[0]
+        dc := &dns.Client{TsigSecret: k.AsMap()}
+
+        m := &dns.Msg{}
+        m.SetUpdate(zone.Name)
+        m.Remove(toRRSet(zone, rrs...))
+        k.Sign(m)
+
+        if _, _, err := dc.Exchange(m, c.Server); err != nil {
+                return err
+        }
+
+        return nil
+}
+
+func (c *DNSClient) RemoveAll(zone *bind.Zone, rrs ...RR) error {
+        k := zone.Keys()[0]
+        dc := &dns.Client{TsigSecret: k.AsMap()}
+
+        m := &dns.Msg{}
+        m.SetUpdate(zone.Name)
+        m.RemoveRRset(toRRSet(zone, rrs...))
+        k.Sign(m)
+
+        if _, _, err := dc.Exchange(m, c.Server); err != nil {
+                return err
+        }
+
+        return nil
+}