acme-dns/main.go

package main

import (
	"fmt"
	"github.com/iris-contrib/middleware/cors"
	"github.com/kataras/iris"
	"github.com/miekg/dns"
	"github.com/op/go-logging"
	"os"
)

// Logging config
var log = logging.MustGetLogger("acme-dns")

// Global configuration struct
var DNSConf DNSConfig

var DB Database

// Static records
var RR Records

func main() {
	// Read global config
	configTmp, err := readConfig("config.cfg")
	if err != nil {
		fmt.Printf("Got error %v\n", DNSConf.Logconfig.File)
		os.Exit(1)
	}
	DNSConf = configTmp
	// Setup logging
	var logformat = logging.MustStringFormatter(DNSConf.Logconfig.Format)
	var logBackend *logging.LogBackend
	switch DNSConf.Logconfig.Logtype {
	default:
		// Setup logging - stdout
		logBackend = logging.NewLogBackend(os.Stdout, "", 0)
	case "file":
		// Logging to file
		logfh, err := os.OpenFile(DNSConf.Logconfig.File, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
		if err != nil {
			fmt.Printf("Could not open log file %s\n", DNSConf.Logconfig.File)
			os.Exit(1)
		}
		defer logfh.Close()
		logBackend = logging.NewLogBackend(logfh, "", 0)
	}
	logFormatter := logging.NewBackendFormatter(logBackend, logformat)
	logLevel := logging.AddModuleLevel(logFormatter)
	switch DNSConf.Logconfig.Level {
	default:
		logLevel.SetLevel(logging.DEBUG, "")
	case "warning":
		logLevel.SetLevel(logging.WARNING, "")
	case "error":
		logLevel.SetLevel(logging.ERROR, "")
	case "info":
		logLevel.SetLevel(logging.INFO, "")
	}
	logging.SetBackend(logFormatter)

	// Read the default records in
	RR.Parse(DNSConf.General.StaticRecords)

	// Open database
	err = DB.Init(DNSConf.Database.Engine, DNSConf.Database.Connection)
	if err != nil {
		log.Errorf("Could not open database [%v]", err)
		os.Exit(1)
	}
	defer DB.DB.Close()

	// DNS server part
	dns.HandleFunc(".", handleRequest)
	server := &dns.Server{Addr: ":53", Net: "udp"}
	go func() {
		err = server.ListenAndServe()
		if err != nil {
			log.Errorf("%v", err)
			os.Exit(1)
		}
	}()

	// API server and endpoints
	api := iris.New()
	api.Config.DisableBanner = true
	crs := cors.New(cors.Options{
		AllowedOrigins:     DNSConf.API.CorsOrigins,
		AllowedMethods:     []string{"GET", "POST"},
		OptionsPassthrough: false,
		Debug:              DNSConf.General.Debug,
	})
	api.Use(crs)
	var ForceAuth = AuthMiddleware{}
	api.Get("/register", WebRegisterGet)
	api.Post("/register", WebRegisterPost)
	api.Post("/update", ForceAuth.Serve, WebUpdatePost)
	// TODO: migrate to api.Serve(iris.LETSENCRYPTPROD("mydomain.com"))
	switch DNSConf.API.TLS {
	case "letsencrypt":
		host := DNSConf.API.Domain + ":" + DNSConf.API.Port
		api.Listen(host)
	case "cert":
		host := DNSConf.API.Domain + ":" + DNSConf.API.Port
		api.ListenTLS(host, DNSConf.API.TLSCertFullchain, DNSConf.API.TLSCertPrivkey)
	default:
		host := DNSConf.API.Domain + ":" + DNSConf.API.Port
		api.Listen(host)
	}
	if err != nil {
		log.Errorf("Error in HTTP server [%v]", err)
	}
	log.Debugf("Shutting down...")
}