guiprep/bash2.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 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 {
		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)
		os.Exit(0)
	}

	arg.Register(&argBash)
}