gertdns/main.go

package main

import (
	"flag"
	"log"
	"os"
	"os/signal"
	"syscall"

	"github.com/MarekWojt/gertdns/auth"
	"github.com/MarekWojt/gertdns/config"
	"github.com/MarekWojt/gertdns/dns"
	"github.com/MarekWojt/gertdns/web"
	rootDns "github.com/miekg/dns"
)

var (
	configFile      = flag.String("config-file", "conf.toml", "Path to configuration file")
	authFile        = flag.String("auth-file", "auth.toml", "Path to authentication file")
	dataPath        = flag.String("data-path", ".", "Where to save data")
	enableDebugMode = flag.Bool("enable-debug-mode", false, "Enables debug mode, will output a list of all registered records on the index page of the HTTP server")
)

type dnsResult struct {
	server *rootDns.Server
	err    error
}

func main() {
	flag.Parse()

	err := config.Load(*configFile)
	if err != nil {
		log.Fatalf("Failed to load configuration: %s\n ", err.Error())
	}

	dns.Init(*dataPath)
	web.Init(*enableDebugMode)
	err = auth.Init(*authFile)
	if err != nil {
		log.Fatalf("Failed to initialize authentication module: %s\n", err.Error())
	}

	webChan := make(chan error)
	dnsChan := make(chan dnsResult)
	go func() {
		server, err := dns.Run()
		if err != nil {
			log.Fatalf("Failed to start DNS server: %s\n ", err.Error())
		}

		dnsChan <- dnsResult{
			server: server,
			err:    err,
		}
	}()

	go func() {
		err := web.RunSocket()
		if err != nil {
			log.Fatalf("Failed to start HTTP socket: %s\n ", err.Error())
		}

		webChan <- err
	}()

	go func() {
		err := web.RunHTTP()
		if err != nil {
			log.Fatalf("Failed to start HTTP server: %s\n ", err.Error())
		}

		webChan <- err
	}()

	c := make(chan os.Signal)
	signal.Notify(c, os.Interrupt, os.Kill, syscall.SIGTERM)
	go func() {
		<-c
		dns.Shutdown()
		os.Exit(0)
	}()

	currentDnsResult := <-dnsChan
	defer currentDnsResult.server.Shutdown()
	defer dns.Shutdown()
	<-webChan
	<-webChan
}