add command helpers

1 files changed, 55 insertions, 0 deletions

diff --git a/commands/msg/utils.go b/commands/msg/utils.go
new file mode 100644
index 00000000..30a4394b
--- /dev/null
+++ b/commands/msg/utils.go
@@ -0,0 +1,55 @@
+package msg
+
+import (
+	"errors"
+
+	"git.sr.ht/~sircmpwn/aerc/commands"
+	"git.sr.ht/~sircmpwn/aerc/lib"
+	"git.sr.ht/~sircmpwn/aerc/models"
+	"git.sr.ht/~sircmpwn/aerc/widgets"
+)
+
+type helper struct {
+	msgProvider widgets.ProvidesMessages
+}
+
+func newHelper(aerc *widgets.Aerc) *helper {
+	return &helper{aerc.SelectedTab().(widgets.ProvidesMessages)}
+}
+
+func (h *helper) uids() ([]uint32, error) {
+	msgs, err := commands.MarkedOrSelected(h.msgProvider)
+	if err != nil {
+		return nil, err
+	}
+	uids := commands.UidsFromMessageInfos(msgs)
+	return uids, nil
+}
+
+func (h *helper) store() (*lib.MessageStore, error) {
+	store := h.msgProvider.Store()
+	if store == nil {
+		return nil, errors.New("Cannot perform action. Messages still loading")
+	}
+	return store, nil
+}
+
+func (h *helper) account() (*widgets.AccountView, error) {
+	acct := h.msgProvider.SelectedAccount()
+	if acct == nil {
+		return nil, errors.New("No account selected")
+	}
+	return acct, nil
+}
+
+func (h *helper) messages() ([]*models.MessageInfo, error) {
+	return commands.MarkedOrSelected(h.msgProvider)
+}
+
+func (h *helper) messageUids() ([]uint32, error) {
+	msgs, err := h.messages()
+	if err != nil {
+		return nil, err
+	}
+	return commands.UidsFromMessageInfos(msgs), nil
+}