add bash autocomplete

3 files changed, 106 insertions, 8 deletions

diff --git a/argv.go b/argv.go
index 7c6e867..d86b6cd 100644
--- a/argv.go
+++ b/argv.go
@@ -7,13 +7,15 @@ package main
 var argv args
 
 type args struct {
-	Trim    bool `arg:"--trim"      default:"true"     help:"trim entries from go.sum"`
-	Verbose bool `arg:"--verbose"                      help:"show more"`
-	Restore bool `arg:"--restore"                      help:"only restore from go/pkg/mod/"`
-	Force   bool `arg:"--force"                        help:"remove things and redo them no matter what"`
-	Strict  bool `arg:"--strict"                       help:"never make go.* files unless everything is perfect"`
-	Purge   bool `arg:"--purge"                        help:"purge all the git notes. this might be bad for you."`
-	Smart   bool `arg:"--smart"                        help:"whatever seems best at the time. never os.Exit(-1)"`
+	Trim     bool     `arg:"--trim"      default:"true"     help:"trim entries from go.sum"`
+	Verbose  bool     `arg:"--verbose"                      help:"show more"`
+	Restore  bool     `arg:"--restore"                      help:"only restore from go/pkg/mod/"`
+	Force    bool     `arg:"--force"                        help:"remove things and redo them no matter what"`
+	Strict   bool     `arg:"--strict"                       help:"never make go.* files unless everything is perfect"`
+	Purge    bool     `arg:"--purge"                        help:"purge all the git notes. this might be bad for you."`
+	Smart    bool     `arg:"--smart"                        help:"whatever seems best at the time. never os.Exit(-1)"`
+	Bash     bool     `arg:"--bash"                               help:"generate bash completion"`
+	BashAuto []string `arg:"--auto-complete"                      help:"todo: move this to go-arg"`
 }
 
 func (args) Version() string {
diff --git a/argvAutoshell.go b/argvAutoshell.go
new file mode 100644
index 0000000..3f712a1
--- /dev/null
+++ b/argvAutoshell.go
@@ -0,0 +1,85 @@
+// 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"
+	"os"
+)
+
+/*
+	handles shell autocomplete
+*/
+
+// used for shell auto completion
+// var ARGNAME string = "forge" // todo: get this from $0 ?
+
+func deleteMatch() {
+	// f := forgedb.InitSimple()
+	fmt.Println("go.wit.com/lib/gui/repostatus todo: need to do this")
+}
+
+func (args) doBashAuto() {
+	argv.doBashHelp()
+	switch argv.BashAuto[0] {
+	case "checkout":
+		fmt.Println("user devel master ")
+	default:
+		if argv.BashAuto[0] == ARGNAME {
+			// list the subcommands here
+			fmt.Println("--bash strict --force")
+		}
+	}
+	os.Exit(0)
+}
+
+// prints help to STDERR // TODO: move everything below this to go-args
+func (args) doBashHelp() {
+	if argv.BashAuto[1] != "''" {
+		// if this is not blank, then the user has typed something
+		return
+	}
+	if argv.BashAuto[0] != ARGNAME {
+		// if this is not the name of the command, the user already started doing something
+		return
+	}
+	if argv.BashAuto[0] == ARGNAME {
+		pp.WriteHelp(os.Stderr)
+		return
+	}
+	fmt.Fprintln(os.Stderr, "")
+	fmt.Fprintln(os.Stderr, "hello world")
+	fmt.Fprintln(os.Stderr, "")
+}
+
+// complete -F forge --bash forge
+func (args) doBash() {
+	fmt.Println("# add this in your bashrc:")
+	fmt.Println("")
+	fmt.Println("# todo: add this to go-arg as a 'hidden' go-arg option --bash")
+	fmt.Println("#")
+	fmt.Println("# todo: can this output work/parse with:")
+	fmt.Println("# complete -C `" + ARGNAME + " --bash` " + ARGNAME)
+	fmt.Println("")
+	fmt.Println("_" + ARGNAME + "_complete()")
+	fmt.Println("{")
+	fmt.Println("	# sets local to this func vars")
+	fmt.Println("	local cur prev all")
+	fmt.Println("	cur=${COMP_WORDS[COMP_CWORD]}")
+	fmt.Println("	prev=${COMP_WORDS[COMP_CWORD-1]}")
+	fmt.Println("	all=${COMP_WORDS[@]}")
+	fmt.Println("")
+	fmt.Println("	# this is where we generate the go-arg output")
+	fmt.Println("        GOARGS=$(" + ARGNAME + " --auto-complete $prev \\'$cur\\' $all)")
+	fmt.Println("")
+	fmt.Println("	# this compares the command line input from the user")
+	fmt.Println("	# to whatever strings we output")
+	fmt.Println("        COMPREPLY=( $(compgen -W \"$GOARGS\" -- $cur) )  # THIS WORKS")
+	fmt.Println("	return 0")
+	fmt.Println("}")
+	fmt.Println("complete -F _" + ARGNAME + "_complete " + ARGNAME)
+	fmt.Println("")
+	fmt.Println("# copy and paste the above into your bash shell should work")
+	os.Exit(0)
+}
diff --git a/main.go b/main.go
index a85de75..a98fb38 100644
--- a/main.go
+++ b/main.go
@@ -14,7 +14,10 @@ import (
 var VERSION string
 var BUILDTIME string
 
-var golangVersion string = "1.21"
+// used for shell auto completion
+var ARGNAME string = "go-mod-clean"
+
+var golangVersion string = "1.22"
 
 var pp *arg.Parser
 var forge *forgepb.Forge
@@ -27,6 +30,14 @@ func main() {
 	log.Info("go-mod-clean version", VERSION, "built on", BUILDTIME)
 	pp = arg.MustParse(&argv)
 
+	if argv.Bash {
+		argv.doBash()
+		os.Exit(0)
+	}
+	if len(argv.BashAuto) != 0 {
+		argv.doBashAuto()
+		os.Exit(0)
+	}
 	// load the ~/.config/forge/ config
 	// this lets you configure repos you have read/write access too
 	forge = forgepb.InitPB() // todo: make this scan only if pb file not found