diff options
Diffstat (limited to 'complete.go')
| -rw-r--r-- | complete.go | 363 |
1 files changed, 363 insertions, 0 deletions
diff --git a/complete.go b/complete.go new file mode 100644 index 0000000..dee60c5 --- /dev/null +++ b/complete.go @@ -0,0 +1,363 @@ +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 (pb *Auto) doHandlePB() 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) + // found.PrintDebug() + pb.Debugf("AUTO HISTORY: age=%s cmd='%s' partial='%s' argv='%v'", shell.FormatDuration(dur), found.Cmd, found.Partial, 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 (pb *Auto) Autocomplete2(sendthis string) { + dur := pb.Duration.AsDuration() + if dur < time.Millisecond*200 { + pb.Debug = true + /* + pb.Debugf("TODO: show extended help here '%s' '%s' %v dur=%v\n", pb.Arg0, pb.Arg1, pb.Argv, shell.FormatDuration(dur)) + pb.PrintDebug() + fmt.Println(" ") + */ + if myAuto.pp == nil { + pb.Debugf("myAuto.pp == nil") + } else { + pb.Debugf("myAuto.pp != nil") + if pb.Cmd == "" { + myAuto.pp.WriteHelp(os.Stderr) + } else { + myAuto.pp.WriteHelpForSubcommand(os.Stderr, pb.Cmd) + } + } + } + + 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, " ")) + /* + if dur > time.Millisecond*200 { + if dur < time.Millisecond*800 { + // fmt.Println("a b") + fmt.Println(pb.Partial + " hello world") + } + } + */ +} + +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.Partial == "'"+newauto.Cmd+"'" { + // not really a command, it's just a partially inputed string from the user + newauto.Cmd = "" + } + // if newauto.Cmd == "" { + // newauto.Cmd = strings.Join(newauto.Argv, "BLAH") + // } + } + return newauto +} + +// also try to parse/send cur (?) +func Bash2(argname string, appAutoFunc func(*Auto)) *Auto { + pb := parseArgv(argname) + if pb.SetupAuto { + // --bash was passed. try to configure bash-completion + doBash2(argname) + os.Exit(0) + } + + if pb.Debug { + // dump debug info + pb.PrintDebug() + } + + if pb.IsAuto { + pb.doHandlePB() + if pb.Debug { + // TODO: + // 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 + } + arg.Register(&argBash) + // flags := []string{pb.Arg3, pb.Arg0} + // arg.InitFlags(flags) + + appAutoFunc(pb) // run the autocomplete function the user made for their application + os.Exit(0) + } + + arg.Register(&argBash) + return pb +} + +// also try to parse/send cur (?) +// func Bash3(appAutoFunc func(*Auto), dest any) *Auto { +func Bash3(dest any) *Auto { + myAuto = new(AutoArgs) + // myAuto.appName = argname + // myAuto.autoFunc = appAutoFunc + newTest(dest) + + pb := parseArgv(myAuto.appName) + if pb.SetupAuto { + // --bash was passed. try to configure bash-completion + doBash2(myAuto.appName) + os.Exit(0) + } + + if pb.Debug { + // dump debug info + pb.PrintDebug() + } + + pb.doHandlePB() // read in the history protobuf file + + // turn on debugging if duration < 200 milliseconds + dur := pb.Duration.AsDuration() + if dur < time.Millisecond*200 { + pb.Debug = true + } + + arg.Register(&argBash) + flags := []string{} + for _, s := range pb.Argv { + if s == "--autodebug" { + continue + } + flags = append(flags, s) + } + // pb.Debug = true + pb.Debugf("DEBUG: MustParse(%v)", flags) + var err error + myAuto.pp, err = arg.ParseFlags(flags, dest) + if err != nil { + pb.Debugf("DEBUG: Parse error: %v", err) + } + + if myAuto.pp == nil { + pb.Debugf("DEBUG: myAuto.pp == nil after ParseFlags()") + } else { + pb.Debugf("DEBUG: myAuto.pp is ok after ParseFlags()") + } + + if pb.IsAuto { + myAuto.autoFunc(pb) // run the autocomplete function the user made for their application + if pb.Debug { + // TODO: + // 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) + myAuto.pp = arg.MustParse(dest) + return pb +} |