From de364846ccaba3b93c383add3846443048f1c2c9 Mon Sep 17 00:00:00 2001
From: Drew DeVault <sir@cmpwn.com>
Date: Thu, 14 Mar 2019 22:19:04 -0400
Subject: Display message subjects in message list

---
 widgets/msglist.go    | 25 ++++++++++++++++++++++---
 widgets/spinner.go    |  2 +-
 worker/imap/fetch.go  | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 worker/imap/worker.go |  2 ++
 4 files changed, 71 insertions(+), 4 deletions(-)
 create mode 100644 worker/imap/fetch.go

diff --git a/widgets/msglist.go b/widgets/msglist.go
index fef396b..22695ec 100644
--- a/widgets/msglist.go
+++ b/widgets/msglist.go
@@ -14,6 +14,8 @@ import (
 type MessageStore struct {
 	DirInfo  types.DirectoryInfo
 	Messages map[uint32]*types.MessageInfo
+	// Ordered list of known UIDs
+	Uids []uint32
 	// Map of uids we've asked the worker to fetch
 	onUpdate       func(store *MessageStore)
 	pendingBodies  map[uint32]interface{}
@@ -67,9 +69,11 @@ func (store *MessageStore) Update(msg types.WorkerMessage) {
 			}
 		}
 		store.Messages = newMap
+		store.Uids = msg.Uids
 		update = true
 		break
 	case *types.MessageInfo:
+		// TODO: merge message info into existing record, if applicable
 		store.Messages[msg.Uid] = msg
 		if _, ok := store.pendingHeaders[msg.Uid]; msg.Envelope != nil && ok {
 			delete(store.pendingHeaders, msg.Uid)
@@ -90,6 +94,7 @@ type MessageList struct {
 	conf         *config.AercConfig
 	logger       *log.Logger
 	onInvalidate func(d ui.Drawable)
+	selected     int
 	spinner      *Spinner
 	store        *MessageStore
 }
@@ -97,8 +102,9 @@ type MessageList struct {
 // TODO: fish in config
 func NewMessageList(logger *log.Logger) *MessageList {
 	ml := &MessageList{
-		logger:  logger,
-		spinner: NewSpinner(),
+		logger:   logger,
+		selected: 0,
+		spinner:  NewSpinner(),
 	}
 	ml.spinner.OnInvalidate(func(_ ui.Drawable) {
 		ml.Invalidate()
@@ -131,7 +137,10 @@ func (ml *MessageList) Draw(ctx *ui.Context) {
 		row          int = 0
 	)
 
-	for uid, msg := range ml.store.Messages {
+	for i := len(ml.store.Uids) - 1; i >= 0; i-- {
+		uid := ml.store.Uids[i]
+		msg := ml.store.Messages[uid]
+
 		if row >= ctx.Height() {
 			break
 		}
@@ -139,7 +148,17 @@ func (ml *MessageList) Draw(ctx *ui.Context) {
 		if msg == nil {
 			needsHeaders = append(needsHeaders, uid)
 			ml.spinner.Draw(ctx.Subcontext(0, row, ctx.Width(), 1))
+			row += 1
+			continue
+		}
+
+		style := tcell.StyleDefault
+		if row == ml.selected {
+			style = style.Background(tcell.ColorWhite).
+				Foreground(tcell.ColorBlack)
 		}
+		ctx.Fill(0, row, ctx.Width(), 1, ' ', style)
+		ctx.Printf(0, row, style, "%s", msg.Envelope.Subject)
 
 		row += 1
 	}
diff --git a/widgets/spinner.go b/widgets/spinner.go
index 812fb2c..bafc712 100644
--- a/widgets/spinner.go
+++ b/widgets/spinner.go
@@ -73,7 +73,7 @@ func (s *Spinner) IsRunning() bool {
 
 func (s *Spinner) Draw(ctx *ui.Context) {
 	if !s.IsRunning() {
-		return
+		s.Start()
 	}
 
 	ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', tcell.StyleDefault)
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
 	}
-- 
cgit v1.2.3