From 121d703bcd3ecbaba031499070907251b5eae1a9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kier=C3=A1n=20Meinhardt?= <kieran.meinhardt@gmail.com>
Date: Tue, 22 Sep 2020 23:37:06 +0200
Subject: Action: add openFold and closeFold

---
 Action.hs | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/Action.hs b/Action.hs
index 08c30a4..7ca023d 100644
--- a/Action.hs
+++ b/Action.hs
@@ -134,18 +134,26 @@ moveCursorDownToNextUnread =
     moveCursorToUnread findNext botOverrun moveTreeUp
 
 
-toggleFold :: State -> IO State
-toggleFold q@State{..} =
-    getNewSubForest >>= return . \case
+openFold :: State -> IO State
+openFold q@State{..} =
+    handle <$> loadSubForest (Z.label cursor)
+  where
+    handle = \case
         Left err ->
             q { flashMessage = SGR [31] $ Plain err }
         Right sf ->
             q { cursor = Z.modifyTree (setSubForest sf) cursor }
-  where
-    getNewSubForest =
-        if hasUnloadedSubForest (Z.tree cursor)
-            then loadSubForest (Z.label cursor)
-            else return $ Right $ unloadSubForest (Z.tree cursor)
+
+closeFold :: State -> IO State
+closeFold q@State{..} =
+    let sf = unloadSubForest (Z.tree cursor)
+     in return q { cursor = Z.modifyTree (setSubForest sf) cursor }
+
+toggleFold :: State -> IO State
+toggleFold q@State{..} =
+    if hasUnloadedSubForest (Z.tree cursor)
+        then openFold q
+        else closeFold q
 
 
 toggleTagAtCursor :: Tag -> State -> IO State
-- 
cgit v1.2.3