init: pristine aerc 0.20.0 source

commands/account/rmdir.go

@@ -0,0 +1,154 @@
+package account
+
+import (
+	"errors"
+	"fmt"
+	"time"
+
+	"git.sr.ht/~rjarry/aerc/app"
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/models"
+	"git.sr.ht/~rjarry/aerc/worker/types"
+	"git.sr.ht/~rjarry/go-opt/v2"
+)
+
+type RemoveDir struct {
+	Force  bool   `opt:"-f" desc:"Remove the directory even if it contains messages."`
+	Folder string `opt:"folder" complete:"CompleteFolder" required:"false" desc:"Folder name."`
+}
+
+func init() {
+	commands.Register(RemoveDir{})
+}
+
+func (RemoveDir) Description() string {
+	return "Remove folder."
+}
+
+func (RemoveDir) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST
+}
+
+func (RemoveDir) Aliases() []string {
+	return []string{"rmdir"}
+}
+
+func (RemoveDir) CompleteFolder(arg string) []string {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return nil
+	}
+	return commands.FilterList(acct.Directories().List(), arg, opt.QuoteArg)
+}
+
+func (r RemoveDir) Execute(args []string) error {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return errors.New("No account selected")
+	}
+
+	current := acct.Directories().SelectedDirectory()
+	toRemove := current
+	if r.Folder != "" {
+		toRemove = acct.Directories().Directory(r.Folder)
+		if toRemove == nil {
+			return fmt.Errorf("No such directory: %s", r.Folder)
+		}
+	}
+
+	role := toRemove.Role
+
+	// Check for any messages in the directory.
+	if role != models.QueryRole && toRemove.Exists > 0 && !r.Force {
+		return errors.New("Refusing to remove non-empty directory; use -f")
+	}
+
+	if role == models.VirtualRole {
+		return errors.New("Cannot remove a virtual node")
+	}
+
+	if toRemove != current {
+		r.remove(acct, toRemove, func() {})
+		return nil
+	}
+
+	curDir := current.Name
+	var newDir string
+	dirFound := false
+
+	oldDir := acct.Directories().Previous()
+	if oldDir != "" {
+		present := false
+		for _, dir := range acct.Directories().List() {
+			if dir == oldDir {
+				present = true
+				break
+			}
+		}
+		if oldDir != curDir && present {
+			newDir = oldDir
+			dirFound = true
+		}
+	}
+
+	defaultDir := acct.AccountConfig().Default
+	if !dirFound && defaultDir != curDir {
+		for _, dir := range acct.Directories().List() {
+			if defaultDir == dir {
+				newDir = dir
+				dirFound = true
+				break
+			}
+		}
+	}
+
+	if !dirFound {
+		for _, dir := range acct.Directories().List() {
+			if dir != curDir {
+				newDir = dir
+				dirFound = true
+				break
+			}
+		}
+	}
+
+	if !dirFound {
+		return errors.New("No directory to move to afterwards!")
+	}
+
+	reopenCurrentDir := func() { acct.Directories().Open(curDir, "", 0, nil, false) }
+
+	acct.Directories().Open(newDir, "", 0, func(msg types.WorkerMessage) {
+		switch msg.(type) {
+		case *types.Done:
+			break
+		case *types.Error:
+			app.PushError("Could not change directory")
+			reopenCurrentDir()
+			return
+		default:
+			return
+		}
+		r.remove(acct, toRemove, reopenCurrentDir)
+	}, false)
+
+	return nil
+}
+
+func (r RemoveDir) remove(acct *app.AccountView, dir *models.Directory, onErr func()) {
+	acct.Worker().PostAction(&types.RemoveDirectory{
+		Directory: dir.Name,
+		Quiet:     r.Force,
+	}, func(msg types.WorkerMessage) {
+		switch msg := msg.(type) {
+		case *types.Done:
+			app.PushStatus("Directory removed.", 10*time.Second)
+		case *types.Error:
+			app.PushError(msg.Error.Error())
+			onErr()
+		case *types.Unsupported:
+			app.PushError(":rmdir is not supported by the backend.")
+			onErr()
+		}
+	})
+}