fix argv build

1 files changed, 15 insertions, 10 deletions

diff --git a/argv.go b/argv.go
index 0a5a453..67a64d4 100644
--- a/argv.go
+++ b/argv.go
@@ -1,8 +1,9 @@
 package main
 
 import (
-	"fmt"
 	"os"
+
+	"go.wit.com/lib/gui/prep"
 )
 
 /*
@@ -41,15 +42,19 @@ go-mod-clean will try to verify your go.* files are using the newest package ver
 `
 }
 
-func (a args) DoAutoComplete(argv []string) {
-	switch argv[0] {
-	case "checkout":
-		fmt.Println("user devel master ")
-	default:
-		if argv[0] == ARGNAME {
-			// list the subcommands here
-			fmt.Println("strict restore purge lax")
-		}
+/*
+	handles shell autocomplete
+*/
+
+func (args) Appname() string {
+	return ARGNAME
+}
+
+func (a args) DoAutoComplete(pb *prep.Auto) {
+	if pb.Cmd == "" {
+		pb.Autocomplete3([]string{"strict", "--restore", "purge", "lax"})
+	} else {
+		pb.SubCommand(pb.Argv...)
 	}
 	os.Exit(0)
 }