ENH use better algorithm for cronpatterns

This commit is contained in:
Nathan Dwarshuis 2023-02-05 10:34:26 -05:00
parent 56a14e5e9e
commit dae132c8b9
8 changed files with 141 additions and 225 deletions

View File

@ -158,12 +158,15 @@ runSync :: MonadUnliftIO m => FilePath -> m ()
runSync c = do runSync c = do
config <- readConfig c config <- readConfig c
handle err $ migrate_ (sqlConfig config) $ do handle err $ migrate_ (sqlConfig config) $ do
s <- getDBState config res <- getDBState config
flip runReaderT (s $ takeDirectory c) $ do case res of
es1 <- insertBudget $ budget config Left es -> throwIO $ InsertException es
es2 <- insertStatements config Right s -> do
let es = es1 ++ es2 flip runReaderT (s $ takeDirectory c) $ do
unless (null es) $ throwIO $ InsertException es es1 <- insertBudget $ budget config
es2 <- insertStatements config
let es = es1 ++ es2
unless (null es) $ throwIO $ InsertException es
where where
err (InsertException es) = do err (InsertException es) = do
liftIO $ mapM_ TI.putStrLn $ concatMap showError es liftIO $ mapM_ TI.putStrLn $ concatMap showError es

View File

@ -14,7 +14,7 @@ let Gregorian = { gYear : Natural, gMonth : Natural, gDay : Natural }
let GregorianM = { gmYear : Natural, gmMonth : Natural } let GregorianM = { gmYear : Natural, gmMonth : Natural }
let Interval = { intStart : Optional Gregorian, intEnd : Optional Gregorian } let Interval = { intStart : Gregorian, intEnd : Optional Gregorian }
let Global = { budgetInterval : Interval, statementInterval : Interval } let Global = { budgetInterval : Interval, statementInterval : Interval }

View File

@ -4,7 +4,7 @@ let List/map =
let T = let T =
./Types.dhall ./Types.dhall
sha256:8aa818fc10bf9468f39cb4fdfc9e05d37fa7ffb6060b1c6e1f420ba61f113a1e sha256:49fae993de77c82248b3a53a7237ee06542331e969b185cc943a56746247d4e1
let nullSplit = let nullSplit =
\(a : T.SplitAcnt) -> \(a : T.SplitAcnt) ->

View File

@ -80,8 +80,8 @@ type CurrencyMap = M.Map CurID CurrencyRId
data DBState = DBState data DBState = DBState
{ kmCurrency :: !CurrencyMap { kmCurrency :: !CurrencyMap
, kmAccount :: !AccountMap , kmAccount :: !AccountMap
, kmBudgetInterval :: !MaybeBounds , kmBudgetInterval :: !Bounds
, kmStatementInterval :: !MaybeBounds , kmStatementInterval :: !Bounds
, kmNewCommits :: ![Int] , kmNewCommits :: ![Int]
, kmConfigDir :: !FilePath , kmConfigDir :: !FilePath
, kmBoundsCache :: !(MVar (M.Map (Bounds, DatePat) [Day])) , kmBoundsCache :: !(MVar (M.Map (Bounds, DatePat) [Day]))

View File

@ -308,7 +308,7 @@ indexAcntRoot r =
getDBState getDBState
:: MonadUnliftIO m :: MonadUnliftIO m
=> Config => Config
-> SqlPersistT m (FilePath -> DBState) -> SqlPersistT m (EitherErrs (FilePath -> DBState))
getDBState c = do getDBState c = do
am <- updateAccounts $ accounts c am <- updateAccounts $ accounts c
cm <- updateCurrencies $ currencies c cm <- updateCurrencies $ currencies c
@ -316,13 +316,16 @@ getDBState c = do
v <- newMVar M.empty v <- newMVar M.empty
-- TODO not sure how I feel about this, probably will change this struct alot -- TODO not sure how I feel about this, probably will change this struct alot
-- in the future so whatever...for now -- in the future so whatever...for now
return $ \f -> return $ concatEither2 bi si $ \b s f ->
DBState DBState
{ kmCurrency = cm { kmCurrency = cm
, kmAccount = am , kmAccount = am
, kmBudgetInterval = intervalMaybeBounds $ budgetInterval $ global c , kmBudgetInterval = b
, kmStatementInterval = intervalMaybeBounds $ statementInterval $ global c , kmStatementInterval = s
, kmNewCommits = hs , kmNewCommits = hs
, kmConfigDir = f , kmConfigDir = f
, kmBoundsCache = v , kmBoundsCache = v
} }
where
bi = resolveBounds $ budgetInterval $ global c
si = resolveBounds $ statementInterval $ global c

View File

@ -17,206 +17,87 @@ import Internal.Statement
import Internal.Types hiding (sign) import Internal.Types hiding (sign)
import Internal.Utils import Internal.Utils
import RIO hiding (to) import RIO hiding (to)
-- import qualified RIO.Map as M
import qualified RIO.List as L
import qualified RIO.Text as T import qualified RIO.Text as T
import RIO.Time import RIO.Time
-------------------------------------------------------------------------------- --------------------------------------------------------------------------------
-- intervals -- intervals
-- expandDatePat :: MonadUnliftIO m => Bounds -> DatePat -> MappingT m [Day] expandDatePat :: Bounds -> DatePat -> EitherErrs [Day]
-- expandDatePat d p = do
-- -- TODO crude memoization
-- v <- asks kmBoundsCache
-- modifyMVar v $ \m -> case M.lookup (d, p) m of
-- Just ds -> return (m, ds)
-- Nothing -> do
-- let res = expandDatePat_ d p
-- return (M.insert (d, p) res m, res)
expandDatePat :: Bounds -> DatePat -> [Day]
expandDatePat b (Cron cp) = expandCronPat b cp expandDatePat b (Cron cp) = expandCronPat b cp
-- expandDatePat (a, b) (Cron cp) = expandDatePat i (Mod mp) = Right $ expandModPat mp i
-- fmap xGregToDay $
-- filter (cronPatternMatches cp) $
-- take (fromIntegral $ diffDays b a) $
-- gregorians a
expandDatePat i (Mod mp) = expandModPat mp i
expandModPat :: ModPat -> Bounds -> [Day] expandModPat :: ModPat -> Bounds -> [Day]
expandModPat expandModPat ModPat {mpStart = s, mpBy = b, mpUnit = u, mpRepeats = r} bs =
ModPat {mpStart = s, mpBy = b, mpUnit = u, mpRepeats = r} takeWhile (<= upper) $
(lower, upper) = (`addFun` start) . (* b')
takeWhile (<= upper) $ <$> maybe id (take . fromIntegral) r [0 ..]
(`addFun` start) . (* b')
<$> maybe id (take . fromIntegral) r [0 ..]
where
start = maybe lower fromGregorian' s
b' = fromIntegral b
addFun = case u of
Day -> addDays
Week -> addDays . (* 7)
Month -> addGregorianMonthsClip
Year -> addGregorianYearsClip
-- nextXGreg_ :: CronPat -> XGregorian -> XGregorian
-- nextXGreg_ c XGregorian {xgYear = y, xgMonth = m, xgDay = d, xgDayOfWeek = w}
-- | m == 12 && d == 31 = XGregorian (y + 1) 1 1 w_
-- | (m == 2 && (not leap && d == 28 || (leap && d == 29)))
-- || (m `elem` [4, 6, 9, 11] && d == 30)
-- || (d == 31) =
-- XGregorian y (m + 1) 1 w_
-- | otherwise = XGregorian y m (d + 1) w_
-- where
-- -- don't use DayOfWeek from Data.Time since this uses mod (which uses a
-- -- division opcode) and thus will be slower than just checking for equality
-- -- and adding
-- w_ = if w == 6 then 0 else w + 1
-- leap = isLeapYear $ fromIntegral y
monthLength :: (Integral a, Integral b, Integral c) => a -> b -> c
monthLength y m
| m == 2 && isLeapYear (fromIntegral y) = 29
| m == 2 = 28
| m `elem` [4, 6, 9, 11] = 30
| otherwise = 31
-- TODO this can be optimized to prevent filtering a bunch of dates for
-- one/a few cron patterns
-- cronPatternMatches :: CronPat -> XGregorian -> Bool
-- cronPatternMatches CronPat {..} XGregorian {..} =
-- testYMD xgYear cronYear
-- && testYMD xgMonth cronMonth
-- && testYMD xgDay cronDay
-- && testW (dayOfWeek_ xgDayOfWeek) cronWeekly
-- where
-- testYMD z = maybe True (mdyPatternMatches (fromIntegral z))
-- testW z = maybe True (`weekdayPatternMatches` z)
expandCronPat :: Bounds -> CronPat -> [Day]
expandCronPat b = L.unfoldr nextCronPat . compileCronPat b
data CompiledCronPat = CompiledCronPat
{ ccpYear :: ![Int]
, ccpMonth :: !(Zipper Int)
, ccpDay :: !(Zipper Int)
, ccpWeekly :: ![Int]
, ccpMonthEnd :: !Int
, ccpDayEnd :: !Int
}
deriving (Show)
data Zipper a = Zipper ![a] ![a] deriving (Show)
initZipper :: [a] -> Zipper a
initZipper = Zipper []
resetZipper :: Zipper a -> Zipper a
resetZipper (Zipper bs as) = initZipper $ reverse bs ++ as
shiftZipperWhile :: (a -> Bool) -> Zipper a -> Zipper a
shiftZipperWhile f z@(Zipper bs as) = case as of
[] -> z
x : xs
| f x -> shiftZipperWhile f $ Zipper (x : bs) xs
| otherwise -> z
zipperCurrent :: Zipper a -> Either (Zipper a) (a, Zipper a)
zipperCurrent z@(Zipper _ []) = Left $ resetZipper z
zipperCurrent (Zipper bs (a : as)) = Right (a, Zipper (a : bs) as)
compileCronPat :: Bounds -> CronPat -> CompiledCronPat
compileCronPat (x, y) CronPat {..} =
CompiledCronPat
{ ccpYear = maybe [y0_ .. y1_] compileMDY_ cronYear
, ccpMonth = compileDY [1 .. 12] m0 cronMonth
, ccpDay = compileDY [1 .. 31] d0 cronDay
, ccpWeekly = maybe [] compileW cronWeekly
, ccpMonthEnd = m1
, ccpDayEnd = d1
}
where where
(y0, m0, d0) = toGregorian x (lower, upper) = expandBounds bs
(y1, m1, d1) = toGregorian y start = maybe lower fromGregorian' s
y0_ = fromIntegral y0 b' = fromIntegral b
y1_ = fromIntegral y1 addFun = case u of
compileDY def k = shiftZipperWhile (< k) . initZipper . maybe def compileMDY_ Day -> addDays
compileMDY_ (Single z) = [fromIntegral z] Week -> addDays . (* 7)
compileMDY_ (Multi zs) = fromIntegral <$> zs Month -> addGregorianMonthsClip
compileMDY_ (Repeat RepeatPat {rpStart = s, rpBy = b, rpRepeats = rs}) = Year -> addGregorianYearsClip
-- TODO minor perf improvement, filter the repeats before filterng <=31
let b' = fromIntegral b
xs = takeWhile (<= 31) $ L.iterate (+ b') $ fromIntegral s
in maybe xs (\r -> take (fromIntegral r) xs) rs
compileW (OnDay w) = [fromEnum w] expandCronPat :: Bounds -> CronPat -> EitherErrs [Day]
compileW (OnDays ws) = fromEnum <$> ws expandCronPat b CronPat {..} = concatEither3 yRes mRes dRes $ \ys ms ds ->
filter validWeekday $
nextCronPat :: CompiledCronPat -> Maybe (Day, CompiledCronPat) mapMaybe (uncurry3 toDay) $
nextCronPat CompiledCronPat {ccpYear = []} = Nothing takeWhile (\((y, _), m, d) -> (y, m, d) <= (yb1, mb1, db1)) $
nextCronPat c@(CompiledCronPat {..}) = dropWhile (\((y, _), m, d) -> (y, m, d) < (yb0, mb0, db0)) $
case zipperCurrent ccpMonth of [(y, m, d) | y <- (\y -> (y, isLeapYear y)) <$> ys, m <- ms, d <- ds]
Left mz -> nextCronPat $ c {ccpYear = ys, ccpMonth = mz, ccpDay = resetZipper ccpDay}
Right (m, mz) -> case zipperCurrent ccpDay of
Left dz -> nextCronPat $ c {ccpMonth = mz, ccpDay = dz}
Right (d, dz)
| null ys && m >= ccpMonthEnd && d >= ccpDayEnd -> Nothing
| otherwise -> case dayMaybe m d of
Nothing -> nextCronPat $ c {ccpMonth = mz, ccpDay = resetZipper dz}
Just day -> Just (day, c {ccpDay = dz})
where where
y : ys = ccpYear yRes = case cronYear of
-- TODO not the most efficient way to check weekdays (most likely) since Nothing -> return [yb0 .. yb1]
-- I have to go through all the trouble of converting to a day and then Just pat -> do
-- doing some complex math to figure out which day of the week it is ys <- expandMDYPat (fromIntegral yb1) pat
validWeekday day = return $ dropWhile (< yb0) $ fromIntegral <$> ys
null ccpWeekly mRes = expandMD 12 cronMonth
|| (not (null ccpWeekly) && dayToWeekday day `elem` ccpWeekly) dRes = expandMD 31 cronDay
dayMaybe m d (s, e) = expandBounds b
| d > monthLength y m = Nothing (yb0, mb0, db0) = toGregorian s
| otherwise = (yb1, mb1, db1) = toGregorian $ addDays (-1) e
let day = fromGregorian (fromIntegral y) m d expandMD lim = fmap (fromIntegral <$>) . maybe (return [1 .. lim]) (expandMDYPat lim)
in if validWeekday day then Just day else Nothing expandW (OnDay x) = [fromEnum x]
expandW (OnDays xs) = fromEnum <$> xs
ws = maybe [] expandW cronWeekly
validWeekday = if null ws then const True else \day -> dayToWeekday day `elem` ws
toDay (y, leap) m d
| m == 2 && (not leap && d > 28 || leap && d > 29) = Nothing
| m `elem` [4, 6, 9, 11] && d > 30 = Nothing
| otherwise = Just $ fromGregorian y m d
expandMDYPat :: Natural -> MDYPat -> EitherErr [Natural]
expandMDYPat _ (Single x) = Right [x]
expandMDYPat _ (Multi xs) = Right xs
expandMDYPat lim (Repeat RepeatPat {rpStart = s, rpBy = b, rpRepeats = r})
| b < 1 = Left $ PatternError s b r ZeroLength
| otherwise = do
k <- limit r
return $ takeWhile (<= k) [s + i * b | i <- [0 ..]]
where
limit Nothing = Right lim
limit (Just n)
-- this guard not only produces the error for the user but also protects
-- from an underflow below it
| n < 1 = Left $ PatternError s b r ZeroRepeats
| otherwise = Right $ min (s + b * (n - 1)) lim
dayToWeekday :: Day -> Int dayToWeekday :: Day -> Int
dayToWeekday (ModifiedJulianDay d) = mod (fromIntegral d + 2) 7 dayToWeekday (ModifiedJulianDay d) = mod (fromIntegral d + 2) 7
-- -- TODO could clean this up by making an enum instance for Weekday
-- dayOfWeek_ :: Int -> Weekday
-- dayOfWeek_ d = case d of
-- 0 -> Sun
-- 1 -> Mon
-- 2 -> Tue
-- 3 -> Wed
-- 4 -> Thu
-- 5 -> Fri
-- _ -> Sat
-- weekdayPatternMatches :: WeekdayPat -> Weekday -> Bool
-- weekdayPatternMatches (OnDay x) = (== x)
-- weekdayPatternMatches (OnDays xs) = (`elem` xs)
-- mdyPatternMatches :: Natural -> MDYPat -> Bool
-- mdyPatternMatches x p = case p of
-- Single y -> x == y
-- Multi xs -> x `elem` xs
-- Repeat (RepeatPat {rpStart = s, rpBy = b, rpRepeats = r}) ->
-- s >= x && mod x b == 0 && maybe True (\y -> x <= s + b * y) r
withDates withDates
:: MonadUnliftIO m :: MonadUnliftIO m
=> DatePat => DatePat
-> (Day -> MappingT m a) -> (Day -> MappingT m a)
-> MappingT m [a] -> MappingT m (EitherErrs [a])
withDates dp f = do withDates dp f = do
bounds <- askBounds bounds <- asks kmBudgetInterval
let days = expandDatePat bounds dp let days = expandDatePat bounds dp
mapM f days mapM (mapM f) days
askBounds :: MonadUnliftIO m => MappingT m Bounds
askBounds = (liftIO . resolveBounds) =<< asks kmBudgetInterval
-------------------------------------------------------------------------------- --------------------------------------------------------------------------------
-- budget -- budget
@ -263,8 +144,8 @@ data BudgetTx = BudgetTx
insertIncome :: MonadUnliftIO m => Income -> MappingT m [InsertError] insertIncome :: MonadUnliftIO m => Income -> MappingT m [InsertError]
insertIncome i@Income {..} = insertIncome i@Income {..} =
whenHash CTIncome i [] $ \c -> whenHash CTIncome i [] $ \c ->
unlessLeft (balanceIncome i) $ \balance -> unlessLeft (balanceIncome i) $ \balance -> do
fmap concat $ withDates incWhen $ \day -> do res <- withDates incWhen $ \day -> do
let meta = BudgetMeta c day incCurrency let meta = BudgetMeta c day incCurrency
let fromAllos b = concatMap (fromAllo meta incFrom (Just b)) let fromAllos b = concatMap (fromAllo meta incFrom (Just b))
let pre = fromAllos PreTax incPretax let pre = fromAllos PreTax incPretax
@ -279,6 +160,7 @@ insertIncome i@Income {..} =
, btDesc = "balance after deductions" , btDesc = "balance after deductions"
} }
fmap concat $ mapM insertBudgetTx $ bal : (pre ++ tax ++ post) fmap concat $ mapM insertBudgetTx $ bal : (pre ++ tax ++ post)
unlessLefts res $ return . concat
fromAllo fromAllo
:: BudgetMeta :: BudgetMeta
@ -329,9 +211,10 @@ sumTaxes = sum . fmap (dec2Rat . taxValue)
insertTransfer :: MonadUnliftIO m => Transfer -> MappingT m [InsertError] insertTransfer :: MonadUnliftIO m => Transfer -> MappingT m [InsertError]
insertTransfer t@Transfer {..} = insertTransfer t@Transfer {..} =
fmap (concat . concat) $ whenHash CTExpense t [] $ \key -> do fmap concat $ whenHash CTExpense t [] $ \key -> do
forM transAmounts $ \(TimeAmount amt pat) -> forM transAmounts $ \(TimeAmount amt pat) -> do
withDates pat $ \day -> insertBudgetTx $ budgetTx amt day key res <- withDates pat $ \day -> insertBudgetTx $ budgetTx amt day key
unlessLefts res $ return . concat
where where
meta d c = BudgetMeta {bmWhen = d, bmCur = transCurrency, bmCommit = c} meta d c = BudgetMeta {bmWhen = d, bmCur = transCurrency, bmCommit = c}
budgetTx (Amount desc v) d c = budgetTx (Amount desc v) d c =
@ -399,10 +282,12 @@ insertManual
, manualDesc = e , manualDesc = e
} = do } = do
whenHash CTManual m [] $ \c -> do whenHash CTManual m [] $ \c -> do
bounds <- (liftIO . resolveBounds) =<< asks kmStatementInterval bounds <- asks kmStatementInterval
let days = expandDatePat bounds dp -- let days = expandDatePat bounds dp
res <- mapM tx days let dayRes = expandDatePat bounds dp
unlessLefts_ (concatEithersL res) $ lift . mapM_ (insertTx c) unlessLefts dayRes $ \days -> do
txRes <- mapM tx days
unlessLefts_ (concatEithersL txRes) $ lift . mapM_ (insertTx c)
where where
tx day = txPair day from to u (dec2Rat v) e tx day = txPair day from to u (dec2Rat v) e
@ -411,8 +296,8 @@ insertImport i = whenHash CTImport i [] $ \c -> do
-- TODO this isn't efficient, the whole file will be read and maybe no -- TODO this isn't efficient, the whole file will be read and maybe no
-- transactions will be desired -- transactions will be desired
recoverIO (readImport i) $ \r -> unlessLefts r $ \bs -> do recoverIO (readImport i) $ \r -> unlessLefts r $ \bs -> do
bounds <- asks kmStatementInterval bounds <- expandBounds <$> asks kmStatementInterval
res <- mapM resolveTx $ filter (inMaybeBounds bounds . txDate) bs res <- mapM resolveTx $ filter (inBounds bounds . txDate) bs
unlessLefts_ (concatEithersL res) $ lift . mapM_ (insertTx c) unlessLefts_ (concatEithersL res) $ lift . mapM_ (insertTx c)
where where
recoverIO x rest = do recoverIO x rest = do

View File

@ -520,9 +520,9 @@ data TxRecord = TxRecord
} }
deriving (Show, Eq, Ord) deriving (Show, Eq, Ord)
type Bounds = (Day, Day) type Bounds = (Day, Natural)
type MaybeBounds = (Maybe Day, Maybe Day) -- type MaybeBounds = (Maybe Day, Maybe Day)
data Keyed a = Keyed data Keyed a = Keyed
{ kKey :: !Int64 { kKey :: !Int64
@ -576,6 +576,8 @@ data AllocationSuberr
| TooManyBlanks | TooManyBlanks
deriving (Show) deriving (Show)
data PatternSuberr = ZeroLength | ZeroRepeats deriving (Show)
data InsertError data InsertError
= RegexError !T.Text = RegexError !T.Text
| MatchValPrecisionError !Natural !Natural | MatchValPrecisionError !Natural !Natural
@ -585,6 +587,8 @@ data InsertError
| LookupError !LookupSuberr !T.Text | LookupError !LookupSuberr !T.Text
| BalanceError !BalanceType !CurID ![RawSplit] | BalanceError !BalanceType !CurID ![RawSplit]
| IncomeError !DatePat | IncomeError !DatePat
| PatternError !Natural !Natural !(Maybe Natural) !PatternSuberr
| BoundsError !Gregorian !(Maybe Gregorian)
| StatementError ![TxRecord] ![MatchRe] | StatementError ![TxRecord] ![MatchRe]
deriving (Show) deriving (Show)

View File

@ -5,7 +5,9 @@
module Internal.Utils module Internal.Utils
( compareDate ( compareDate
, intervalMaybeBounds -- , intervalMaybeBounds
, inBounds
, expandBounds
, fmtRational , fmtRational
, matches , matches
, fromGregorian' , fromGregorian'
@ -23,7 +25,7 @@ module Internal.Utils
, unlessLefts_ , unlessLefts_
, unlessLeft , unlessLeft
, unlessLefts , unlessLefts
, inMaybeBounds -- , inMaybeBounds
, acntPath2Text , acntPath2Text
, showT , showT
, lookupErr , lookupErr
@ -119,29 +121,26 @@ compareDate (In md offset) x = do
| otherwise = if (start + fromIntegral offset - 1) < z then GT else EQ | otherwise = if (start + fromIntegral offset - 1) < z then GT else EQ
toMonth year month = (year * 12) + fromIntegral month toMonth year month = (year * 12) + fromIntegral month
-- boundsFromGregorian :: (Gregorian, Gregorian) -> Bounds
-- boundsFromGregorian (a, b) = (fromGregorian' a, fromGregorian' b)
fromGregorian' :: Gregorian -> Day fromGregorian' :: Gregorian -> Day
fromGregorian' = uncurry3 fromGregorian . gregTup fromGregorian' = uncurry3 fromGregorian . gregTup
-- inBounds :: Bounds -> Day -> Bool -- TODO misleading name
-- inBounds (d0, d1) x = d0 <= x && x <= d1 inBounds :: (Day, Day) -> Day -> Bool
inBounds (d0, d1) x = d0 <= x && x < d1
inMaybeBounds :: MaybeBounds -> Day -> Bool resolveBounds :: Interval -> EitherErr Bounds
inMaybeBounds (d0, d1) x = maybe True (x >=) d0 && maybe True (x <=) d1 resolveBounds Interval {intStart = s, intEnd = e} =
case fromGregorian' <$> e of
intervalMaybeBounds :: Interval -> MaybeBounds Nothing -> Right $ toBounds $ fromGregorian' $ s {gYear = gYear s + 50}
intervalMaybeBounds Interval {intStart = s, intEnd = e} = Just e_
(fmap fromGregorian' s, fmap fromGregorian' e) | s_ < e_ -> Right $ toBounds e_
| otherwise -> Left $ BoundsError s e
resolveBounds :: MonadUnliftIO m => MaybeBounds -> m Bounds
resolveBounds (s, e) = do
s' <- maybe getDay return s
e' <- maybe (addGregorianYearsClip 50 <$> getDay) return e
return (s', e')
where where
getDay = utctDay <$> getCurrentTime s_ = fromGregorian' s
toBounds end = (s_, fromIntegral $ diffDays end s_ - 1)
expandBounds :: Bounds -> (Day, Day)
expandBounds (d, n) = (d, addDays (fromIntegral n + 1) d)
-------------------------------------------------------------------------------- --------------------------------------------------------------------------------
-- matching -- matching
@ -319,6 +318,25 @@ acntPath2Text (AcntPath t cs) = T.intercalate "/" (atName t : cs)
showError :: InsertError -> [T.Text] showError :: InsertError -> [T.Text]
showError (StatementError ts ms) = (showTx <$> ts) ++ (showMatch <$> ms) showError (StatementError ts ms) = (showTx <$> ts) ++ (showMatch <$> ms)
showError other = (: []) $ case other of showError other = (: []) $ case other of
(BoundsError a b) ->
T.unwords ["Could not create bounds from", showGregorian_ a, "and", showGreg b]
where
showGreg (Just g) = showGregorian_ g
showGreg Nothing = "Inf"
(PatternError s b r p) -> T.unwords [msg, "in pattern: ", pat]
where
pat =
keyVals $
[ (k, v)
| (k, Just v) <-
[ ("start", Just $ showT s)
, ("by", Just $ showT b)
, ("repeats", showT <$> r)
]
]
msg = case p of
ZeroLength -> "Zero repeat length"
ZeroRepeats -> "Zero repeats"
(RegexError re) -> T.append "could not make regex from pattern: " re (RegexError re) -> T.append "could not make regex from pattern: " re
(ConversionError x) -> T.append "Could not convert to rational number: " x (ConversionError x) -> T.append "Could not convert to rational number: " x
(InsertIOError msg) -> T.append "IO Error: " msg (InsertIOError msg) -> T.append "IO Error: " msg
@ -353,6 +371,9 @@ showError other = (: []) $ case other of
NotOneBlank -> "Exactly one split must be blank" NotOneBlank -> "Exactly one split must be blank"
splits = T.intercalate ", " $ fmap (singleQuote . showSplit) rss splits = T.intercalate ", " $ fmap (singleQuote . showSplit) rss
showGregorian_ :: Gregorian -> T.Text
showGregorian_ Gregorian {..} = T.intercalate "-" $ showT <$> [gYear, gMonth, gDay]
showTx :: TxRecord -> T.Text showTx :: TxRecord -> T.Text
showTx TxRecord {trDate = d, trAmount = v, trDesc = e, trFile = f} = showTx TxRecord {trDate = d, trAmount = v, trDesc = e, trFile = f} =
T.append "Unmatched transaction: " $ T.append "Unmatched transaction: " $