2 files changed, 48 insertions, 0 deletions
diff --git a/worker/imap/fetch.go b/worker/imap/fetch.go
new file mode 100644
index 0000000..383a8a8
--- /dev/null
+++ b/worker/imap/fetch.go
@@ -0,0 +1,46 @@
+package imap
+
+import (
+	"github.com/emersion/go-imap"
+
+	"git.sr.ht/~sircmpwn/aerc2/worker/types"
+)
+
+func (imapw *IMAPWorker) handleFetchMessageHeaders(
+	msg *types.FetchMessageHeaders) {
+
+	imapw.worker.Logger.Printf("Fetching message headers")
+
+	go func() {
+		messages := make(chan *imap.Message)
+		done := make(chan error, 1)
+		items := []imap.FetchItem{
+			imap.FetchEnvelope,
+			imap.FetchInternalDate,
+			imap.FetchFlags,
+			imap.FetchUid,
+		}
+		go func() {
+			done <- imapw.client.UidFetch(&msg.Uids, items, messages)
+		}()
+		go func() {
+			for msg := range messages {
+				imapw.worker.PostMessage(&types.MessageInfo{
+					Envelope:     msg.Envelope,
+					Flags:        msg.Flags,
+					InternalDate: msg.InternalDate,
+					Uid:          msg.Uid,
+				}, nil)
+			}
+			if err := <-done; err != nil {
+				imapw.worker.PostMessage(&types.Error{
+					Message: types.RespondTo(msg),
+					Error:   err,
+				}, nil)
+			} else {
+				imapw.worker.PostMessage(
+					&types.Done{types.RespondTo(msg)}, nil)
+			}
+		}()
+	}()
+}
diff --git a/worker/imap/worker.go b/worker/imap/worker.go
index 51cb221..1646165 100644
--- a/worker/imap/worker.go
+++ b/worker/imap/worker.go
@@ -154,6 +154,8 @@ func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
 		w.handleOpenDirectory(msg)
 	case *types.FetchDirectoryContents:
 		w.handleFetchDirectoryContents(msg)
+	case *types.FetchMessageHeaders:
+		w.handleFetchMessageHeaders(msg)
 	default:
 		return errUnsupported
 	}