ddns/ddns.go

71 lines
1.2 KiB
Go
Raw Normal View History

package main
import (
2014-07-15 21:48:15 +02:00
"flag"
"github.com/pboehm/ddns/connection"
"log"
2014-07-12 23:37:01 +02:00
"strings"
)
func HandleErr(err error) {
if err != nil {
log.Fatal(err)
}
}
var (
2014-07-15 23:41:54 +02:00
DdnsDomain string
DdnsWebListenSocket string
)
func init() {
flag.StringVar(&DdnsDomain, "domain", "",
"The subdomain which should be handled by DDNS")
2014-07-15 23:41:54 +02:00
flag.StringVar(&DdnsWebListenSocket, "listen", ":8080",
"Which socket should the web service use to bind itself")
}
2014-07-15 21:48:15 +02:00
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 {
2014-07-15 21:48:15 +02:00
flag.Parse()
ValidateCommandArgs()
2014-07-15 21:48:15 +02:00
if len(flag.Args()) != 1 {
usage()
}
2014-07-15 21:48:15 +02:00
cmd := flag.Args()[0]
return cmd
}
func main() {
2014-07-15 23:41:54 +02:00
cmd := PrepareForExecution()
conn := connection.OpenConnection()
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]")
}