yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-29 22:55:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Create struct CmdLine for yggdrasilctl

Browse source
This commit is contained in:
Alex Kotov 2021-07-23 15:23:34 +05:00
parent 7e98367d43
commit b9a4a9749c
1 changed files with 21 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

33
cmd/yggdrasilctl/main.go
View file

@ -24,6 +24,14 @@ import (
type admin_info map[string]interface{} type admin_info map[string]interface{}
type CmdLine struct {
args []string
server *string
injson *bool
verbose *bool
ver *bool
}
func main() { func main() {
// makes sure we can use defer and still return an error code to the OS // makes sure we can use defer and still return an error code to the OS
os.Exit(run()) os.Exit(run())
@ -62,26 +70,27 @@ func run() int {
fmt.Println(" - ", os.Args[0], "-endpoint=unix:///var/run/ygg.sock getDHT") fmt.Println(" - ", os.Args[0], "-endpoint=unix:///var/run/ygg.sock getDHT")
} }
server := flag.String("endpoint", endpoint, "Admin socket endpoint") var cmdline CmdLine
injson := flag.Bool("json", false, "Output in JSON format (as opposed to pretty-print)") cmdline.server = flag.String("endpoint", endpoint, "Admin socket endpoint")
verbose := flag.Bool("v", false, "Verbose output (includes public keys)") cmdline.injson = flag.Bool("json", false, "Output in JSON format (as opposed to pretty-print)")
ver := flag.Bool("version", false, "Prints the version of this build") cmdline.verbose = flag.Bool("v", false, "Verbose output (includes public keys)")
cmdline.ver = flag.Bool("version", false, "Prints the version of this build")
flag.Parse() flag.Parse()
args := flag.Args() cmdline.args = flag.Args()
if *ver { if *cmdline.ver {
fmt.Println("Build name:", version.BuildName()) fmt.Println("Build name:", version.BuildName())
fmt.Println("Build version:", version.BuildVersion()) fmt.Println("Build version:", version.BuildVersion())
fmt.Println("To get the version number of the running Yggdrasil node, run", os.Args[0], "getSelf") fmt.Println("To get the version number of the running Yggdrasil node, run", os.Args[0], "getSelf")
return 0 return 0
} }
if len(args) == 0 { if len(cmdline.args) == 0 {
flag.Usage() flag.Usage()
return 0 return 0
} }
if *server == endpoint { if *cmdline.server == endpoint {
if config, err := ioutil.ReadFile(defaults.GetDefaults().DefaultConfigFile); err == nil { if config, err := ioutil.ReadFile(defaults.GetDefaults().DefaultConfigFile); err == nil {
if bytes.Equal(config[0:2], []byte{0xFF, 0xFE}) || if bytes.Equal(config[0:2], []byte{0xFF, 0xFE}) ||
bytes.Equal(config[0:2], []byte{0xFE, 0xFF}) { bytes.Equal(config[0:2], []byte{0xFE, 0xFF}) {
@ -109,7 +118,7 @@ func run() int {
logger.Println("Falling back to platform default", defaults.GetDefaults().DefaultAdminListen) logger.Println("Falling back to platform default", defaults.GetDefaults().DefaultAdminListen)
} }
} else { } else {
endpoint = *server endpoint = *cmdline.server
logger.Println("Using endpoint", endpoint, "from command line") logger.Println("Using endpoint", endpoint, "from command line")
} }
@ -145,7 +154,7 @@ func run() int {
send := make(admin_info) send := make(admin_info)
recv := make(admin_info) recv := make(admin_info)
for c, a := range args { for c, a := range cmdline.args {
if c == 0 { if c == 0 {
if strings.HasPrefix(a, "-") { if strings.HasPrefix(a, "-") {
logger.Printf("Ignoring flag %s as it should be specified before other parameters\n", a) logger.Printf("Ignoring flag %s as it should be specified before other parameters\n", a)
@ -204,14 +213,14 @@ func run() int {
} }
res := recv["response"].(map[string]interface{}) res := recv["response"].(map[string]interface{})
if *injson { if *cmdline.injson {
if json, err := json.MarshalIndent(res, "", " "); err == nil { if json, err := json.MarshalIndent(res, "", " "); err == nil {
fmt.Println(string(json)) fmt.Println(string(json))
} }
return 0 return 0
} }
runAll(recv, verbose) runAll(recv, cmdline.verbose)
} else { } else {
logger.Println("Error receiving response:", err) logger.Println("Error receiving response:", err)
} }