--- * module
--- ** doc
-- In Emacs, use TAB on lines beginning with "-- *" to collapse/expand sections.
{-|

A reader for a CSV rules file. 
This reads the actual data from a file specified by a `source` rule
or from a similarly-named file in the same directory.

Most of the code for reading rules files and csv files is in this module.
-}
-- Lots of haddocks in this file are for non-exported types.
-- Here's a command that will render them:
-- stack haddock hledger-lib --fast --no-haddock-deps --haddock-arguments='--ignore-all-exports' --open

--- ** language
{-# LANGUAGE FlexibleInstances    #-}
{-# LANGUAGE OverloadedStrings    #-}
{-# LANGUAGE RecordWildCards      #-}
{-# LANGUAGE ScopedTypeVariables  #-}
{-# LANGUAGE ViewPatterns         #-}
{-# OPTIONS_GHC -Wno-unrecognised-pragmas #-}

--- ** exports
module Hledger.Read.RulesReader (
  -- * Reader
  reader,
  -- * Misc.
  readJournalFromCsv,
  -- readRulesFile,
  -- parseCsvRules,
  -- validateCsvRules,
  -- CsvRules,
  dataFileFor,
  rulesFileFor,
  -- * Tests
  tests_RulesReader,
)
where

--- ** imports
import Prelude hiding (Applicative(..))
import Control.Applicative (Applicative(..))
import Control.Monad              (unless, when, void)
import Control.Monad.Except       (ExceptT(..), liftEither, throwError)
import qualified Control.Monad.Fail as Fail
import Control.Monad.IO.Class     (MonadIO, liftIO)
import Control.Monad.State.Strict (StateT, get, modify', evalStateT)
import Control.Monad.Trans.Class  (lift)
import Data.Char                  (toLower, isDigit, isSpace, isAlphaNum, ord)
import Data.Bifunctor             (first)
import Data.Functor               ((<&>))
import Data.List (elemIndex, foldl', mapAccumL, nub, sortOn)
import Data.List.Extra (groupOn)
import Data.Maybe (catMaybes, fromMaybe, isJust)
import Data.MemoUgly (memo)
import qualified Data.Set as S
import Data.Text (Text)
import qualified Data.Text as T
import qualified Data.Text.Encoding as T
import qualified Data.Text.IO as T
import Data.Time ( Day, TimeZone, UTCTime, LocalTime, ZonedTime(ZonedTime),
  defaultTimeLocale, getCurrentTimeZone, localDay, parseTimeM, utcToLocalTime, localTimeToUTC, zonedTimeToUTC)
import Safe (atMay, headMay, lastMay, readMay)
import System.FilePath ((</>), takeDirectory, takeExtension, stripExtension, takeFileName)
import qualified Data.Csv as Cassava
import qualified Data.Csv.Parser.Megaparsec as CassavaMegaparsec
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as BL
import Data.Foldable (asum, toList)
import Text.Megaparsec hiding (match, parse)
import Text.Megaparsec.Char (char, newline, string)
import Text.Megaparsec.Custom (parseErrorAt)
import Text.Printf (printf)

import Hledger.Data
import Hledger.Utils
import Hledger.Read.Common (aliasesFromOpts, Reader(..), InputOpts(..), amountp, statusp, journalFinalise, accountnamep )
import Hledger.Read.CsvUtils
import System.Directory (doesFileExist, getHomeDirectory)

--- ** doctest setup
-- $setup
-- >>> :set -XOverloadedStrings

--- ** reader
_READER__________________________________________ :: a
_READER__________________________________________ = forall a. HasCallStack => a
undefined  -- VSCode outline separator


reader :: MonadIO m => Reader m
reader :: forall (m :: * -> *). MonadIO m => Reader m
reader = Reader
  {rFormat :: String
rFormat     = String
"rules"
  ,rExtensions :: [String]
rExtensions = [String
"rules"]
  ,rReadFn :: InputOpts -> String -> CsvAmountString -> ExceptT String IO Journal
rReadFn     = InputOpts -> String -> CsvAmountString -> ExceptT String IO Journal
parse
  ,rParser :: MonadIO m => ErroringJournalParser m Journal
rParser     = forall a. String -> a
error' String
"sorry, rules files can't be included"  -- PARTIAL:
  }

isFileName :: String -> Bool
isFileName String
f = String -> String
takeFileName String
f forall a. Eq a => a -> a -> Bool
== String
f

getDownloadDir :: IO String
getDownloadDir = do
  String
home <- IO String
getHomeDirectory
  forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ String
home String -> String -> String
</> String
"Downloads"  -- XXX

-- | Parse and post-process a "Journal" from the given rules file path, or give an error.
-- A data file is inferred from the @source@ rule, otherwise from a similarly-named file
-- in the same directory.
-- The source rule can specify a glob pattern and supports ~ for home directory.
-- If it is a bare filename it will be relative to the defaut download directory
-- on this system. If is a relative file path it will be relative to the rules
-- file's directory. When a glob pattern matches multiple files, the alphabetically
-- last is used. (Eg in case of multiple numbered downloads, the highest-numbered
-- will be used.)
-- The provided text, or a --rules-file option, are ignored by this reader.
-- Balance assertions are not checked.
parse :: InputOpts -> FilePath -> Text -> ExceptT String IO Journal
parse :: InputOpts -> String -> CsvAmountString -> ExceptT String IO Journal
parse InputOpts
iopts String
f CsvAmountString
_ = do
  CsvRules
rules <- String -> ExceptT String IO CsvRules
readRulesFile forall a b. (a -> b) -> a -> b
$ forall a. Show a => String -> a -> a
dbg4 String
"reading rules file" String
f
  -- XXX higher-than usual debug level for file reading to bypass excessive noise from elsewhere, normally 6 or 7
  Maybe String
mdatafile <- forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO forall a b. (a -> b) -> a -> b
$ do
    String
dldir <- IO String
getDownloadDir
    let rulesdir :: String
rulesdir = String -> String
takeDirectory String
f
    let msource :: Maybe String
msource = CsvAmountString -> String
T.unpack forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"source" CsvRules
rules
    [String]
fs <- case Maybe String
msource of
            Just String
src -> String -> String -> IO [String]
expandGlob String
dir (forall a. Show a => String -> a -> a
dbg4 String
"source" String
src) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= [String] -> IO [String]
sortByModTime forall (f :: * -> *) a b. Functor f => f a -> (a -> b) -> f b
<&> forall a. Show a => String -> a -> a
dbg4 (String
"matched files"forall a. Semigroup a => a -> a -> a
<>String
descforall a. Semigroup a => a -> a -> a
<>String
", newest first")
              where (String
dir,String
desc) = if String -> Bool
isFileName String
src then (String
dldir,String
" in download directory") else (String
rulesdir,String
"")
            Maybe String
Nothing  -> forall (m :: * -> *) a. Monad m => a -> m a
return [forall b a. b -> (a -> b) -> Maybe a -> b
maybe forall {a}. a
err (forall a. Show a => String -> a -> a
dbg4 String
"inferred source") forall a b. (a -> b) -> a -> b
$ String -> Maybe String
dataFileFor String
f]  -- shouldn't fail, f has .rules extension
              where err :: a
err = forall a. String -> a
error' forall a b. (a -> b) -> a -> b
$ String
"could not infer a data file for " forall a. Semigroup a => a -> a -> a
<> String
f
    forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ forall a. Show a => String -> a -> a
dbg4 String
"data file" forall a b. (a -> b) -> a -> b
$ forall a. [a] -> Maybe a
headMay [String]
fs
  case Maybe String
mdatafile of
    Maybe String
Nothing -> forall (m :: * -> *) a. Monad m => a -> m a
return Journal
nulljournal  -- data file specified by source rule was not found
    Just String
dat -> do
      Bool
exists <- forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO forall a b. (a -> b) -> a -> b
$ String -> IO Bool
doesFileExist String
dat
      if Bool -> Bool
not (String
datforall a. Eq a => a -> a -> Bool
==String
"-" Bool -> Bool -> Bool
|| Bool
exists)
      then forall (m :: * -> *) a. Monad m => a -> m a
return Journal
nulljournal      -- data file inferred from rules file name was not found
      else do
        CsvAmountString
t <- forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO forall a b. (a -> b) -> a -> b
$ String -> IO CsvAmountString
readFileOrStdinPortably String
dat
        Maybe (Either CsvRules String)
-> String -> CsvAmountString -> ExceptT String IO Journal
readJournalFromCsv (forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ forall a b. a -> Either a b
Left CsvRules
rules) String
dat CsvAmountString
t
        -- apply any command line account aliases. Can fail with a bad replacement pattern.
        forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall e (m :: * -> *) a. MonadError e m => Either e a -> m a
liftEither forall b c a. (b -> c) -> (a -> b) -> a -> c
. [AccountAlias] -> Journal -> Either String Journal
journalApplyAliases (InputOpts -> [AccountAlias]
aliasesFromOpts InputOpts
iopts)
            -- journalFinalise assumes the journal's items are
            -- reversed, as produced by JournalReader's parser.
            -- But here they are already properly ordered. So we'd
            -- better preemptively reverse them once more. XXX inefficient
            forall b c a. (b -> c) -> (a -> b) -> a -> c
. Journal -> Journal
journalReverse
        forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= InputOpts
-> String
-> CsvAmountString
-> Journal
-> ExceptT String IO Journal
journalFinalise InputOpts
iopts{balancingopts_ :: BalancingOpts
balancingopts_=(InputOpts -> BalancingOpts
balancingopts_ InputOpts
iopts){ignore_assertions_ :: Bool
ignore_assertions_=Bool
True}} String
f CsvAmountString
""

--- ** reading rules files
--- *** rules utilities
_RULES_READING__________________________________________ :: a
_RULES_READING__________________________________________ = forall a. HasCallStack => a
undefined

-- | Given a rules file path, what would be the corresponding data file ?
-- (Remove a .rules extension.)
dataFileFor :: FilePath -> Maybe FilePath
dataFileFor :: String -> Maybe String
dataFileFor = String -> String -> Maybe String
stripExtension String
"rules"

-- | Given a csv file path, what would be the corresponding rules file ?
-- (Add a .rules extension.)
rulesFileFor :: FilePath -> FilePath
rulesFileFor :: String -> String
rulesFileFor = (forall a. [a] -> [a] -> [a]
++ String
".rules")

-- | An exception-throwing IO action that reads and validates
-- the specified CSV rules file (which may include other rules files).
readRulesFile :: FilePath -> ExceptT String IO CsvRules
readRulesFile :: String -> ExceptT String IO CsvRules
readRulesFile String
f =
  forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (do
    forall (m :: * -> *) a. (MonadIO m, Show a) => String -> a -> m ()
dbg6IO String
"using conversion rules file" String
f
    String -> IO CsvAmountString
readFilePortably String
f forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= String -> CsvAmountString -> IO CsvAmountString
expandIncludes (String -> String
takeDirectory String
f)
  ) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> CsvAmountString -> Either String CsvRules
parseAndValidateCsvRules String
f

-- | Inline all files referenced by include directives in this hledger CSV rules text, recursively.
-- Included file paths may be relative to the directory of the provided file path.
-- This is done as a pre-parse step to simplify the CSV rules parser.
expandIncludes :: FilePath -> Text -> IO Text
expandIncludes :: String -> CsvAmountString -> IO CsvAmountString
expandIncludes String
dir0 CsvAmountString
content = forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (String -> CsvAmountString -> IO CsvAmountString
expandLine String
dir0) (CsvAmountString -> [CsvAmountString]
T.lines CsvAmountString
content) forall (f :: * -> *) a b. Functor f => f a -> (a -> b) -> f b
<&> [CsvAmountString] -> CsvAmountString
T.unlines
  where
    expandLine :: String -> CsvAmountString -> IO CsvAmountString
expandLine String
dir1 CsvAmountString
line =
      case CsvAmountString
line of
        (CsvAmountString -> CsvAmountString -> Maybe CsvAmountString
T.stripPrefix CsvAmountString
"include " -> Just CsvAmountString
f) -> String -> CsvAmountString -> IO CsvAmountString
expandIncludes String
dir2 forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< String -> IO CsvAmountString
T.readFile String
f'
          where
            f' :: String
f' = String
dir1 String -> String -> String
</> CsvAmountString -> String
T.unpack ((Char -> Bool) -> CsvAmountString -> CsvAmountString
T.dropWhile Char -> Bool
isSpace CsvAmountString
f)
            dir2 :: String
dir2 = String -> String
takeDirectory String
f'
        CsvAmountString
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return CsvAmountString
line

-- defaultRulesText :: FilePath -> Text
-- defaultRulesText _csvfile = T.pack $ unlines
--   ["# hledger csv conversion rules" --  for " ++ csvFileFor (takeFileName csvfile)
--   ,"# cf http://hledger.org/hledger.html#csv"
--   ,""
--   ,"account1 assets:bank:checking"
--   ,""
--   ,"fields date, description, amount1"
--   ,""
--   ,"#skip 1"
--   ,"#newest-first"
--   ,""
--   ,"#date-format %-d/%-m/%Y"
--   ,"#date-format %-m/%-d/%Y"
--   ,"#date-format %Y-%h-%d"
--   ,""
--   ,"#currency $"
--   ,""
--   ,"if ITUNES"
--   ," account2 expenses:entertainment"
--   ,""
--   ,"if (TO|FROM) SAVINGS"
--   ," account2 assets:bank:savings\n"
--   ]

-- | An error-throwing IO action that parses this text as CSV conversion rules
-- and runs some extra validation checks. The file path is used in error messages.
parseAndValidateCsvRules :: FilePath -> T.Text -> Either String CsvRules
parseAndValidateCsvRules :: String -> CsvAmountString -> Either String CsvRules
parseAndValidateCsvRules String
rulesfile CsvAmountString
s =
  case String
-> CsvAmountString
-> Either
     (ParseErrorBundle CsvAmountString HledgerParseErrorData) CsvRules
parseCsvRules String
rulesfile CsvAmountString
s of
    Left ParseErrorBundle CsvAmountString HledgerParseErrorData
err    -> forall a b. a -> Either a b
Left forall a b. (a -> b) -> a -> b
$ ParseErrorBundle CsvAmountString HledgerParseErrorData -> String
customErrorBundlePretty ParseErrorBundle CsvAmountString HledgerParseErrorData
err
    Right CsvRules
rules -> forall (p :: * -> * -> *) a b c.
Bifunctor p =>
(a -> b) -> p a c -> p b c
first String -> String
makeFancyParseError forall a b. (a -> b) -> a -> b
$ CsvRules -> Either String CsvRules
validateCsvRules CsvRules
rules
  where
    makeFancyParseError :: String -> String
    makeFancyParseError :: String -> String
makeFancyParseError String
errorString =
      forall s e.
(VisualStream s, ShowErrorComponent e) =>
ParseError s e -> String
parseErrorPretty (forall s e. CsvFieldIndex -> Set (ErrorFancy e) -> ParseError s e
FancyError CsvFieldIndex
0 (forall a. a -> Set a
S.singleton forall a b. (a -> b) -> a -> b
$ forall e. String -> ErrorFancy e
ErrorFail String
errorString) :: ParseError Text String)

instance ShowErrorComponent String where
  showErrorComponent :: String -> String
showErrorComponent = forall a. a -> a
id

-- | Parse this text as CSV conversion rules. The file path is for error messages.
parseCsvRules :: FilePath -> T.Text -> Either (ParseErrorBundle T.Text HledgerParseErrorData) CsvRules
-- parseCsvRules rulesfile s = runParser csvrulesfile nullrules{baseAccount=takeBaseName rulesfile} rulesfile s
parseCsvRules :: String
-> CsvAmountString
-> Either
     (ParseErrorBundle CsvAmountString HledgerParseErrorData) CsvRules
parseCsvRules = forall e s a.
Parsec e s a -> String -> s -> Either (ParseErrorBundle s e) a
runParser (forall (m :: * -> *) s a. Monad m => StateT s m a -> s -> m a
evalStateT CsvRulesParser CsvRules
rulesp CsvRulesParsed
defrules)

-- | Return the validated rules, or an error.
validateCsvRules :: CsvRules -> Either String CsvRules
validateCsvRules :: CsvRules -> Either String CsvRules
validateCsvRules CsvRules
rules = do
  forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
unless (CsvAmountString -> Bool
isAssigned CsvAmountString
"date")   forall a b. (a -> b) -> a -> b
$ forall a b. a -> Either a b
Left String
"Please specify (at top level) the date field. Eg: date %1"
  forall a b. b -> Either a b
Right CsvRules
rules
  where
    isAssigned :: CsvAmountString -> Bool
isAssigned CsvAmountString
f = forall a. Maybe a -> Bool
isJust forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [] CsvAmountString
f

--- *** rules types
_RULES_TYPES__________________________________________ :: a
_RULES_TYPES__________________________________________ = forall a. HasCallStack => a
undefined

-- | A set of data definitions and account-matching patterns sufficient to
-- convert a particular CSV data file into meaningful journal transactions.
data CsvRules' a = CsvRules' {
  forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives        :: [(DirectiveName,Text)],
    -- ^ top-level rules, as (keyword, value) pairs
  forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes   :: [(CsvFieldName, CsvFieldIndex)],
    -- ^ csv field names and their column number, if declared by a fields list
  forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments       :: [(HledgerFieldName, FieldTemplate)],
    -- ^ top-level assignments to hledger fields, as (field name, value template) pairs
  forall a. CsvRules' a -> [ConditionalBlock]
rconditionalblocks :: [ConditionalBlock],
    -- ^ conditional blocks, which containing additional assignments/rules to apply to matched csv records
  forall a. CsvRules' a -> a
rblocksassigning :: a -- (String -> [ConditionalBlock])
    -- ^ all conditional blocks which can potentially assign field with a given name (memoized)
}

-- | Type used by parsers. Directives, assignments and conditional blocks
-- are in the reverse order compared to what is in the file and rblocksassigning is non-functional,
-- could not be used for processing CSV records yet
type CsvRulesParsed = CsvRules' ()

-- | Type used after parsing is done. Directives, assignments and conditional blocks
-- are in the same order as they were in the unput file and rblocksassigning is functional.
-- Ready to be used for CSV record processing
type CsvRules = CsvRules' (Text -> [ConditionalBlock])  -- XXX simplify

instance Eq CsvRules where
  CsvRules
r1 == :: CsvRules -> CsvRules -> Bool
== CsvRules
r2 = (forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives CsvRules
r1, forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes CsvRules
r1, forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRules
r1) forall a. Eq a => a -> a -> Bool
==
             (forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives CsvRules
r2, forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes CsvRules
r2, forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRules
r2)

-- Custom Show instance used for debug output: omit the rblocksassigning field, which isn't showable.
instance Show CsvRules where
  show :: CsvRules -> String
show CsvRules
r = String
"CsvRules { rdirectives = " forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show (forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives CsvRules
r) forall a. [a] -> [a] -> [a]
++
           String
", rcsvfieldindexes = "     forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show (forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes CsvRules
r) forall a. [a] -> [a] -> [a]
++
           String
", rassignments = "         forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show (forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRules
r) forall a. [a] -> [a] -> [a]
++
           String
", rconditionalblocks = "   forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show (forall a. CsvRules' a -> [ConditionalBlock]
rconditionalblocks CsvRules
r) forall a. [a] -> [a] -> [a]
++
           String
" }"

type CsvRulesParser a = StateT CsvRulesParsed SimpleTextParser a

-- | The keyword of a CSV rule - "fields", "skip", "if", etc.
type DirectiveName    = Text

-- | CSV field name.
type CsvFieldName     = Text

-- | 1-based CSV column number.
type CsvFieldIndex    = Int

-- | Percent symbol followed by a CSV field name or column number. Eg: %date, %1.
type CsvFieldReference = Text

-- | One of the standard hledger fields or pseudo-fields that can be assigned to.
-- Eg date, account1, amount, amount1-in, date-format.
type HledgerFieldName = Text

-- | A text value to be assigned to a hledger field, possibly
-- containing csv field references to be interpolated.
type FieldTemplate    = Text

-- | A strptime date parsing pattern, as supported by Data.Time.Format.
type DateFormat       = Text

-- | A prefix for a matcher test, either & or none (implicit or).
data MatcherPrefix = And | None
  deriving (CsvFieldIndex -> MatcherPrefix -> String -> String
[MatcherPrefix] -> String -> String
MatcherPrefix -> String
forall a.
(CsvFieldIndex -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
showList :: [MatcherPrefix] -> String -> String
$cshowList :: [MatcherPrefix] -> String -> String
show :: MatcherPrefix -> String
$cshow :: MatcherPrefix -> String
showsPrec :: CsvFieldIndex -> MatcherPrefix -> String -> String
$cshowsPrec :: CsvFieldIndex -> MatcherPrefix -> String -> String
Show, MatcherPrefix -> MatcherPrefix -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: MatcherPrefix -> MatcherPrefix -> Bool
$c/= :: MatcherPrefix -> MatcherPrefix -> Bool
== :: MatcherPrefix -> MatcherPrefix -> Bool
$c== :: MatcherPrefix -> MatcherPrefix -> Bool
Eq)

-- | A single test for matching a CSV record, in one way or another.
data Matcher =
    RecordMatcher MatcherPrefix Regexp                          -- ^ match if this regexp matches the overall CSV record
  | FieldMatcher MatcherPrefix CsvFieldReference Regexp         -- ^ match if this regexp matches the referenced CSV field's value
  deriving (CsvFieldIndex -> Matcher -> String -> String
[Matcher] -> String -> String
Matcher -> String
forall a.
(CsvFieldIndex -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
showList :: [Matcher] -> String -> String
$cshowList :: [Matcher] -> String -> String
show :: Matcher -> String
$cshow :: Matcher -> String
showsPrec :: CsvFieldIndex -> Matcher -> String -> String
$cshowsPrec :: CsvFieldIndex -> Matcher -> String -> String
Show, Matcher -> Matcher -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Matcher -> Matcher -> Bool
$c/= :: Matcher -> Matcher -> Bool
== :: Matcher -> Matcher -> Bool
$c== :: Matcher -> Matcher -> Bool
Eq)

-- | A conditional block: a set of CSV record matchers, and a sequence
-- of rules which will be enabled only if one or more of the matchers
-- succeeds.
--
-- Three types of rule are allowed inside conditional blocks: field
-- assignments, skip, end. (A skip or end rule is stored as if it was
-- a field assignment, and executed in validateCsv. XXX)
data ConditionalBlock = CB {
   ConditionalBlock -> [Matcher]
cbMatchers    :: [Matcher]
  ,ConditionalBlock -> [(CsvAmountString, CsvAmountString)]
cbAssignments :: [(HledgerFieldName, FieldTemplate)]
  } deriving (CsvFieldIndex -> ConditionalBlock -> String -> String
[ConditionalBlock] -> String -> String
ConditionalBlock -> String
forall a.
(CsvFieldIndex -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
showList :: [ConditionalBlock] -> String -> String
$cshowList :: [ConditionalBlock] -> String -> String
show :: ConditionalBlock -> String
$cshow :: ConditionalBlock -> String
showsPrec :: CsvFieldIndex -> ConditionalBlock -> String -> String
$cshowsPrec :: CsvFieldIndex -> ConditionalBlock -> String -> String
Show, ConditionalBlock -> ConditionalBlock -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: ConditionalBlock -> ConditionalBlock -> Bool
$c/= :: ConditionalBlock -> ConditionalBlock -> Bool
== :: ConditionalBlock -> ConditionalBlock -> Bool
$c== :: ConditionalBlock -> ConditionalBlock -> Bool
Eq)

defrules :: CsvRulesParsed
defrules :: CsvRulesParsed
defrules = CsvRules' {
  rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives=[],
  rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[],
  rassignments :: [(CsvAmountString, CsvAmountString)]
rassignments=[],
  rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[],
  rblocksassigning :: ()
rblocksassigning = ()
  }

-- | Create CsvRules from the content parsed out of the rules file
mkrules :: CsvRulesParsed -> CsvRules
mkrules :: CsvRulesParsed -> CsvRules
mkrules CsvRulesParsed
rules =
  let conditionalblocks :: [ConditionalBlock]
conditionalblocks = forall a. [a] -> [a]
reverse forall a b. (a -> b) -> a -> b
$ forall a. CsvRules' a -> [ConditionalBlock]
rconditionalblocks CsvRulesParsed
rules
      maybeMemo :: (CsvAmountString -> b) -> CsvAmountString -> b
maybeMemo = if forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [ConditionalBlock]
conditionalblocks forall a. Ord a => a -> a -> Bool
>= CsvFieldIndex
15 then forall a b. Ord a => (a -> b) -> a -> b
memo else forall a. a -> a
id
  in
    CsvRules' {
    rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives=forall a. [a] -> [a]
reverse forall a b. (a -> b) -> a -> b
$ forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives CsvRulesParsed
rules,
    rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes CsvRulesParsed
rules,
    rassignments :: [(CsvAmountString, CsvAmountString)]
rassignments=forall a. [a] -> [a]
reverse forall a b. (a -> b) -> a -> b
$ forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRulesParsed
rules,
    rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[ConditionalBlock]
conditionalblocks,
    rblocksassigning :: CsvAmountString -> [ConditionalBlock]
rblocksassigning = forall {b}. (CsvAmountString -> b) -> CsvAmountString -> b
maybeMemo (\CsvAmountString
f -> forall a. (a -> Bool) -> [a] -> [a]
filter (forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any ((forall a. Eq a => a -> a -> Bool
==CsvAmountString
f)forall b c a. (b -> c) -> (a -> b) -> a -> c
.forall a b. (a, b) -> a
fst) forall b c a. (b -> c) -> (a -> b) -> a -> c
. ConditionalBlock -> [(CsvAmountString, CsvAmountString)]
cbAssignments) [ConditionalBlock]
conditionalblocks)
    }

--- *** rules parsers
_RULES_PARSING__________________________________________ :: a
_RULES_PARSING__________________________________________ = forall a. HasCallStack => a
undefined

{-
Grammar for the CSV conversion rules, more or less:

RULES: RULE*

RULE: ( SOURCE | FIELD-LIST | FIELD-ASSIGNMENT | CONDITIONAL-BLOCK | SKIP | TIMEZONE | NEWEST-FIRST | INTRA-DAY-REVERSED | DATE-FORMAT | DECIMAL-MARK | COMMENT | BLANK ) NEWLINE

SOURCE: source SPACE FILEPATH

FIELD-LIST: fields SPACE FIELD-NAME ( SPACE? , SPACE? FIELD-NAME )*

FIELD-NAME: QUOTED-FIELD-NAME | BARE-FIELD-NAME

QUOTED-FIELD-NAME: " (any CHAR except double-quote)+ "

BARE-FIELD-NAME: any CHAR except space, tab, #, ;

FIELD-ASSIGNMENT: JOURNAL-FIELD ASSIGNMENT-SEPARATOR FIELD-VALUE

JOURNAL-FIELD: date | date2 | status | code | description | comment | account1 | account2 | amount | JOURNAL-PSEUDO-FIELD

JOURNAL-PSEUDO-FIELD: amount-in | amount-out | currency

ASSIGNMENT-SEPARATOR: SPACE | ( : SPACE? )

FIELD-VALUE: VALUE (possibly containing CSV-FIELD-REFERENCEs)

CSV-FIELD-REFERENCE: % CSV-FIELD

CSV-FIELD: ( FIELD-NAME | FIELD-NUMBER ) (corresponding to a CSV field)

FIELD-NUMBER: DIGIT+

CONDITIONAL-BLOCK: if ( FIELD-MATCHER NEWLINE )+ INDENTED-BLOCK

FIELD-MATCHER: ( CSV-FIELD-NAME SPACE? )? ( MATCHOP SPACE? )? PATTERNS

MATCHOP: ~

PATTERNS: ( NEWLINE REGEXP )* REGEXP

INDENTED-BLOCK: ( SPACE ( FIELD-ASSIGNMENT | COMMENT ) NEWLINE )+

REGEXP: ( NONSPACE CHAR* ) SPACE?

VALUE: SPACE? ( CHAR* ) SPACE?

COMMENT: SPACE? COMMENT-CHAR VALUE

COMMENT-CHAR: # | ; | *

NONSPACE: any CHAR not a SPACE-CHAR

BLANK: SPACE?

SPACE: SPACE-CHAR+

SPACE-CHAR: space | tab

CHAR: any character except newline

DIGIT: 0-9

-}

addDirective :: (DirectiveName, Text) -> CsvRulesParsed -> CsvRulesParsed
addDirective :: (CsvAmountString, CsvAmountString)
-> CsvRulesParsed -> CsvRulesParsed
addDirective (CsvAmountString, CsvAmountString)
d CsvRulesParsed
r = CsvRulesParsed
r{rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives=(CsvAmountString, CsvAmountString)
dforall a. a -> [a] -> [a]
:forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives CsvRulesParsed
r}

addAssignment :: (HledgerFieldName, FieldTemplate) -> CsvRulesParsed -> CsvRulesParsed
addAssignment :: (CsvAmountString, CsvAmountString)
-> CsvRulesParsed -> CsvRulesParsed
addAssignment (CsvAmountString, CsvAmountString)
a CsvRulesParsed
r = CsvRulesParsed
r{rassignments :: [(CsvAmountString, CsvAmountString)]
rassignments=(CsvAmountString, CsvAmountString)
aforall a. a -> [a] -> [a]
:forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRulesParsed
r}

setIndexesAndAssignmentsFromList :: [CsvFieldName] -> CsvRulesParsed -> CsvRulesParsed
setIndexesAndAssignmentsFromList :: [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
setIndexesAndAssignmentsFromList [CsvAmountString]
fs = [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
addAssignmentsFromList [CsvAmountString]
fs forall b c a. (b -> c) -> (a -> b) -> a -> c
. [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
setCsvFieldIndexesFromList [CsvAmountString]
fs
  where
    setCsvFieldIndexesFromList :: [CsvFieldName] -> CsvRulesParsed -> CsvRulesParsed
    setCsvFieldIndexesFromList :: [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
setCsvFieldIndexesFromList [CsvAmountString]
fs' CsvRulesParsed
r = CsvRulesParsed
r{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=forall a b. [a] -> [b] -> [(a, b)]
zip [CsvAmountString]
fs' [CsvFieldIndex
1..]}

    addAssignmentsFromList :: [CsvFieldName] -> CsvRulesParsed -> CsvRulesParsed
    addAssignmentsFromList :: [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
addAssignmentsFromList [CsvAmountString]
fs' CsvRulesParsed
r = forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' CsvRulesParsed -> CsvAmountString -> CsvRulesParsed
maybeAddAssignment CsvRulesParsed
r [CsvAmountString]
journalfieldnames
      where
        maybeAddAssignment :: CsvRulesParsed -> CsvAmountString -> CsvRulesParsed
maybeAddAssignment CsvRulesParsed
rules CsvAmountString
f = (forall b a. b -> (a -> b) -> Maybe a -> b
maybe forall a. a -> a
id forall {a}.
(Show a, Num a) =>
a -> CsvRulesParsed -> CsvRulesParsed
addAssignmentFromIndex forall a b. (a -> b) -> a -> b
$ forall a. Eq a => a -> [a] -> Maybe CsvFieldIndex
elemIndex CsvAmountString
f [CsvAmountString]
fs') CsvRulesParsed
rules
          where
            addAssignmentFromIndex :: a -> CsvRulesParsed -> CsvRulesParsed
addAssignmentFromIndex a
i = (CsvAmountString, CsvAmountString)
-> CsvRulesParsed -> CsvRulesParsed
addAssignment (CsvAmountString
f, String -> CsvAmountString
T.pack forall a b. (a -> b) -> a -> b
$ Char
'%'forall a. a -> [a] -> [a]
:forall a. Show a => a -> String
show (a
iforall a. Num a => a -> a -> a
+a
1))

addConditionalBlock :: ConditionalBlock -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlock :: ConditionalBlock -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlock ConditionalBlock
b CsvRulesParsed
r = CsvRulesParsed
r{rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=ConditionalBlock
bforall a. a -> [a] -> [a]
:forall a. CsvRules' a -> [ConditionalBlock]
rconditionalblocks CsvRulesParsed
r}

addConditionalBlocks :: [ConditionalBlock] -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlocks :: [ConditionalBlock] -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlocks [ConditionalBlock]
bs CsvRulesParsed
r = CsvRulesParsed
r{rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[ConditionalBlock]
bsforall a. [a] -> [a] -> [a]
++forall a. CsvRules' a -> [ConditionalBlock]
rconditionalblocks CsvRulesParsed
r}

rulesp :: CsvRulesParser CsvRules
rulesp :: CsvRulesParser CsvRules
rulesp = do
  [()]
_ <- forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many forall a b. (a -> b) -> a -> b
$ forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice
    [StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
blankorcommentlinep                                                forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"blank or comment line"
    ,(CsvRulesParser (CsvAmountString, CsvAmountString)
directivep        forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify' forall b c a. (b -> c) -> (a -> b) -> a -> c
. (CsvAmountString, CsvAmountString)
-> CsvRulesParsed -> CsvRulesParsed
addDirective)                     forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"directive"
    ,(CsvRulesParser [CsvAmountString]
fieldnamelistp    forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify' forall b c a. (b -> c) -> (a -> b) -> a -> c
. [CsvAmountString] -> CsvRulesParsed -> CsvRulesParsed
setIndexesAndAssignmentsFromList) forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"field name list"
    ,(CsvRulesParser (CsvAmountString, CsvAmountString)
fieldassignmentp  forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify' forall b c a. (b -> c) -> (a -> b) -> a -> c
. (CsvAmountString, CsvAmountString)
-> CsvRulesParsed -> CsvRulesParsed
addAssignment)                    forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"field assignment"
    -- conditionalblockp backtracks because it shares "if" prefix with conditionaltablep.
    ,forall e s (m :: * -> *) a. MonadParsec e s m => m a -> m a
try (CsvRulesParser ConditionalBlock
conditionalblockp forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify' forall b c a. (b -> c) -> (a -> b) -> a -> c
. ConditionalBlock -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlock)          forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"conditional block"
    -- 'reverse' is there to ensure that conditions are added in the order they listed in the file
    ,(CsvRulesParser [ConditionalBlock]
conditionaltablep forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify' forall b c a. (b -> c) -> (a -> b) -> a -> c
. [ConditionalBlock] -> CsvRulesParsed -> CsvRulesParsed
addConditionalBlocks forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. [a] -> [a]
reverse)   forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"conditional table"
    ]
  forall e s (m :: * -> *). MonadParsec e s m => m ()
eof
  CsvRulesParsed -> CsvRules
mkrules forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall s (m :: * -> *). MonadState s m => m s
get

blankorcommentlinep :: CsvRulesParser ()
blankorcommentlinep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
blankorcommentlinep = forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying blankorcommentlinep") forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall s (m :: * -> *) a.
[StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a]
-> StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a
choiceInState [StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
blanklinep, StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
commentlinep]

blanklinep :: CsvRulesParser ()
blanklinep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
blanklinep = forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
newline forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return () forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"blank line"

commentlinep :: CsvRulesParser ()
commentlinep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
commentlinep = forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  Char
commentcharp forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m String
restofline forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return () forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"comment line"

commentcharp :: CsvRulesParser Char
commentcharp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  Char
commentcharp = forall (f :: * -> *) e s (m :: * -> *).
(Foldable f, MonadParsec e s m) =>
f (Token s) -> m (Token s)
oneOf (String
";#*" :: [Char])

directivep :: CsvRulesParser (DirectiveName, Text)
directivep :: CsvRulesParser (CsvAmountString, CsvAmountString)
directivep = (do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying directive"
  CsvAmountString
d <- forall s (m :: * -> *) a.
[StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a]
-> StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a
choiceInState forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string) [CsvAmountString]
directives
  CsvAmountString
v <- (((forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
':' forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many forall s (m :: * -> *).
(Stream s, Char ~ Token s) =>
ParsecT HledgerParseErrorData s m Char
spacenonewline)) forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some forall s (m :: * -> *).
(Stream s, Char ~ Token s) =>
ParsecT HledgerParseErrorData s m Char
spacenonewline)) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
directivevalp)
       forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> (forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional (forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
':') forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return CsvAmountString
"")
  forall (m :: * -> *) a. Monad m => a -> m a
return (CsvAmountString
d, CsvAmountString
v)
  ) forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"directive"

directives :: [Text]
directives :: [CsvAmountString]
directives =
  [CsvAmountString
"source"
  ,CsvAmountString
"date-format"
  ,CsvAmountString
"decimal-mark"
  ,CsvAmountString
"separator"
  -- ,"default-account"
  -- ,"default-currency"
  ,CsvAmountString
"skip"
  ,CsvAmountString
"timezone"
  ,CsvAmountString
"newest-first"
  ,CsvAmountString
"intra-day-reversed"
  , CsvAmountString
"balance-type"
  ]

directivevalp :: CsvRulesParser Text
directivevalp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
directivevalp = String -> CsvAmountString
T.pack forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
`manyTill` forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof

fieldnamelistp :: CsvRulesParser [CsvFieldName]
fieldnamelistp :: CsvRulesParser [CsvAmountString]
fieldnamelistp = (do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying fieldnamelist"
  forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens CsvAmountString
"fields"
  forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional forall a b. (a -> b) -> a -> b
$ forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
':'
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces1
  let separator :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
separator = forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
',' forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces
  CsvAmountString
f <- forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldnamep
  [CsvAmountString]
fs <- forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some forall a b. (a -> b) -> a -> b
$ (StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
separator forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldnamep)
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m String
restofline
  forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
map CsvAmountString -> CsvAmountString
T.toLower forall a b. (a -> b) -> a -> b
$ CsvAmountString
fforall a. a -> [a] -> [a]
:[CsvAmountString]
fs
  ) forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"field name list"

fieldnamep :: CsvRulesParser Text
fieldnamep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldnamep = StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
quotedfieldnamep forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
barefieldnamep

quotedfieldnamep :: CsvRulesParser Text
quotedfieldnamep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
quotedfieldnamep =
    forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
'"' forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P forall a. Maybe a
Nothing (forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`notElem` (String
"\"\n:;#~" :: [Char])) forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
'"'

barefieldnamep :: CsvRulesParser Text
barefieldnamep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
barefieldnamep = forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P forall a. Maybe a
Nothing (forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`notElem` (String
" \t\n,;#~" :: [Char]))

fieldassignmentp :: CsvRulesParser (HledgerFieldName, FieldTemplate)
fieldassignmentp :: CsvRulesParser (CsvAmountString, CsvAmountString)
fieldassignmentp = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying fieldassignmentp"
  CsvAmountString
f <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
journalfieldnamep
  CsvAmountString
v <- forall s (m :: * -> *) a.
[StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a]
-> StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a
choiceInState [ StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
assignmentseparatorp forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldvalp
                     , forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return CsvAmountString
""
                     ]
  forall (m :: * -> *) a. Monad m => a -> m a
return (CsvAmountString
f,CsvAmountString
v)
  forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"field assignment"

journalfieldnamep :: CsvRulesParser Text
journalfieldnamep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
journalfieldnamep = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying journalfieldnamep")
  forall s (m :: * -> *) a.
[StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a]
-> StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a
choiceInState forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string) [CsvAmountString]
journalfieldnames

maxpostings :: CsvFieldIndex
maxpostings = CsvFieldIndex
99

-- Transaction fields and pseudo fields for CSV conversion.
-- Names must precede any other name they contain, for the parser
-- (amount-in before amount; date2 before date). TODO: fix
journalfieldnames :: [CsvAmountString]
journalfieldnames =
  forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat [[ CsvAmountString
"account" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i
          ,CsvAmountString
"amount" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i forall a. Semigroup a => a -> a -> a
<> CsvAmountString
"-in"
          ,CsvAmountString
"amount" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i forall a. Semigroup a => a -> a -> a
<> CsvAmountString
"-out"
          ,CsvAmountString
"amount" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i
          ,CsvAmountString
"balance" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i
          ,CsvAmountString
"comment" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i
          ,CsvAmountString
"currency" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
i
          ] | CsvFieldIndex
x <- [CsvFieldIndex
maxpostings, (CsvFieldIndex
maxpostingsforall a. Num a => a -> a -> a
-CsvFieldIndex
1)..CsvFieldIndex
1], let i :: CsvAmountString
i = String -> CsvAmountString
T.pack forall a b. (a -> b) -> a -> b
$ forall a. Show a => a -> String
show CsvFieldIndex
x]
  forall a. [a] -> [a] -> [a]
++
  [CsvAmountString
"amount-in"
  ,CsvAmountString
"amount-out"
  ,CsvAmountString
"amount"
  ,CsvAmountString
"balance"
  ,CsvAmountString
"code"
  ,CsvAmountString
"comment"
  ,CsvAmountString
"currency"
  ,CsvAmountString
"date2"
  ,CsvAmountString
"date"
  ,CsvAmountString
"description"
  ,CsvAmountString
"status"
  ,CsvAmountString
"skip" -- skip and end are not really fields, but we list it here to allow conditional rules that skip records
  ,CsvAmountString
"end"
  ]

assignmentseparatorp :: CsvRulesParser ()
assignmentseparatorp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
assignmentseparatorp = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying assignmentseparatorp"
  ()
_ <- forall s (m :: * -> *) a.
[StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a]
-> StateT s (ParsecT HledgerParseErrorData CsvAmountString m) a
choiceInState [ forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
':' forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces
                     , forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces1
                     ]
  forall (m :: * -> *) a. Monad m => a -> m a
return ()

fieldvalp :: CsvRulesParser Text
fieldvalp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldvalp = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying fieldvalp"
  String -> CsvAmountString
T.pack forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
`manyTill` forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof

-- A conditional block: one or more matchers, one per line, followed by one or more indented rules.
conditionalblockp :: CsvRulesParser ConditionalBlock
conditionalblockp :: CsvRulesParser ConditionalBlock
conditionalblockp = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying conditionalblockp"
  -- "if\nMATCHER" or "if    \nMATCHER" or "if MATCHER"
  CsvFieldIndex
start <- forall e s (m :: * -> *). MonadParsec e s m => m CsvFieldIndex
getOffset
  forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens CsvAmountString
"if" forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> ( (forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
newline forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return forall a. Maybe a
Nothing)
                  forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces1 forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
newline))
  [Matcher]
ms <- forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some CsvRulesParser Matcher
matcherp
  [(CsvAmountString, CsvAmountString)]
as <- forall a. [Maybe a] -> [a]
catMaybes forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>
    forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces1 forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
          forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice [ forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return forall a. Maybe a
Nothing
                 , forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap forall a. a -> Maybe a
Just CsvRulesParser (CsvAmountString, CsvAmountString)
fieldassignmentp
                 ])
  forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (forall (t :: * -> *) a. Foldable t => t a -> Bool
null [(CsvAmountString, CsvAmountString)]
as) forall a b. (a -> b) -> a -> b
$
    forall e s (m :: * -> *) a. MonadParsec e s m => e -> m a
customFailure forall a b. (a -> b) -> a -> b
$ CsvFieldIndex -> String -> HledgerParseErrorData
parseErrorAt CsvFieldIndex
start forall a b. (a -> b) -> a -> b
$  String
"start of conditional block found, but no assignment rules afterward\n(assignment rules in a conditional block should be indented)"
  forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ CB{cbMatchers :: [Matcher]
cbMatchers=[Matcher]
ms, cbAssignments :: [(CsvAmountString, CsvAmountString)]
cbAssignments=[(CsvAmountString, CsvAmountString)]
as}
  forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"conditional block"

-- A conditional table: "if" followed by separator, followed by some field names,
-- followed by many lines, each of which has:
-- one matchers, followed by field assignments (as many as there were fields)
conditionaltablep :: CsvRulesParser [ConditionalBlock]
conditionaltablep :: CsvRulesParser [ConditionalBlock]
conditionaltablep = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying conditionaltablep"
  CsvFieldIndex
start <- forall e s (m :: * -> *). MonadParsec e s m => m CsvFieldIndex
getOffset
  forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens CsvAmountString
"if"
  Char
sep <- forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall e s (m :: * -> *).
MonadParsec e s m =>
(Token s -> Bool) -> m (Token s)
satisfy (\Token CsvAmountString
c -> Bool -> Bool
not (Char -> Bool
isAlphaNum Token CsvAmountString
c Bool -> Bool -> Bool
|| Char -> Bool
isSpace Token CsvAmountString
c))
  [CsvAmountString]
fields <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
journalfieldnamep forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
`sepBy1` (forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
sep)
  forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
newline
  [(Matcher, [CsvAmountString])]
body <- forall a b c. (a -> b -> c) -> b -> a -> c
flip forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
manyTill (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof) forall a b. (a -> b) -> a -> b
$ do
    CsvFieldIndex
off <- forall e s (m :: * -> *). MonadParsec e s m => m CsvFieldIndex
getOffset
    Matcher
m <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
matcherp' forall a b. (a -> b) -> a -> b
$ forall (f :: * -> *) a. Functor f => f a -> f ()
void forall a b. (a -> b) -> a -> b
$ forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
sep
    [CsvAmountString]
vs <- (Char -> Bool) -> CsvAmountString -> [CsvAmountString]
T.split (forall a. Eq a => a -> a -> Bool
==Char
sep) forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> CsvAmountString
T.pack forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m String
restofline
    if (forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [CsvAmountString]
vs forall a. Eq a => a -> a -> Bool
/= forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [CsvAmountString]
fields)
      then forall e s (m :: * -> *) a. MonadParsec e s m => e -> m a
customFailure forall a b. (a -> b) -> a -> b
$ CsvFieldIndex -> String -> HledgerParseErrorData
parseErrorAt CsvFieldIndex
off forall a b. (a -> b) -> a -> b
$ ((forall r. PrintfType r => String -> r
printf String
"line of conditional table should have %d values, but this one has only %d" (forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [CsvAmountString]
fields) (forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [CsvAmountString]
vs)) :: String)
      else forall (m :: * -> *) a. Monad m => a -> m a
return (Matcher
m,[CsvAmountString]
vs)
  forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (forall (t :: * -> *) a. Foldable t => t a -> Bool
null [(Matcher, [CsvAmountString])]
body) forall a b. (a -> b) -> a -> b
$
    forall e s (m :: * -> *) a. MonadParsec e s m => e -> m a
customFailure forall a b. (a -> b) -> a -> b
$ CsvFieldIndex -> String -> HledgerParseErrorData
parseErrorAt CsvFieldIndex
start forall a b. (a -> b) -> a -> b
$ String
"start of conditional table found, but no assignment rules afterward"
  forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ forall a b c. (a -> b -> c) -> b -> a -> c
flip forall a b. (a -> b) -> [a] -> [b]
map [(Matcher, [CsvAmountString])]
body forall a b. (a -> b) -> a -> b
$ \(Matcher
m,[CsvAmountString]
vs) ->
    CB{cbMatchers :: [Matcher]
cbMatchers=[Matcher
m], cbAssignments :: [(CsvAmountString, CsvAmountString)]
cbAssignments=forall a b. [a] -> [b] -> [(a, b)]
zip [CsvAmountString]
fields [CsvAmountString]
vs}
  forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"conditional table"

-- A single matcher, on one line.
matcherp' :: CsvRulesParser () -> CsvRulesParser Matcher
matcherp' :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
matcherp' StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end = forall e s (m :: * -> *) a. MonadParsec e s m => m a -> m a
try (StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
fieldmatcherp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end) forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
recordmatcherp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end

matcherp :: CsvRulesParser Matcher
matcherp :: CsvRulesParser Matcher
matcherp = StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
matcherp' (forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m ()
eolof)

-- A single whole-record matcher.
-- A pattern on the whole line, not beginning with a csv field reference.
recordmatcherp :: CsvRulesParser () -> CsvRulesParser Matcher
recordmatcherp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
recordmatcherp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying recordmatcherp"
  -- pos <- currentPos
  -- _  <- optional (matchoperatorp >> lift skipNonNewlineSpaces >> optional newline)
  MatcherPrefix
p <- CsvRulesParser MatcherPrefix
matcherprefixp
  Regexp
r <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Regexp
regexp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end
  forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
p Regexp
r
  -- when (null ps) $
  --   Fail.fail "start of record matcher found, but no patterns afterward\n(patterns should not be indented)"
  forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"record matcher"

-- | A single matcher for a specific field. A csv field reference
-- (like %date or %1), and a pattern on the rest of the line,
-- optionally space-separated. Eg:
-- %description chez jacques
fieldmatcherp :: CsvRulesParser () -> CsvRulesParser Matcher
fieldmatcherp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Matcher
fieldmatcherp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying fieldmatcher"
  -- An optional fieldname (default: "all")
  -- f <- fromMaybe "all" `fmap` (optional $ do
  --        f' <- fieldnamep
  --        lift skipNonNewlineSpaces
  --        return f')
  MatcherPrefix
p <- CsvRulesParser MatcherPrefix
matcherprefixp
  CsvAmountString
f <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
csvfieldreferencep forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces
  -- optional operator.. just ~ (case insensitive infix regex) for now
  -- _op <- fromMaybe "~" <$> optional matchoperatorp
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces
  Regexp
r <- StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Regexp
regexp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end
  forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
p CsvAmountString
f Regexp
r
  forall e s (m :: * -> *) a.
MonadParsec e s m =>
m a -> String -> m a
<?> String
"field matcher"

matcherprefixp :: CsvRulesParser MatcherPrefix
matcherprefixp :: CsvRulesParser MatcherPrefix
matcherprefixp = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying matcherprefixp"
  (forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
'&' forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall s (m :: * -> *).
(Stream s, Token s ~ Char) =>
ParsecT HledgerParseErrorData s m ()
skipNonNewlineSpaces forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return MatcherPrefix
And) forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> forall (m :: * -> *) a. Monad m => a -> m a
return MatcherPrefix
None

csvfieldreferencep :: CsvRulesParser CsvFieldReference
csvfieldreferencep :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
csvfieldreferencep = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying csvfieldreferencep"
  forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
'%'
  Char -> CsvAmountString -> CsvAmountString
T.cons Char
'%' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString
textQuoteIfNeeded forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
fieldnamep

-- A single regular expression
regexp :: CsvRulesParser () -> CsvRulesParser Regexp
regexp :: StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
-> CsvRulesParser Regexp
regexp StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end = do
  forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *). CsvFieldIndex -> String -> TextParser m ()
dbgparse CsvFieldIndex
8 String
"trying regexp"
  -- notFollowedBy matchoperatorp
  Char
c <- forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall (m :: * -> *). TextParser m Char
nonspace
  String
cs <- forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
`manyTill` StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  ()
end
  case CsvAmountString -> Either String Regexp
toRegexCI forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString
T.strip forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> CsvAmountString
T.pack forall a b. (a -> b) -> a -> b
$ Char
cforall a. a -> [a] -> [a]
:String
cs of
       Left String
x -> forall (m :: * -> *) a. MonadFail m => String -> m a
Fail.fail forall a b. (a -> b) -> a -> b
$ String
"CSV parser: " forall a. [a] -> [a] -> [a]
++ String
x
       Right Regexp
x -> forall (m :: * -> *) a. Monad m => a -> m a
return Regexp
x

-- -- A match operator, indicating the type of match to perform.
-- -- Currently just ~ meaning case insensitive infix regex match.
-- matchoperatorp :: CsvRulesParser String
-- matchoperatorp = fmap T.unpack $ choiceInState $ map string
--   ["~"
--   -- ,"!~"
--   -- ,"="
--   -- ,"!="
--   ]

_RULES_LOOKUP__________________________________________ :: a
_RULES_LOOKUP__________________________________________ = forall a. HasCallStack => a
undefined

getDirective :: DirectiveName -> CsvRules -> Maybe FieldTemplate
getDirective :: CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
directivename = forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CsvAmountString
directivename forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rdirectives

-- | Look up the value (template) of a csv rule by rule keyword.
csvRule :: CsvRules -> DirectiveName -> Maybe FieldTemplate
csvRule :: CsvRules -> CsvAmountString -> Maybe CsvAmountString
csvRule CsvRules
rules = (CsvAmountString -> CsvRules -> Maybe CsvAmountString
`getDirective` CsvRules
rules)

-- | Look up the value template assigned to a hledger field by field
-- list/field assignment rules, taking into account the current record and
-- conditional rules.
hledgerField :: CsvRules -> CsvRecord -> HledgerFieldName -> Maybe FieldTemplate
hledgerField :: CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerField = CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment

-- | Look up the final value assigned to a hledger field, with csv field
-- references interpolated.
hledgerFieldValue :: CsvRules -> CsvRecord -> HledgerFieldName -> Maybe Text
hledgerFieldValue :: CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerFieldValue CsvRules
rules [CsvAmountString]
record = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
renderTemplate CsvRules
rules [CsvAmountString]
record) forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerField CsvRules
rules [CsvAmountString]
record

-- | Given the conversion rules, a CSV record and a hledger field name, find
-- the value template ultimately assigned to this field, if any, by a field
-- assignment at top level or in a conditional block matching this record.
--
-- Note conditional blocks' patterns are matched against an approximation of the
-- CSV record: all the field values, without enclosing quotes, comma-separated.
--
getEffectiveAssignment :: CsvRules -> CsvRecord -> HledgerFieldName -> Maybe FieldTemplate
getEffectiveAssignment :: CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString]
record CsvAmountString
f = forall a. [a] -> Maybe a
lastMay forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$ [(CsvAmountString, CsvAmountString)]
assignments
  where
    -- all active assignments to field f, in order
    assignments :: [(CsvAmountString, CsvAmountString)]
assignments = forall a. Show a => String -> a -> a
dbg9 String
"csv assignments" forall a b. (a -> b) -> a -> b
$ forall a. (a -> Bool) -> [a] -> [a]
filter ((forall a. Eq a => a -> a -> Bool
==CsvAmountString
f)forall b c a. (b -> c) -> (a -> b) -> a -> c
.forall a b. (a, b) -> a
fst) forall a b. (a -> b) -> a -> b
$ [(CsvAmountString, CsvAmountString)]
toplevelassignments forall a. [a] -> [a] -> [a]
++ [(CsvAmountString, CsvAmountString)]
conditionalassignments
      where
        -- all top level field assignments
        toplevelassignments :: [(CsvAmountString, CsvAmountString)]
toplevelassignments    = forall a. CsvRules' a -> [(CsvAmountString, CsvAmountString)]
rassignments CsvRules
rules
        -- all field assignments in conditional blocks assigning to field f and active for the current csv record
        conditionalassignments :: [(CsvAmountString, CsvAmountString)]
conditionalassignments = forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap ConditionalBlock -> [(CsvAmountString, CsvAmountString)]
cbAssignments forall a b. (a -> b) -> a -> b
$ forall a. (a -> Bool) -> [a] -> [a]
filter ConditionalBlock -> Bool
isBlockActive forall a b. (a -> b) -> a -> b
$ (forall a. CsvRules' a -> a
rblocksassigning CsvRules
rules) CsvAmountString
f
          where
            -- does this conditional block match the current csv record ?
            isBlockActive :: ConditionalBlock -> Bool
            isBlockActive :: ConditionalBlock -> Bool
isBlockActive CB{[(CsvAmountString, CsvAmountString)]
[Matcher]
cbAssignments :: [(CsvAmountString, CsvAmountString)]
cbMatchers :: [Matcher]
cbAssignments :: ConditionalBlock -> [(CsvAmountString, CsvAmountString)]
cbMatchers :: ConditionalBlock -> [Matcher]
..} = forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any (forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
all Matcher -> Bool
matcherMatches) forall a b. (a -> b) -> a -> b
$ [Matcher] -> [[Matcher]]
groupedMatchers [Matcher]
cbMatchers
              where
                -- does this individual matcher match the current csv record ?
                matcherMatches :: Matcher -> Bool
                matcherMatches :: Matcher -> Bool
matcherMatches (RecordMatcher MatcherPrefix
_ Regexp
pat) = Regexp -> CsvAmountString -> Bool
regexMatchText Regexp
pat' CsvAmountString
wholecsvline
                  where
                    pat' :: Regexp
pat' = forall a. Show a => String -> a -> a
dbg7 String
"regex" Regexp
pat
                    -- A synthetic whole CSV record to match against. Note, this can be
                    -- different from the original CSV data:
                    -- - any whitespace surrounding field values is preserved
                    -- - any quotes enclosing field values are removed
                    -- - and the field separator is always comma
                    -- which means that a field containing a comma will look like two fields.
                    wholecsvline :: CsvAmountString
wholecsvline = forall a. Show a => String -> a -> a
dbg7 String
"wholecsvline" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> [CsvAmountString] -> CsvAmountString
T.intercalate CsvAmountString
"," [CsvAmountString]
record
                matcherMatches (FieldMatcher MatcherPrefix
_ CsvAmountString
csvfieldref Regexp
pat) = Regexp -> CsvAmountString -> Bool
regexMatchText Regexp
pat CsvAmountString
csvfieldvalue
                  where
                    -- the value of the referenced CSV field to match against.
                    csvfieldvalue :: CsvAmountString
csvfieldvalue = forall a. Show a => String -> a -> a
dbg7 String
"csvfieldvalue" forall a b. (a -> b) -> a -> b
$ CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
replaceCsvFieldReference CsvRules
rules [CsvAmountString]
record CsvAmountString
csvfieldref

                -- | Group matchers into associative pairs based on prefix, e.g.:
                --   A
                --   & B
                --   C
                --   D
                --   & E
                --   => [[A, B], [C], [D, E]]
                groupedMatchers :: [Matcher] -> [[Matcher]]
                groupedMatchers :: [Matcher] -> [[Matcher]]
groupedMatchers [] = []
                groupedMatchers (Matcher
x:[Matcher]
xs) = (Matcher
xforall a. a -> [a] -> [a]
:[Matcher]
ys) forall a. a -> [a] -> [a]
: [Matcher] -> [[Matcher]]
groupedMatchers [Matcher]
zs
                  where
                    ([Matcher]
ys, [Matcher]
zs) = forall a. (a -> Bool) -> [a] -> ([a], [a])
span (\Matcher
y -> Matcher -> MatcherPrefix
matcherPrefix Matcher
y forall a. Eq a => a -> a -> Bool
== MatcherPrefix
And) [Matcher]
xs
                    matcherPrefix :: Matcher -> MatcherPrefix
                    matcherPrefix :: Matcher -> MatcherPrefix
matcherPrefix (RecordMatcher MatcherPrefix
prefix Regexp
_) = MatcherPrefix
prefix
                    matcherPrefix (FieldMatcher MatcherPrefix
prefix CsvAmountString
_ Regexp
_) = MatcherPrefix
prefix

-- | Render a field assignment's template, possibly interpolating referenced
-- CSV field values. Outer whitespace is removed from interpolated values.
renderTemplate ::  CsvRules -> CsvRecord -> FieldTemplate -> Text
renderTemplate :: CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
renderTemplate CsvRules
rules [CsvAmountString]
record CsvAmountString
t = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
t forall a. Monoid a => [a] -> a
mconcat forall a b. (a -> b) -> a -> b
$ forall e s a. (Ord e, Stream s) => Parsec e s a -> s -> Maybe a
parseMaybe
    (forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many forall a b. (a -> b) -> a -> b
$ forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P forall a. Maybe a
Nothing (forall a. Eq a => a -> a -> Bool
/=Char
'%')
        forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
replaceCsvFieldReference CsvRules
rules [CsvAmountString]
record forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> ParsecT
  HledgerParseErrorData CsvAmountString Identity CsvAmountString
referencep)
    CsvAmountString
t
  where
    referencep :: ParsecT
  HledgerParseErrorData CsvAmountString Identity CsvAmountString
referencep = forall (f :: * -> *) a b c.
Applicative f =>
(a -> b -> c) -> f a -> f b -> f c
liftA2 Char -> CsvAmountString -> CsvAmountString
T.cons (forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
'%') (forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P (forall a. a -> Maybe a
Just String
"reference") Char -> Bool
isFieldNameChar) :: Parsec HledgerParseErrorData Text Text
    isFieldNameChar :: Char -> Bool
isFieldNameChar Char
c = Char -> Bool
isAlphaNum Char
c Bool -> Bool -> Bool
|| Char
c forall a. Eq a => a -> a -> Bool
== Char
'_' Bool -> Bool -> Bool
|| Char
c forall a. Eq a => a -> a -> Bool
== Char
'-'

-- | Replace something that looks like a reference to a csv field ("%date" or "%1)
-- with that field's value. If it doesn't look like a field reference, or if we
-- can't find such a field, replace it with the empty string.
replaceCsvFieldReference :: CsvRules -> CsvRecord -> CsvFieldReference -> Text
replaceCsvFieldReference :: CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
replaceCsvFieldReference CsvRules
rules [CsvAmountString]
record CsvAmountString
s = case CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
s of
    Just (Char
'%', CsvAmountString
fieldname) -> forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
csvFieldValue CsvRules
rules [CsvAmountString]
record CsvAmountString
fieldname
    Maybe (Char, CsvAmountString)
_                     -> CsvAmountString
s

-- | Get the (whitespace-stripped) value of a CSV field, identified by its name or
-- column number, ("date" or "1"), from the given CSV record, if such a field exists.
csvFieldValue :: CsvRules -> CsvRecord -> CsvFieldName -> Maybe Text
csvFieldValue :: CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
csvFieldValue CsvRules
rules [CsvAmountString]
record CsvAmountString
fieldname = do
  CsvFieldIndex
fieldindex <-
    if (Char -> Bool) -> CsvAmountString -> Bool
T.all Char -> Bool
isDigit CsvAmountString
fieldname
    then forall a. Read a => String -> Maybe a
readMay forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
fieldname
    else forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup (CsvAmountString -> CsvAmountString
T.toLower CsvAmountString
fieldname) forall a b. (a -> b) -> a -> b
$ forall a. CsvRules' a -> [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes CsvRules
rules
  CsvAmountString -> CsvAmountString
T.strip forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> forall a. [a] -> CsvFieldIndex -> Maybe a
atMay [CsvAmountString]
record (CsvFieldIndex
fieldindexforall a. Num a => a -> a -> a
-CsvFieldIndex
1)

_CSV_READING__________________________________________ :: a
_CSV_READING__________________________________________ = forall a. HasCallStack => a
undefined

-- | Read a Journal from the given CSV data (and filename, used for error
-- messages), or return an error. Proceed as follows:
--
-- 1. Conversion rules are provided, or they are parsed from the specified
--    rules file, or from the default rules file for the CSV data file.
--    If rules parsing fails, or the required rules file does not exist, throw an error.
--
-- 2. Parse the CSV data using the rules, or throw an error.
--
-- 3. Convert the CSV records to hledger transactions using the rules.
--
-- 4. Return the transactions as a Journal.
--
readJournalFromCsv :: Maybe (Either CsvRules FilePath) -> FilePath -> Text -> ExceptT String IO Journal
readJournalFromCsv :: Maybe (Either CsvRules String)
-> String -> CsvAmountString -> ExceptT String IO Journal
readJournalFromCsv Maybe (Either CsvRules String)
Nothing String
"-" CsvAmountString
_ = forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError String
"please use --rules-file when reading CSV from stdin"
readJournalFromCsv Maybe (Either CsvRules String)
merulesfile String
csvfile CsvAmountString
csvtext = do
    -- for now, correctness is the priority here, efficiency not so much

    CsvRules
rules <- case Maybe (Either CsvRules String)
merulesfile of
      Just (Left CsvRules
rs)         -> forall (m :: * -> *) a. Monad m => a -> m a
return CsvRules
rs
      Just (Right String
rulesfile) -> String -> ExceptT String IO CsvRules
readRulesFile String
rulesfile
      Maybe (Either CsvRules String)
Nothing                -> String -> ExceptT String IO CsvRules
readRulesFile forall a b. (a -> b) -> a -> b
$ String -> String
rulesFileFor String
csvfile
    forall (m :: * -> *) a. (MonadIO m, Show a) => String -> a -> m ()
dbg6IO String
"csv rules" CsvRules
rules

    -- convert the csv data to lines and remove all empty/blank lines
    let csvlines1 :: [CsvAmountString]
csvlines1 = forall a. Show a => String -> a -> a
dbg9 String
"csvlines1" forall a b. (a -> b) -> a -> b
$ forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> Bool
T.null forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString
T.strip) forall a b. (a -> b) -> a -> b
$ forall a. Show a => String -> a -> a
dbg9 String
"csvlines0" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> [CsvAmountString]
T.lines CsvAmountString
csvtext

    -- if there is a top-level skip rule, skip the specified number of non-empty lines
    CsvFieldIndex
skiplines <- case CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"skip" CsvRules
rules of
                      Maybe CsvAmountString
Nothing -> forall (m :: * -> *) a. Monad m => a -> m a
return CsvFieldIndex
0
                      Just CsvAmountString
"" -> forall (m :: * -> *) a. Monad m => a -> m a
return CsvFieldIndex
1
                      Just CsvAmountString
s  -> forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError forall a b. (a -> b) -> a -> b
$ String
"could not parse skip value: " forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show CsvAmountString
s) forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Read a => String -> Maybe a
readMay forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
s
    let csvlines2 :: [CsvAmountString]
csvlines2 = forall a. Show a => String -> a -> a
dbg9 String
"csvlines2" forall a b. (a -> b) -> a -> b
$ forall a. CsvFieldIndex -> [a] -> [a]
drop CsvFieldIndex
skiplines [CsvAmountString]
csvlines1

    -- convert back to text and parse as csv records
    let
      csvtext1 :: CsvAmountString
csvtext1 = [CsvAmountString] -> CsvAmountString
T.unlines [CsvAmountString]
csvlines2
      separator :: Char
separator =
        case CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"separator" CsvRules
rules forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= CsvAmountString -> Maybe Char
parseSeparator of
          Just Char
c           -> Char
c
          Maybe Char
_ | String
ext forall a. Eq a => a -> a -> Bool
== String
"ssv" -> Char
';'
          Maybe Char
_ | String
ext forall a. Eq a => a -> a -> Bool
== String
"tsv" -> Char
'\t'
          Maybe Char
_                -> Char
','
          where
            ext :: String
ext = forall a b. (a -> b) -> [a] -> [b]
map Char -> Char
toLower forall a b. (a -> b) -> a -> b
$ forall a. CsvFieldIndex -> [a] -> [a]
drop CsvFieldIndex
1 forall a b. (a -> b) -> a -> b
$ String -> String
takeExtension String
csvfile
      -- parsec seemed to fail if you pass it "-" here   -- TODO: try again with megaparsec
      parsecfilename :: String
parsecfilename = if String
csvfile forall a. Eq a => a -> a -> Bool
== String
"-" then String
"(stdin)" else String
csvfile
    forall (m :: * -> *) a. (MonadIO m, Show a) => String -> a -> m ()
dbg6IO String
"using separator" Char
separator
    -- parse csv records
    [[CsvAmountString]]
csvrecords0 <- forall a. Show a => String -> a -> a
dbg7 String
"parseCsv" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Char
-> String
-> CsvAmountString
-> ExceptT String IO [[CsvAmountString]]
parseCsv Char
separator String
parsecfilename CsvAmountString
csvtext1
    -- remove any records skipped by conditional skip or end rules
    let csvrecords1 :: [[CsvAmountString]]
csvrecords1 = CsvRules -> [[CsvAmountString]] -> [[CsvAmountString]]
applyConditionalSkips CsvRules
rules [[CsvAmountString]]
csvrecords0
    -- and check the remaining records for any obvious problems
    [[CsvAmountString]]
csvrecords <- forall e (m :: * -> *) a. MonadError e m => Either e a -> m a
liftEither forall a b. (a -> b) -> a -> b
$ forall a. Show a => String -> a -> a
dbg7 String
"validateCsv" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [[CsvAmountString]] -> Either String [[CsvAmountString]]
validateCsv [[CsvAmountString]]
csvrecords1
    forall (m :: * -> *) a. (MonadIO m, Show a) => String -> a -> m ()
dbg6IO String
"first 3 csv records" forall a b. (a -> b) -> a -> b
$ forall a. CsvFieldIndex -> [a] -> [a]
take CsvFieldIndex
3 [[CsvAmountString]]
csvrecords

    -- XXX identify header lines some day ?
    -- let (headerlines, datalines) = identifyHeaderLines csvrecords'
    --     mfieldnames = lastMay headerlines

    TimeZone
tzout <- forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO IO TimeZone
getCurrentTimeZone
    Maybe TimeZone
mtzin <- case CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"timezone" CsvRules
rules of
              Maybe CsvAmountString
Nothing -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a. Maybe a
Nothing
              Just CsvAmountString
s  ->
                forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError forall a b. (a -> b) -> a -> b
$ String
"could not parse time zone: " forall a. [a] -> [a] -> [a]
++ CsvAmountString -> String
T.unpack CsvAmountString
s) (forall (m :: * -> *) a. Monad m => a -> m a
returnforall b c a. (b -> c) -> (a -> b) -> a -> c
.forall a. a -> Maybe a
Just) forall a b. (a -> b) -> a -> b
$
                forall (m :: * -> *) t.
(MonadFail m, ParseTime t) =>
Bool -> TimeLocale -> String -> String -> m t
parseTimeM Bool
False TimeLocale
defaultTimeLocale String
"%Z" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
s
    let
      -- convert CSV records to transactions, saving the CSV line numbers for error positions
      txns :: [Transaction]
txns = forall a. Show a => String -> a -> a
dbg7 String
"csv txns" forall a b. (a -> b) -> a -> b
$ forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$ forall (t :: * -> *) s a b.
Traversable t =>
(s -> a -> (s, b)) -> s -> t a -> (s, t b)
mapAccumL
                     (\SourcePos
pos [CsvAmountString]
r ->
                        let
                          SourcePos String
name Pos
line Pos
col = SourcePos
pos
                          line' :: Pos
line' = (CsvFieldIndex -> Pos
mkPos forall b c a. (b -> c) -> (a -> b) -> a -> c
. (forall a. Num a => a -> a -> a
+CsvFieldIndex
1) forall b c a. (b -> c) -> (a -> b) -> a -> c
. Pos -> CsvFieldIndex
unPos) Pos
line
                          pos' :: SourcePos
pos' = String -> Pos -> Pos -> SourcePos
SourcePos String
name Pos
line' Pos
col
                        in
                          (SourcePos
pos', Bool
-> Maybe TimeZone
-> TimeZone
-> SourcePos
-> CsvRules
-> [CsvAmountString]
-> Transaction
transactionFromCsvRecord Bool
timesarezoned Maybe TimeZone
mtzin TimeZone
tzout SourcePos
pos CsvRules
rules [CsvAmountString]
r)
                     )
                     (String -> SourcePos
initialPos String
parsecfilename) [[CsvAmountString]]
csvrecords
        where
          timesarezoned :: Bool
timesarezoned =
            case CsvRules -> CsvAmountString -> Maybe CsvAmountString
csvRule CsvRules
rules CsvAmountString
"date-format" of
              Just CsvAmountString
f | forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any (CsvAmountString -> CsvAmountString -> Bool
`T.isInfixOf` CsvAmountString
f) [CsvAmountString
"%Z",CsvAmountString
"%z",CsvAmountString
"%EZ",CsvAmountString
"%Ez"] -> Bool
True
              Maybe CsvAmountString
_ -> Bool
False

      -- Do our best to ensure transactions will be ordered chronologically,
      -- from oldest to newest. This is done in several steps:
      -- 1. Intra-day order: if there's an "intra-day-reversed" rule,
      -- assume each day's CSV records were ordered in reverse of the overall date order,
      -- so reverse each day's txns.
      intradayreversed :: Bool
intradayreversed = forall a. Show a => String -> a -> a
dbg6 String
"intra-day-reversed" forall a b. (a -> b) -> a -> b
$ forall a. Maybe a -> Bool
isJust forall a b. (a -> b) -> a -> b
$ CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"intra-day-reversed" CsvRules
rules
      txns1 :: [Transaction]
txns1 = forall a. Show a => String -> a -> a
dbg7 String
"txns1" forall a b. (a -> b) -> a -> b
$
        (if Bool
intradayreversed then forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap forall a. [a] -> [a]
reverse forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall k a. Eq k => (a -> k) -> [a] -> [[a]]
groupOn Transaction -> Day
tdate else forall a. a -> a
id) [Transaction]
txns
      -- 2. Overall date order: now if there's a "newest-first" rule,
      -- or if there's multiple dates and the first is more recent than the last,
      -- assume CSV records were ordered newest dates first,
      -- so reverse all txns.
      newestfirst :: Bool
newestfirst = forall a. Show a => String -> a -> a
dbg6 String
"newest-first" forall a b. (a -> b) -> a -> b
$ forall a. Maybe a -> Bool
isJust forall a b. (a -> b) -> a -> b
$ CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"newest-first" CsvRules
rules
      mdatalooksnewestfirst :: Maybe Bool
mdatalooksnewestfirst = forall a. Show a => String -> a -> a
dbg6 String
"mdatalooksnewestfirst" forall a b. (a -> b) -> a -> b
$
        case forall a. Eq a => [a] -> [a]
nub forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map Transaction -> Day
tdate [Transaction]
txns of
          [Day]
ds | forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length [Day]
ds forall a. Ord a => a -> a -> Bool
> CsvFieldIndex
1 -> forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ forall a. [a] -> a
head [Day]
ds forall a. Ord a => a -> a -> Bool
> forall a. [a] -> a
last [Day]
ds
          [Day]
_                  -> forall a. Maybe a
Nothing
      txns2 :: [Transaction]
txns2 = forall a. Show a => String -> a -> a
dbg7 String
"txns2" forall a b. (a -> b) -> a -> b
$
        (if Bool
newestfirst Bool -> Bool -> Bool
|| Maybe Bool
mdatalooksnewestfirst forall a. Eq a => a -> a -> Bool
== forall a. a -> Maybe a
Just Bool
True then forall a. [a] -> [a]
reverse else forall a. a -> a
id) [Transaction]
txns1
      -- 3. Disordered dates: in case the CSV records were ordered by chaos,
      -- do a final sort by date. If it was only a few records out of order,
      -- this will hopefully refine any good ordering done by steps 1 and 2.
      txns3 :: [Transaction]
txns3 = forall a. Show a => String -> a -> a
dbg7 String
"date-sorted csv txns" forall a b. (a -> b) -> a -> b
$ forall b a. Ord b => (a -> b) -> [a] -> [a]
sortOn Transaction -> Day
tdate [Transaction]
txns2

    forall (m :: * -> *) a. Monad m => a -> m a
return Journal
nulljournal{jtxns :: [Transaction]
jtxns=[Transaction]
txns3}

-- | Parse special separator names TAB and SPACE, or return the first
-- character. Return Nothing on empty string
parseSeparator :: Text -> Maybe Char
parseSeparator :: CsvAmountString -> Maybe Char
parseSeparator = CsvAmountString -> Maybe Char
specials forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString
T.toLower
  where specials :: CsvAmountString -> Maybe Char
specials CsvAmountString
"space" = forall a. a -> Maybe a
Just Char
' '
        specials CsvAmountString
"tab"   = forall a. a -> Maybe a
Just Char
'\t'
        specials CsvAmountString
xs      = forall a b. (a, b) -> a
fst forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
xs

-- Call parseCassava on a file or stdin, converting the result to ExceptT.
parseCsv :: Char -> FilePath -> Text -> ExceptT String IO [CsvRecord]
parseCsv :: Char
-> String
-> CsvAmountString
-> ExceptT String IO [[CsvAmountString]]
parseCsv Char
separator String
filePath CsvAmountString
csvtext = forall e (m :: * -> *) a. m (Either e a) -> ExceptT e m a
ExceptT forall a b. (a -> b) -> a -> b
$
  case String
filePath of
    String
"-" -> Char
-> String -> CsvAmountString -> Either String [[CsvAmountString]]
parseCassava Char
separator String
"(stdin)" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IO CsvAmountString
T.getContents
    String
_   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ if CsvAmountString -> Bool
T.null CsvAmountString
csvtext then forall a b. b -> Either a b
Right forall a. Monoid a => a
mempty else Char
-> String -> CsvAmountString -> Either String [[CsvAmountString]]
parseCassava Char
separator String
filePath CsvAmountString
csvtext

-- Parse text into CSV records, using Cassava and the given field separator.
parseCassava :: Char -> FilePath -> Text -> Either String [CsvRecord]
parseCassava :: Char
-> String -> CsvAmountString -> Either String [[CsvAmountString]]
parseCassava Char
separator String
path CsvAmountString
content =
  -- XXX we now remove all blank lines before parsing; will Cassava will still produce [""] records ?
  -- filter (/=[""])
  forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either (forall a b. a -> Either a b
Left forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall s e.
(VisualStream s, TraversableStream s, ShowErrorComponent e) =>
ParseErrorBundle s e -> String
errorBundlePretty) (forall a b. b -> Either a b
Right forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (t :: * -> *).
(Foldable t, Functor t) =>
t (t ByteString) -> [[CsvAmountString]]
parseResultToCsv) forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>
  forall a.
FromRecord a =>
DecodeOptions
-> HasHeader
-> String
-> ByteString
-> Either (ParseErrorBundle ByteString ConversionError) (Vector a)
CassavaMegaparsec.decodeWith DecodeOptions
decodeOptions HasHeader
Cassava.NoHeader String
path forall a b. (a -> b) -> a -> b
$
  ByteString -> ByteString
BL.fromStrict forall a b. (a -> b) -> a -> b
$ CsvAmountString -> ByteString
T.encodeUtf8 CsvAmountString
content
  where
    decodeOptions :: DecodeOptions
decodeOptions = DecodeOptions
Cassava.defaultDecodeOptions {
                      decDelimiter :: Word8
Cassava.decDelimiter = forall a b. (Integral a, Num b) => a -> b
fromIntegral (Char -> CsvFieldIndex
ord Char
separator)
                    }
    parseResultToCsv :: (Foldable t, Functor t) => t (t B.ByteString) -> [CsvRecord]
    parseResultToCsv :: forall (t :: * -> *).
(Foldable t, Functor t) =>
t (t ByteString) -> [[CsvAmountString]]
parseResultToCsv = forall {a}. t (t a) -> [[a]]
toListList forall b c a. (b -> c) -> (a -> b) -> a -> c
. t (t ByteString) -> t (t CsvAmountString)
unpackFields
      where
        toListList :: t (t a) -> [[a]]
toListList = forall (t :: * -> *) a. Foldable t => t a -> [a]
toList forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap forall (t :: * -> *) a. Foldable t => t a -> [a]
toList
        unpackFields :: t (t ByteString) -> t (t CsvAmountString)
unpackFields  = (forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap) ByteString -> CsvAmountString
T.decodeUtf8

-- | Scan for csv records where a conditional `skip` or `end` rule applies,
-- and apply that rule, removing one or more following records.
applyConditionalSkips :: CsvRules -> [CsvRecord] -> [CsvRecord]
applyConditionalSkips :: CsvRules -> [[CsvAmountString]] -> [[CsvAmountString]]
applyConditionalSkips CsvRules
_ [] = []
applyConditionalSkips CsvRules
rules ([CsvAmountString]
r:[[CsvAmountString]]
rest) =
  case forall {a}.
(Bounded a, Num a, Read a) =>
[CsvAmountString] -> Maybe a
skipnum [CsvAmountString]
r of
    Maybe CsvFieldIndex
Nothing -> [CsvAmountString]
r forall a. a -> [a] -> [a]
: CsvRules -> [[CsvAmountString]] -> [[CsvAmountString]]
applyConditionalSkips CsvRules
rules [[CsvAmountString]]
rest
    Just CsvFieldIndex
cnt -> CsvRules -> [[CsvAmountString]] -> [[CsvAmountString]]
applyConditionalSkips CsvRules
rules forall a b. (a -> b) -> a -> b
$ forall a. CsvFieldIndex -> [a] -> [a]
drop (CsvFieldIndex
cntforall a. Num a => a -> a -> a
-CsvFieldIndex
1) [[CsvAmountString]]
rest
  where
    skipnum :: [CsvAmountString] -> Maybe a
skipnum [CsvAmountString]
r1 =
      case (CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString]
r1 CsvAmountString
"end", CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString]
r1 CsvAmountString
"skip") of
        (Maybe CsvAmountString
Nothing, Maybe CsvAmountString
Nothing) -> forall a. Maybe a
Nothing
        (Just CsvAmountString
_, Maybe CsvAmountString
_) -> forall a. a -> Maybe a
Just forall a. Bounded a => a
maxBound
        (Maybe CsvAmountString
Nothing, Just CsvAmountString
"") -> forall a. a -> Maybe a
Just a
1
        (Maybe CsvAmountString
Nothing, Just CsvAmountString
x) -> forall a. a -> Maybe a
Just (forall a. Read a => String -> a
read forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
x)

-- | Do some validation on the parsed CSV records:
-- check that they all have at least two fields.
validateCsv :: [CsvRecord] -> Either String [CsvRecord]
validateCsv :: [[CsvAmountString]] -> Either String [[CsvAmountString]]
validateCsv [] = forall a b. b -> Either a b
Right []
validateCsv rs :: [[CsvAmountString]]
rs@([CsvAmountString]
_first:[[CsvAmountString]]
_) =
  case Maybe [CsvAmountString]
lessthan2 of
    Just [CsvAmountString]
r  -> forall a b. a -> Either a b
Left forall a b. (a -> b) -> a -> b
$ forall r. PrintfType r => String -> r
printf String
"CSV record %s has less than two fields" (forall a. Show a => a -> String
show [CsvAmountString]
r)
    Maybe [CsvAmountString]
Nothing -> forall a b. b -> Either a b
Right [[CsvAmountString]]
rs
  where
    lessthan2 :: Maybe [CsvAmountString]
lessthan2 = forall a. [a] -> Maybe a
headMay forall a b. (a -> b) -> a -> b
$ forall a. (a -> Bool) -> [a] -> [a]
filter ((forall a. Ord a => a -> a -> Bool
<CsvFieldIndex
2)forall b c a. (b -> c) -> (a -> b) -> a -> c
.forall (t :: * -> *) a. Foldable t => t a -> CsvFieldIndex
length) [[CsvAmountString]]
rs

-- -- | The highest (0-based) field index referenced in the field
-- -- definitions, or -1 if no fields are defined.
-- maxFieldIndex :: CsvRules -> Int
-- maxFieldIndex r = maximumDef (-1) $ catMaybes [
--                    dateField r
--                   ,statusField r
--                   ,codeField r
--                   ,amountField r
--                   ,amountInField r
--                   ,amountOutField r
--                   ,currencyField r
--                   ,accountField r
--                   ,account2Field r
--                   ,date2Field r
--                   ]

--- ** converting csv records to transactions

transactionFromCsvRecord :: Bool -> Maybe TimeZone -> TimeZone -> SourcePos -> CsvRules -> CsvRecord -> Transaction
transactionFromCsvRecord :: Bool
-> Maybe TimeZone
-> TimeZone
-> SourcePos
-> CsvRules
-> [CsvAmountString]
-> Transaction
transactionFromCsvRecord Bool
timesarezoned Maybe TimeZone
mtzin TimeZone
tzout SourcePos
sourcepos CsvRules
rules [CsvAmountString]
record = Transaction
t
  where
    ----------------------------------------------------------------------
    -- 1. Define some helpers:

    rule :: CsvAmountString -> Maybe CsvAmountString
rule     = CsvRules -> CsvAmountString -> Maybe CsvAmountString
csvRule           CsvRules
rules        :: DirectiveName    -> Maybe FieldTemplate
    -- ruleval  = csvRuleValue      rules record :: DirectiveName    -> Maybe String
    field :: CsvAmountString -> Maybe CsvAmountString
field    = CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerField      CsvRules
rules [CsvAmountString]
record :: HledgerFieldName -> Maybe FieldTemplate
    fieldval :: CsvAmountString -> Maybe CsvAmountString
fieldval = CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerFieldValue CsvRules
rules [CsvAmountString]
record :: HledgerFieldName -> Maybe Text
    mdateformat :: Maybe CsvAmountString
mdateformat = CsvAmountString -> Maybe CsvAmountString
rule CsvAmountString
"date-format"
    parsedate :: CsvAmountString -> Maybe Day
parsedate = Bool
-> Maybe TimeZone
-> TimeZone
-> Maybe CsvAmountString
-> CsvAmountString
-> Maybe Day
parseDateWithCustomOrDefaultFormats Bool
timesarezoned Maybe TimeZone
mtzin TimeZone
tzout Maybe CsvAmountString
mdateformat
    mkdateerror :: CsvAmountString
-> CsvAmountString -> Maybe CsvAmountString -> String
mkdateerror CsvAmountString
datefield CsvAmountString
datevalue Maybe CsvAmountString
mdateformat' = CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ [CsvAmountString] -> CsvAmountString
T.unlines
      [CsvAmountString
"error: could not parse \""forall a. Semigroup a => a -> a -> a
<>CsvAmountString
datevalueforall a. Semigroup a => a -> a -> a
<>CsvAmountString
"\" as a date using date format "
        forall a. Semigroup a => a -> a -> a
<>forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"\"YYYY/M/D\", \"YYYY-M-D\" or \"YYYY.M.D\"" (String -> CsvAmountString
T.pack forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Show a => a -> String
show) Maybe CsvAmountString
mdateformat'
      ,[CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
record
      ,CsvAmountString
"the "forall a. Semigroup a => a -> a -> a
<>CsvAmountString
datefieldforall a. Semigroup a => a -> a -> a
<>CsvAmountString
" rule is:   "forall a. Semigroup a => a -> a -> a
<>(forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"required, but missing" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
field CsvAmountString
datefield)
      ,CsvAmountString
"the date-format is: "forall a. Semigroup a => a -> a -> a
<>forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"unspecified" Maybe CsvAmountString
mdateformat'
      ,CsvAmountString
"you may need to "
        forall a. Semigroup a => a -> a -> a
<>CsvAmountString
"change your "forall a. Semigroup a => a -> a -> a
<>CsvAmountString
datefieldforall a. Semigroup a => a -> a -> a
<>CsvAmountString
" rule, "
        forall a. Semigroup a => a -> a -> a
<>forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"add a" (forall a b. a -> b -> a
const CsvAmountString
"change your") Maybe CsvAmountString
mdateformat'forall a. Semigroup a => a -> a -> a
<>CsvAmountString
" date-format rule, "
        forall a. Semigroup a => a -> a -> a
<>CsvAmountString
"or "forall a. Semigroup a => a -> a -> a
<>forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"add a" (forall a b. a -> b -> a
const CsvAmountString
"change your") Maybe CsvAmountString
mskipforall a. Semigroup a => a -> a -> a
<>CsvAmountString
" skip rule"
      ,CsvAmountString
"for m/d/y or d/m/y dates, use date-format %-m/%-d/%Y or date-format %-d/%-m/%Y"
      ]
      where
        mskip :: Maybe CsvAmountString
mskip = CsvAmountString -> Maybe CsvAmountString
rule CsvAmountString
"skip"

    ----------------------------------------------------------------------
    -- 2. Gather values needed for the transaction itself, by evaluating the
    -- field assignment rules using the CSV record's data, and parsing a bit
    -- more where needed (dates, status).

    date :: CsvAmountString
date        = forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"date"
    -- PARTIAL:
    date' :: Day
date'       = forall a. a -> Maybe a -> a
fromMaybe (forall a. String -> a
error' forall a b. (a -> b) -> a -> b
$ CsvAmountString
-> CsvAmountString -> Maybe CsvAmountString -> String
mkdateerror CsvAmountString
"date" CsvAmountString
date Maybe CsvAmountString
mdateformat) forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe Day
parsedate CsvAmountString
date
    mdate2 :: Maybe CsvAmountString
mdate2      = CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"date2"
    mdate2' :: Maybe Day
mdate2'     = (forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall a. String -> a
error' forall a b. (a -> b) -> a -> b
$ CsvAmountString
-> CsvAmountString -> Maybe CsvAmountString -> String
mkdateerror CsvAmountString
"date2" (forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" Maybe CsvAmountString
mdate2) Maybe CsvAmountString
mdateformat) forall a. a -> Maybe a
Just forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> Maybe Day
parsedate) forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< Maybe CsvAmountString
mdate2
    status :: Status
status      =
      case CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"status" of
        Maybe CsvAmountString
Nothing -> Status
Unmarked
        Just CsvAmountString
s  -> forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either forall {c}.
ParseErrorBundle CsvAmountString HledgerParseErrorData -> c
statuserror forall a. a -> a
id forall a b. (a -> b) -> a -> b
$ forall e s a.
Parsec e s a -> String -> s -> Either (ParseErrorBundle s e) a
runParser (forall (m :: * -> *). TextParser m Status
statusp forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall e s (m :: * -> *). MonadParsec e s m => m ()
eof) String
"" CsvAmountString
s
          where
            statuserror :: ParseErrorBundle CsvAmountString HledgerParseErrorData -> c
statuserror ParseErrorBundle CsvAmountString HledgerParseErrorData
err = forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ [CsvAmountString] -> CsvAmountString
T.unlines
              [CsvAmountString
"error: could not parse \""forall a. Semigroup a => a -> a -> a
<>CsvAmountString
sforall a. Semigroup a => a -> a -> a
<>CsvAmountString
"\" as a cleared status (should be *, ! or empty)"
              ,CsvAmountString
"the parse error is:      "forall a. Semigroup a => a -> a -> a
<>String -> CsvAmountString
T.pack (ParseErrorBundle CsvAmountString HledgerParseErrorData -> String
customErrorBundlePretty ParseErrorBundle CsvAmountString HledgerParseErrorData
err)
              ]
    code :: CsvAmountString
code        = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"" CsvAmountString -> CsvAmountString
singleline' forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"code"
    description :: CsvAmountString
description = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"" CsvAmountString -> CsvAmountString
singleline' forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"description"
    comment :: CsvAmountString
comment     = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"" CsvAmountString -> CsvAmountString
unescapeNewlines forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"comment"
    precomment :: CsvAmountString
precomment  = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"" CsvAmountString -> CsvAmountString
unescapeNewlines forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"precomment"

    singleline' :: CsvAmountString -> CsvAmountString
singleline' = [CsvAmountString] -> CsvAmountString
T.unwords forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> Bool
T.null) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
map CsvAmountString -> CsvAmountString
T.strip forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> [CsvAmountString]
T.lines
    unescapeNewlines :: CsvAmountString -> CsvAmountString
unescapeNewlines = CsvAmountString -> [CsvAmountString] -> CsvAmountString
T.intercalate CsvAmountString
"\n" forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString -> [CsvAmountString]
T.splitOn CsvAmountString
"\\n"

    ----------------------------------------------------------------------
    -- 3. Generate the postings for which an account has been assigned
    -- (possibly indirectly due to an amount or balance assignment)

    p1IsVirtual :: Bool
p1IsVirtual = (CsvAmountString -> PostingType
accountNamePostingType forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"account1") forall a. Eq a => a -> a -> Bool
== forall a. a -> Maybe a
Just PostingType
VirtualPosting
    ps :: [Posting]
ps = [Posting
p | CsvFieldIndex
n <- [CsvFieldIndex
1..CsvFieldIndex
maxpostings]
         ,let cmt :: CsvAmountString
cmt  = forall b a. b -> (a -> b) -> Maybe a -> b
maybe CsvAmountString
"" CsvAmountString -> CsvAmountString
unescapeNewlines forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Maybe CsvAmountString
fieldval (CsvAmountString
"comment"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n))
         ,let currency :: CsvAmountString
currency = forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" (CsvAmountString -> Maybe CsvAmountString
fieldval (CsvAmountString
"currency"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n)) forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"currency")
         ,let mamount :: Maybe MixedAmount
mamount  = CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> Bool
-> CsvFieldIndex
-> Maybe MixedAmount
getAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency Bool
p1IsVirtual CsvFieldIndex
n
         ,let mbalance :: Maybe (Amount, SourcePos)
mbalance = CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvFieldIndex
-> Maybe (Amount, SourcePos)
getBalance CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvFieldIndex
n
         ,Just (CsvAmountString
acct,Bool
isfinal) <- [CsvRules
-> [CsvAmountString]
-> Maybe MixedAmount
-> Maybe (Amount, SourcePos)
-> CsvFieldIndex
-> Maybe (CsvAmountString, Bool)
getAccount CsvRules
rules [CsvAmountString]
record Maybe MixedAmount
mamount Maybe (Amount, SourcePos)
mbalance CsvFieldIndex
n]  -- skips Nothings
         ,let acct' :: CsvAmountString
acct' | Bool -> Bool
not Bool
isfinal Bool -> Bool -> Bool
&& CsvAmountString
acctforall a. Eq a => a -> a -> Bool
==CsvAmountString
unknownExpenseAccount Bool -> Bool -> Bool
&&
                      forall a. a -> Maybe a -> a
fromMaybe Bool
False (Maybe MixedAmount
mamount forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= MixedAmount -> Maybe Bool
isNegativeMixedAmount) = CsvAmountString
unknownIncomeAccount
                    | Bool
otherwise = CsvAmountString
acct
         ,let p :: Posting
p = Posting
nullposting{paccount :: CsvAmountString
paccount          = CsvAmountString -> CsvAmountString
accountNameWithoutPostingType CsvAmountString
acct'
                             ,pamount :: MixedAmount
pamount           = forall a. a -> Maybe a -> a
fromMaybe MixedAmount
missingmixedamt Maybe MixedAmount
mamount
                             ,ptransaction :: Maybe Transaction
ptransaction      = forall a. a -> Maybe a
Just Transaction
t
                             ,pbalanceassertion :: Maybe BalanceAssertion
pbalanceassertion = CsvRules
-> [CsvAmountString] -> (Amount, SourcePos) -> BalanceAssertion
mkBalanceAssertion CsvRules
rules [CsvAmountString]
record forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe (Amount, SourcePos)
mbalance
                             ,pcomment :: CsvAmountString
pcomment          = CsvAmountString
cmt
                             ,ptype :: PostingType
ptype             = CsvAmountString -> PostingType
accountNamePostingType CsvAmountString
acct
                             }
         ]

    ----------------------------------------------------------------------
    -- 4. Build the transaction (and name it, so the postings can reference it).

    t :: Transaction
t = Transaction
nulltransaction{
           tsourcepos :: (SourcePos, SourcePos)
tsourcepos        = (SourcePos
sourcepos, SourcePos
sourcepos)  -- the CSV line number
          ,tdate :: Day
tdate             = Day
date'
          ,tdate2 :: Maybe Day
tdate2            = Maybe Day
mdate2'
          ,tstatus :: Status
tstatus           = Status
status
          ,tcode :: CsvAmountString
tcode             = CsvAmountString
code
          ,tdescription :: CsvAmountString
tdescription      = CsvAmountString
description
          ,tcomment :: CsvAmountString
tcomment          = CsvAmountString
comment
          ,tprecedingcomment :: CsvAmountString
tprecedingcomment = CsvAmountString
precomment
          ,tpostings :: [Posting]
tpostings         = [Posting]
ps
          }

-- | Parse the date string using the specified date-format, or if unspecified
-- the "simple date" formats (YYYY/MM/DD, YYYY-MM-DD, YYYY.MM.DD, leading
-- zeroes optional). If a timezone is provided, we assume the DateFormat
-- produces a zoned time and we localise that to the given timezone.
parseDateWithCustomOrDefaultFormats :: Bool -> Maybe TimeZone -> TimeZone -> Maybe DateFormat -> Text -> Maybe Day
parseDateWithCustomOrDefaultFormats :: Bool
-> Maybe TimeZone
-> TimeZone
-> Maybe CsvAmountString
-> CsvAmountString
-> Maybe Day
parseDateWithCustomOrDefaultFormats Bool
timesarezoned Maybe TimeZone
mtzin TimeZone
tzout Maybe CsvAmountString
mformat CsvAmountString
s = UTCTime -> Day
localdate forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe UTCTime
mutctime
  -- this time code can probably be simpler, I'm just happy to get out alive
  where
    UTCTime -> Day
localdate :: UTCTime -> Day =
      LocalTime -> Day
localDay forall b c a. (b -> c) -> (a -> b) -> a -> c
.
      forall a. Show a => String -> a -> a
dbg7 (String
"time in output timezone "forall a. [a] -> [a] -> [a]
++forall a. Show a => a -> String
show TimeZone
tzout) forall b c a. (b -> c) -> (a -> b) -> a -> c
.
      TimeZone -> UTCTime -> LocalTime
utcToLocalTime TimeZone
tzout
    Maybe UTCTime
mutctime :: Maybe UTCTime = forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
asum forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map String -> Maybe UTCTime
parseWithFormat [String]
formats

    parseWithFormat :: String -> Maybe UTCTime
    parseWithFormat :: String -> Maybe UTCTime
parseWithFormat String
fmt =
      if Bool
timesarezoned
      then
        forall a. Show a => String -> a -> a
dbg7 String
"zoned CSV time, expressed as UTC" forall a b. (a -> b) -> a -> b
$
        forall (m :: * -> *) t.
(MonadFail m, ParseTime t) =>
Bool -> TimeLocale -> String -> String -> m t
parseTimeM Bool
True TimeLocale
defaultTimeLocale String
fmt forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
s :: Maybe UTCTime
      else
        -- parse as a local day and time; then if an input timezone is provided,
        -- assume it's in that, otherwise assume it's in the output timezone;
        -- then convert to UTC like the above
        let
          mlocaltime :: Maybe LocalTime
mlocaltime =
            forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (forall a. Show a => String -> a -> a
dbg7 String
"unzoned CSV time") forall a b. (a -> b) -> a -> b
$
            forall (m :: * -> *) t.
(MonadFail m, ParseTime t) =>
Bool -> TimeLocale -> String -> String -> m t
parseTimeM Bool
True TimeLocale
defaultTimeLocale String
fmt forall a b. (a -> b) -> a -> b
$ CsvAmountString -> String
T.unpack CsvAmountString
s :: Maybe LocalTime
          localTimeAsZonedTime :: TimeZone -> LocalTime -> ZonedTime
localTimeAsZonedTime TimeZone
tz LocalTime
lt =  LocalTime -> TimeZone -> ZonedTime
ZonedTime LocalTime
lt TimeZone
tz
        in
          case Maybe TimeZone
mtzin of
            Just TimeZone
tzin ->
              (forall a. Show a => String -> a -> a
dbg7 (String
"unzoned CSV time, declared as "forall a. [a] -> [a] -> [a]
++forall a. Show a => a -> String
show TimeZone
tzinforall a. [a] -> [a] -> [a]
++ String
", expressed as UTC") forall b c a. (b -> c) -> (a -> b) -> a -> c
.
              TimeZone -> LocalTime -> UTCTime
localTimeToUTC TimeZone
tzin)
              forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe LocalTime
mlocaltime
            Maybe TimeZone
Nothing ->
              (forall a. Show a => String -> a -> a
dbg7 (String
"unzoned CSV time, treated as "forall a. [a] -> [a] -> [a]
++forall a. Show a => a -> String
show TimeZone
tzoutforall a. [a] -> [a] -> [a]
++ String
", expressed as UTC") forall b c a. (b -> c) -> (a -> b) -> a -> c
.
                ZonedTime -> UTCTime
zonedTimeToUTC forall b c a. (b -> c) -> (a -> b) -> a -> c
.
                TimeZone -> LocalTime -> ZonedTime
localTimeAsZonedTime TimeZone
tzout)
              forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe LocalTime
mlocaltime

    formats :: [String]
formats = forall a b. (a -> b) -> [a] -> [b]
map CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ forall b a. b -> (a -> b) -> Maybe a -> b
maybe
               [CsvAmountString
"%Y/%-m/%-d"
               ,CsvAmountString
"%Y-%-m-%-d"
               ,CsvAmountString
"%Y.%-m.%-d"
               -- ,"%-m/%-d/%Y"
                -- ,parseTimeM TruedefaultTimeLocale "%Y/%m/%e" (take 5 s ++ "0" ++ drop 5 s)
                -- ,parseTimeM TruedefaultTimeLocale "%Y-%m-%e" (take 5 s ++ "0" ++ drop 5 s)
                -- ,parseTimeM TruedefaultTimeLocale "%m/%e/%Y" ('0':s)
                -- ,parseTimeM TruedefaultTimeLocale "%m-%e-%Y" ('0':s)
               ]
               (forall a. a -> [a] -> [a]
:[])
                Maybe CsvAmountString
mformat

-- | Figure out the amount specified for posting N, if any.
-- A currency symbol to prepend to the amount, if any, is provided,
-- and whether posting 1 requires balancing or not.
-- This looks for a non-empty amount value assigned to "amountN", "amountN-in", or "amountN-out".
-- For postings 1 or 2 it also looks at "amount", "amount-in", "amount-out".
-- If more than one of these has a value, it looks for one that is non-zero.
-- If there's multiple non-zeros, or no non-zeros but multiple zeros, it throws an error.
getAmount :: CsvRules -> CsvRecord -> Text -> Bool -> Int -> Maybe MixedAmount
getAmount :: CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> Bool
-> CsvFieldIndex
-> Maybe MixedAmount
getAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency Bool
p1IsVirtual CsvFieldIndex
n =
  -- Warning! Many tricky corner cases here.
  -- Keep synced with:
  -- hledger_csv.m4.md -> CSV FORMAT -> "amount", "Setting amounts",
  -- hledger/test/csv.test -> 13, 31-34
  let
    unnumberedfieldnames :: [CsvAmountString]
unnumberedfieldnames = [CsvAmountString
"amount",CsvAmountString
"amount-in",CsvAmountString
"amount-out"]

    -- amount field names which can affect this posting
    fieldnames :: [CsvAmountString]
fieldnames = forall a b. (a -> b) -> [a] -> [b]
map ((CsvAmountString
"amount"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n))forall a. Semigroup a => a -> a -> a
<>) [CsvAmountString
"",CsvAmountString
"-in",CsvAmountString
"-out"]
                 -- For posting 1, also recognise the old amount/amount-in/amount-out names.
                 -- For posting 2, the same but only if posting 1 needs balancing.
                 forall a. [a] -> [a] -> [a]
++ if CsvFieldIndex
nforall a. Eq a => a -> a -> Bool
==CsvFieldIndex
1 Bool -> Bool -> Bool
|| CsvFieldIndex
nforall a. Eq a => a -> a -> Bool
==CsvFieldIndex
2 Bool -> Bool -> Bool
&& Bool -> Bool
not Bool
p1IsVirtual then [CsvAmountString]
unnumberedfieldnames else []

    -- assignments to any of these field names with non-empty values
    assignments :: [(CsvAmountString, MixedAmount)]
assignments = [(CsvAmountString
f,MixedAmount
a') | CsvAmountString
f <- [CsvAmountString]
fieldnames
                          , Just CsvAmountString
v <- [CsvAmountString -> CsvAmountString
T.strip forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvRules -> [CsvAmountString] -> CsvAmountString -> CsvAmountString
renderTemplate CsvRules
rules [CsvAmountString]
record forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerField CsvRules
rules [CsvAmountString]
record CsvAmountString
f]
                          , Bool -> Bool
not forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Bool
T.null CsvAmountString
v
                          -- XXX maybe ignore rule-generated values like "", "-", "$", "-$", "$-" ? cf CSV FORMAT -> "amount", "Setting amounts",
                          , let a :: MixedAmount
a = CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvAmountString
-> MixedAmount
parseAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvAmountString
v
                          -- With amount/amount-in/amount-out, in posting 2,
                          -- flip the sign and convert to cost, as they did before 1.17
                          , let a' :: MixedAmount
a' = if CsvAmountString
f forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [CsvAmountString]
unnumberedfieldnames Bool -> Bool -> Bool
&& CsvFieldIndex
nforall a. Eq a => a -> a -> Bool
==CsvFieldIndex
2 then MixedAmount -> MixedAmount
mixedAmountCost (MixedAmount -> MixedAmount
maNegate MixedAmount
a) else MixedAmount
a
                          ]

    -- if any of the numbered field names are present, discard all the unnumbered ones
    discardUnnumbered :: [(CsvAmountString, b)] -> [(CsvAmountString, b)]
discardUnnumbered [(CsvAmountString, b)]
xs = if forall (t :: * -> *) a. Foldable t => t a -> Bool
null [(CsvAmountString, b)]
numbered then [(CsvAmountString, b)]
xs else [(CsvAmountString, b)]
numbered
      where
        numbered :: [(CsvAmountString, b)]
numbered = forall a. (a -> Bool) -> [a] -> [a]
filter ((Char -> Bool) -> CsvAmountString -> Bool
T.any Char -> Bool
isDigit forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a, b) -> a
fst) [(CsvAmountString, b)]
xs

    -- discard all zero amounts, unless all amounts are zero, in which case discard all but the first
    discardExcessZeros :: [(a, MixedAmount)] -> [(a, MixedAmount)]
discardExcessZeros [(a, MixedAmount)]
xs = if forall (t :: * -> *) a. Foldable t => t a -> Bool
null [(a, MixedAmount)]
nonzeros then forall a. CsvFieldIndex -> [a] -> [a]
take CsvFieldIndex
1 [(a, MixedAmount)]
xs else [(a, MixedAmount)]
nonzeros
      where
        nonzeros :: [(a, MixedAmount)]
nonzeros = forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not forall b c a. (b -> c) -> (a -> b) -> a -> c
. MixedAmount -> Bool
mixedAmountLooksZero forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a, b) -> b
snd) [(a, MixedAmount)]
xs

    -- for -out fields, flip the sign  XXX unless it's already negative ? back compat issues / too confusing ?
    negateIfOut :: CsvAmountString -> MixedAmount -> MixedAmount
negateIfOut CsvAmountString
f = if CsvAmountString
"-out" CsvAmountString -> CsvAmountString -> Bool
`T.isSuffixOf` CsvAmountString
f then MixedAmount -> MixedAmount
maNegate else forall a. a -> a
id

  in case forall {a}. [(a, MixedAmount)] -> [(a, MixedAmount)]
discardExcessZeros forall a b. (a -> b) -> a -> b
$ forall {b}. [(CsvAmountString, b)] -> [(CsvAmountString, b)]
discardUnnumbered [(CsvAmountString, MixedAmount)]
assignments of
      []      -> forall a. Maybe a
Nothing
      [(CsvAmountString
f,MixedAmount
a)] -> forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ CsvAmountString -> MixedAmount -> MixedAmount
negateIfOut CsvAmountString
f MixedAmount
a
      [(CsvAmountString, MixedAmount)]
fs      -> forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> CsvAmountString
textChomp forall b c a. (b -> c) -> (a -> b) -> a -> c
. [CsvAmountString] -> CsvAmountString
T.unlines forall a b. (a -> b) -> a -> b
$  -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
          -- PARTIAL:
        [CsvAmountString
"in CSV rules:"
        ,CsvAmountString
"While processing " forall a. Semigroup a => a -> a -> a
<> [CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
record
        ,CsvAmountString
"while calculating amount for posting " forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n)
        ] forall a. [a] -> [a] -> [a]
++
        [CsvAmountString
"rule \"" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
f forall a. Semigroup a => a -> a -> a
<> CsvAmountString
" " forall a. Semigroup a => a -> a -> a
<>
          forall a. a -> Maybe a -> a
fromMaybe CsvAmountString
"" (CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerField CsvRules
rules [CsvAmountString]
record CsvAmountString
f) forall a. Semigroup a => a -> a -> a
<>
          CsvAmountString
"\" assigned value \"" forall a. Semigroup a => a -> a -> a
<> WideBuilder -> CsvAmountString
wbToText (AmountDisplayOpts -> MixedAmount -> WideBuilder
showMixedAmountB AmountDisplayOpts
noColour MixedAmount
a) forall a. Semigroup a => a -> a -> a
<> CsvAmountString
"\"" -- XXX not sure this is showing all the right info
          | (CsvAmountString
f,MixedAmount
a) <- [(CsvAmountString, MixedAmount)]
fs
        ] forall a. [a] -> [a] -> [a]
++
        [CsvAmountString
""
        ,CsvAmountString
"Multiple non-zero amounts were assigned for an amount field."
        ,CsvAmountString
"Please ensure just one non-zero amount is assigned, perhaps with an if rule."
        ,CsvAmountString
"See also: https://hledger.org/hledger.html#setting-amounts"
        ,CsvAmountString
"(hledger manual -> CSV format -> Tips -> Setting amounts)"
        ]
-- | Figure out the expected balance (assertion or assignment) specified for posting N,
-- if any (and its parse position).
getBalance :: CsvRules -> CsvRecord -> Text -> Int -> Maybe (Amount, SourcePos)
getBalance :: CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvFieldIndex
-> Maybe (Amount, SourcePos)
getBalance CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvFieldIndex
n = do
  CsvAmountString
v <- (CsvAmountString -> Maybe CsvAmountString
fieldval (CsvAmountString
"balance"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n))
        -- for posting 1, also recognise the old field name
        forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> if CsvFieldIndex
nforall a. Eq a => a -> a -> Bool
==CsvFieldIndex
1 then CsvAmountString -> Maybe CsvAmountString
fieldval CsvAmountString
"balance" else forall a. Maybe a
Nothing)
  case CsvAmountString
v of
    CsvAmountString
"" -> forall a. Maybe a
Nothing
    CsvAmountString
s  -> forall a. a -> Maybe a
Just (
            CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvFieldIndex
-> CsvAmountString
-> Amount
parseBalanceAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvFieldIndex
n CsvAmountString
s
           ,String -> SourcePos
initialPos String
""  -- parse position to show when assertion fails,
           )               -- XXX the csv record's line number would be good
  where
    fieldval :: CsvAmountString -> Maybe CsvAmountString
fieldval = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap CsvAmountString -> CsvAmountString
T.strip forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerFieldValue CsvRules
rules [CsvAmountString]
record :: HledgerFieldName -> Maybe Text

-- | Given a non-empty amount string (from CSV) to parse, along with a
-- possibly non-empty currency symbol to prepend,
-- parse as a hledger MixedAmount (as in journal format), or raise an error.
-- The whole CSV record is provided for the error message.
parseAmount :: CsvRules -> CsvRecord -> Text -> Text -> MixedAmount
parseAmount :: CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvAmountString
-> MixedAmount
parseAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvAmountString
s =
    forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either forall {c}.
ParseErrorBundle CsvAmountString HledgerParseErrorData -> c
mkerror Amount -> MixedAmount
mixedAmount forall a b. (a -> b) -> a -> b
$  -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
      -- PARTIAL:
    forall e s a.
Parsec e s a -> String -> s -> Either (ParseErrorBundle s e) a
runParser (forall (m :: * -> *) s a. Monad m => StateT s m a -> s -> m a
evalStateT (forall (m :: * -> *). JournalParser m Amount
amountp forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall e s (m :: * -> *). MonadParsec e s m => m ()
eof) Journal
journalparsestate) String
"" forall a b. (a -> b) -> a -> b
$
    CsvAmountString
currency forall a. Semigroup a => a -> a -> a
<> CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
s
  where
    journalparsestate :: Journal
journalparsestate = Journal
nulljournal{jparsedecimalmark :: Maybe Char
jparsedecimalmark=CsvRules -> Maybe Char
parseDecimalMark CsvRules
rules}
    mkerror :: ParseErrorBundle CsvAmountString HledgerParseErrorData -> c
mkerror ParseErrorBundle CsvAmountString HledgerParseErrorData
e = forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ [CsvAmountString] -> CsvAmountString
T.unlines
      [CsvAmountString
"error: could not parse \"" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
s forall a. Semigroup a => a -> a -> a
<> CsvAmountString
"\" as an amount"
      ,[CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
record
      ,CsvRules -> [CsvAmountString] -> CsvAmountString
showRules CsvRules
rules [CsvAmountString]
record
      -- ,"the default-currency is: "++fromMaybe "unspecified" (getDirective "default-currency" rules)
      ,CsvAmountString
"the parse error is:      " forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (ParseErrorBundle CsvAmountString HledgerParseErrorData -> String
customErrorBundlePretty ParseErrorBundle CsvAmountString HledgerParseErrorData
e)
      ,CsvAmountString
"you may need to \
        \change your amount*, balance*, or currency* rules, \
        \or add or change your skip rule"
      ]

-- | Show the values assigned to each journal field.
showRules :: CsvRules -> [CsvAmountString] -> CsvAmountString
showRules CsvRules
rules [CsvAmountString]
record = [CsvAmountString] -> CsvAmountString
T.unlines forall a b. (a -> b) -> a -> b
$ forall a. [Maybe a] -> [a]
catMaybes
  [ ((CsvAmountString
"the "forall a. Semigroup a => a -> a -> a
<>CsvAmountString
fldforall a. Semigroup a => a -> a -> a
<>CsvAmountString
" rule is: ")forall a. Semigroup a => a -> a -> a
<>) forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>
    CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString]
record CsvAmountString
fld | CsvAmountString
fld <- [CsvAmountString]
journalfieldnames ]

-- | Show a (approximate) recreation of the original CSV record.
showRecord :: CsvRecord -> Text
showRecord :: [CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
r = CsvAmountString
"CSV record: "forall a. Semigroup a => a -> a -> a
<>CsvAmountString -> [CsvAmountString] -> CsvAmountString
T.intercalate CsvAmountString
"," (forall a b. (a -> b) -> [a] -> [b]
map (CsvAmountString
-> CsvAmountString -> CsvAmountString -> CsvAmountString
wrap CsvAmountString
"\"" CsvAmountString
"\"") [CsvAmountString]
r)

-- XXX unify these ^v

-- | Almost but not quite the same as parseAmount.
-- Given a non-empty amount string (from CSV) to parse, along with a
-- possibly non-empty currency symbol to prepend,
-- parse as a hledger Amount (as in journal format), or raise an error.
-- The CSV record and the field's numeric suffix are provided for the error message.
parseBalanceAmount :: CsvRules -> CsvRecord -> Text -> Int -> Text -> Amount
parseBalanceAmount :: CsvRules
-> [CsvAmountString]
-> CsvAmountString
-> CsvFieldIndex
-> CsvAmountString
-> Amount
parseBalanceAmount CsvRules
rules [CsvAmountString]
record CsvAmountString
currency CsvFieldIndex
n CsvAmountString
s =
  forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either (forall {a} {c}.
Show a =>
a
-> CsvAmountString
-> ParseErrorBundle CsvAmountString HledgerParseErrorData
-> c
mkerror CsvFieldIndex
n CsvAmountString
s) forall a. a -> a
id forall a b. (a -> b) -> a -> b
$
    forall e s a.
Parsec e s a -> String -> s -> Either (ParseErrorBundle s e) a
runParser (forall (m :: * -> *) s a. Monad m => StateT s m a -> s -> m a
evalStateT (forall (m :: * -> *). JournalParser m Amount
amountp forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall e s (m :: * -> *). MonadParsec e s m => m ()
eof) Journal
journalparsestate) String
"" forall a b. (a -> b) -> a -> b
$
    CsvAmountString
currency forall a. Semigroup a => a -> a -> a
<> CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
s
                  -- the csv record's line number would be good
  where
    journalparsestate :: Journal
journalparsestate = Journal
nulljournal{jparsedecimalmark :: Maybe Char
jparsedecimalmark=CsvRules -> Maybe Char
parseDecimalMark CsvRules
rules}
    mkerror :: a
-> CsvAmountString
-> ParseErrorBundle CsvAmountString HledgerParseErrorData
-> c
mkerror a
n' CsvAmountString
s' ParseErrorBundle CsvAmountString HledgerParseErrorData
e = forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ [CsvAmountString] -> CsvAmountString
T.unlines
      [CsvAmountString
"error: could not parse \"" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
s' forall a. Semigroup a => a -> a -> a
<> CsvAmountString
"\" as balance"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show a
n') forall a. Semigroup a => a -> a -> a
<> CsvAmountString
" amount"
      ,[CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
record
      ,CsvRules -> [CsvAmountString] -> CsvAmountString
showRules CsvRules
rules [CsvAmountString]
record
      -- ,"the default-currency is: "++fromMaybe "unspecified" mdefaultcurrency
      ,CsvAmountString
"the parse error is:      "forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (ParseErrorBundle CsvAmountString HledgerParseErrorData -> String
customErrorBundlePretty ParseErrorBundle CsvAmountString HledgerParseErrorData
e)
      ]

-- Read a valid decimal mark from the decimal-mark rule, if any.
-- If the rule is present with an invalid argument, raise an error.
parseDecimalMark :: CsvRules -> Maybe DecimalMark
parseDecimalMark :: CsvRules -> Maybe Char
parseDecimalMark CsvRules
rules = do
    CsvAmountString
s <- CsvRules
rules CsvRules -> CsvAmountString -> Maybe CsvAmountString
`csvRule` CsvAmountString
"decimal-mark"
    case CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
s of
        Just (Char
c, CsvAmountString
rest) | CsvAmountString -> Bool
T.null CsvAmountString
rest Bool -> Bool -> Bool
&& Char -> Bool
isDecimalMark Char
c -> forall (m :: * -> *) a. Monad m => a -> m a
return Char
c
        Maybe (Char, CsvAmountString)
_ -> forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ CsvAmountString
"decimal-mark's argument should be \".\" or \",\" (not \""forall a. Semigroup a => a -> a -> a
<>CsvAmountString
sforall a. Semigroup a => a -> a -> a
<>CsvAmountString
"\")"

-- | Make a balance assertion for the given amount, with the given parse
-- position (to be shown in assertion failures), with the assertion type
-- possibly set by a balance-type rule.
-- The CSV rules and current record are also provided, to be shown in case
-- balance-type's argument is bad (XXX refactor).
mkBalanceAssertion :: CsvRules -> CsvRecord -> (Amount, SourcePos) -> BalanceAssertion
mkBalanceAssertion :: CsvRules
-> [CsvAmountString] -> (Amount, SourcePos) -> BalanceAssertion
mkBalanceAssertion CsvRules
rules [CsvAmountString]
record (Amount
amt, SourcePos
pos) = BalanceAssertion
assrt{baamount :: Amount
baamount=Amount
amt, baposition :: SourcePos
baposition=SourcePos
pos}
  where
    assrt :: BalanceAssertion
assrt =
      case CsvAmountString -> CsvRules -> Maybe CsvAmountString
getDirective CsvAmountString
"balance-type" CsvRules
rules of
        Maybe CsvAmountString
Nothing    -> BalanceAssertion
nullassertion
        Just CsvAmountString
"="   -> BalanceAssertion
nullassertion
        Just CsvAmountString
"=="  -> BalanceAssertion
nullassertion{batotal :: Bool
batotal=Bool
True}
        Just CsvAmountString
"=*"  -> BalanceAssertion
nullassertion{bainclusive :: Bool
bainclusive=Bool
True}
        Just CsvAmountString
"==*" -> BalanceAssertion
nullassertion{batotal :: Bool
batotal=Bool
True, bainclusive :: Bool
bainclusive=Bool
True}
        Just CsvAmountString
x     -> forall a. String -> a
error' forall b c a. (b -> c) -> (a -> b) -> a -> c
. CsvAmountString -> String
T.unpack forall a b. (a -> b) -> a -> b
$ [CsvAmountString] -> CsvAmountString
T.unlines  -- PARTIAL:
          [ CsvAmountString
"balance-type \"" forall a. Semigroup a => a -> a -> a
<> CsvAmountString
x forall a. Semigroup a => a -> a -> a
<>CsvAmountString
"\" is invalid. Use =, ==, =* or ==*."
          , [CsvAmountString] -> CsvAmountString
showRecord [CsvAmountString]
record
          , CsvRules -> [CsvAmountString] -> CsvAmountString
showRules CsvRules
rules [CsvAmountString]
record
          ]

-- | Figure out the account name specified for posting N, if any.
-- And whether it is the default unknown account (which may be
-- improved later) or an explicitly set account (which may not).
getAccount :: CsvRules -> CsvRecord -> Maybe MixedAmount -> Maybe (Amount, SourcePos) -> Int -> Maybe (AccountName, Bool)
getAccount :: CsvRules
-> [CsvAmountString]
-> Maybe MixedAmount
-> Maybe (Amount, SourcePos)
-> CsvFieldIndex
-> Maybe (CsvAmountString, Bool)
getAccount CsvRules
rules [CsvAmountString]
record Maybe MixedAmount
mamount Maybe (Amount, SourcePos)
mbalance CsvFieldIndex
n =
  let
    fieldval :: CsvAmountString -> Maybe CsvAmountString
fieldval = CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
hledgerFieldValue CsvRules
rules [CsvAmountString]
record :: HledgerFieldName -> Maybe Text
    maccount :: Maybe CsvAmountString
maccount = CsvAmountString -> CsvAmountString
T.strip forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> CsvAmountString -> Maybe CsvAmountString
fieldval (CsvAmountString
"account"forall a. Semigroup a => a -> a -> a
<> String -> CsvAmountString
T.pack (forall a. Show a => a -> String
show CsvFieldIndex
n))
  in case Maybe CsvAmountString
maccount of
    -- accountN is set to the empty string - no posting will be generated
    Just CsvAmountString
"" -> forall a. Maybe a
Nothing
    -- accountN is set (possibly to "expenses:unknown"! #1192) - mark it final
    Just CsvAmountString
a  ->
      -- Check it and reject if invalid.. sometimes people try
      -- to set an amount or comment along with the account name.
      case forall e a.
Parsec e CsvAmountString a
-> CsvAmountString -> Either (ParseErrorBundle CsvAmountString e) a
parsewith (forall (m :: * -> *). TextParser m CsvAmountString
accountnamep forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall e s (m :: * -> *). MonadParsec e s m => m ()
eof) CsvAmountString
a of
        Left ParseErrorBundle CsvAmountString HledgerParseErrorData
e  -> forall a. String -> a
usageError forall a b. (a -> b) -> a -> b
$ forall s e.
(VisualStream s, TraversableStream s, ShowErrorComponent e) =>
ParseErrorBundle s e -> String
errorBundlePretty ParseErrorBundle CsvAmountString HledgerParseErrorData
e
        Right ()
_ -> forall a. a -> Maybe a
Just (CsvAmountString
a, Bool
True)
    -- accountN is unset
    Maybe CsvAmountString
Nothing ->
      case (Maybe MixedAmount
mamount, Maybe (Amount, SourcePos)
mbalance) of
        -- amountN is set, or implied by balanceN - set accountN to
        -- the default unknown account ("expenses:unknown") and
        -- allow it to be improved later
        (Just MixedAmount
_, Maybe (Amount, SourcePos)
_) -> forall a. a -> Maybe a
Just (CsvAmountString
unknownExpenseAccount, Bool
False)
        (Maybe MixedAmount
_, Just (Amount, SourcePos)
_) -> forall a. a -> Maybe a
Just (CsvAmountString
unknownExpenseAccount, Bool
False)
        -- amountN is also unset - no posting will be generated
        (Maybe MixedAmount
Nothing, Maybe (Amount, SourcePos)
Nothing) -> forall a. Maybe a
Nothing

-- | Default account names to use when needed.
unknownExpenseAccount :: CsvAmountString
unknownExpenseAccount = CsvAmountString
"expenses:unknown"
unknownIncomeAccount :: CsvAmountString
unknownIncomeAccount  = CsvAmountString
"income:unknown"

type CsvAmountString = Text

-- | Canonicalise the sign in a CSV amount string.
-- Such strings can have a minus sign, parentheses (equivalent to minus),
-- or any two of these (which cancel out),
-- or a plus sign (which is removed),
-- or any sign by itself with no following number (which is removed).
-- See hledger > CSV FORMAT > Tips > Setting amounts.
--
-- These are supported (note, not every possibile combination):
--
-- >>> simplifySign "1"
-- "1"
-- >>> simplifySign "+1"
-- "1"
-- >>> simplifySign "-1"
-- "-1"
-- >>> simplifySign "(1)"
-- "-1"
-- >>> simplifySign "--1"
-- "1"
-- >>> simplifySign "-(1)"
-- "1"
-- >>> simplifySign "-+1"
-- "-1"
-- >>> simplifySign "(-1)"
-- "1"
-- >>> simplifySign "((1))"
-- "1"
-- >>> simplifySign "-"
-- ""
-- >>> simplifySign "()"
-- ""
-- >>> simplifySign "+"
-- ""
simplifySign :: CsvAmountString -> CsvAmountString
simplifySign :: CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
amtstr
  | Just (Char
' ',CsvAmountString
t) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr = CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
t
  | Just (CsvAmountString
t,Char
' ') <- CsvAmountString -> Maybe (CsvAmountString, Char)
T.unsnoc CsvAmountString
amtstr = CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
t
  | Just (Char
'(',CsvAmountString
t) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr, Just (CsvAmountString
amt,Char
')') <- CsvAmountString -> Maybe (CsvAmountString, Char)
T.unsnoc CsvAmountString
t = CsvAmountString -> CsvAmountString
simplifySign forall a b. (a -> b) -> a -> b
$ CsvAmountString -> CsvAmountString
negateStr CsvAmountString
amt
  | Just (Char
'-',CsvAmountString
b) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr, Just (Char
'(',CsvAmountString
t) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
b, Just (CsvAmountString
amt,Char
')') <- CsvAmountString -> Maybe (CsvAmountString, Char)
T.unsnoc CsvAmountString
t = CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
amt
  | Just (Char
'-',CsvAmountString
m) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr, Just (Char
'-',CsvAmountString
amt) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
m = CsvAmountString
amt
  | Just (Char
'-',CsvAmountString
m) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr, Just (Char
'+',CsvAmountString
amt) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
m = CsvAmountString -> CsvAmountString
negateStr CsvAmountString
amt
  | CsvAmountString
amtstr forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [CsvAmountString
"-",CsvAmountString
"+",CsvAmountString
"()"] = CsvAmountString
""
  | Just (Char
'+',CsvAmountString
amt) <- CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr = CsvAmountString -> CsvAmountString
simplifySign CsvAmountString
amt
  | Bool
otherwise = CsvAmountString
amtstr

negateStr :: Text -> Text
negateStr :: CsvAmountString -> CsvAmountString
negateStr CsvAmountString
amtstr = case CsvAmountString -> Maybe (Char, CsvAmountString)
T.uncons CsvAmountString
amtstr of
    Just (Char
'-',CsvAmountString
s) -> CsvAmountString
s
    Maybe (Char, CsvAmountString)
_            -> Char -> CsvAmountString -> CsvAmountString
T.cons Char
'-' CsvAmountString
amtstr

--- ** tests
_TESTS__________________________________________ :: a
_TESTS__________________________________________ = forall a. HasCallStack => a
undefined

tests_RulesReader :: TestTree
tests_RulesReader = String -> [TestTree] -> TestTree
testGroup String
"RulesReader" [
   String -> [TestTree] -> TestTree
testGroup String
"parseCsvRules" [
     String -> Assertion -> TestTree
testCase String
"empty file" forall a b. (a -> b) -> a -> b
$
      String
-> CsvAmountString
-> Either
     (ParseErrorBundle CsvAmountString HledgerParseErrorData) CsvRules
parseCsvRules String
"unknown" CsvAmountString
"" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= forall a b. b -> Either a b
Right (CsvRulesParsed -> CsvRules
mkrules CsvRulesParsed
defrules)
   ]
  ,String -> [TestTree] -> TestTree
testGroup String
"rulesp" [
     String -> Assertion -> TestTree
testCase String
"trailing comments" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser CsvRules
rulesp CsvAmountString
"skip\n# \n#\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= forall a b. b -> Either a b
Right (CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives = [(CsvAmountString
"skip",CsvAmountString
"")]})

    ,String -> Assertion -> TestTree
testCase String
"trailing blank lines" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser CsvRules
rulesp CsvAmountString
"skip\n\n  \n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right (CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives = [(CsvAmountString
"skip",CsvAmountString
"")]}))

    ,String -> Assertion -> TestTree
testCase String
"no final newline" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser CsvRules
rulesp CsvAmountString
"skip" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right (CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rdirectives :: [(CsvAmountString, CsvAmountString)]
rdirectives=[(CsvAmountString
"skip",CsvAmountString
"")]}))

    ,String -> Assertion -> TestTree
testCase String
"assignment with empty value" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser CsvRules
rulesp CsvAmountString
"account1 \nif foo\n  account2 foo\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?=
        (forall a b. b -> Either a b
Right (CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rassignments :: [(CsvAmountString, CsvAmountString)]
rassignments = [(CsvAmountString
"account1",CsvAmountString
"")], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks = [CB{cbMatchers :: [Matcher]
cbMatchers=[MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
None (CsvAmountString -> Regexp
toRegex' CsvAmountString
"foo")],cbAssignments :: [(CsvAmountString, CsvAmountString)]
cbAssignments=[(CsvAmountString
"account2",CsvAmountString
"foo")]}]}))
   ]
  ,String -> [TestTree] -> TestTree
testGroup String
"conditionalblockp" [
    String -> Assertion -> TestTree
testCase String
"space after conditional" forall a b. (a -> b) -> a -> b
$ -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
       -- #1120
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser ConditionalBlock
conditionalblockp CsvAmountString
"if a\n account2 b\n \n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?=
        (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ CB{cbMatchers :: [Matcher]
cbMatchers=[MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
None forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"a"],cbAssignments :: [(CsvAmountString, CsvAmountString)]
cbAssignments=[(CsvAmountString
"account2",CsvAmountString
"b")]})

  ],

  String -> [TestTree] -> TestTree
testGroup String
"csvfieldreferencep" [
    String -> Assertion -> TestTree
testCase String
"number" forall a b. (a -> b) -> a -> b
$ forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
csvfieldreferencep CsvAmountString
"%1" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right CsvAmountString
"%1")
   ,String -> Assertion -> TestTree
testCase String
"name" forall a b. (a -> b) -> a -> b
$ forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
csvfieldreferencep CsvAmountString
"%date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right CsvAmountString
"%date")
   ,String -> Assertion -> TestTree
testCase String
"quoted name" forall a b. (a -> b) -> a -> b
$ forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules StateT
  CsvRulesParsed
  (ParsecT HledgerParseErrorData CsvAmountString Identity)
  CsvAmountString
csvfieldreferencep CsvAmountString
"%\"csv date\"" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right CsvAmountString
"%\"csv date\"")
   ]

  ,String -> [TestTree] -> TestTree
testGroup String
"matcherp" [

    String -> Assertion -> TestTree
testCase String
"recordmatcherp" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser Matcher
matcherp CsvAmountString
"A A\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
None forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"A A")

   ,String -> Assertion -> TestTree
testCase String
"recordmatcherp.starts-with-&" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser Matcher
matcherp CsvAmountString
"& A A\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
And forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"A A")

   ,String -> Assertion -> TestTree
testCase String
"fieldmatcherp.starts-with-%" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser Matcher
matcherp CsvAmountString
"description A A\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> Regexp -> Matcher
RecordMatcher MatcherPrefix
None forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"description A A")

   ,String -> Assertion -> TestTree
testCase String
"fieldmatcherp" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser Matcher
matcherp CsvAmountString
"%description A A\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"A A")

   ,String -> Assertion -> TestTree
testCase String
"fieldmatcherp.starts-with-&" forall a b. (a -> b) -> a -> b
$
      forall s st e a.
Stream s =>
st
-> StateT st (ParsecT e s Identity) a
-> s
-> Either (ParseErrorBundle s e) a
parseWithState' CsvRulesParsed
defrules CsvRulesParser Matcher
matcherp CsvAmountString
"& %description A A\n" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
And CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegexCI' CsvAmountString
"A A")

   -- ,testCase "fieldmatcherp with operator" $
   --    parseWithState' defrules matcherp "%description ~ A A\n" @?= (Right $ FieldMatcher "%description" "A A")

   ]

 ,String -> [TestTree] -> TestTree
testGroup String
"getEffectiveAssignment" [
    let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules {rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1)],rassignments :: [(CsvAmountString, CsvAmountString)]
rassignments=[(CsvAmountString
"date",CsvAmountString
"%csvdate")]}

    in String -> Assertion -> TestTree
testCase String
"toplevel" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"a",CsvAmountString
"b"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ,let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1)], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[[Matcher]
-> [(CsvAmountString, CsvAmountString)] -> ConditionalBlock
CB [MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%csvdate" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"a"] [(CsvAmountString
"date",CsvAmountString
"%csvdate")]]}
    in String -> Assertion -> TestTree
testCase String
"conditional" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"a",CsvAmountString
"b"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ,let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1),(CsvAmountString
"description",CsvFieldIndex
2)], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[[Matcher]
-> [(CsvAmountString, CsvAmountString)] -> ConditionalBlock
CB [MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%csvdate" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"a", MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"b"] [(CsvAmountString
"date",CsvAmountString
"%csvdate")]]}
    in String -> Assertion -> TestTree
testCase String
"conditional-with-or-a" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"a"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ,let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1),(CsvAmountString
"description",CsvFieldIndex
2)], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[[Matcher]
-> [(CsvAmountString, CsvAmountString)] -> ConditionalBlock
CB [MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%csvdate" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"a", MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"b"] [(CsvAmountString
"date",CsvAmountString
"%csvdate")]]}
    in String -> Assertion -> TestTree
testCase String
"conditional-with-or-b" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"_", CsvAmountString
"b"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ,let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1),(CsvAmountString
"description",CsvFieldIndex
2)], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[[Matcher]
-> [(CsvAmountString, CsvAmountString)] -> ConditionalBlock
CB [MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%csvdate" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"a", MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
And CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"b"] [(CsvAmountString
"date",CsvAmountString
"%csvdate")]]}
    in String -> Assertion -> TestTree
testCase String
"conditional.with-and" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"a", CsvAmountString
"b"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ,let rules :: CsvRules
rules = CsvRulesParsed -> CsvRules
mkrules forall a b. (a -> b) -> a -> b
$ CsvRulesParsed
defrules{rcsvfieldindexes :: [(CsvAmountString, CsvFieldIndex)]
rcsvfieldindexes=[(CsvAmountString
"csvdate",CsvFieldIndex
1),(CsvAmountString
"description",CsvFieldIndex
2)], rconditionalblocks :: [ConditionalBlock]
rconditionalblocks=[[Matcher]
-> [(CsvAmountString, CsvAmountString)] -> ConditionalBlock
CB [MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%csvdate" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"a", MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
And CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"b", MatcherPrefix -> CsvAmountString -> Regexp -> Matcher
FieldMatcher MatcherPrefix
None CsvAmountString
"%description" forall a b. (a -> b) -> a -> b
$ CsvAmountString -> Regexp
toRegex' CsvAmountString
"c"] [(CsvAmountString
"date",CsvAmountString
"%csvdate")]]}
    in String -> Assertion -> TestTree
testCase String
"conditional.with-and-or" forall a b. (a -> b) -> a -> b
$ CsvRules
-> [CsvAmountString] -> CsvAmountString -> Maybe CsvAmountString
getEffectiveAssignment CsvRules
rules [CsvAmountString
"_", CsvAmountString
"c"] CsvAmountString
"date" forall a. (Eq a, Show a, HasCallStack) => a -> a -> Assertion
@?= (forall a. a -> Maybe a
Just CsvAmountString
"%csvdate")

   ]

 ]