diff --git a/hledger-ui/Hledger/UI/AccountsScreen.hs b/hledger-ui/Hledger/UI/AccountsScreen.hs index 7cc7cb782..5678bc10f 100644 --- a/hledger-ui/Hledger/UI/AccountsScreen.hs +++ b/hledger-ui/Hledger/UI/AccountsScreen.hs @@ -104,7 +104,7 @@ drawAccountsItem fmt _sel item = render $ str $ showitem item handleAccountsScreen :: AppState -> Vty.Event -> EventM (Next AppState) -handleAccountsScreen st@AppState{aScreen=scr@AccountsScreen{asState=is}} e = do +handleAccountsScreen st@AppState{aargs=args, aScreen=scr@AccountsScreen{asState=is}} e = do d <- liftIO getCurrentDay -- c <- getContext -- let h = c^.availHeightL @@ -114,13 +114,13 @@ handleAccountsScreen st@AppState{aScreen=scr@AccountsScreen{asState=is}} e = do Vty.EvKey (Vty.KChar 'q') [] -> halt st Vty.EvKey (Vty.KLeft) [] -> continue $ popScreen st Vty.EvKey (Vty.KRight) [] -> do - let st' = screenEnter d args RS2.screen st + let st' = screenEnter d args RS2.screen{rs2Acct=acct} st vScrollToBeginning $ viewportScroll "register" continue st' where - args = case listSelectedElement is of - Just (_, ((acct, _, _), _)) -> ["acct:"++accountNameToAccountRegex acct] - Nothing -> [] + acct = case listSelectedElement is of + Just (_, ((a, _, _), _)) -> a + Nothing -> "" -- Vty.EvKey (Vty.KPageDown) [] -> continue $ st{aScreen=scr{asState=moveSel h is}} -- Vty.EvKey (Vty.KPageUp) [] -> continue $ st{aScreen=scr{asState=moveSel (-h) is}} diff --git a/hledger-ui/Hledger/UI/RegisterScreen2.hs b/hledger-ui/Hledger/UI/RegisterScreen2.hs index 0d4454812..e3d8be2ad 100644 --- a/hledger-ui/Hledger/UI/RegisterScreen2.hs +++ b/hledger-ui/Hledger/UI/RegisterScreen2.hs @@ -8,8 +8,9 @@ where import Control.Lens ((^.)) -- import Control.Monad.IO.Class (liftIO) --- import Data.List +import Data.List import Data.List.Split (splitOn) +import Data.Monoid -- import Data.Maybe import Data.Time.Calendar (Day) import qualified Data.Vector as V @@ -30,13 +31,14 @@ import Hledger.UI.UIUtils screen = RegisterScreen2{ rs2State = list "register" V.empty 1 + ,rs2Acct = "" ,sInitFn = initRegisterScreen2 ,sDrawFn = drawRegisterScreen2 ,sHandleFn = handleRegisterScreen2 } initRegisterScreen2 :: Day -> [String] -> AppState -> AppState -initRegisterScreen2 d args st@AppState{aopts=opts, ajournal=j, aScreen=s@RegisterScreen2{}} = +initRegisterScreen2 d args st@AppState{aopts=opts, ajournal=j, aScreen=s@RegisterScreen2{rs2Acct=acct}} = st{aScreen=s{rs2State=l}} where -- gather arguments and queries @@ -46,14 +48,13 @@ initRegisterScreen2 d args st@AppState{aopts=opts, ajournal=j, aScreen=s@Registe balancetype_=HistoricalBalance } -- XXX temp - curacct = drop 5 $ head args -- should be "acct:..." - thisacctq = Acct $ curacct -- XXX why is this excluding subs: accountNameToAccountRegex curacct + thisacctq = Acct $ accountNameToAccountRegex acct -- XXX why is this excluding subs: accountNameToAccountRegex curacct q = queryFromOpts d ropts -- query_="cur:\\$"} -- XXX limit to one commodity to ensure one-line items --{query_=unwords' $ locArgs l} -- run a transactions report, most recent last - (_label,items') = accountTransactionsReport ropts j thisacctq q + (_label,items') = accountTransactionsReport ropts j q thisacctq items = reverse items' -- pre-render all items; these will be the List elements. This helps calculate column widths. @@ -78,13 +79,18 @@ initRegisterScreen2 d args st@AppState{aopts=opts, ajournal=j, aScreen=s@Registe initRegisterScreen2 _ _ _ = error "init function called with wrong screen type, should not happen" drawRegisterScreen2 :: AppState -> [Widget] -drawRegisterScreen2 AppState{aopts=_opts, aScreen=RegisterScreen2{rs2State=l}} = [ui] +drawRegisterScreen2 AppState{aopts=_uopts@UIOpts{cliopts_=_copts@CliOpts{reportopts_=_ropts@ReportOpts{query_=querystr}}}, + aargs=_args, aScreen=RegisterScreen2{rs2State=l,rs2Acct=acct}} = [ui] where - label = str "Transaction " + label = str "Transactions in " + <+> withAttr ("border" <> "bold") (str acct) + <+> borderQuery querystr + -- <+> str " and subs" + <+> str " (" <+> cur <+> str " of " <+> total - <+> str " to/from this account" -- " <+> str query <+> "and subaccounts" + <+> str ")" cur = str $ case l^.listSelectedL of Nothing -> "-" Just i -> show (i + 1) diff --git a/hledger-ui/Hledger/UI/UITypes.hs b/hledger-ui/Hledger/UI/UITypes.hs index 94c6786b7..adfd25cea 100644 --- a/hledger-ui/Hledger/UI/UITypes.hs +++ b/hledger-ui/Hledger/UI/UITypes.hs @@ -22,7 +22,7 @@ data AppState = AppState { -- | Types of screen available within the app, along with their state. -- Screen types are distinguished by their constructor and by the type --- of their state (hence the unique accessor names for the latter). +-- of their state (which must have unique accessor names). data Screen = AccountsScreen { asState :: List BalanceReportItem -- ^ the screen's state (data being displayed and widget state) @@ -38,6 +38,7 @@ data Screen = } | RegisterScreen2 { rs2State :: List (String,String,String,String,String) + ,rs2Acct :: AccountName -- ^ the account we are showing a register for ,sInitFn :: Day -> [String] -> AppState -> AppState ,sHandleFn :: AppState -> V.Event -> EventM (Next AppState) ,sDrawFn :: AppState -> [Widget]