From ebc402288826774de3a8c226c279cd115bcd08ca Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Thu, 3 Sep 2015 20:51:05 -0700
Subject: [PATCH] ui: after exiting --register, select that account

---
 hledger-ui/Hledger/UI/AccountsScreen.hs | 4 +++-
 hledger-ui/Hledger/UI/Main.hs           | 5 ++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/hledger-ui/Hledger/UI/AccountsScreen.hs b/hledger-ui/Hledger/UI/AccountsScreen.hs
index 7a34ac186..05626c34e 100644
--- a/hledger-ui/Hledger/UI/AccountsScreen.hs
+++ b/hledger-ui/Hledger/UI/AccountsScreen.hs
@@ -4,7 +4,9 @@
 {-# LANGUAGE RecordWildCards #-}
 
 module Hledger.UI.AccountsScreen
- (screen)
+ (screen
+ ,initAccountsScreen
+ )
 where
 
 import Control.Lens ((^.))
diff --git a/hledger-ui/Hledger/UI/Main.hs b/hledger-ui/Hledger/UI/Main.hs
index 195a546d7..409c1157d 100644
--- a/hledger-ui/Hledger/UI/Main.hs
+++ b/hledger-ui/Hledger/UI/Main.hs
@@ -101,13 +101,12 @@ runBrickUi uopts@UIOpts{cliopts_=copts@CliOpts{reportopts_=ropts}} j = do
                  $ filter (regexMatches apat) $ journalAccountNames j
           -- Initialising the accounts screen is awkward, requiring
           -- another temporary AppState value..
-          ascr = AS.screen
           ascr' = aScreen $
-                  (sInitFn ascr) d
+                  AS.initAccountsScreen (Just acct) d  -- acct will be selected
                   AppState{
                     aopts=uopts'
                    ,ajournal=j
-                   ,aScreen=ascr
+                   ,aScreen=AS.screen
                    ,aPrevScreens=[]
                    }