netbox: Support arbitrary tag definitions

README.md

@@ -14,6 +14,7 @@ in the environment:
     global data.
   * `NETBOX_TOKEN` - Token with suitable permissions to read netbox
     data.
+  * `NETBOX_TAG` - Only service hosts that have a specific tag.
   * `NETBOX_URL` - URL to the netbox server.
   * `DNSMASQ_TEMPLATE` - A go template expression for the dhcp-hosts
     file.  Defaults to a suitable configuration for IPv4.  The default

main.go

@@ -48,14 +48,15 @@ func main() {
 		os.Exit(1)
 	}
 
-	site := os.Getenv("NETBOX_SITE")
 	nb, err := netbox.NewClient(netbox.WithNetBoxURL(netboxURL), netbox.WithToken(token))
 	if err != nil {
 		log.Println("Error initializing client:", err)
 		os.Exit(1)
 	}
 
-	devices, err := nb.ListDevices(site)
+	site := os.Getenv("NETBOX_SITE")
+	tag := os.Getenv("NETBOX_TAG")
+	devices, err := nb.ListDevices(site, tag)
 	if err != nil {
 		log.Println("Error listing devices:", err)
 		os.Exit(1)

netbox/netbox.go

@@ -39,14 +39,17 @@ func NewClient(opts ...Option) (*Client, error) {
 }
 
 // ListDevices searches for netboxes that match the given options.
-func (nb *Client) ListDevices(site string) ([]Device, error) {
+func (nb *Client) ListDevices(site, tag string) ([]Device, error) {
 	queryURL := *nb.baseURL
 	queryURL.Path = "/api/dcim/devices/"
 
 	queryVals := url.Values{}
-	queryVals.Add("tag", "pxe-enable")
 	queryVals.Add("has_primary_ip", "yes")
 
+	if tag != "" {
+		queryVals.Add("tag", tag)
+	}
+
 	if site != "" {
 		queryVals.Add("site", site)
 	}