mesh.cat/meshname
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor flag parsing

Browse Source
This commit is contained in:
George 2020-09-16 11:29:09 -04:00
parent ffe97b4c42
commit bbdc18ed1b
1 changed files with 23 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
cmd/meshnamed/main.go
View File

@ -28,13 +28,21 @@ func parseNetworks(networksconf string) (map[string]*net.IPNet, error) {
return networks, nil return networks, nil
} }
var (
genconf, subdomain, useconffile, listenAddr, networksconf string
debug bool
)
func init() {
flag.StringVar(&genconf, "genconf", "", "generate a new config for IP address")
flag.StringVar(&subdomain, "subdomain", "meshname.", "subdomain used to generate config")
flag.StringVar(&useconffile, "useconffile", "", "run daemon with a config file")
flag.StringVar(&listenAddr, "listenaddr", "[::1]:53535", "address to listen on")
flag.StringVar(&networksconf, "networks", "ygg=200::/7,cjd=fc00::/8,meshname=::/0", "TLD=subnet list separated by comma")
flag.BoolVar(&debug,"debug", false, "enable debug logging")
}
func main() { func main() {
genconf := flag.String("genconf", "", "generate a new config for IP address")
subdomain := flag.String("subdomain", "meshname.", "subdomain used to generate config")
useconffile := flag.String("useconffile", "", "run daemon with a config file")
listenAddr := flag.String("listenaddr", "[::1]:53535", "address to listen on")
networksconf := flag.String("networks", "ygg=200::/7,cjd=fc00::/8,meshname=::/0", "TLD=subnet list separated by comma")
debug := flag.Bool("debug", false, "enable debug logging")
flag.Parse() flag.Parse()
var logger *log.Logger var logger *log.Logger
@ -43,12 +51,12 @@ func main() {
logger.EnableLevel("error") logger.EnableLevel("error")
logger.EnableLevel("warn") logger.EnableLevel("warn")
logger.EnableLevel("info") logger.EnableLevel("info")
if *debug { if debug {
logger.EnableLevel("debug") logger.EnableLevel("debug")
} }
if *genconf != "" { if genconf != "" {
if conf, err := meshname.GenConf(*genconf, *subdomain); err == nil { if conf, err := meshname.GenConf(genconf, subdomain); err == nil {
fmt.Println(conf) fmt.Println(conf)
} else { } else {
logger.Errorln(err) logger.Errorln(err)
@ -58,16 +66,16 @@ func main() {
s := new(meshname.MeshnameServer) s := new(meshname.MeshnameServer)
s.Init(logger, *listenAddr) s.Init(logger, listenAddr)
if networks, err := parseNetworks(*networksconf); err == nil { if networks, err := parseNetworks(networksconf); err == nil {
s.SetNetworks(networks) s.SetNetworks(networks)
} else { } else {
logger.Errorln(err) logger.Errorln(err)
} }
if *useconffile != "" { if useconffile != "" {
s.LoadConfig(*useconffile) s.LoadConfig(useconffile)
} }
s.Start() s.Start()
@ -82,8 +90,8 @@ func main() {
case _ = <-c: case _ = <-c:
return return
case _ = <-r: case _ = <-r:
if *useconffile != "" { if useconffile != "" {
s.LoadConfig(*useconffile) s.LoadConfig(useconffile)
} }
} }
} }