From af33a5ac3dfe2e23ca6f84bd6ac670c9086777da Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Sun, 13 Sep 2015 14:34:46 -0700
Subject: [PATCH] ui: f key toggles flat/tree mode

---
 hledger-ui/Hledger/UI/AccountsScreen.hs | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/hledger-ui/Hledger/UI/AccountsScreen.hs b/hledger-ui/Hledger/UI/AccountsScreen.hs
index 2b0482c51..f87639404 100644
--- a/hledger-ui/Hledger/UI/AccountsScreen.hs
+++ b/hledger-ui/Hledger/UI/AccountsScreen.hs
@@ -121,6 +121,7 @@ drawAccountsScreen _st@AppState{aopts=uopts, ajournal=j, aScreen=AccountsScreen{
       bottomlabel = borderKeysStr [
          -- ("up/down/pgup/pgdown/home/end", "move")
          ("-+=1234567890", "adjust/clear depth limit")
+        ,("f", "flat/tree mode")
         ,("right/enter", "show transactions")
         ,("g", "reload")
         ,("q", "quit")
@@ -195,7 +196,7 @@ drawAccountsItem (acctwidth, balwidth) selected (indent, _fullacct, displayacct,
 handleAccountsScreen :: AppState -> Vty.Event -> EventM (Next AppState)
 handleAccountsScreen st@AppState{
   aScreen=scr@AccountsScreen{asState=l}
-  ,aopts=UIOpts{cliopts_=copts}
+  ,aopts=UIOpts{cliopts_=copts@CliOpts{reportopts_=ropts}}
   ,ajournal=j
   } e = do
     d <- liftIO getCurrentDay
@@ -231,6 +232,15 @@ handleAccountsScreen st@AppState{
         Vty.EvKey (Vty.KChar '8') [] -> reload $ setDepth 8 st
         Vty.EvKey (Vty.KChar '9') [] -> reload $ setDepth 9 st
         Vty.EvKey (Vty.KChar '0') [] -> reload $ setDepth 0 st
+        Vty.EvKey (Vty.KChar 'f') [] -> reload $ st'
+          where
+            st' = st{
+              aopts=(aopts st){
+                 cliopts_=copts{
+                    reportopts_=toggleFlatMode ropts
+                    }
+                 }
+              }
         Vty.EvKey (Vty.KLeft) []     -> continue $ popScreen st
         Vty.EvKey (k) [] | k `elem` [Vty.KRight, Vty.KEnter] -> do
           let st' = screenEnter d RS.screen{rsAcct=acct} st
@@ -247,6 +257,11 @@ handleAccountsScreen st@AppState{
                                  -- continue =<< handleEventLensed st someLens ev
 handleAccountsScreen _ _ = error "event handler called with wrong screen type, should not happen"
 
+-- | Toggle between flat and tree mode. If in the third "default" mode, go to flat mode.
+toggleFlatMode :: ReportOpts -> ReportOpts
+toggleFlatMode ropts@ReportOpts{accountlistmode_=ALFlat} = ropts{accountlistmode_=ALTree}
+toggleFlatMode ropts = ropts{accountlistmode_=ALFlat}
+
 -- | Get the maximum account depth in the current journal.
 maxDepth :: AppState -> Int
 maxDepth AppState{ajournal=j} = maximum $ map accountNameLevel $ journalAccountNames j