mirror of
https://github.com/simonmichael/hledger.git
synced 2024-09-19 02:08:01 +03:00
95 lines
2.7 KiB
Haskell
95 lines
2.7 KiB
Haskell
#!/usr/bin/env runhaskell
|
|
{-
|
|
hledger - ledger-compatible money management tool (& haskell study)
|
|
GPLv3, (c) Simon Michael & contributors
|
|
A port of John Wiegley's ledger at http://newartisans.com/ledger.html
|
|
|
|
See Types.hs for a code overview.
|
|
-}
|
|
|
|
module Main
|
|
where
|
|
import System
|
|
import Text.ParserCombinators.Parsec (ParseError)
|
|
|
|
import Options
|
|
import Models
|
|
import Parse
|
|
import Tests
|
|
import Utils hiding (test)
|
|
|
|
|
|
main :: IO ()
|
|
main = do
|
|
(opts, (cmd:args)) <- getArgs >>= parseOptions
|
|
let pats = parsePatternArgs args
|
|
run cmd opts pats
|
|
where run cmd opts pats
|
|
| Help `elem` opts = putStr usage
|
|
| cmd `isPrefixOf` "test" = test opts pats
|
|
| cmd `isPrefixOf` "print" = printcmd opts pats
|
|
| cmd `isPrefixOf` "register" = register opts pats
|
|
| cmd `isPrefixOf` "balance" = balance opts pats
|
|
| otherwise = putStr usage
|
|
|
|
doWithFilteredLedger :: [Flag] -> FilterPatterns -> (Ledger -> IO ()) -> IO ()
|
|
doWithFilteredLedger opts pats cmd = do
|
|
ledgerFilePath opts >>= parseLedgerFile >>= doWithParsed pats cmd
|
|
|
|
doWithParsed :: FilterPatterns -> (Ledger -> IO ()) -> (Either ParseError LedgerFile) -> IO ()
|
|
doWithParsed pats cmd parsed = do
|
|
case parsed of Left e -> parseError e
|
|
Right l -> cmd $ cacheLedger pats l
|
|
|
|
type Command = [Flag] -> FilterPatterns -> IO ()
|
|
|
|
test :: Command
|
|
test opts pats = do
|
|
Tests.hunit
|
|
Tests.quickcheck
|
|
return ()
|
|
|
|
printcmd :: Command
|
|
printcmd opts pats = do
|
|
doWithFilteredLedger opts pats printentries
|
|
where
|
|
printentries l = putStr $ showEntries $ setprecision $ entries $ rawledger l
|
|
where
|
|
setprecision = map (entrySetPrecision (lprecision l))
|
|
|
|
register :: Command
|
|
register opts pats = do
|
|
doWithFilteredLedger opts pats printregister
|
|
where
|
|
printregister l =
|
|
putStr $ showTransactionsWithBalances
|
|
(sortBy (comparing date) $ ledgerTransactions l)
|
|
nullamt{precision=lprecision l}
|
|
|
|
balance :: Command
|
|
balance opts pats = do
|
|
doWithFilteredLedger opts pats printbalance
|
|
where
|
|
printbalance l =
|
|
putStr $ showLedgerAccounts l depth
|
|
where
|
|
showsubs = (ShowSubs `elem` opts)
|
|
depth = case (pats, showsubs) of
|
|
((Nothing,_), False) -> 1
|
|
otherwise -> 9999
|
|
|
|
{-
|
|
interactive testing:
|
|
|
|
*Main> p <- ledgerFilePath [File "./test.dat"] >>= parseLedgerFile
|
|
*Main> let r = either (\_ -> LedgerFile [] [] []) id p
|
|
*Main> let l = cacheLedger r
|
|
*Main> let ant = accountnametree l
|
|
*Main> let at = accounts l
|
|
*Main> putStr $ drawTree $ treemap show $ ant
|
|
*Main> putStr $ showLedgerAccounts l [] False 1
|
|
*Main> :m +Tests
|
|
*Main Tests> l7
|
|
|
|
-}
|