diff options
Diffstat (limited to 'worker')
| -rw-r--r-- | worker/imap/fetch.go | 24 | ||||
| -rw-r--r-- | worker/imap/worker.go | 12 |
2 files changed, 34 insertions, 2 deletions
diff --git a/worker/imap/fetch.go b/worker/imap/fetch.go index 2d5bb7f..7d1bfcf 100644 --- a/worker/imap/fetch.go +++ b/worker/imap/fetch.go @@ -40,7 +40,11 @@ func (imapw *IMAPWorker) handleFetchMessageBodyPart( imapw.worker.Logger.Printf("Fetching message part") section := &imap.BodySectionName{} section.Path = msg.Part - items := []imap.FetchItem{section.FetchItem()} + items := []imap.FetchItem{ + imap.FetchFlags, + imap.FetchUid, + section.FetchItem(), + } uids := imap.SeqSet{} uids.AddNum(msg.Uid) imapw.handleFetchMessages(msg, &uids, items, section) @@ -51,7 +55,11 @@ func (imapw *IMAPWorker) handleFetchFullMessages( imapw.worker.Logger.Printf("Fetching full messages") section := &imap.BodySectionName{} - items := []imap.FetchItem{section.FetchItem()} + items := []imap.FetchItem{ + imap.FetchFlags, + imap.FetchUid, + section.FetchItem(), + } imapw.handleFetchMessages(msg, &msg.Uids, items, section) } @@ -89,6 +97,12 @@ func (imapw *IMAPWorker) handleFetchMessages( Reader: reader, Uid: _msg.Uid, }, nil) + // Update flags (to mark message as read) + imapw.worker.PostMessage(&types.MessageInfo{ + Message: types.RespondTo(msg), + Flags: _msg.Flags, + Uid: _msg.Uid, + }, nil) case *types.FetchMessageBodyPart: reader := _msg.GetBody(section) imapw.worker.PostMessage(&types.MessageBodyPart{ @@ -96,6 +110,12 @@ func (imapw *IMAPWorker) handleFetchMessages( Reader: reader, Uid: _msg.Uid, }, nil) + // Update flags (to mark message as read) + imapw.worker.PostMessage(&types.MessageInfo{ + Message: types.RespondTo(msg), + Flags: _msg.Flags, + Uid: _msg.Uid, + }, nil) } } done <- nil diff --git a/worker/imap/worker.go b/worker/imap/worker.go index f71a950..373072b 100644 --- a/worker/imap/worker.go +++ b/worker/imap/worker.go @@ -171,6 +171,18 @@ func (w *IMAPWorker) handleImapUpdate(update client.Update) { Recent: int(status.Recent), Unseen: int(status.Unseen), }, nil) + case *client.MessageUpdate: + msg := update.Message + if msg.Uid == 0 { + msg.Uid = w.seqMap[msg.SeqNum-1] + } + w.worker.PostMessage(&types.MessageInfo{ + BodyStructure: msg.BodyStructure, + Envelope: msg.Envelope, + Flags: msg.Flags, + InternalDate: msg.InternalDate, + Uid: msg.Uid, + }, nil) case *client.ExpungeUpdate: i := update.SeqNum - 1 uid := w.seqMap[i] |