61 lines
1.9 KiB
Haskell
Executable File
61 lines
1.9 KiB
Haskell
Executable File
#!/usr/bin/env stack
|
|
{- stack runghc --verbosity info
|
|
--package hledger
|
|
--package here
|
|
-}
|
|
|
|
{-# LANGUAGE QuasiQuotes #-}
|
|
|
|
import Data.Maybe
|
|
import Data.String.Here
|
|
import Data.Time
|
|
import qualified Data.Text as T
|
|
import Control.Monad
|
|
import Hledger.Cli
|
|
|
|
------------------------------------------------------------------------------
|
|
cmdmode =
|
|
let m = defAddonCommandMode "hledger-prices"
|
|
in m {
|
|
modeHelp = [here|
|
|
Print all prices from the journal.
|
|
|]
|
|
,modeHelpSuffix=lines [here|
|
|
|]
|
|
,modeArgs = ([], Nothing)
|
|
,modeGroupFlags = (modeGroupFlags m) {
|
|
groupUnnamed = [
|
|
flagNone ["costs"] (setboolopt "costs") "print transaction prices from postings instead of market prices"
|
|
]
|
|
}
|
|
}
|
|
------------------------------------------------------------------------------
|
|
|
|
showPrice :: MarketPrice -> String
|
|
showPrice mp = unwords ["P", show $ mpdate mp, T.unpack . quoteCommoditySymbolIfNeeded $ mpcommodity mp, showAmountWithZeroCommodity $ mpamount mp]
|
|
|
|
amountCost :: Day -> Amount -> Maybe MarketPrice
|
|
amountCost d a =
|
|
case aprice a of
|
|
NoPrice -> Nothing
|
|
UnitPrice pa -> Just
|
|
MarketPrice { mpdate = d, mpcommodity = acommodity a, mpamount = pa }
|
|
TotalPrice pa -> Just
|
|
MarketPrice { mpdate = d, mpcommodity = acommodity a, mpamount = pa `divideAmount` abs (aquantity a) }
|
|
|
|
postingCosts :: Posting -> [MarketPrice]
|
|
postingCosts p = mapMaybe (amountCost date) . amounts $ pamount p where
|
|
date = fromMaybe (tdate . fromJust $ ptransaction p) $ pdate p
|
|
|
|
allPostsings :: Journal -> [Posting]
|
|
allPostsings = concatMap tpostings . jtxns
|
|
|
|
main :: IO ()
|
|
main = do
|
|
opts <- getHledgerCliOpts cmdmode
|
|
withJournalDo opts{ ignore_assertions_ = True } $ \_ j -> do
|
|
let cprices = concatMap postingCosts . allPostsings $ j
|
|
printPrices = mapM_ (putStrLn . showPrice)
|
|
when (boolopt "costs" $ rawopts_ opts) $ printPrices cprices
|
|
printPrices $ jmarketprices j
|