Handle incoming emails gracefully

5 files changed, 36 insertions, 9 deletions

diff --git a/lib/msgstore.go b/lib/msgstore.go
index b39d0bb..fb13346 100644
--- a/lib/msgstore.go
+++ b/lib/msgstore.go
@@ -1,6 +1,7 @@
 package lib
 
 import (
+	"fmt"
 	"io"
 	"sync"
 	"time"
@@ -140,6 +141,11 @@ func (store *MessageStore) Update(msg types.WorkerMessage) {
 	switch msg := msg.(type) {
 	case *types.DirectoryInfo:
 		store.DirInfo = *msg
+		fmt.Printf("got dirinfo, %d exists, %d known\n",
+			store.DirInfo.Exists, len(store.Uids))
+		if store.DirInfo.Exists != len(store.Uids) {
+			store.worker.PostAction(&types.FetchDirectoryContents{}, nil)
+		}
 		update = true
 	case *types.DirectoryContents:
 		newMap := make(map[uint32]*types.MessageInfo)
diff --git a/widgets/account.go b/widgets/account.go
index c01b15f..ba2b8d3 100644
--- a/widgets/account.go
+++ b/widgets/account.go
@@ -161,17 +161,17 @@ func (acct *AccountView) onMessage(msg types.WorkerMessage) {
 			} else {
 				acct.msglist.SetStore(nil)
 			}
-			acct.worker.PostAction(&types.FetchDirectoryContents{},
-				func(msg types.WorkerMessage) {
-					store := acct.msgStores[acct.dirlist.selected]
-					acct.msglist.SetStore(store)
-				})
 		}
 	case *types.DirectoryInfo:
 		if store, ok := acct.msgStores[msg.Name]; ok {
 			store.Update(msg)
 		} else {
-			acct.msgStores[msg.Name] = lib.NewMessageStore(acct.worker, msg)
+			store = lib.NewMessageStore(acct.worker, msg)
+			acct.msgStores[msg.Name] = store
+			store.OnUpdate(func(_ *lib.MessageStore) {
+				store.OnUpdate(nil)
+				acct.msglist.SetStore(store)
+			})
 		}
 	case *types.DirectoryContents:
 		store := acct.msgStores[acct.dirlist.selected]
diff --git a/widgets/msglist.go b/widgets/msglist.go
index eeadec7..65473a8 100644
--- a/widgets/msglist.go
+++ b/widgets/msglist.go
@@ -109,6 +109,7 @@ func (ml *MessageList) Height() int {
 }
 
 func (ml *MessageList) storeUpdate(store *lib.MessageStore) {
+	ml.logger.Println("message store updated")
 	if ml.Store() != store {
 		return
 	}
diff --git a/worker/imap/open.go b/worker/imap/open.go
index dc5d6d1..ab54e88 100644
--- a/worker/imap/open.go
+++ b/worker/imap/open.go
@@ -17,6 +17,9 @@ func (imapw *IMAPWorker) handleOpenDirectory(msg *types.OpenDirectory) {
 		}, nil)
 	} else {
 		imapw.worker.PostMessage(&types.Done{types.RespondTo(msg)}, nil)
+		if imapw.idleStop == nil {
+			imapw.idleStop = make(chan struct{})
+		}
 	}
 }
 
diff --git a/worker/imap/worker.go b/worker/imap/worker.go
index 4354ab9..3d4555f 100644
--- a/worker/imap/worker.go
+++ b/worker/imap/worker.go
@@ -18,7 +18,7 @@ var errUnsupported = fmt.Errorf("unsupported command")
 
 type imapClient struct {
 	*client.Client
-	*idle.IdleClient
+	idle *idle.IdleClient
 }
 
 type IMAPWorker struct {
@@ -30,6 +30,8 @@ type IMAPWorker struct {
 	}
 
 	client   *imapClient
+	idleStop chan struct{}
+	idleDone chan error
 	selected imap.MailboxStatus
 	updates  chan client.Update
 	worker   *types.Worker
@@ -39,8 +41,9 @@ type IMAPWorker struct {
 
 func NewIMAPWorker(worker *types.Worker) *IMAPWorker {
 	return &IMAPWorker{
-		updates: make(chan client.Update, 50),
-		worker:  worker,
+		idleDone: make(chan error),
+		updates:  make(chan client.Update, 50),
+		worker:   worker,
 	}
 }
 
@@ -80,6 +83,13 @@ func (w *IMAPWorker) verifyPeerCert(msg types.WorkerMessage) func(
 }
 
 func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
+	if w.idleStop != nil {
+		close(w.idleStop)
+		if err := <-w.idleDone; err != nil {
+			w.worker.PostMessage(&types.Error{Error: err}, nil)
+		}
+	}
+
 	switch msg := msg.(type) {
 	case *types.Unsupported:
 		// No-op
@@ -167,6 +177,13 @@ func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
 	default:
 		return errUnsupported
 	}
+
+	if w.idleStop != nil {
+		w.idleStop = make(chan struct{})
+		go func() {
+			w.idleDone <- w.client.idle.IdleWithFallback(w.idleStop, 0)
+		}()
+	}
 	return nil
 }