summaryrefslogtreecommitdiff
path: root/Hledger
diff options
context:
space:
mode:
Diffstat (limited to 'Hledger')
-rw-r--r--Hledger/RegisterCSV.hs115
1 files changed, 115 insertions, 0 deletions
diff --git a/Hledger/RegisterCSV.hs b/Hledger/RegisterCSV.hs
new file mode 100644
index 0000000..e8cdfad
--- /dev/null
+++ b/Hledger/RegisterCSV.hs
@@ -0,0 +1,115 @@
+-- sflc-ledger-reports: web interface to hledger-based reports
+-- Copyright (C) 2013 Clint Adams
+--
+-- This program is free software: you can redistribute it and/or modify
+-- it under the terms of the GNU Affero General Public License as
+-- published by the Free Software Foundation, either version 3 of the
+-- License, or (at your option) any later version.
+--
+-- This program is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+-- GNU Affero General Public License for more details.
+--
+-- You should have received a copy of the GNU Affero General Public License
+-- along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+{-# LANGUAGE FlexibleContexts #-}
+
+module Hledger.RegisterCSV (convertJournalToRegisterCSV) where
+
+import Prelude (String, Double, Show, Eq, (||), Bool(..), const, (<), (>), snd, (.), ($), maybe, show, id, length, (==), (/=), head, map, FilePath, IO, Either(..), Maybe(..), return, concatMap, Char, otherwise)
+
+import Control.Applicative (many)
+import Control.Monad.IO.Class (MonadIO)
+import Control.Monad.Trans.Control (MonadBaseControl)
+import Control.Monad.Trans.Resource (MonadUnsafeIO, MonadThrow)
+
+import Data.List (isPrefixOf)
+import Data.Bifunctor (bimap)
+import qualified Data.Text as T
+
+import Hledger.Data.Amount (showMixedAmount)
+import Hledger.Data.Journal (journalAccountNamesUsed)
+import Hledger.Data.Types (Journal(..), Transaction(..), Posting(..), MixedAmount(..))
+import Hledger.Query (Query(..))
+import Hledger.Read (readJournalFile)
+import Hledger.Reports (postingsReport, defreportopts, PostingsReportItem)
+
+import Data.Conduit (($=),($$), runResourceT)
+import qualified Data.Conduit.List as CL
+import Data.CSV.Conduit (fromCSV, defCSVSettings, CSV)
+
+import Text.Parsec.Char (char, digit, satisfy, space)
+import Text.Parsec.Combinator (many1)
+import Text.Parsec.Prim (ParsecT, Stream, parse)
+
+data AcctReg = AcctReg {
+ _acct :: String
+ , _openingBalance :: Double
+ , _items :: [PostingsReportItem]
+} deriving (Show, Eq)
+
+cordGuess :: String -> MixedAmount -> T.Text
+cordGuess acct
+ | "Assets:" `isPrefixOf` acct || "Liabilities:" `isPrefixOf` acct || "Equity:" `isPrefixOf` acct = cordGuest' True
+ | "Income:" `isPrefixOf` acct || "Expenses:" `isPrefixOf` acct = cordGuest' False
+ | otherwise = const (T.pack "INVALID")
+ where
+ cordGuest' True amt | amt > 0 = T.pack "Deposit"
+ cordGuest' True _ = T.pack "Check"
+ cordGuest' False amt | amt > 0 = T.pack "Check"
+ cordGuest' False _ = T.pack "Deposit"
+
+acctReg :: String -> Journal -> AcctReg
+acctReg a j = AcctReg a 0 (snd . postingsReport defreportopts (Acct a) $ j)
+
+priToLine :: PostingsReportItem -> [T.Text]
+priToLine (mds, p, mixa) = [T.empty, cord, date, num, name, memo, split, amt, bal]
+ where
+ (date, desc') = maybe (T.empty, "") (bimap (T.pack . show) id) mds
+ (c', desc) = parseDesc desc'
+ cord = cordGuess (paccount p) (pamount p)
+ num = T.pack c'
+ name = T.empty
+ memo = T.pack desc
+ split = maybe (T.pack "ERROR") (splitField . tpostings) (ptransaction p)
+ amt = T.pack . showMixedAmount . pamount $ p
+ bal = T.pack . showMixedAmount $ mixa
+
+splitField :: [Posting] -> T.Text
+splitField ps
+ | length ps < 2 = T.pack "ERROR"
+ | length ps == 2 = T.pack (paccount . head $ ps)
+ | otherwise = T.pack "-SPLIT-"
+
+arToLines :: AcctReg -> [[T.Text]]
+arToLines ar = [T.pack . _acct $ ar,T.empty,T.empty,T.empty,T.empty,T.empty,T.empty,T.empty,T.pack "0.000000"]:map priToLine (_items ar)
+
+convertJournalToRegisterCSV :: FilePath -> IO (T.Text)
+convertJournalToRegisterCSV fp = do
+ Right j <- readJournalFile Nothing Nothing fp
+ csv <- journalToRegisterCSV j
+ return $ T.concat csv
+
+journalToRegisterCSV :: (MonadUnsafeIO m, MonadThrow m, MonadIO m, CSV a [T.Text], MonadBaseControl IO m) => Journal -> m [a]
+journalToRegisterCSV j = do
+ let accts = journalAccountNamesUsed j
+ precsv' = concatMap (\x -> arToLines $ acctReg x j) accts
+ precsv = [T.empty, T.pack "Type",T.pack "Date",T.pack "Num",T.pack "Name",T.pack "Memo",T.pack "Split",T.pack "Amount",T.pack "Balance"]:precsv'
+ csv <- runResourceT $ CL.sourceList precsv $= fromCSV defCSVSettings $$ CL.consume
+ return csv
+
+parseDesc :: String -> (String, String)
+parseDesc d = case parse descParser' "desc" d of
+ Left _ -> ("", d)
+ Right x -> x
+
+descParser' :: Stream s m Char => ParsecT s u m (String, String)
+descParser' = do
+ desc' <- many1 (satisfy (/='('))
+ _ <- char '('
+ checknum <- many1 digit
+ _ <- char ')'
+ _ <- many space
+ return (checknum, desc')