aboutsummaryrefslogtreecommitdiffstats
path: root/commands/msg/utils.go
diff options
context:
space:
mode:
Diffstat (limited to 'commands/msg/utils.go')
-rw-r--r--commands/msg/utils.go55
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
+}