diff options
Diffstat (limited to 'worker')
-rw-r--r-- | worker/imap/fetch.go | 1 | ||||
-rw-r--r-- | worker/imap/flags.go | 43 | ||||
-rw-r--r-- | worker/imap/open.go | 1 | ||||
-rw-r--r-- | worker/imap/worker.go | 6 | ||||
-rw-r--r-- | worker/types/messages.go | 10 |
5 files changed, 60 insertions, 1 deletions
diff --git a/worker/imap/fetch.go b/worker/imap/fetch.go index 383a8a8f..489dbe4d 100644 --- a/worker/imap/fetch.go +++ b/worker/imap/fetch.go @@ -25,6 +25,7 @@ func (imapw *IMAPWorker) handleFetchMessageHeaders( }() go func() { for msg := range messages { + imapw.seqMap[msg.SeqNum-1] = msg.Uid imapw.worker.PostMessage(&types.MessageInfo{ Envelope: msg.Envelope, Flags: msg.Flags, diff --git a/worker/imap/flags.go b/worker/imap/flags.go new file mode 100644 index 00000000..cb9b3b1e --- /dev/null +++ b/worker/imap/flags.go @@ -0,0 +1,43 @@ +package imap + +import ( + "github.com/emersion/go-imap" + + "git.sr.ht/~sircmpwn/aerc2/worker/types" +) + +func (imapw *IMAPWorker) handleDeleteMessages(msg *types.DeleteMessages) { + item := imap.FormatFlagsOp(imap.AddFlags, true) + flags := []interface{}{imap.DeletedFlag} + if err := imapw.client.UidStore(&msg.Uids, item, flags, nil); err != nil { + imapw.worker.PostMessage(&types.Error{ + Message: types.RespondTo(msg), + Error: err, + }, nil) + return + } + var deleted []uint32 + ch := make(chan uint32) + done := make(chan interface{}) + go func() { + for seqNum := range ch { + i := seqNum - 1 + deleted = append(deleted, imapw.seqMap[i]) + imapw.seqMap = append(imapw.seqMap[:i], imapw.seqMap[i+1:]...) + } + done <- nil + }() + if err := imapw.client.Expunge(ch); err != nil { + imapw.worker.PostMessage(&types.Error{ + Message: types.RespondTo(msg), + Error: err, + }, nil) + } else { + <-done + imapw.worker.PostMessage(&types.MessagesDeleted{ + Message: types.RespondTo(msg), + Uids: deleted, + }, nil) + imapw.worker.PostMessage(&types.Done{types.RespondTo(msg)}, nil) + } +} diff --git a/worker/imap/open.go b/worker/imap/open.go index 87c4fb33..3705bc0d 100644 --- a/worker/imap/open.go +++ b/worker/imap/open.go @@ -39,6 +39,7 @@ func (imapw *IMAPWorker) handleFetchDirectoryContents( }, nil) } else { imapw.worker.Logger.Printf("Found %d UIDs", len(uids)) + imapw.seqMap = make([]uint32, len(uids)) imapw.worker.PostMessage(&types.DirectoryContents{ Message: types.RespondTo(msg), Uids: uids, diff --git a/worker/imap/worker.go b/worker/imap/worker.go index 16461651..ea7f317a 100644 --- a/worker/imap/worker.go +++ b/worker/imap/worker.go @@ -33,12 +33,14 @@ type IMAPWorker struct { selected imap.MailboxStatus updates chan client.Update worker *types.Worker + // Map of sequence numbers to UIDs, index 0 is seq number 1 + seqMap []uint32 } func NewIMAPWorker(worker *types.Worker) *IMAPWorker { return &IMAPWorker{ - worker: worker, updates: make(chan client.Update, 50), + worker: worker, } } @@ -156,6 +158,8 @@ func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error { w.handleFetchDirectoryContents(msg) case *types.FetchMessageHeaders: w.handleFetchMessageHeaders(msg) + case *types.DeleteMessages: + w.handleDeleteMessages(msg) default: return errUnsupported } diff --git a/worker/types/messages.go b/worker/types/messages.go index 3f1a39f7..ff2c36bb 100644 --- a/worker/types/messages.go +++ b/worker/types/messages.go @@ -86,6 +86,11 @@ type FetchMessageBodies struct { Uids imap.SeqSet } +type DeleteMessages struct { + Message + Uids imap.SeqSet +} + // Messages type CertificateApprovalRequest struct { @@ -122,3 +127,8 @@ type MessageInfo struct { Size uint32 Uid uint32 } + +type MessagesDeleted struct { + Message + Uids []uint32 +} |