mirror of https://git.sr.ht/~rjarry/aerc
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package commands
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"git.sr.ht/~rjarry/aerc/lib/xdg"
|
|
)
|
|
|
|
type Zoxide struct {
|
|
Args []string `opt:"..." required:"false" metavar:"<query>..." complete:"CompleteFolder" desc:"Target folder."`
|
|
}
|
|
|
|
func ZoxideAdd(arg string) error {
|
|
zargs := []string{"add", arg}
|
|
cmd := exec.Command("zoxide", zargs...)
|
|
err := cmd.Run()
|
|
return err
|
|
}
|
|
|
|
func ZoxideQuery(args []string) (string, error) {
|
|
zargs := append([]string{"query"}, args...)
|
|
cmd := exec.Command("zoxide", zargs...)
|
|
res, err := cmd.Output()
|
|
return strings.TrimSuffix(string(res), "\n"), err
|
|
}
|
|
|
|
func init() {
|
|
_, err := exec.LookPath("zoxide")
|
|
if err == nil {
|
|
Register(Zoxide{})
|
|
}
|
|
}
|
|
|
|
func (Zoxide) Description() string {
|
|
return "Change aerc's current working directory using zoxide."
|
|
}
|
|
|
|
func (Zoxide) Context() CommandContext {
|
|
return GLOBAL
|
|
}
|
|
|
|
func (Zoxide) Aliases() []string {
|
|
return []string{"z"}
|
|
}
|
|
|
|
func (*Zoxide) CompleteFolder(arg string) []string {
|
|
return CompletePath(arg, true)
|
|
}
|
|
|
|
// Execute calls zoxide add and query and delegates actually changing the
|
|
// directory to ChangeDirectory
|
|
func (z Zoxide) Execute(args []string) error {
|
|
if len(z.Args) == 0 {
|
|
z.Args = []string{"~"}
|
|
}
|
|
if len(z.Args) == 1 && (z.Args[0] == "~" || z.Args[0] == "-") {
|
|
if previousDir != "" {
|
|
err := ZoxideAdd(previousDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return ChangeDirectory{Target: z.Args[0]}.Execute(args)
|
|
} else {
|
|
target := xdg.ExpandHome(z.Args[0])
|
|
_, err := os.Stat(target)
|
|
if err != nil || len(z.Args) > 1 {
|
|
// not a file, assume zoxide query
|
|
res, err := ZoxideQuery(z.Args)
|
|
if err != nil {
|
|
return errors.New("zoxide: no match found")
|
|
} else {
|
|
err := ZoxideAdd(res)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cd := ChangeDirectory{Target: res}
|
|
return cd.Execute([]string{"z", res})
|
|
}
|
|
|
|
} else {
|
|
err := ZoxideAdd(target)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ChangeDirectory{Target: target}.Execute(args)
|
|
}
|
|
|
|
}
|
|
}
|