Handle incoming emails gracefully

2 files changed, 7 insertions, 6 deletions

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
 	}