{-# LANGUAGE DeriveDataTypeable #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TupleSections #-}
module Hledger.Reports.PostingsReport (
PostingsReport,
PostingsReportItem,
postingsReport,
mkpostingsReportItem,
tests_PostingsReport
)
where
import Data.List
import Data.Maybe
import Data.Ord (comparing)
import qualified Data.Text as T
import Data.Time.Calendar
import Safe (headMay, lastMay)
import Hledger.Data
import Hledger.Query
import Hledger.Utils
import Hledger.Reports.ReportOptions
type PostingsReport = (String
,[PostingsReportItem]
)
type PostingsReportItem = (Maybe Day
,Maybe Day
,Maybe String
,Posting
,MixedAmount
)
type SummaryPosting = (Posting, Day)
postingsReport :: ReportOpts -> Query -> Journal -> PostingsReport
postingsReport :: ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ropts :: ReportOpts
ropts@ReportOpts{..} q :: Query
q j :: Journal
j@Journal{..} =
(String
totallabel, [PostingsReportItem]
items)
where
reportspan :: DateSpan
reportspan = ReportOpts -> Query -> Journal -> DateSpan
adjustReportDates ReportOpts
ropts Query
q Journal
j
whichdate :: WhichDate
whichdate = ReportOpts -> WhichDate
whichDateFromOpts ReportOpts
ropts
depth :: Int
depth = Query -> Int
queryDepth Query
q
styles :: Map Text AmountStyle
styles = Journal -> Map Text AmountStyle
journalCommodityStyles Journal
j
priceoracle :: PriceOracle
priceoracle = Journal -> PriceOracle
journalPriceOracle Journal
j
multiperiod :: Bool
multiperiod = Interval
interval_ Interval -> Interval -> Bool
forall a. Eq a => a -> a -> Bool
/= Interval
NoInterval
today :: Day
today = Day -> Maybe Day -> Day
forall a. a -> Maybe a -> a
fromMaybe (String -> Day
forall a. String -> a
error' "postingsReport: could not pick a valuation date, ReportOpts today_ is unset") Maybe Day
today_
(precedingps :: [Posting]
precedingps, reportps :: [Posting]
reportps) = ReportOpts
-> Query -> Journal -> DateSpan -> ([Posting], [Posting])
matchedPostingsBeforeAndDuring ReportOpts
ropts Query
q Journal
j DateSpan
reportspan
[(Posting, Maybe Day)]
displayps :: [(Posting, Maybe Day)]
| Bool
multiperiod =
let summaryps :: [SummaryPosting]
summaryps = Interval
-> WhichDate
-> Int
-> Bool
-> DateSpan
-> [Posting]
-> [SummaryPosting]
summarisePostingsByInterval Interval
interval_ WhichDate
whichdate Int
depth Bool
showempty DateSpan
reportspan [Posting]
reportps
in [(Posting -> Day -> Posting
pvalue Posting
p Day
lastday, Day -> Maybe Day
forall a. a -> Maybe a
Just Day
periodend) | (p :: Posting
p, periodend :: Day
periodend) <- [SummaryPosting]
summaryps, let lastday :: Day
lastday = Year -> Day -> Day
addDays (-1) Day
periodend]
| Bool
otherwise =
[(Posting -> Day -> Posting
pvalue Posting
p Day
reportorjournallast, Maybe Day
forall a. Maybe a
Nothing) | Posting
p <- [Posting]
reportps]
where
showempty :: Bool
showempty = Bool
empty_ Bool -> Bool -> Bool
|| Bool
average_
pvalue :: Posting -> Day -> Posting
pvalue p :: Posting
p periodlast :: Day
periodlast = Posting
-> (ValuationType -> Posting) -> Maybe ValuationType -> Posting
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Posting
p (PriceOracle
-> Map Text AmountStyle
-> Day
-> Maybe Day
-> Day
-> Bool
-> Posting
-> ValuationType
-> Posting
postingApplyValuation PriceOracle
priceoracle Map Text AmountStyle
styles Day
periodlast Maybe Day
mreportlast Day
today Bool
multiperiod Posting
p) Maybe ValuationType
value_
where
mreportlast :: Maybe Day
mreportlast = ReportOpts -> Maybe Day
reportPeriodLastDay ReportOpts
ropts
reportorjournallast :: Day
reportorjournallast =
Day -> Maybe Day -> Day
forall a. a -> Maybe a -> a
fromMaybe (String -> Day
forall a. String -> a
error' "postingsReport: expected a non-empty journal") (Maybe Day -> Day) -> Maybe Day -> Day
forall a b. (a -> b) -> a -> b
$
ReportOpts -> Journal -> Maybe Day
reportPeriodOrJournalLastDay ReportOpts
ropts Journal
j
items :: [PostingsReportItem]
items =
String -> [PostingsReportItem] -> [PostingsReportItem]
forall a. Show a => String -> a -> a
dbg1 "postingsReport items" ([PostingsReportItem] -> [PostingsReportItem])
-> [PostingsReportItem] -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$
[(Posting, Maybe Day)]
-> (Posting, Maybe Day)
-> WhichDate
-> Int
-> MixedAmount
-> (Int -> MixedAmount -> MixedAmount -> MixedAmount)
-> Int
-> [PostingsReportItem]
postingsReportItems [(Posting, Maybe Day)]
displayps (Posting
nullposting,Maybe Day
forall a. Maybe a
Nothing) WhichDate
whichdate Int
depth MixedAmount
startbal Int -> MixedAmount -> MixedAmount -> MixedAmount
runningcalc Int
startnum
where
historical :: Bool
historical = BalanceType
balancetype_ BalanceType -> BalanceType -> Bool
forall a. Eq a => a -> a -> Bool
== BalanceType
HistoricalBalance
startbal :: MixedAmount
startbal | Bool
average_ = if Bool
historical then MixedAmount -> MixedAmount
bvalue MixedAmount
precedingavg else 0
| Bool
otherwise = if Bool
historical then MixedAmount -> MixedAmount
bvalue MixedAmount
precedingsum else 0
where
precedingsum :: MixedAmount
precedingsum = [Posting] -> MixedAmount
sumPostings [Posting]
precedingps
precedingavg :: MixedAmount
precedingavg | [Posting] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [Posting]
precedingps = 0
| Bool
otherwise = Quantity -> MixedAmount -> MixedAmount
divideMixedAmount (Int -> Quantity
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Int -> Quantity) -> Int -> Quantity
forall a b. (a -> b) -> a -> b
$ [Posting] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Posting]
precedingps) MixedAmount
precedingsum
bvalue :: MixedAmount -> MixedAmount
bvalue = (MixedAmount -> MixedAmount)
-> (ValuationType -> MixedAmount -> MixedAmount)
-> Maybe ValuationType
-> MixedAmount
-> MixedAmount
forall b a. b -> (a -> b) -> Maybe a -> b
maybe MixedAmount -> MixedAmount
forall a. a -> a
id (PriceOracle
-> Map Text AmountStyle
-> Day
-> Maybe Day
-> Day
-> Bool
-> ValuationType
-> MixedAmount
-> MixedAmount
mixedAmountApplyValuation PriceOracle
priceoracle Map Text AmountStyle
styles Day
daybeforereportstart Maybe Day
forall a. Maybe a
Nothing Day
today Bool
multiperiod) Maybe ValuationType
value_
where
daybeforereportstart :: Day
daybeforereportstart =
Day -> (Day -> Day) -> Maybe Day -> Day
forall b a. b -> (a -> b) -> Maybe a -> b
maybe (String -> Day
forall a. String -> a
error' "postingsReport: expected a non-empty journal")
(Year -> Day -> Day
addDays (-1))
(Maybe Day -> Day) -> Maybe Day -> Day
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Journal -> Maybe Day
reportPeriodOrJournalStart ReportOpts
ropts Journal
j
runningcalc :: Int -> MixedAmount -> MixedAmount -> MixedAmount
runningcalc = ReportOpts -> Int -> MixedAmount -> MixedAmount -> MixedAmount
registerRunningCalculationFn ReportOpts
ropts
startnum :: Int
startnum = if Bool
historical then [Posting] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Posting]
precedingps Int -> Int -> Int
forall a. Num a => a -> a -> a
+ 1 else 1
registerRunningCalculationFn :: ReportOpts -> (Int -> MixedAmount -> MixedAmount -> MixedAmount)
registerRunningCalculationFn :: ReportOpts -> Int -> MixedAmount -> MixedAmount -> MixedAmount
registerRunningCalculationFn ropts :: ReportOpts
ropts
| ReportOpts -> Bool
average_ ReportOpts
ropts = \i :: Int
i avg :: MixedAmount
avg amt :: MixedAmount
amt -> MixedAmount
avg MixedAmount -> MixedAmount -> MixedAmount
forall a. Num a => a -> a -> a
+ Quantity -> MixedAmount -> MixedAmount
divideMixedAmount (Int -> Quantity
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
i) (MixedAmount
amt MixedAmount -> MixedAmount -> MixedAmount
forall a. Num a => a -> a -> a
- MixedAmount
avg)
| Bool
otherwise = \_ bal :: MixedAmount
bal amt :: MixedAmount
amt -> MixedAmount
bal MixedAmount -> MixedAmount -> MixedAmount
forall a. Num a => a -> a -> a
+ MixedAmount
amt
totallabel :: String
totallabel = "Total"
adjustReportDates :: ReportOpts -> Query -> Journal -> DateSpan
adjustReportDates :: ReportOpts -> Query -> Journal -> DateSpan
adjustReportDates opts :: ReportOpts
opts q :: Query
q j :: Journal
j = DateSpan
reportspan
where
requestedspan :: DateSpan
requestedspan = String -> DateSpan -> DateSpan
forall a. Show a => String -> a -> a
dbg1 "requestedspan" (DateSpan -> DateSpan) -> DateSpan -> DateSpan
forall a b. (a -> b) -> a -> b
$ Query -> DateSpan
queryDateSpan' Query
q
journalspan :: DateSpan
journalspan = String -> DateSpan -> DateSpan
forall a. Show a => String -> a -> a
dbg1 "journalspan" (DateSpan -> DateSpan) -> DateSpan -> DateSpan
forall a b. (a -> b) -> a -> b
$ DateSpan
dates DateSpan -> DateSpan -> DateSpan
`spanUnion` DateSpan
date2s
where
dates :: DateSpan
dates = Bool -> Journal -> DateSpan
journalDateSpan Bool
False Journal
j
date2s :: DateSpan
date2s = Bool -> Journal -> DateSpan
journalDateSpan Bool
True Journal
j
requestedspanclosed :: DateSpan
requestedspanclosed = String -> DateSpan -> DateSpan
forall a. Show a => String -> a -> a
dbg1 "requestedspanclosed" (DateSpan -> DateSpan) -> DateSpan -> DateSpan
forall a b. (a -> b) -> a -> b
$ DateSpan
requestedspan DateSpan -> DateSpan -> DateSpan
`spanDefaultsFrom` DateSpan
journalspan
intervalspans :: [DateSpan]
intervalspans = String -> [DateSpan] -> [DateSpan]
forall a. Show a => String -> a -> a
dbg1 "intervalspans" ([DateSpan] -> [DateSpan]) -> [DateSpan] -> [DateSpan]
forall a b. (a -> b) -> a -> b
$ Interval -> DateSpan -> [DateSpan]
splitSpan (ReportOpts -> Interval
interval_ ReportOpts
opts) DateSpan
requestedspanclosed
mreportstart :: Maybe Day
mreportstart = String -> Maybe Day -> Maybe Day
forall a. Show a => String -> a -> a
dbg1 "reportstart" (Maybe Day -> Maybe Day) -> Maybe Day -> Maybe Day
forall a b. (a -> b) -> a -> b
$ Maybe Day -> (DateSpan -> Maybe Day) -> Maybe DateSpan -> Maybe Day
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Maybe Day
forall a. Maybe a
Nothing DateSpan -> Maybe Day
spanStart (Maybe DateSpan -> Maybe Day) -> Maybe DateSpan -> Maybe Day
forall a b. (a -> b) -> a -> b
$ [DateSpan] -> Maybe DateSpan
forall a. [a] -> Maybe a
headMay [DateSpan]
intervalspans
mreportend :: Maybe Day
mreportend = String -> Maybe Day -> Maybe Day
forall a. Show a => String -> a -> a
dbg1 "reportend" (Maybe Day -> Maybe Day) -> Maybe Day -> Maybe Day
forall a b. (a -> b) -> a -> b
$ Maybe Day -> (DateSpan -> Maybe Day) -> Maybe DateSpan -> Maybe Day
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Maybe Day
forall a. Maybe a
Nothing DateSpan -> Maybe Day
spanEnd (Maybe DateSpan -> Maybe Day) -> Maybe DateSpan -> Maybe Day
forall a b. (a -> b) -> a -> b
$ [DateSpan] -> Maybe DateSpan
forall a. [a] -> Maybe a
lastMay [DateSpan]
intervalspans
reportspan :: DateSpan
reportspan = String -> DateSpan -> DateSpan
forall a. Show a => String -> a -> a
dbg1 "reportspan" (DateSpan -> DateSpan) -> DateSpan -> DateSpan
forall a b. (a -> b) -> a -> b
$ Maybe Day -> Maybe Day -> DateSpan
DateSpan Maybe Day
mreportstart Maybe Day
mreportend
matchedPostingsBeforeAndDuring :: ReportOpts -> Query -> Journal -> DateSpan -> ([Posting],[Posting])
matchedPostingsBeforeAndDuring :: ReportOpts
-> Query -> Journal -> DateSpan -> ([Posting], [Posting])
matchedPostingsBeforeAndDuring opts :: ReportOpts
opts q :: Query
q j :: Journal
j (DateSpan mstart :: Maybe Day
mstart mend :: Maybe Day
mend) =
String -> ([Posting], [Posting]) -> ([Posting], [Posting])
forall a. Show a => String -> a -> a
dbg1 "beforeps, duringps" (([Posting], [Posting]) -> ([Posting], [Posting]))
-> ([Posting], [Posting]) -> ([Posting], [Posting])
forall a b. (a -> b) -> a -> b
$ (Posting -> Bool) -> [Posting] -> ([Posting], [Posting])
forall a. (a -> Bool) -> [a] -> ([a], [a])
span (Query
beforestartq Query -> Posting -> Bool
`matchesPosting`) [Posting]
beforeandduringps
where
beforestartq :: Query
beforestartq = String -> Query -> Query
forall a. Show a => String -> a -> a
dbg1 "beforestartq" (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ DateSpan -> Query
dateqtype (DateSpan -> Query) -> DateSpan -> Query
forall a b. (a -> b) -> a -> b
$ Maybe Day -> Maybe Day -> DateSpan
DateSpan Maybe Day
forall a. Maybe a
Nothing Maybe Day
mstart
beforeandduringps :: [Posting]
beforeandduringps =
String -> [Posting] -> [Posting]
forall a. Show a => String -> a -> a
dbg1 "ps5" ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$ (Posting -> Posting -> Ordering) -> [Posting] -> [Posting]
forall a. (a -> a -> Ordering) -> [a] -> [a]
sortBy ((Posting -> Day) -> Posting -> Posting -> Ordering
forall a b. Ord a => (b -> a) -> b -> b -> Ordering
comparing Posting -> Day
sortdate) ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$
String -> [Posting] -> [Posting]
forall a. Show a => String -> a -> a
dbg1 "ps4" ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$ (if ReportOpts -> Bool
invert_ ReportOpts
opts then (Posting -> Posting) -> [Posting] -> [Posting]
forall a b. (a -> b) -> [a] -> [b]
map Posting -> Posting
negatePostingAmount else [Posting] -> [Posting]
forall a. a -> a
id) ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$
String -> [Posting] -> [Posting]
forall a. Show a => String -> a -> a
dbg1 "ps3" ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$ (Posting -> Posting) -> [Posting] -> [Posting]
forall a b. (a -> b) -> [a] -> [b]
map (Query -> Posting -> Posting
filterPostingAmount Query
symq) ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$
String -> [Posting] -> [Posting]
forall a. Show a => String -> a -> a
dbg1 "ps2" ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$ (if ReportOpts -> Bool
related_ ReportOpts
opts then (Posting -> [Posting]) -> [Posting] -> [Posting]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap Posting -> [Posting]
relatedPostings else [Posting] -> [Posting]
forall a. a -> a
id) ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$
String -> [Posting] -> [Posting]
forall a. Show a => String -> a -> a
dbg1 "ps1" ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$ (Posting -> Bool) -> [Posting] -> [Posting]
forall a. (a -> Bool) -> [a] -> [a]
filter (Query
beforeandduringq Query -> Posting -> Bool
`matchesPosting`) ([Posting] -> [Posting]) -> [Posting] -> [Posting]
forall a b. (a -> b) -> a -> b
$
Journal -> [Posting]
journalPostings (Journal -> [Posting]) -> Journal -> [Posting]
forall a b. (a -> b) -> a -> b
$
ReportOpts -> Journal -> Journal
journalSelectingAmountFromOpts ReportOpts
opts Journal
j
where
beforeandduringq :: Query
beforeandduringq = String -> Query -> Query
forall a. Show a => String -> a -> a
dbg1 "beforeandduringq" (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ [Query] -> Query
And [Query -> Query
depthless (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ Query -> Query
dateless Query
q, Query
beforeendq]
where
depthless :: Query -> Query
depthless = (Query -> Bool) -> Query -> Query
filterQuery (Bool -> Bool
not (Bool -> Bool) -> (Query -> Bool) -> Query -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Query -> Bool
queryIsDepth)
dateless :: Query -> Query
dateless = (Query -> Bool) -> Query -> Query
filterQuery (Bool -> Bool
not (Bool -> Bool) -> (Query -> Bool) -> Query -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Query -> Bool
queryIsDateOrDate2)
beforeendq :: Query
beforeendq = DateSpan -> Query
dateqtype (DateSpan -> Query) -> DateSpan -> Query
forall a b. (a -> b) -> a -> b
$ Maybe Day -> Maybe Day -> DateSpan
DateSpan Maybe Day
forall a. Maybe a
Nothing Maybe Day
mend
sortdate :: Posting -> Day
sortdate = if ReportOpts -> Bool
date2_ ReportOpts
opts then Posting -> Day
postingDate2 else Posting -> Day
postingDate
symq :: Query
symq = String -> Query -> Query
forall a. Show a => String -> a -> a
dbg1 "symq" (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ (Query -> Bool) -> Query -> Query
filterQuery Query -> Bool
queryIsSym Query
q
dateqtype :: DateSpan -> Query
dateqtype
| Query -> Bool
queryIsDate2 Query
dateq Bool -> Bool -> Bool
|| (Query -> Bool
queryIsDate Query
dateq Bool -> Bool -> Bool
&& ReportOpts -> Bool
date2_ ReportOpts
opts) = DateSpan -> Query
Date2
| Bool
otherwise = DateSpan -> Query
Date
where
dateq :: Query
dateq = String -> Query -> Query
forall a. Show a => String -> a -> a
dbg1 "dateq" (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ (Query -> Bool) -> Query -> Query
filterQuery Query -> Bool
queryIsDateOrDate2 (Query -> Query) -> Query -> Query
forall a b. (a -> b) -> a -> b
$ String -> Query -> Query
forall a. Show a => String -> a -> a
dbg1 "q" Query
q
postingsReportItems :: [(Posting,Maybe Day)] -> (Posting,Maybe Day) -> WhichDate -> Int -> MixedAmount -> (Int -> MixedAmount -> MixedAmount -> MixedAmount) -> Int -> [PostingsReportItem]
postingsReportItems :: [(Posting, Maybe Day)]
-> (Posting, Maybe Day)
-> WhichDate
-> Int
-> MixedAmount
-> (Int -> MixedAmount -> MixedAmount -> MixedAmount)
-> Int
-> [PostingsReportItem]
postingsReportItems [] _ _ _ _ _ _ = []
postingsReportItems ((p :: Posting
p,menddate :: Maybe Day
menddate):ps :: [(Posting, Maybe Day)]
ps) (pprev :: Posting
pprev,menddateprev :: Maybe Day
menddateprev) wd :: WhichDate
wd d :: Int
d b :: MixedAmount
b runningcalcfn :: Int -> MixedAmount -> MixedAmount -> MixedAmount
runningcalcfn itemnum :: Int
itemnum = PostingsReportItem
iPostingsReportItem -> [PostingsReportItem] -> [PostingsReportItem]
forall a. a -> [a] -> [a]
:([(Posting, Maybe Day)]
-> (Posting, Maybe Day)
-> WhichDate
-> Int
-> MixedAmount
-> (Int -> MixedAmount -> MixedAmount -> MixedAmount)
-> Int
-> [PostingsReportItem]
postingsReportItems [(Posting, Maybe Day)]
ps (Posting
p,Maybe Day
menddate) WhichDate
wd Int
d MixedAmount
b' Int -> MixedAmount -> MixedAmount -> MixedAmount
runningcalcfn (Int
itemnumInt -> Int -> Int
forall a. Num a => a -> a -> a
+1))
where
i :: PostingsReportItem
i = Bool
-> Bool
-> WhichDate
-> Maybe Day
-> Posting
-> MixedAmount
-> PostingsReportItem
mkpostingsReportItem Bool
showdate Bool
showdesc WhichDate
wd Maybe Day
menddate Posting
p' MixedAmount
b'
(showdate :: Bool
showdate, showdesc :: Bool
showdesc) | Maybe Day -> Bool
forall a. Maybe a -> Bool
isJust Maybe Day
menddate = (Maybe Day
menddate Maybe Day -> Maybe Day -> Bool
forall a. Eq a => a -> a -> Bool
/= Maybe Day
menddateprev, Bool
False)
| Bool
otherwise = (Bool
isfirstintxn Bool -> Bool -> Bool
|| Bool
isdifferentdate, Bool
isfirstintxn)
isfirstintxn :: Bool
isfirstintxn = Posting -> Maybe Transaction
ptransaction Posting
p Maybe Transaction -> Maybe Transaction -> Bool
forall a. Eq a => a -> a -> Bool
/= Posting -> Maybe Transaction
ptransaction Posting
pprev
isdifferentdate :: Bool
isdifferentdate = case WhichDate
wd of PrimaryDate -> Posting -> Day
postingDate Posting
p Day -> Day -> Bool
forall a. Eq a => a -> a -> Bool
/= Posting -> Day
postingDate Posting
pprev
SecondaryDate -> Posting -> Day
postingDate2 Posting
p Day -> Day -> Bool
forall a. Eq a => a -> a -> Bool
/= Posting -> Day
postingDate2 Posting
pprev
p' :: Posting
p' = Posting
p{paccount :: Text
paccount= Int -> Text -> Text
clipOrEllipsifyAccountName Int
d (Text -> Text) -> Text -> Text
forall a b. (a -> b) -> a -> b
$ Posting -> Text
paccount Posting
p}
b' :: MixedAmount
b' = Int -> MixedAmount -> MixedAmount -> MixedAmount
runningcalcfn Int
itemnum MixedAmount
b (Posting -> MixedAmount
pamount Posting
p)
mkpostingsReportItem :: Bool -> Bool -> WhichDate -> Maybe Day -> Posting -> MixedAmount -> PostingsReportItem
mkpostingsReportItem :: Bool
-> Bool
-> WhichDate
-> Maybe Day
-> Posting
-> MixedAmount
-> PostingsReportItem
mkpostingsReportItem showdate :: Bool
showdate showdesc :: Bool
showdesc wd :: WhichDate
wd menddate :: Maybe Day
menddate p :: Posting
p b :: MixedAmount
b =
(if Bool
showdate then Day -> Maybe Day
forall a. a -> Maybe a
Just Day
date else Maybe Day
forall a. Maybe a
Nothing
,Maybe Day
menddate
,if Bool
showdesc then String -> Maybe String
forall a. a -> Maybe a
Just String
desc else Maybe String
forall a. Maybe a
Nothing
,Posting
p
,MixedAmount
b
)
where
date :: Day
date = case WhichDate
wd of PrimaryDate -> Posting -> Day
postingDate Posting
p
SecondaryDate -> Posting -> Day
postingDate2 Posting
p
desc :: String
desc = Text -> String
T.unpack (Text -> String) -> Text -> String
forall a b. (a -> b) -> a -> b
$ Text -> (Transaction -> Text) -> Maybe Transaction -> Text
forall b a. b -> (a -> b) -> Maybe a -> b
maybe "" Transaction -> Text
tdescription (Maybe Transaction -> Text) -> Maybe Transaction -> Text
forall a b. (a -> b) -> a -> b
$ Posting -> Maybe Transaction
ptransaction Posting
p
summarisePostingsByInterval :: Interval -> WhichDate -> Int -> Bool -> DateSpan -> [Posting] -> [SummaryPosting]
summarisePostingsByInterval :: Interval
-> WhichDate
-> Int
-> Bool
-> DateSpan
-> [Posting]
-> [SummaryPosting]
summarisePostingsByInterval interval :: Interval
interval wd :: WhichDate
wd depth :: Int
depth showempty :: Bool
showempty reportspan :: DateSpan
reportspan ps :: [Posting]
ps = (DateSpan -> [SummaryPosting]) -> [DateSpan] -> [SummaryPosting]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap DateSpan -> [SummaryPosting]
summarisespan ([DateSpan] -> [SummaryPosting]) -> [DateSpan] -> [SummaryPosting]
forall a b. (a -> b) -> a -> b
$ Interval -> DateSpan -> [DateSpan]
splitSpan Interval
interval DateSpan
reportspan
where
summarisespan :: DateSpan -> [SummaryPosting]
summarisespan s :: DateSpan
s = DateSpan
-> WhichDate -> Int -> Bool -> [Posting] -> [SummaryPosting]
summarisePostingsInDateSpan DateSpan
s WhichDate
wd Int
depth Bool
showempty (DateSpan -> [Posting]
postingsinspan DateSpan
s)
postingsinspan :: DateSpan -> [Posting]
postingsinspan s :: DateSpan
s = (Posting -> Bool) -> [Posting] -> [Posting]
forall a. (a -> Bool) -> [a] -> [a]
filter (WhichDate -> DateSpan -> Posting -> Bool
isPostingInDateSpan' WhichDate
wd DateSpan
s) [Posting]
ps
summarisePostingsInDateSpan :: DateSpan -> WhichDate -> Int -> Bool -> [Posting] -> [SummaryPosting]
summarisePostingsInDateSpan :: DateSpan
-> WhichDate -> Int -> Bool -> [Posting] -> [SummaryPosting]
summarisePostingsInDateSpan (DateSpan b :: Maybe Day
b e :: Maybe Day
e) wd :: WhichDate
wd depth :: Int
depth showempty :: Bool
showempty ps :: [Posting]
ps
| [Posting] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [Posting]
ps Bool -> Bool -> Bool
&& (Maybe Day -> Bool
forall a. Maybe a -> Bool
isNothing Maybe Day
b Bool -> Bool -> Bool
|| Maybe Day -> Bool
forall a. Maybe a -> Bool
isNothing Maybe Day
e) = []
| [Posting] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [Posting]
ps Bool -> Bool -> Bool
&& Bool
showempty = [(Posting
summaryp, Day
e')]
| Bool
otherwise = [SummaryPosting]
summarypes
where
postingdate :: Posting -> Day
postingdate = if WhichDate
wd WhichDate -> WhichDate -> Bool
forall a. Eq a => a -> a -> Bool
== WhichDate
PrimaryDate then Posting -> Day
postingDate else Posting -> Day
postingDate2
b' :: Day
b' = Day -> Maybe Day -> Day
forall a. a -> Maybe a -> a
fromMaybe (Day -> (Posting -> Day) -> Maybe Posting -> Day
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Day
nulldate Posting -> Day
postingdate (Maybe Posting -> Day) -> Maybe Posting -> Day
forall a b. (a -> b) -> a -> b
$ [Posting] -> Maybe Posting
forall a. [a] -> Maybe a
headMay [Posting]
ps) Maybe Day
b
e' :: Day
e' = Day -> Maybe Day -> Day
forall a. a -> Maybe a -> a
fromMaybe (Day -> (Posting -> Day) -> Maybe Posting -> Day
forall b a. b -> (a -> b) -> Maybe a -> b
maybe (Year -> Day -> Day
addDays 1 Day
nulldate) Posting -> Day
postingdate (Maybe Posting -> Day) -> Maybe Posting -> Day
forall a b. (a -> b) -> a -> b
$ [Posting] -> Maybe Posting
forall a. [a] -> Maybe a
lastMay [Posting]
ps) Maybe Day
e
summaryp :: Posting
summaryp = Posting
nullposting{pdate :: Maybe Day
pdate=Day -> Maybe Day
forall a. a -> Maybe a
Just Day
b'}
clippedanames :: [Text]
clippedanames | Int
depth Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> 0 = [Text] -> [Text]
forall a. Eq a => [a] -> [a]
nub ([Text] -> [Text]) -> [Text] -> [Text]
forall a b. (a -> b) -> a -> b
$ (Text -> Text) -> [Text] -> [Text]
forall a b. (a -> b) -> [a] -> [b]
map (Int -> Text -> Text
clipAccountName Int
depth) [Text]
anames
| Bool
otherwise = ["..."]
summaryps :: [Posting]
summaryps | Int
depth Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> 0 = [Posting
summaryp{paccount :: Text
paccount=Text
a,pamount :: MixedAmount
pamount=Text -> MixedAmount
balance Text
a} | Text
a <- [Text]
clippedanames]
| Bool
otherwise = [Posting
summaryp{paccount :: Text
paccount="...",pamount :: MixedAmount
pamount=[MixedAmount] -> MixedAmount
forall (t :: * -> *) a. (Foldable t, Num a) => t a -> a
sum ([MixedAmount] -> MixedAmount) -> [MixedAmount] -> MixedAmount
forall a b. (a -> b) -> a -> b
$ (Posting -> MixedAmount) -> [Posting] -> [MixedAmount]
forall a b. (a -> b) -> [a] -> [b]
map Posting -> MixedAmount
pamount [Posting]
ps}]
summarypes :: [SummaryPosting]
summarypes = (Posting -> SummaryPosting) -> [Posting] -> [SummaryPosting]
forall a b. (a -> b) -> [a] -> [b]
map (, Day
e') ([Posting] -> [SummaryPosting]) -> [Posting] -> [SummaryPosting]
forall a b. (a -> b) -> a -> b
$ (if Bool
showempty then [Posting] -> [Posting]
forall a. a -> a
id else (Posting -> Bool) -> [Posting] -> [Posting]
forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not (Bool -> Bool) -> (Posting -> Bool) -> Posting -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. MixedAmount -> Bool
isZeroMixedAmount (MixedAmount -> Bool)
-> (Posting -> MixedAmount) -> Posting -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Posting -> MixedAmount
pamount)) [Posting]
summaryps
anames :: [Text]
anames = [Text] -> [Text]
forall a. Ord a => [a] -> [a]
sort ([Text] -> [Text]) -> [Text] -> [Text]
forall a b. (a -> b) -> a -> b
$ [Text] -> [Text]
forall a. Eq a => [a] -> [a]
nub ([Text] -> [Text]) -> [Text] -> [Text]
forall a b. (a -> b) -> a -> b
$ (Posting -> Text) -> [Posting] -> [Text]
forall a b. (a -> b) -> [a] -> [b]
map Posting -> Text
paccount [Posting]
ps
accts :: [Account]
accts = [Posting] -> [Account]
accountsFromPostings [Posting]
ps
balance :: Text -> MixedAmount
balance a :: Text
a = MixedAmount
-> (Account -> MixedAmount) -> Maybe Account -> MixedAmount
forall b a. b -> (a -> b) -> Maybe a -> b
maybe MixedAmount
nullmixedamt Account -> MixedAmount
bal (Maybe Account -> MixedAmount) -> Maybe Account -> MixedAmount
forall a b. (a -> b) -> a -> b
$ Text -> [Account] -> Maybe Account
lookupAccount Text
a [Account]
accts
where
bal :: Account -> MixedAmount
bal = if Text -> Bool
isclipped Text
a then Account -> MixedAmount
aibalance else Account -> MixedAmount
aebalance
isclipped :: Text -> Bool
isclipped a :: Text
a = Text -> Int
accountNameLevel Text
a Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
>= Int
depth
negatePostingAmount :: Posting -> Posting
negatePostingAmount :: Posting -> Posting
negatePostingAmount p :: Posting
p = Posting
p { pamount :: MixedAmount
pamount = MixedAmount -> MixedAmount
forall a. Num a => a -> a
negate (MixedAmount -> MixedAmount) -> MixedAmount -> MixedAmount
forall a b. (a -> b) -> a -> b
$ Posting -> MixedAmount
pamount Posting
p }
tests_PostingsReport :: TestTree
tests_PostingsReport = String -> [TestTree] -> TestTree
tests "PostingsReport" [
String -> Assertion -> TestTree
test "postingsReport" (Assertion -> TestTree) -> Assertion -> TestTree
forall a b. (a -> b) -> a -> b
$ do
let (query :: Query
query, journal :: Journal
journal) gives :: (Query, Journal) -> Int -> Assertion
`gives` n :: Int
n = ([PostingsReportItem] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([PostingsReportItem] -> Int) -> [PostingsReportItem] -> Int
forall a b. (a -> b) -> a -> b
$ PostingsReport -> [PostingsReportItem]
forall a b. (a, b) -> b
snd (PostingsReport -> [PostingsReportItem])
-> PostingsReport -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ReportOpts
defreportopts Query
query Journal
journal) Int -> Int -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= Int
n
(Query
Any, Journal
nulljournal) (Query, Journal) -> Int -> Assertion
`gives` 0
(Query
Any, Journal
samplejournal) (Query, Journal) -> Int -> Assertion
`gives` 13
(Int -> Query
Depth 2, Journal
samplejournal) (Query, Journal) -> Int -> Assertion
`gives` 13
([Query] -> Query
And [Int -> Query
Depth 1, Status -> Query
StatusQ Status
Cleared, String -> Query
Acct "expenses"], Journal
samplejournal) (Query, Journal) -> Int -> Assertion
`gives` 2
([Query] -> Query
And [[Query] -> Query
And [Int -> Query
Depth 1, Status -> Query
StatusQ Status
Cleared], String -> Query
Acct "expenses"], Journal
samplejournal) (Query, Journal) -> Int -> Assertion
`gives` 2
([PostingsReportItem] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([PostingsReportItem] -> Int) -> [PostingsReportItem] -> Int
forall a b. (a -> b) -> a -> b
$ PostingsReport -> [PostingsReportItem]
forall a b. (a, b) -> b
snd (PostingsReport -> [PostingsReportItem])
-> PostingsReport -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ReportOpts
defreportopts Query
Any Journal
samplejournal) Int -> Int -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= 13
([PostingsReportItem] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([PostingsReportItem] -> Int) -> [PostingsReportItem] -> Int
forall a b. (a -> b) -> a -> b
$ PostingsReport -> [PostingsReportItem]
forall a b. (a, b) -> b
snd (PostingsReport -> [PostingsReportItem])
-> PostingsReport -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ReportOpts
defreportopts{interval_ :: Interval
interval_=Int -> Interval
Months 1} Query
Any Journal
samplejournal) Int -> Int -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= 11
([PostingsReportItem] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([PostingsReportItem] -> Int) -> [PostingsReportItem] -> Int
forall a b. (a -> b) -> a -> b
$ PostingsReport -> [PostingsReportItem]
forall a b. (a, b) -> b
snd (PostingsReport -> [PostingsReportItem])
-> PostingsReport -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ReportOpts
defreportopts{interval_ :: Interval
interval_=Int -> Interval
Months 1, empty_ :: Bool
empty_=Bool
True} Query
Any Journal
samplejournal) Int -> Int -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= 20
([PostingsReportItem] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length ([PostingsReportItem] -> Int) -> [PostingsReportItem] -> Int
forall a b. (a -> b) -> a -> b
$ PostingsReport -> [PostingsReportItem]
forall a b. (a, b) -> b
snd (PostingsReport -> [PostingsReportItem])
-> PostingsReport -> [PostingsReportItem]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Query -> Journal -> PostingsReport
postingsReport ReportOpts
defreportopts (String -> Query
Acct "assets:bank:checking") Journal
samplejournal) Int -> Int -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= 5
,String -> Assertion -> TestTree
test "summarisePostingsByInterval" (Assertion -> TestTree) -> Assertion -> TestTree
forall a b. (a -> b) -> a -> b
$
Interval
-> WhichDate
-> Int
-> Bool
-> DateSpan
-> [Posting]
-> [SummaryPosting]
summarisePostingsByInterval (Int -> Interval
Quarters 1) WhichDate
PrimaryDate 99999 Bool
False (Maybe Day -> Maybe Day -> DateSpan
DateSpan Maybe Day
forall a. Maybe a
Nothing Maybe Day
forall a. Maybe a
Nothing) [] [SummaryPosting] -> [SummaryPosting] -> Assertion
forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= []
]