someday I'll actually get to work on zookeeperHEADv0.0.113v0.0.112masterdevel

1 files changed, 64 insertions, 0 deletions

diff --git a/complete.go b/complete.go
new file mode 100644
index 0000000..b045ce9
--- /dev/null
+++ b/complete.go
@@ -0,0 +1,64 @@
+// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
+// Use of this source code is governed by the GPL 3.0
+
+package main
+
+import (
+	"fmt"
+	"strings"
+
+	"go.wit.com/dev/alexflint/arg"
+	"go.wit.com/lib/protobuf/argvpb"
+	"go.wit.com/log"
+)
+
+// sent via -ldflags
+var VERSION string
+var BUILDTIME string
+
+// used for shell auto completion
+var APPNAME string = "zood"
+
+func (a args) Description() string {
+	return `
+         this daemon talks to zookeeper
+`
+}
+
+var NOW *log.LogFlag
+var INFO *log.LogFlag
+var PING *log.LogFlag
+var WARN *log.LogFlag
+
+func init() {
+	full := "go.wit.com/apps/zood"
+	short := "zood"
+
+	NOW = log.NewFlag("NOW", true, full, short, "useful while doing debugging")
+	INFO = log.NewFlag("INFO", false, full, short, "general zood")
+	PING = log.NewFlag("PING", false, full, short, "show pings to the zookeeper")
+	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) 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
+		}
+	}
+	err = me.pp.WriteHelpForAutocomplete(argvpb.PB.Partial, argvpb.PB.Real...)
+	return err
+}