module Penny.Brenner
( FitAcct(..)
, Config(..)
, R.GroupSpecs(..)
, R.GroupSpec(..)
, Y.Translator(..)
, L.Side(..)
, L.SpaceBetween(..)
, usePayeeOrDesc
, brennerMain
, ofxParser
) where
import qualified Penny.Brenner.Types as Y
import Control.Monad (join)
import Data.Either (partitionEithers)
import qualified Data.Text as X
import qualified Data.Version as V
import qualified Penny.Liberty as Ly
import qualified Penny.Lincoln as L
import qualified Penny.Lincoln.Builders as Bd
import qualified Penny.Copper.Render as R
import qualified Penny.Brenner.Clear as C
import qualified Penny.Brenner.Database as D
import qualified Penny.Brenner.Import as I
import qualified Penny.Brenner.Info as Info
import qualified Penny.Brenner.Merge as M
import qualified Penny.Brenner.OFX as O
import qualified Penny.Brenner.Print as P
import qualified Penny.Brenner.Util as U
import qualified System.Console.MultiArg as MA
import qualified Control.Monad.Exception.Synchronous as Ex
brennerMain
:: V.Version
-> Config
-> IO ()
brennerMain v cf = do
let cf' = convertConfig cf
join $ MA.modesWithHelp (help False) (globalOpts v)
(preProcessor cf')
globalOpts
:: V.Version
-> [MA.OptSpec (Either (IO ()) Y.FitAcctName)]
globalOpts v =
[ MA.OptSpec ["fit-account"] "f"
(MA.OneArg (Right . Y.FitAcctName . X.pack))
, fmap Left (Ly.version v)
]
preProcessor
:: Y.Config
-> [Either (IO ()) Y.FitAcctName]
-> Either (a -> IO ()) [MA.Mode (IO ())]
preProcessor cf args =
let (vers, as) = partitionEithers args
in case vers of
[] -> makeModes Nothing cf as
x:_ -> Left (const x)
makeModes
:: Maybe Y.ConfigLocation
-> Y.Config
-> [Y.FitAcctName]
-> Either (a -> IO ()) [MA.Mode (IO ())]
makeModes cl cf as = Ex.toEither . Ex.mapException (const . U.errExit) $ do
mayFi <- case as of
[] -> return $ Y.defaultFitAcct cf
_ ->
let pdct a = Y.fitAcctName a == s
s = last as
toFilter = case Y.defaultFitAcct cf of
Nothing -> Y.moreFitAccts cf
Just d -> d : Y.moreFitAccts cf
in case filter pdct toFilter of
[] -> Ex.throw $
"financial institution account "
++ (X.unpack . Y.unFitAcctName $ s) ++ " not configured."
c:[] -> return $ Just c
_ -> Ex.throw $
"more than one financial institution account "
++ "named " ++ (X.unpack . Y.unFitAcctName $ s)
++ " configured."
return . map (fmap (\f -> f cl cf mayFi)) $ allModes
type ModeFunc
= Maybe Y.ConfigLocation
-> Y.Config
-> Maybe Y.FitAcct
-> IO ()
allModes :: [MA.Mode ModeFunc]
allModes =
fmap (\f cl cf _ -> f cl cf) Info.mode
: map (fmap (const . const))
[C.mode, I.mode, M.mode, P.mode, D.mode]
help
:: Bool
-> String
-> String
help dyn n = unlines ls
where
ls = [ "usage: " ++ n ++ " [global-options]"
++ " COMMAND [local-options]"
++ " ARGS..."
, ""
, "where COMMAND is one of:"
, "import, merge, clear, database, print, info"
, ""
, "For help on an individual command and its"
++ " local options, use "
, n ++ " COMMAND --help"
, ""
, "Global Options:"
, "-f, --fit-account ACCOUNT"
, " use the given financial institution account"
, " (use the \"info\" command to see which are available)."
, " If this option does not appear,"
, " the default account is used if there is one."
] ++ if not dyn then [] else
[ ""
, "-c, --config-file FILENAME"
, " Specify configuration file location"
]
data FitAcct = FitAcct
{ fitAcctName :: String
, fitAcctDesc :: String
, dbLocation :: String
, pennyAcct :: String
, defaultAcct :: String
, currency :: String
, groupSpecs :: R.GroupSpecs
, translator :: Y.Translator
, side :: L.Side
, spaceBetween :: L.SpaceBetween
, parser :: ( Y.ParserDesc
, Y.FitFileLocation -> IO (Ex.Exceptional String [Y.Posting]))
, toLincolnPayee :: Y.Desc -> Y.Payee -> L.Payee
} deriving Show
convertFitAcct :: FitAcct -> Y.FitAcct
convertFitAcct (FitAcct fn fd db ax df cy gs tl sd sb ps tlp) = Y.FitAcct
{ Y.fitAcctName = Y.FitAcctName . X.pack $ fn
, Y.fitAcctDesc = Y.FitAcctDesc . X.pack $ fd
, Y.dbLocation = Y.DbLocation . X.pack $ db
, Y.pennyAcct = Y.PennyAcct . Bd.account . X.pack $ ax
, Y.defaultAcct = Y.DefaultAcct . Bd.account . X.pack $ df
, Y.currency = Y.Currency . L.Commodity . X.pack $ cy
, Y.groupSpecs = gs
, Y.translator = tl
, Y.side = sd
, Y.spaceBetween = sb
, Y.parser = ps
, Y.toLincolnPayee = tlp
}
data Config = Config
{ defaultFitAcct :: Maybe FitAcct
, moreFitAccts :: [FitAcct]
} deriving Show
convertConfig :: Config -> Y.Config
convertConfig (Config d m) = Y.Config
{ Y.defaultFitAcct = fmap convertFitAcct d
, Y.moreFitAccts = map convertFitAcct m
}
usePayeeOrDesc :: Y.Desc -> Y.Payee -> L.Payee
usePayeeOrDesc (Y.Desc d) (Y.Payee p) = L.Payee $
if X.null p then d else p
ofxParser :: (Y.ParserDesc, Y.ParserFn)
ofxParser = O.parser