1 files changed, 64 insertions, 0 deletions
diff --git a/complete.go b/complete.go
new file mode 100644
index 0000000..5ad12b1
--- /dev/null
+++ b/complete.go
@@ -0,0 +1,64 @@
+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
+
+var APPNAME string = "virtigo"
+
+func (a args) Description() string {
+	return `
+	virtigo: control your cluster
+
+This maintains a master list of all your vm's (aka 'droplets')
+in your homelab cloud.  You can import libvirt xml files. 
+This app talks to your hypervisors via the virtigod daemon. 
+`
+}
+
+var INFO *log.LogFlag
+var POLL *log.LogFlag
+var WARN *log.LogFlag
+var EVENT *log.LogFlag
+
+func init() {
+	full := "go.wit.com/apps/virtigo"
+	short := "virtigo"
+
+	INFO = log.NewFlag("INFO", false, full, short, "general virtigo")
+	POLL = log.NewFlag("POLL", false, full, short, "virtigo polling")
+	WARN = log.NewFlag("WARN", true, full, short, "bad things")
+	EVENT = log.NewFlag("EVENT", true, full, short, "hypeprvisor/droplet events")
+}
+
+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{"--version", "list", "droplet"}
+		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
+}