init: pristine aerc 0.20.0 source

commands/account/expand-folder.go

@@ -0,0 +1,52 @@
+package account
+
+import (
+	"errors"
+
+	"git.sr.ht/~rjarry/aerc/app"
+	"git.sr.ht/~rjarry/aerc/commands"
+)
+
+type ExpandCollapseFolder struct {
+	Folder string `opt:"folder" required:"false" complete:"CompleteFolder" desc:"Folder name."`
+}
+
+func init() {
+	commands.Register(ExpandCollapseFolder{})
+}
+
+func (ExpandCollapseFolder) Description() string {
+	return "Expand or collapse the current folder."
+}
+
+func (ExpandCollapseFolder) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST
+}
+
+func (ExpandCollapseFolder) Aliases() []string {
+	return []string{"expand-folder", "collapse-folder"}
+}
+
+func (*ExpandCollapseFolder) CompleteFolder(arg string) []string {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return nil
+	}
+	return commands.FilterList(acct.Directories().List(), arg, nil)
+}
+
+func (e ExpandCollapseFolder) Execute(args []string) error {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return errors.New("No account selected")
+	}
+	if e.Folder == "" {
+		e.Folder = acct.Directories().Selected()
+	}
+	if args[0] == "expand-folder" {
+		acct.Directories().ExpandFolder(e.Folder)
+	} else {
+		acct.Directories().CollapseFolder(e.Folder)
+	}
+	return nil
+}