1 files changed, 23 insertions, 0 deletions

diff --git a/clients/netbox/client.go b/clients/netbox/client.go
index 3b7fbaf..aceebed 100644
--- a/clients/netbox/client.go
+++ b/clients/netbox/client.go
@@ -11,6 +11,7 @@ type NetboxClient interface {
         GetSitePrefixesWithTag(ctx context.Context, site string, tag string) ([]*net.IPNet, error)
         GetPrefixesWithTag(ctx context.Context, tag string) ([]*net.IPNet, error)
         GetServicesForVm(ctx context.Context, vmName string) ([]*Service, error)
+        GetDnsServersForSite(ctx context.Context, site string) ([]net.IP, error)
 }
 
 type BasicNetboxClient struct {
@@ -145,3 +146,25 @@ func (c *BasicNetboxClient) GetServicesForVm(ctx context.Context, vmName string)
 
         return out, nil
 }
+
+func (c *BasicNetboxClient) GetDnsServersForSite(ctx context.Context, site string) ([]net.IP, error) {
+        var m dnsServerGQLResponse
+        if err := c.Do(ctx, &NetboxGraphQLRequest{dnsServerGQLQuery}, &m); err != nil {
+                return nil, err
+        }
+
+        out := []net.IP{}
+        for _, a := range m.Data.AddressList {
+                if a.AssignedObject.VirtualMachine.Site.Name == site ||
+                        a.AssignedObject.Device.Site.Name == site {
+
+                        ip, _, err := net.ParseCIDR(a.Address)
+                        if err != nil {
+                                return nil, err
+                        }
+
+                        out = append(out, ip)
+                }
+        }
+        return out, nil
+}