85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
|
||
|
|
"github.com/darkhal/autarch-dns/api"
|
||
|
|
"github.com/darkhal/autarch-dns/config"
|
||
|
|
"github.com/darkhal/autarch-dns/server"
|
||
|
|
)
|
||
|
|
|
||
|
|
var version = "2.1.0"
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
configPath := flag.String("config", "config.json", "Path to config file")
|
||
|
|
listenDNS := flag.String("dns", "", "DNS listen address (overrides config)")
|
||
|
|
listenAPI := flag.String("api", "", "API listen address (overrides config)")
|
||
|
|
apiToken := flag.String("token", "", "API auth token (overrides config)")
|
||
|
|
showVersion := flag.Bool("version", false, "Show version")
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
if *showVersion {
|
||
|
|
fmt.Printf("autarch-dns v%s\n", version)
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load config
|
||
|
|
cfg := config.DefaultConfig()
|
||
|
|
if data, err := os.ReadFile(*configPath); err == nil {
|
||
|
|
if err := json.Unmarshal(data, cfg); err != nil {
|
||
|
|
log.Printf("Warning: invalid config file: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// CLI overrides
|
||
|
|
if *listenDNS != "" {
|
||
|
|
cfg.ListenDNS = *listenDNS
|
||
|
|
}
|
||
|
|
if *listenAPI != "" {
|
||
|
|
cfg.ListenAPI = *listenAPI
|
||
|
|
}
|
||
|
|
if *apiToken != "" {
|
||
|
|
cfg.APIToken = *apiToken
|
||
|
|
}
|
||
|
|
|
||
|
|
// Initialize zone store
|
||
|
|
store := server.NewZoneStore(cfg.ZonesDir)
|
||
|
|
if err := store.LoadAll(); err != nil {
|
||
|
|
log.Printf("Warning: loading zones: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Start DNS server
|
||
|
|
dnsServer := server.NewDNSServer(cfg, store)
|
||
|
|
go func() {
|
||
|
|
log.Printf("DNS server listening on %s (UDP+TCP)", cfg.ListenDNS)
|
||
|
|
if err := dnsServer.Start(); err != nil {
|
||
|
|
log.Fatalf("DNS server error: %v", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
// Start API server
|
||
|
|
apiServer := api.NewAPIServer(cfg, store, dnsServer)
|
||
|
|
go func() {
|
||
|
|
log.Printf("API server listening on %s", cfg.ListenAPI)
|
||
|
|
if err := apiServer.Start(); err != nil {
|
||
|
|
log.Fatalf("API server error: %v", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
log.Printf("autarch-dns v%s started", version)
|
||
|
|
|
||
|
|
// Wait for shutdown signal
|
||
|
|
sig := make(chan os.Signal, 1)
|
||
|
|
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
|
||
|
|
<-sig
|
||
|
|
|
||
|
|
log.Println("Shutting down...")
|
||
|
|
dnsServer.Stop()
|
||
|
|
}
|