package msg import ( "errors" "git.sr.ht/~rjarry/aerc/commands" "git.sr.ht/~rjarry/aerc/lib" "git.sr.ht/~rjarry/aerc/models" "git.sr.ht/~rjarry/aerc/widgets" ) type helper struct { msgProvider widgets.ProvidesMessages } func newHelper(aerc *widgets.Aerc) *helper { return &helper{aerc.SelectedTabContent().(widgets.ProvidesMessages)} } func (h *helper) markedOrSelectedUids() ([]uint32, error) { return commands.MarkedOrSelected(h.msgProvider) } 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) { uid, err := commands.MarkedOrSelected(h.msgProvider) if err != nil { return nil, err } store, err := h.store() if err != nil { return nil, err } return commands.MsgInfoFromUids(store, uid) }