store: implement thread folding

Implement thread folding on the message store level.

Signed-off-by: Koni Marti <koni.marti@gmail.com>
Acked-by: Robin Jarry <robin@jarry.cc>
Tested-by: inwit <inwit@sindominio.net>

2 files changed, 39 insertions, 1 deletions

diff --git a/lib/msgstore.go b/lib/msgstore.go
index c6ff5eb2..7c6f91c5 100644
--- a/lib/msgstore.go
+++ b/lib/msgstore.go
@@ -478,6 +478,38 @@ func (store *MessageStore) SelectedThread() (*types.Thread, error) {
 	return store.Thread(store.SelectedUid())
 }
 
+func (store *MessageStore) Fold(uid uint32) error {
+	return store.doThreadFolding(uid, true)
+}
+
+func (store *MessageStore) Unfold(uid uint32) error {
+	return store.doThreadFolding(uid, false)
+}
+
+func (store *MessageStore) doThreadFolding(uid uint32, hidden bool) error {
+	thread, err := store.Thread(uid)
+	if err != nil {
+		return err
+	}
+	err = thread.Walk(func(t *types.Thread, _ int, __ error) error {
+		if t.Uid != uid {
+			t.Hidden = hidden
+		}
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+	if store.builder == nil {
+		return errors.New("No thread builder available")
+	}
+	store.Select(uid)
+	store.threadsMutex.Lock()
+	store.builder.RebuildUids(store.threads, store.ReverseThreadOrder())
+	store.threadsMutex.Unlock()
+	return nil
+}
+
 func (store *MessageStore) Delete(uids []uint32,
 	cb func(msg types.WorkerMessage),
 ) {
diff --git a/lib/threadbuilder.go b/lib/threadbuilder.go
index 793034df..1f2d72c6 100644
--- a/lib/threadbuilder.go
+++ b/lib/threadbuilder.go
@@ -198,11 +198,17 @@ func (builder *ThreadBuilder) RebuildUids(threads []*types.Thread, inverse bool)
 		var threaduids []uint32
 		_ = iterT.Value().(*types.Thread).Walk(
 			func(t *types.Thread, level int, currentErr error) error {
+				stored, ok := builder.threadMap[t.Uid]
+				if ok {
+					// make this info persistent
+					t.Hidden = stored.Hidden
+					t.Deleted = stored.Deleted
+				}
+				builder.threadMap[t.Uid] = t
 				if t.Deleted || t.Hidden {
 					return nil
 				}
 				threaduids = append(threaduids, t.Uid)
-				builder.threadMap[t.Uid] = t
 				return nil
 			})
 		if inverse {