1 files changed, 56 insertions, 0 deletions

diff --git a/clients/netbox/config_file_client.go b/clients/netbox/config_file_client.go
index d84dd80..c692b9f 100644
--- a/clients/netbox/config_file_client.go
+++ b/clients/netbox/config_file_client.go
@@ -150,6 +150,18 @@ func parseValues(v []string) ([]*net.IPNet, error) {
         return out, nil
 }
 
+func parseIps(v []string) ([]net.IP, error) {
+        out := make([]net.IP, len(v))
+        for i, n := range v {
+                ip := net.ParseIP(n)
+                if ip == nil {
+                        return nil, fmt.Errorf("Invalid IP '%s'", n)
+                }
+                out[i] = ip
+        }
+        return out, nil
+}
+
 // GetServicesForVm returns a list of services for a named VM
 //
 // The data format of the config file should be, under the key "services":
@@ -183,3 +195,47 @@ func (c *ConfigFileNetboxClient) GetServicesForVm(ctx context.Context, vmName st
 
         return vm, nil
 }
+
+// GetDnsServersForSite gets a list of IP address for a site that are
+// assigned to DNS servers.
+//
+// The data format of the config file should be, under the key "sites":
+//
+//      map[string]map[string][]string{
+//              "site_name": map[string][]string{
+//                      "dns-servers": []string{
+//                              "127.0.0.1",
+//                      },
+//              },
+//      }
+//
+// In JSON format:
+//
+//      { "sites": {
+//          "site": {
+//            "dns-servers": [ "127.0.0.1" ]
+//          }
+//      }
+func (c *ConfigFileNetboxClient) GetDnsServersForSite(ctx context.Context, site string) ([]net.IP, error) {
+        s, ok := c.cfg["sites"]
+        if !ok {
+                return nil, fmt.Errorf("No key 'sites' in config file")
+        }
+
+        sites := map[string]map[string][]string{}
+        if err := mapstructure.Decode(s, &sites); err != nil {
+                return nil, fmt.Errorf("Key 'sites' not in correct format: %w", err)
+        }
+
+        tags, ok := sites[site]
+        if !ok {
+                return nil, fmt.Errorf("Site %s not in sites", site)
+        }
+
+        values, ok := tags["dns-servers"]
+        if !ok {
+                return nil, fmt.Errorf("Site %s not in sites", site)
+        }
+
+        return parseIps(values)
+}