config.go

package main

import (
	"os"
	"strings"
)

type Domain []string

func NewDomain(s string) Domain {
	s = strings.Replace(s, "..", ".", -1)
	if s[:1] == "." {
		s = s[1:]
	}
	if s[len(s)-1:] == "." {
		s = s[:len(s)-1]
	}
	return Domain(strings.Split(s, "."))
}

func (d *Domain) String() string {
	return strings.Join([]string(*d), ".")
}

type Config struct {
	nameserver string
	dnsAddr    string
	domain     Domain
	dockerHost string
	tlsVerify  bool
	tlsCaCert  string
	tlsCert    string
	tlsKey     string
	verbose    bool
	httpAddr   string
	ttl        int
}

func NewConfig() *Config {
	dockerHost := os.Getenv("DOCKER_HOST")
	if len(dockerHost) == 0 {
		dockerHost = "unix:///var/run/docker.sock"
	}
	tlsVerify := len(os.Getenv("DOCKER_TLS_VERIFY")) != 0
	dockerCerts := os.Getenv("DOCKER_CERT_PATH")
	if len(dockerCerts) == 0 {
		dockerCerts = os.Getenv("HOME") + "/.docker"
	}

	return &Config{
		nameserver: "8.8.8.8:53",
		dnsAddr:    ":53",
		domain:     NewDomain("docker"),
		dockerHost: dockerHost,
		httpAddr:   ":80",
		tlsVerify:  tlsVerify,
		tlsCaCert:  dockerCerts + "/ca.pem",
		tlsCert:    dockerCerts + "/cert.pem",
		tlsKey:     dockerCerts + "/key.pem",
	}

}