2 files changed, 27 insertions, 4 deletions
diff --git a/app/dirlist.go b/app/dirlist.go
index d77ace42..708eb824 100644
--- a/app/dirlist.go
+++ b/app/dirlist.go
@@ -26,6 +26,7 @@ type DirectoryLister interface {
 
 	Selected() string
 	Select(string)
+	Open(string, time.Duration, func(types.WorkerMessage))
 
 	Update(types.WorkerMessage)
 	List() []string
@@ -172,11 +173,16 @@ func (dirlist *DirectoryList) ExpandFolder() {
 }
 
 func (dirlist *DirectoryList) Select(name string) {
+	dirlist.Open(name, dirlist.UiConfig(name).DirListDelay, nil)
+}
+
+func (dirlist *DirectoryList) Open(name string, delay time.Duration,
+	cb func(types.WorkerMessage),
+) {
 	dirlist.selecting = name
 
 	dirlist.cancel()
 	dirlist.ctx, dirlist.cancel = context.WithCancel(context.Background())
-	delay := dirlist.UiConfig(name).DirListDelay
 
 	go func(ctx context.Context) {
 		defer log.PanicHandler()
@@ -198,6 +204,9 @@ func (dirlist *DirectoryList) Select(name string) {
 					case *types.Cancelled:
 						log.Debugf("OpenDirectory cancelled")
 					}
+					if cb != nil {
+						cb(msg)
+					}
 				})
 		case <-ctx.Done():
 			log.Tracef("dirlist: skip %s", name)
diff --git a/app/dirtree.go b/app/dirtree.go
index 50f5797f..6a3b34c0 100644
--- a/app/dirtree.go
+++ b/app/dirtree.go
@@ -5,6 +5,7 @@ import (
 	"sort"
 	"strconv"
 	"strings"
+	"time"
 
 	"git.sr.ht/~rjarry/aerc/config"
 	"git.sr.ht/~rjarry/aerc/lib"
@@ -50,11 +51,14 @@ func (dt *DirectoryTree) Update(msg types.WorkerMessage) {
 	switch msg := msg.(type) {
 
 	case *types.Done:
-		switch msg.InResponseTo().(type) {
+		switch resp := msg.InResponseTo().(type) {
 		case *types.RemoveDirectory, *types.ListDirectories, *types.CreateDirectory:
 			dt.DirectoryList.Update(msg)
 			dt.buildTree()
 			dt.Invalidate()
+		case *types.OpenDirectory:
+			dt.reindex(resp.Directory)
+			dt.DirectoryList.Update(msg)
 		default:
 			dt.DirectoryList.Update(msg)
 		}
@@ -196,7 +200,7 @@ func (dt *DirectoryTree) SelectedMsgStore() (*lib.MessageStore, bool) {
 	return dt.DirectoryList.SelectedMsgStore()
 }
 
-func (dt *DirectoryTree) Select(name string) {
+func (dt *DirectoryTree) reindex(name string) {
 	idx := findString(dt.treeDirs, name)
 	if idx >= 0 {
 		selIdx, node := dt.getTreeNode(uint32(idx))
@@ -205,14 +209,24 @@ func (dt *DirectoryTree) Select(name string) {
 			dt.listIdx = selIdx
 		}
 	}
+}
 
+func (dt *DirectoryTree) Select(name string) {
 	if name == "" {
 		return
 	}
-
+	dt.reindex(name)
 	dt.DirectoryList.Select(name)
 }
 
+func (dt *DirectoryTree) Open(name string, delay time.Duration, cb func(types.WorkerMessage)) {
+	if name == "" {
+		return
+	}
+	dt.reindex(name)
+	dt.DirectoryList.Open(name, delay, cb)
+}
+
 func (dt *DirectoryTree) NextPrev(delta int) {
 	newIdx := dt.listIdx
 	ndirs := len(dt.list)