Flesh out command parsing & handling

1 files changed, 31 insertions, 3 deletions

diff --git a/commands/next-folder.go b/commands/next-folder.go
index 02975e1..ed11615 100644
--- a/commands/next-folder.go
+++ b/commands/next-folder.go
@@ -1,15 +1,43 @@
 package commands
 
 import (
+	"errors"
+	"fmt"
+	"strconv"
+
 	"git.sr.ht/~sircmpwn/aerc2/widgets"
 )
 
 func init() {
-	Register("next-folder", NextFolder)
+	Register("next-folder", NextPrevFolder)
+	Register("prev-folder", NextPrevFolder)
+}
+
+func usage(cmd string) error {
+	return errors.New(fmt.Sprintf("Usage: %s [n]", cmd))
 }
 
-func NextFolder(aerc *widgets.Aerc, cmd string) error {
+func NextPrevFolder(aerc *widgets.Aerc, args []string) error {
+	if len(args) > 2 {
+		return usage(args[0])
+	}
+	var (
+		n   int = 1
+		err error
+	)
+	if len(args) > 1 {
+		n, err = strconv.Atoi(args[1])
+		if err != nil {
+			return usage(args[0])
+		}
+	}
 	acct := aerc.SelectedAccount()
-	acct.Directories().Next()
+	for ; n > 0; n-- {
+		if args[0] == "prev-folder" {
+			acct.Directories().Prev()
+		} else {
+			acct.Directories().Next()
+		}
+	}
 	return nil
 }