hledger/hledger.hs

#!/usr/bin/env runhaskell
-- hledger - ledger-compatible money management utilities (& haskell study)
-- GPLv3, (c) Simon Michael & contributors, 
-- John Wiegley's ledger is at http://newartisans.com/ledger.html .

module Main -- almost all IO is handled here
where

import System (getArgs)
import Data.List (isPrefixOf)
import Test.HUnit (runTestTT)
import Test.QuickCheck (quickCheck)
import Text.ParserCombinators.Parsec (parseFromFile, ParseError)

import Options
import Types
import Parse
import Tests

main :: IO ()
main = do
  (opts, args) <- (getArgs >>= getOptions)
  if args == []
    then register []
    else
      let (command, args') = (head args, tail args) in
      if "reg" `isPrefixOf` command then (register args')
      else if "bal" `isPrefixOf` command then balance args'
           else if "test" `isPrefixOf` command then test
                else error "could not recognise your command"

-- commands

test :: IO ()      
test = do
  putStrLn "hunit "
  runTestTT tests
  putStr "quickcheck "
  mapM quickCheck props
  return ()

register :: [String] -> IO ()
register args = do 
  p <- parseLedgerFile ledgerFilePath
  case p of Left e -> parseError e
            Right l -> printRegister l

balance :: [String] -> IO ()
balance args = do 
  p <- parseLedgerFile ledgerFilePath
  case p of Left e -> parseError e
            Right l -> printBalances l

-- utils

parseLedgerFile :: IO String -> IO (Either ParseError Ledger)
parseLedgerFile f = f >>= parseFromFile ledger

printRegister :: Ledger -> IO ()
printRegister l = putStr $ showRegisterEntries (entries l) 0

printBalances :: Ledger -> IO ()
printBalances l = putStr $ showRegisterEntries (entries l) 0