364 lines
8.4 KiB
Go
364 lines
8.4 KiB
Go
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
|
|
}
|