try to make --restore workv0.0.7

1 files changed, 52 insertions, 30 deletions

diff --git a/argvAutoshell.go b/argvAutoshell.go
index 85d841b..37937ec 100644
--- a/argvAutoshell.go
+++ b/argvAutoshell.go
@@ -6,6 +6,9 @@ package main
 import (
 	"fmt"
 	"os"
+	"path/filepath"
+
+	"go.wit.com/lib/gui/shell"
 )
 
 /*
@@ -28,7 +31,7 @@ func (args) doBashAuto() {
 	default:
 		if argv.BashAuto[0] == ARGNAME {
 			// list the subcommands here
-			fmt.Println("--restore save dump")
+			fmt.Println("--restore save dump dumpx")
 		}
 	}
 	os.Exit(0)
@@ -36,6 +39,10 @@ func (args) doBashAuto() {
 
 // prints help to STDERR // TODO: move everything below this to go-args
 func (args) doBashHelp() {
+	if len(argv.BashAuto) < 2 {
+		fmt.Fprintf(os.Stderr, "something went wrong with the GO args autocomplete in %s\n", ARGNAME)
+		return
+	}
 	if argv.BashAuto[1] != "''" {
 		// if this is not blank, then the user has typed something
 		return
@@ -55,33 +62,48 @@ func (args) doBashHelp() {
 
 // 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("# Put the below in the file: ~/.local/share/bash-completion/completions/" + ARGNAME)
-	fmt.Println("#")
-	fmt.Println("# todo: make this output work/parse with:")
-	fmt.Println("# complete -C " + ARGNAME + " --bash go")
-	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)
+	if homeDir, err := os.UserHomeDir(); err == nil {
+		filename := filepath.Join(homeDir, ".local/share/bash-completion/completions", ARGNAME)
+		if !shell.Exists(filename) {
+			if f, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644); err == nil {
+				f.Write([]byte(makeBashCompletionText(ARGNAME)))
+				f.Close()
+			}
+		}
+	}
+	fmt.Println(makeBashCompletionText(ARGNAME))
+}
+
+func makeBashCompletionText(argname string) string {
+	var out string
+
+	out += fmt.Sprintf("# add this in your bashrc:\n")
+	out += fmt.Sprintf("\n")
+	out += fmt.Sprintf("# todo: add this to go-arg as a 'hidden' go-arg option --bash\n")
+	out += fmt.Sprintf("#\n")
+	out += fmt.Sprintf("# Put the below in the file: ~/.local/share/bash-completion/completions/%s\n", argname)
+	out += fmt.Sprintf("#\n")
+	out += fmt.Sprintf("# todo: make this output work/parse with:\n")
+	out += fmt.Sprintf("# complete -C " + argname + " --bash go\n")
+	out += fmt.Sprintf("\n")
+	out += fmt.Sprintf("_" + argname + "_complete()\n")
+	out += fmt.Sprintf("{\n")
+	out += fmt.Sprintf("	# sets local to this func vars\n")
+	out += fmt.Sprintf("	local cur prev all\n")
+	out += fmt.Sprintf("	cur=${COMP_WORDS[COMP_CWORD]}\n")
+	out += fmt.Sprintf("	prev=${COMP_WORDS[COMP_CWORD-1]}\n")
+	out += fmt.Sprintf("	all=${COMP_WORDS[@]}\n")
+	out += fmt.Sprintf("\n")
+	out += fmt.Sprintf("	# this is where we generate the go-arg output\n")
+	out += fmt.Sprintf("        GOARGS=$(" + argname + " --auto-complete $prev \\'$cur\\' $all)\n")
+	out += fmt.Sprintf("\n")
+	out += fmt.Sprintf("	# this compares the command line input from the user\n")
+	out += fmt.Sprintf("	# to whatever strings we output\n")
+	out += fmt.Sprintf("        COMPREPLY=( $(compgen -W \"$GOARGS\" -- $cur) )  # THIS WORKS\n")
+	out += fmt.Sprintf("	return 0\n")
+	out += fmt.Sprintf("}\n")
+	out += fmt.Sprintf("complete -F _%s_complete %s\n", argname, argname)
+	out += fmt.Sprintf("\n")
+	out += fmt.Sprintf("# copy and paste the above into your bash shell should work\n")
+	return out
 }