diff options
Diffstat (limited to 'bash.new.go')
| -rw-r--r-- | bash.new.go | 242 |
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) +} |
