From 881a9081a850f39294ed97c10a1a50f8d852c9a5 Mon Sep 17 00:00:00 2001
From: Reto Brunner <reto@labrat.space>
Date: Wed, 18 Dec 2019 06:34:00 +0100
Subject: add command helpers

---
 commands/msg/utils.go | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 commands/msg/utils.go

(limited to 'commands/msg')

diff --git a/commands/msg/utils.go b/commands/msg/utils.go
new file mode 100644
index 0000000..30a4394
--- /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
+}
-- 
cgit v1.2.3