74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/pboehm/ddns/connection"
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
func HandleErr(err error) {
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
var (
|
|
DdnsDomain string
|
|
DdnsWebListenSocket string
|
|
DdnsRedisHost string
|
|
)
|
|
|
|
func init() {
|
|
flag.StringVar(&DdnsDomain, "domain", "",
|
|
"The subdomain which should be handled by DDNS")
|
|
|
|
flag.StringVar(&DdnsWebListenSocket, "listen", ":8080",
|
|
"Which socket should the web service use to bind itself")
|
|
|
|
flag.StringVar(&DdnsRedisHost, "redis", ":6379",
|
|
"The Redis socket that should be used")
|
|
}
|
|
|
|
func ValidateCommandArgs() {
|
|
if DdnsDomain == "" {
|
|
log.Fatal("You have to supply the domain via --domain=DOMAIN")
|
|
} else if !strings.HasPrefix(DdnsDomain, ".") {
|
|
// get the domain in the right format
|
|
DdnsDomain = "." + DdnsDomain
|
|
}
|
|
}
|
|
|
|
func PrepareForExecution() string {
|
|
flag.Parse()
|
|
ValidateCommandArgs()
|
|
|
|
if len(flag.Args()) != 1 {
|
|
usage()
|
|
}
|
|
cmd := flag.Args()[0]
|
|
|
|
return cmd
|
|
}
|
|
|
|
func main() {
|
|
cmd := PrepareForExecution()
|
|
|
|
conn := connection.OpenConnection(DdnsRedisHost)
|
|
defer conn.Close()
|
|
|
|
switch cmd {
|
|
case "backend":
|
|
log.Printf("Starting PDNS Backend\n")
|
|
RunBackend(conn)
|
|
case "web":
|
|
log.Printf("Starting Web Service\n")
|
|
RunWebService(conn)
|
|
default:
|
|
usage()
|
|
}
|
|
}
|
|
|
|
func usage() {
|
|
log.Fatal("Usage: ./ddns [backend|web]")
|
|
}
|