2008-10-10 14:04:26 +04:00
|
|
|
{-|
|
|
|
|
|
2008-10-12 13:17:21 +04:00
|
|
|
Utilities for top-level modules and/or ghci. See also "Ledger.Utils".
|
2008-10-10 14:04:26 +04:00
|
|
|
|
|
|
|
-}
|
|
|
|
|
|
|
|
module Utils
|
|
|
|
where
|
2008-12-08 04:49:31 +03:00
|
|
|
import Control.Monad.Error
|
2008-10-10 14:04:26 +04:00
|
|
|
import qualified Data.Map as Map (lookup)
|
2009-01-24 21:48:28 +03:00
|
|
|
import Data.Time.Clock
|
2008-11-22 23:22:59 +03:00
|
|
|
import Text.ParserCombinators.Parsec
|
2008-11-29 23:00:21 +03:00
|
|
|
import System.IO
|
2008-10-10 14:04:26 +04:00
|
|
|
import Options
|
|
|
|
import Ledger
|
|
|
|
|
|
|
|
|
2008-11-29 23:00:21 +03:00
|
|
|
-- | Convert a RawLedger to a canonicalised, cached and filtered Ledger
|
2009-01-24 21:48:28 +03:00
|
|
|
-- based on the command-line options/arguments and the current date/time.
|
|
|
|
prepareLedger :: [Opt] -> [String] -> UTCTime -> String -> RawLedger -> Ledger
|
|
|
|
prepareLedger opts args reftime rawtext rl = l{rawledgertext=rawtext}
|
2008-11-29 23:00:21 +03:00
|
|
|
where
|
2008-12-10 10:29:08 +03:00
|
|
|
l = cacheLedger apats $ filterRawLedger span dpats c r $ canonicaliseAmounts cb rl
|
2008-11-29 23:00:21 +03:00
|
|
|
(apats,dpats) = parseAccountDescriptionArgs [] args
|
2009-01-24 21:48:28 +03:00
|
|
|
span = dateSpanFromOpts (utctDay reftime) opts
|
2008-11-29 23:00:21 +03:00
|
|
|
c = Cleared `elem` opts
|
|
|
|
r = Real `elem` opts
|
|
|
|
cb = CostBasis `elem` opts
|
|
|
|
|
2008-11-22 23:22:59 +03:00
|
|
|
-- | Get a RawLedger from the given string, or raise an error.
|
2008-12-08 04:49:31 +03:00
|
|
|
rawledgerfromstring :: String -> IO RawLedger
|
|
|
|
rawledgerfromstring = liftM (either error id) . runErrorT . parseLedger "(string)"
|
2008-11-22 23:22:59 +03:00
|
|
|
|
2008-12-05 11:51:14 +03:00
|
|
|
-- | Get a Ledger from the given string and options, or raise an error.
|
2009-01-24 21:48:28 +03:00
|
|
|
ledgerfromstringwithopts :: [Opt] -> [String] -> UTCTime -> String -> IO Ledger
|
|
|
|
ledgerfromstringwithopts opts args reftime s =
|
|
|
|
liftM (prepareLedger opts args reftime s) $ rawledgerfromstring s
|
2008-10-10 14:04:26 +04:00
|
|
|
|
2008-12-05 11:51:14 +03:00
|
|
|
-- | Get a Ledger from the given file path and options, or raise an error.
|
|
|
|
ledgerfromfilewithopts :: [Opt] -> [String] -> FilePath -> IO Ledger
|
|
|
|
ledgerfromfilewithopts opts args f = do
|
2008-12-10 10:29:08 +03:00
|
|
|
s <- readFile f
|
|
|
|
rl <- rawledgerfromstring s
|
2009-01-24 21:48:28 +03:00
|
|
|
reftime <- now
|
|
|
|
return $ prepareLedger opts args reftime s rl
|
2008-11-22 08:51:48 +03:00
|
|
|
|
2008-12-05 11:51:14 +03:00
|
|
|
-- | Get a Ledger from your default ledger file, or raise an error.
|
|
|
|
-- Assumes no options.
|
2008-10-10 14:04:26 +04:00
|
|
|
myledger :: IO Ledger
|
2008-12-05 11:51:14 +03:00
|
|
|
myledger = ledgerFilePathFromOpts [] >>= ledgerfromfilewithopts [] []
|
2008-12-08 04:49:31 +03:00
|
|
|
|
|
|
|
parseWithCtx :: GenParser Char LedgerFileCtx a -> String -> Either ParseError a
|
|
|
|
parseWithCtx p ts = runParser p emptyCtx "" ts
|