159 lines
4.6 KiB
Go
159 lines
4.6 KiB
Go
package main
|
|
|
|
/*
|
|
This simply parses the command line arguments using the default golang
|
|
package called 'flag'. This can be used as a simple template to parse
|
|
command line arguments in other programs.
|
|
|
|
It puts everything in the 'config' package which I think is a good
|
|
wrapper around the 'flags' package and doesn't need a whole mess of
|
|
global variables
|
|
*/
|
|
|
|
import "log"
|
|
import "os"
|
|
import "os/user"
|
|
import "flag"
|
|
import "fmt"
|
|
import "runtime"
|
|
|
|
import "github.com/gookit/config"
|
|
import "github.com/gookit/config/yaml"
|
|
import "github.com/gookit/config/json"
|
|
|
|
// import "github.com/davecgh/go-spew/spew"
|
|
|
|
// always override the debugging flag from the command line
|
|
var debugging *bool
|
|
|
|
var customUsage = func() {
|
|
fmt.Fprintf(flag.CommandLine.Output(), "Usage of %s:\n", os.Args[0])
|
|
flag.PrintDefaults()
|
|
|
|
fmt.Println("")
|
|
fmt.Println("EXAMPLES:")
|
|
fmt.Println("")
|
|
fmt.Println(os.Args[0] + "--hostname test.hostname.wit.com")
|
|
fmt.Println("")
|
|
}
|
|
|
|
func parseFlags() {
|
|
var version string
|
|
var hostname string
|
|
var height int
|
|
|
|
flag.StringVar (&version, "version", "v0.1", "Set compiled in version string")
|
|
flag.StringVar (&hostname, "hostname", "localhost", "Your full hostname")
|
|
flag.IntVar (&height, "height", 600, "Height of the Window")
|
|
|
|
width := flag.Int ("width", 400, "Width of the Window")
|
|
debugging = flag.Bool("debugging", false, "Enable debugging")
|
|
nogui := flag.Bool("nogui", false, "Do not display the GUI")
|
|
|
|
// Set the output if something fails to stdout rather than stderr
|
|
flag.CommandLine.SetOutput(os.Stdout)
|
|
|
|
flag.Usage = customUsage
|
|
flag.Parse()
|
|
|
|
if flag.Parsed() {
|
|
log.Println("flag.Parse() worked")
|
|
} else {
|
|
log.Println("flag.Parse() failed")
|
|
onExit(nil)
|
|
}
|
|
|
|
log.Println("ENABLE width =", *width)
|
|
log.Println("ENABLE nogui =", *nogui)
|
|
log.Println("ENABLE debugging =", *debugging)
|
|
|
|
// if (debugging == true) {
|
|
if (*debugging) {
|
|
log.Println("ENABLE width =", *width)
|
|
log.Println("ENABLE debugging =", *debugging)
|
|
} else {
|
|
log.Println("DISABLE DEBUG debugging =", *debugging)
|
|
}
|
|
// os.Exit(0)
|
|
|
|
config.Set("width", *width)
|
|
config.Set("height", height)
|
|
config.Set("hostname", hostname)
|
|
config.Set("nogui", *nogui)
|
|
config.Set("debugging", *debugging)
|
|
log.Println("config.Set width", config.String("width"))
|
|
log.Println("config.Set debugging", config.Bool("debugging"))
|
|
}
|
|
|
|
func parseConfig(defaultConfig string) {
|
|
config.WithOptions(config.ParseEnv)
|
|
parseFlags()
|
|
|
|
log.Println("config.Set width", config.String("width"))
|
|
log.Println("config.Set debugging", config.Bool("debugging"))
|
|
|
|
config.LoadOSEnv([]string{"MAIL"})
|
|
config.LoadOSEnv([]string{"USER"})
|
|
config.LoadOSEnv([]string{"BUILDDEBUG"})
|
|
|
|
// add driver for support of yaml and json
|
|
config.AddDriver(yaml.Driver)
|
|
config.AddDriver(json.Driver)
|
|
|
|
// look up the user information to try to figure out where the config file is stored
|
|
user, err := user.Current()
|
|
if err != nil {
|
|
onExit(err)
|
|
}
|
|
|
|
// TODO: figure out how to look for this file in the right
|
|
// place under Linux, MacOS and Windows
|
|
if runtime.GOOS == "linux" {
|
|
log.Println("OS: Linux")
|
|
config.Set("configfile", user.HomeDir + "/.config/cloud-control-panel.json")
|
|
} else if runtime.GOOS == "windows" {
|
|
log.Println("OS: Windows")
|
|
config.Set("configfile", user.HomeDir + "\\cloud-control-panel.json")
|
|
} else {
|
|
log.Println("OS: " + runtime.GOOS)
|
|
config.Set("configfile", user.HomeDir + "/cloud-control-panel.json")
|
|
}
|
|
|
|
// TODO: load the default config file from the resources dir if the config file doesn't exist yet
|
|
// config.LoadFiles("config.json")
|
|
|
|
filename := config.String("configfile")
|
|
config.LoadFiles(filename)
|
|
|
|
// always override the debugging flag from the command line
|
|
config.Set("debugging", *debugging)
|
|
|
|
for _, addr := range config.Strings("arr1") {
|
|
log.Println("addr =", addr)
|
|
}
|
|
|
|
map1 := config.StringMap("map1")
|
|
fmt.Printf("%#v\n",map1)
|
|
|
|
log.Println(config.String("map1.key1.jwc1"))
|
|
|
|
for account, _ := range config.StringMap("cloud") {
|
|
port := config.String("accounts." + account + ".port")
|
|
proto := config.String("accounts." + account + ".proto")
|
|
hostname := config.String("accounts." + account + ".hostname")
|
|
fmt.Println(account, hostname, port, proto)
|
|
}
|
|
|
|
if (config.String("nogui") == "true") {
|
|
log.Println("DO NOT DISPLAY THE GUI")
|
|
}
|
|
|
|
if (config.String("debugging") == "true") {
|
|
log.Println("ENABLE DEBUG", config.String("debugging"))
|
|
log.Println("ENABLE DEBUG", config.Bool("debugging"))
|
|
} else {
|
|
log.Println("DISABLE DEBUG", config.String("debugging"))
|
|
log.Println("DISABLE DEBUG", config.Bool("debugging"))
|
|
}
|
|
}
|