summaryrefslogtreecommitdiff
path: root/bash.new.go
diff options
context:
space:
mode:
Diffstat (limited to 'bash.new.go')
-rw-r--r--bash.new.go242
1 files changed, 242 insertions, 0 deletions
diff --git a/bash.new.go b/bash.new.go
new file mode 100644
index 0000000..631e5b4
--- /dev/null
+++ b/bash.new.go
@@ -0,0 +1,242 @@
+package prep
+
+// initializes logging and command line options
+
+import (
+ "fmt"
+ "os"
+ "path/filepath"
+ "strings"
+ "time"
+
+ "go.wit.com/dev/alexflint/arg"
+ "go.wit.com/lib/gui/shell"
+ "go.wit.com/log"
+ durationpb "google.golang.org/protobuf/types/known/durationpb"
+ timestamppb "google.golang.org/protobuf/types/known/timestamppb"
+)
+
+// makes a bash autocomplete file for your command
+func doBash2(argname string) {
+ fmt.Println(makeBashCompletionText2(argname))
+
+ homeDir, err := os.UserHomeDir()
+ if err != nil {
+ log.Printf("%v\n", err)
+ os.Exit(0)
+ }
+ filename := filepath.Join(homeDir, ".local/share/bash-completion/completions", argname)
+ if shell.Exists(filename) {
+ log.Println(filename, "file already exists")
+ os.Exit(0)
+ }
+ basedir, _ := filepath.Split(filename)
+ if !shell.IsDir(basedir) {
+ os.MkdirAll(basedir, os.ModePerm)
+ }
+
+ if f, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644); err == nil {
+ f.Write([]byte(makeBashCompletionText2(argname)))
+ f.Close()
+ log.Println("bash file created:", filename)
+ log.Println("restart bash")
+ } else {
+ log.Info(filename, err)
+ }
+ os.Exit(0)
+}
+
+func (pb *Auto) Debugf(fmts string, parts ...any) {
+ fmts = strings.TrimSpace(fmts)
+ fmts += "\n"
+ // NOTE: env doesn't work probably most (all?) the time because bash
+ // doesn't send all the ENV to autocomplete. so, trap on a "--autodebug" command line arg
+ if os.Getenv("AUTOCOMPLETE_VERBOSE") == "true" || pb.Debug {
+ if !pb.Newline {
+ fmt.Fprintf(os.Stderr, "\n")
+ pb.Newline = true
+ }
+ fmt.Fprintf(os.Stderr, fmts, parts...)
+ } else {
+ // fmt.Fprintf(os.Stderr, "NOT DOING ANYTHING\n")
+ }
+}
+
+// makes a bash autocomplete file for your command
+func doHandlePB(pb *Auto) error {
+ homeDir, err := os.UserHomeDir()
+ if err != nil {
+ return err
+ }
+ basedir := filepath.Join(homeDir, ".cache/autocomplete")
+ os.MkdirAll(basedir, os.ModePerm)
+ fullname := filepath.Join(basedir, pb.Argname+".pb")
+
+ all := NewAutos()
+ var last *Auto
+ data, err := os.ReadFile(fullname)
+ if err == nil {
+ err = all.Unmarshal(data)
+ if err == nil {
+ for found := range all.IterAll() {
+ dur := time.Since(found.Ctime.AsTime())
+ pb.Duration = durationpb.New(dur)
+ pb.Debugf("AUTO HISTORY: ctime='%v' age=%s argv='%v'", found.Ctime, shell.FormatDuration(dur), found.Argv)
+ last = found
+ }
+ }
+ }
+
+ if all.Len() > 15 {
+ pb.Debugf("DEBUG: trim() history is over 100 len=%d vs new=%d", all.Len(), all.Len()-90)
+ all.Autos = all.Autos[all.Len()-10:]
+ // newall.Autos = all.Autos[0:10]
+ // for _, found := range all.Autos[0:10] {
+ // newall.Append(found)
+ // }
+ }
+
+ // need this for the first time the user runs autocomplete
+ if last == nil {
+ last = new(Auto)
+ }
+
+ now := time.Now()
+ pb.Ctime = timestamppb.New(now)
+ duration := time.Since(last.Ctime.AsTime())
+ all.Append(pb)
+
+ data, err = all.Marshal()
+ if err != nil {
+ return err
+ }
+
+ f, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
+ defer f.Close()
+ if err != nil {
+ return err
+ }
+ _, err = f.Write(data)
+ pb.Debugf("WRITE DEBUG: write PB='%s' len(pb)=%d len(data)=%d dur=%v err=%v", fullname, all.Len(), len(data), duration, err)
+ return err
+}
+
+/*
+// 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 {
+ me.pp.WriteHelp(os.Stderr)
+ return
+ }
+ fmt.Fprintln(os.Stderr, "")
+ fmt.Fprintln(os.Stderr, "hello world")
+ fmt.Fprintln(os.Stderr, "")
+}
+*/
+
+func (pb *Auto) Autocomplete(notsure any, sendthis string) {
+ parts := strings.Split(sendthis, " ")
+ var all []string
+ for _, part := range parts {
+ var found bool
+ for _, s := range os.Args {
+ if s == part {
+ found = true
+ }
+ }
+ if found {
+ continue
+ }
+ all = append(all, part)
+ }
+ fmt.Printf("%s", strings.Join(all, " "))
+}
+
+func parseArgv(argname string) *Auto {
+ newauto := new(Auto)
+ newauto.Argname = argname
+ if len(os.Args) == 0 {
+ return newauto
+ }
+
+ if len(os.Args) > 1 && os.Args[1] == "--bash" {
+ newauto.SetupAuto = true
+ return newauto
+ }
+
+ // HACK: set debug flag if --autodebug is passed
+ for _, s := range os.Args {
+ if s == "--autodebug" {
+ newauto.Debug = true
+ }
+ }
+
+ // should we do auto complete here?
+ if len(os.Args) > 1 && os.Args[1] == "--auto-complete" {
+ newauto.IsAuto = true
+ newauto.Arg0 = os.Args[0]
+ newauto.Arg1 = os.Args[1]
+ newauto.Partial = os.Args[2]
+ newauto.Arg3 = os.Args[3]
+ if len(os.Args) < 5 {
+ // the user is doing autocomplete on the command itself
+ newauto.Partial = ""
+ newauto.Cmd = ""
+ newauto.Argv = []string{""}
+ return newauto
+ }
+ newauto.Argv = os.Args[4:]
+ if newauto.Partial == "''" {
+ newauto.Partial = ""
+ newauto.Cmd = "todo:findme"
+ }
+ // set pb.Cmd to the first thing that doesn't have a '-' arg
+ for _, s := range newauto.Argv {
+ if strings.HasPrefix(s, "-") {
+ continue
+ }
+ newauto.Cmd = s
+ break
+ }
+ if newauto.Cmd == "" {
+ newauto.Cmd = strings.Join(newauto.Argv, "BLAH")
+ }
+ return newauto
+ }
+ return newauto
+}
+
+// also try to parse/send cur (?)
+func Bash2(argname string, autocomplete func(*Auto)) {
+ newauto := parseArgv(argname)
+ if newauto.SetupAuto {
+ // --bash was passed. try to configure bash-completion
+ doBash2(argname)
+ os.Exit(0)
+ }
+
+ if newauto.Debug {
+ // dump debug info
+ newauto.AutoDebug()
+ }
+
+ if len(os.Args) > 1 && os.Args[1] == "--auto-complete" {
+ doHandlePB(newauto)
+ autocomplete(newauto)
+ if newauto.Debug {
+ // check here to see if there was any completion text sent
+ // if not, send "reset bash newline\n" to cause bash to redraw PS1 for the user
+ }
+ os.Exit(0)
+ }
+
+ arg.Register(&argBash)
+}