diff options
| -rw-r--r-- | argv.go (renamed from argv.struct.go) | 0 | ||||
| -rw-r--r-- | argv.template.go | 81 | ||||
| -rw-r--r-- | complete.go (renamed from argv.custom.go) | 30 | ||||
| -rw-r--r-- | main.go | 6 | ||||
| -rw-r--r-- | structs.go | 2 |
5 files changed, 25 insertions, 94 deletions
diff --git a/argv.template.go b/argv.template.go deleted file mode 100644 index 6948afe..0000000 --- a/argv.template.go +++ /dev/null @@ -1,81 +0,0 @@ -package main - -// these are stubbed in functions needed -// just copy this file from another working app for now -// you shouldn't need to change anything here -// TODO: clean this up in argv - -import ( - "os" - - "go.wit.com/dev/alexflint/arg" - "go.wit.com/gui" - "go.wit.com/lib/env" - "go.wit.com/lib/fhelp" - "go.wit.com/log" -) - -func (args) InitArgv() (string, string, string) { - return APPNAME, BUILDTIME, VERSION -} - -// this function will send the current argv PB to go-args for parsing -func (args) ParseFlags(flags []string) error { - var err error - if me.pp == nil { - // log.Info("Parse Flags GOT flags:", flags) - me.pp, err = arg.ParseFlags(flags, &argv) - // panic("got to the app's ParseFlags()") - } else { - panic("me.pp was not nil") - } - return err -} - -// add this funcgion: this will print the help -func (args) WriteHelpForSubcommand(cmd string) error { - me.pp.WriteHelpForSubcommand(os.Stderr, cmd) - return nil -} - -// this will print the help for the subcmd -func (args) WriteHelpForAutocomplete(part string, subcmd ...string) error { - return me.pp.WriteHelpForAutocomplete(os.Stderr, os.Stdout, part, subcmd...) -} - -func (args) WriteHelpForAutocompleteDebug(part string, subcmd ...string) error { - f, _ := os.OpenFile("/tmp/argv.junk", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) - return me.pp.WriteHelpForAutocomplete(f, os.Stdout, part, subcmd...) -} - -// add this funcgion: this will print the help -func (args) WriteHelp() error { - me.pp.WriteHelp(os.Stderr) - return nil -} - -func (args) InitGui() error { - // panic("got here") - arg.Register(&gui.ArgvGui) - // me.gui = gui.PreInit() - me.myGui = fhelp.Gui() - return nil -} - -func (args) Exit() { - gui.UnloadToolkits() - if env.Verbose() { - log.Info("argv.Exit() called", APPNAME+".Exit()") - } - // remove this from the template for your app (or make one for youself if you need it) - // forgeExit() // custom forge shutdown function -} - -func (args) Help() string { - return "got app help" -} - -func (args) MustParse() error { - me.pp = arg.MustParse(&argv) - return nil -} diff --git a/argv.custom.go b/complete.go index d3e592b..b045ce9 100644 --- a/argv.custom.go +++ b/complete.go @@ -4,8 +4,10 @@ package main import ( - "os" + "fmt" + "strings" + "go.wit.com/dev/alexflint/arg" "go.wit.com/lib/protobuf/argvpb" "go.wit.com/log" ) @@ -38,13 +40,25 @@ func init() { WARN = log.NewFlag("WARN", true, full, short, "bad things") } +func (args) MustParse() error { + me.pp = arg.MustParse(&argv) + return nil +} + // sends the strings to bash or zsh that will be your options -func (a args) SendCompletionStrings(pb *argvpb.Argv) { - if pb.GetCmd() == "" { - base := []string{"status", "test", "--version", "--daemon"} - pb.SendStrings(base) - } else { - pb.SubCommand(pb.Goargs...) +func (a args) DoAutoComplete() error { + if argvpb.PB.GetCmd() == "" { + matches := []string{"status", "test", "--version", "--daemon"} + fmt.Fprintf(argvpb.Stdout, " %s", strings.Join(matches, " ")) + return nil + } + var err error + if me.pp == nil { + me.pp, err = arg.ParseFlagsArgv(&argv) + if err != nil { + return err + } } - os.Exit(0) + err = me.pp.WriteHelpForAutocomplete(argvpb.PB.Partial, argvpb.PB.Real...) + return err } @@ -21,7 +21,7 @@ var resources embed.FS func main() { me = new(zoodStruct) - me.argv = argvpb.Autocomplete(&argv) // adds shell auto complete to go-args + argvpb.Init(&argv, APPNAME, BUILDTIME, VERSION) // adds shell auto-complete if argv.Daemon { // turn off timestamps for STDOUT (systemd adds them) @@ -33,13 +33,13 @@ func main() { if argv.Status != nil { log.Info("todo: show status here") - me.argv.GoodExit("says hi") + argvpb.GoodExit("says hi") } if err := testZoo(); err != nil { log.Info("failed to connect. sleeping for 3 minutes") time.Sleep(3 * time.Minute) - me.argv.BadExit("failed to connect to zookeeper", err) + argvpb.BadExit("failed to connect to zookeeper", err) } me.pollDelay = 3 * time.Second me.failcountmax = 20 // die every minute if zookeeper can't be found @@ -8,7 +8,6 @@ import ( "go.wit.com/dev/alexflint/arg" "go.wit.com/lib/fhelp" - "go.wit.com/lib/protobuf/argvpb" "go.wit.com/lib/protobuf/zoopb" ) @@ -16,7 +15,6 @@ var me *zoodStruct // this app's variables type zoodStruct struct { - argv *argvpb.Argv // more experiments for bash handling pp *arg.Parser // for parsing the command line args. Yay to alexf lint! myGui *fhelp.GuiPrep // for initializing the GUI toolkits urlbase string // the dns name for the zookeeper |