Autarch Will Control The Internet

services/dns-server/main.go

@@ -0,0 +1,84 @@
+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()
+}