diff options
Diffstat (limited to 'commands/msg/move.go')
| -rw-r--r-- | commands/msg/move.go | 171 |
1 files changed, 145 insertions, 26 deletions
diff --git a/commands/msg/move.go b/commands/msg/move.go index 000e2b2a..defe94d1 100644 --- a/commands/msg/move.go +++ b/commands/msg/move.go @@ -1,6 +1,7 @@ package msg import ( + "bytes" "fmt" "time" @@ -9,12 +10,14 @@ import ( "git.sr.ht/~rjarry/aerc/config" "git.sr.ht/~rjarry/aerc/lib" "git.sr.ht/~rjarry/aerc/lib/ui" + "git.sr.ht/~rjarry/aerc/log" "git.sr.ht/~rjarry/aerc/models" "git.sr.ht/~rjarry/aerc/worker/types" ) type Move struct { CreateFolders bool `opt:"-p"` + Account string `opt:"-a" complete:"CompleteAccount"` Folder string `opt:"folder" complete:"CompleteFolder"` } @@ -30,8 +33,21 @@ func (Move) Aliases() []string { return []string{"mv", "move"} } -func (*Move) CompleteFolder(arg string) []string { - return commands.GetFolders(arg) +func (*Move) CompleteAccount(arg string) []string { + return commands.FilterList(app.AccountNames(), arg, commands.QuoteSpace) +} + +func (m *Move) CompleteFolder(arg string) []string { + var acct *app.AccountView + if len(m.Account) > 0 { + acct, _ = app.Account(m.Account) + } else { + acct = app.SelectedAccount() + } + if acct == nil { + return nil + } + return commands.FilterList(acct.Directories().List(), arg, nil) } func (m Move) Execute(args []string) error { @@ -44,47 +60,150 @@ func (m Move) Execute(args []string) error { if err != nil { return err } - msgs, err := h.messages() + uids, err := h.markedOrSelectedUids() if err != nil { return err } - var uids []uint32 - for _, msg := range msgs { - uids = append(uids, msg.Uid) + + if len(m.Account) == 0 { + store.Move(uids, m.Folder, m.CreateFolders, func(msg types.WorkerMessage) { + m.CallBack(msg, acct, uids, false) + }) + return nil } - marker := store.Marker() - marker.ClearVisualMark() - next := findNextNonDeleted(uids, store) - store.Move(uids, m.Folder, m.CreateFolders, func( - msg types.WorkerMessage, - ) { - switch msg := msg.(type) { - case *types.Done: - var s string - if len(uids) > 1 { - s = "%d messages moved to %s" - } else { - s = "%d message moved to %s" - } - handleDone(acct, next, fmt.Sprintf(s, len(uids), m.Folder), store) - case *types.Error: - app.PushError(msg.Error.Error()) - marker.Remark() + destAcct, err := app.Account(m.Account) + if err != nil { + return err + } + + destStore := destAcct.Store() + if destStore == nil { + app.PushError(fmt.Sprintf("No message store in %s", m.Account)) + return nil + } + + var messages []*types.FullMessage + fetchDone := make(chan bool, 1) + store.FetchFull(uids, func(fm *types.FullMessage) { + messages = append(messages, fm) + if len(messages) == len(uids) { + fetchDone <- true } }) + // Since this operation can take some time with some backends + // (e.g. IMAP), provide some feedback to inform the user that + // something is happening + app.PushStatus("Moving messages...", 10*time.Second) + + var appended []uint32 + var timeout bool + go func() { + defer log.PanicHandler() + + select { + case <-fetchDone: + break + case <-time.After(30 * time.Second): + // TODO: find a better way to determine if store.FetchFull() + // has finished with some errors. + app.PushError("Failed to fetch all messages") + if len(messages) == 0 { + return + } + } + + AppendLoop: + for _, fm := range messages { + done := make(chan bool, 1) + uid := fm.Content.Uid + buf := new(bytes.Buffer) + _, err = buf.ReadFrom(fm.Content.Reader) + if err != nil { + log.Errorf("could not get reader for uid %d", uid) + break + } + destStore.Append( + m.Folder, + models.SeenFlag, + time.Now(), + buf, + buf.Len(), + func(msg types.WorkerMessage) { + switch msg := msg.(type) { + case *types.Done: + appended = append(appended, uid) + done <- true + case *types.Error: + log.Errorf("AppendMessage failed: %v", msg.Error) + done <- false + } + }, + ) + select { + case ok := <-done: + if !ok { + break AppendLoop + } + case <-time.After(30 * time.Second): + log.Warnf("timed-out: appended %d of %d", len(appended), len(messages)) + timeout = true + break AppendLoop + } + } + if len(appended) > 0 { + store.Delete(appended, func(msg types.WorkerMessage) { + m.CallBack(msg, acct, appended, timeout) + }) + } + }() return nil } +func (m Move) CallBack(msg types.WorkerMessage, acct *app.AccountView, uids []uint32, timeout bool) { + store := acct.Store() + sel := store.Selected() + marker := store.Marker() + marker.ClearVisualMark() + next := findNextNonDeleted(uids, store) + + dest := m.Folder + if len(m.Account) > 0 { + dest = fmt.Sprintf("%s in %s", m.Folder, m.Account) + } + + switch msg := msg.(type) { + case *types.Done: + var s string + if len(uids) > 1 { + s = "%d messages moved to %s" + } else { + s = "%d message moved to %s" + } + if timeout { + s = "timed-out: only " + s + app.PushError(fmt.Sprintf(s, len(uids), dest)) + } else { + app.PushStatus(fmt.Sprintf(s, len(uids), dest), 10*time.Second) + } + handleDone(acct, next, store) + case *types.Error: + app.PushError(msg.Error.Error()) + marker.Remark() + case *types.Unsupported: + marker.Remark() + store.Select(sel.Uid) + app.PushError("error, unsupported for this worker") + } +} + func handleDone( acct *app.AccountView, next *models.MessageInfo, - message string, store *lib.MessageStore, ) { h := newHelper() - app.PushStatus(message, 10*time.Second) mv, isMsgView := h.msgProvider.(*app.MessageViewer) switch { case isMsgView && !config.Ui.NextMessageOnDelete: |