slawek/netbox-dns-updater
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added dry-run option and refactered FetchNetboxIPAddresses to accept API Base URL.

Browse Source
This commit is contained in:
Sławek Koszewski
2025-06-16 20:01:23 +02:00
parent b33dffbbcf
commit b017332f6b
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
netbox-dns-updater.go
View File

@ -7,8 +7,10 @@ import (
"io"
"log"
"net/http"
"net/url"
"os"
"os/exec"
"path"
"strings"
)
@ -21,8 +23,10 @@ type NetBoxResponse struct {
Results []IPAddress `json:"results"`
}
func FetchNetboxIPAddresses(apiURL, token string) ([]IPAddress, error) {
req, err := http.NewRequest("GET", apiURL, nil)
func FetchNetboxIPAddresses(apiBaseURL, token string) ([]IPAddress, error) {
u, _ := url.Parse(apiBaseURL)
u.Path = path.Join(u.Path, "ipam/ip-addresses")
req, err := http.NewRequest("GET", u.String(), nil)
if err != nil {
return nil, err
}
@ -57,7 +61,7 @@ func FetchNetboxIPAddresses(apiURL, token string) ([]IPAddress, error) {
return filtered, nil
}
func CreateDnsMasqConfig() {
func CreateDnsMasqConfig(writeConfig bool) {
token := os.Getenv("NETBOX_TOKEN")
if token == "" {
home := os.Getenv("HOME")
@ -77,13 +81,13 @@ func CreateDnsMasqConfig() {
}
}
ips, err := FetchNetboxIPAddresses(apiURL, token)
ips, err := FetchNetboxIPAddresses(apiBaseURL, token)
if err != nil {
log.Fatalf("Error fetching IP addresses: %v", err)
}
dir := "/etc/dnsmasq.d"
if stat, err := os.Stat(dir); err == nil && stat.IsDir() {
if stat, err := os.Stat(dir); err == nil && stat.IsDir() && !writeConfig {
file, err := os.Create(dir + "/netbox.conf")
if err != nil {
log.Fatalf("Failed to create netbox.conf: %v", err)
@ -99,20 +103,27 @@ func CreateDnsMasqConfig() {
}
}
var apiURL string
var apiBaseURL string
func main() {
listenAddr := flag.String("listen", ":8080", "address and port to listen on (e.g. :8080 or 127.0.0.1:8080)")
flag.StringVar(&apiURL, "api-url", "https://netbox.koszewscy.waw.pl/api/ipam/ip-addresses/", "NetBox API URL to fetch IP addresses")
flag.StringVar(&apiBaseURL, "api-url", "https://netbox.koszewscy.waw.pl/api", "NetBox API URL to fetch IP addresses")
dryRun := flag.Bool("dry-run", false, "if set, do not write to dnsmasq config file, just print the output")
flag.Parse()
if *dryRun {
log.Println("Dry run mode enabled, not writing to dnsmasq config file.")
CreateDnsMasqConfig(true)
return
}
http.HandleFunc("/update-dnsmasq", func(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
w.WriteHeader(http.StatusMethodNotAllowed)
w.Write([]byte("Method not allowed"))
return
}
CreateDnsMasqConfig()
CreateDnsMasqConfig(false)
cmd := exec.Command("systemctl", "restart", "dnsmasq.service")
if err := cmd.Run(); err != nil {
log.Printf("Failed to restart dnsmasq.service: %v", err)