commands: add zoxide support via :z

zoxide is a command line utility, supported by many CLI programs. Enable
zoxide support via the :z command which is a drop-in replacement for :cd
(and calls ChangeDirectory in the background), but also manages adding
paths to and querying from the zoxide database. The command is not
registered if zoxide is not on $PATH.

Link: https://github.com/ajeetdsouza/zoxide
Signed-off-by: Bence Ferdinandy <bence@ferdinandy.com>
Acked-by: Robin Jarry <robin@jarry.cc>

1 files changed, 85 insertions, 0 deletions

diff --git a/commands/z.go b/commands/z.go
new file mode 100644
index 00000000..ca982ba7
--- /dev/null
+++ b/commands/z.go
@@ -0,0 +1,85 @@
+package commands
+
+import (
+	"errors"
+	"os"
+	"os/exec"
+	"strings"
+
+	"git.sr.ht/~rjarry/aerc/widgets"
+)
+
+type Zoxide struct{}
+
+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[1:]...)
+	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) Aliases() []string {
+	return []string{"z"}
+}
+
+func (Zoxide) Complete(aerc *widgets.Aerc, args []string) []string {
+	return ChangeDirectory{}.Complete(aerc, args)
+}
+
+// Execute calls zoxide add and query and delegates actually changing the
+// directory to ChangeDirectory
+func (Zoxide) Execute(aerc *widgets.Aerc, args []string) error {
+	if len(args) < 1 {
+		return errors.New("Usage: z [directory or zoxide query]")
+	}
+	target := strings.Join(args[1:], " ")
+	switch target {
+	case "":
+		return ChangeDirectory{}.Execute(aerc, args)
+	case "-":
+		if previousDir != "" {
+			err := ZoxideAdd(previousDir)
+			if err != nil {
+				return err
+			}
+		}
+		return ChangeDirectory{}.Execute(aerc, args)
+	default:
+		_, err := os.Stat(target)
+		if err != nil {
+			// not a file, assume zoxide query
+			res, err := ZoxideQuery(args)
+			if err != nil {
+				return errors.New("zoxide: no match found")
+			} else {
+				err := ZoxideAdd(res)
+				if err != nil {
+					return err
+				}
+				return ChangeDirectory{}.Execute(aerc, []string{"z", res})
+			}
+
+		} else {
+			err := ZoxideAdd(target)
+			if err != nil {
+				return err
+			}
+			return ChangeDirectory{}.Execute(aerc, args)
+		}
+
+	}
+}