{-# LANGUAGE NoOverloadedStrings #-} -- prevent trouble if turned on in ghci
{-# LANGUAGE TemplateHaskell #-}

module Hledger.Cli.Commands.Checkdates (
  checkdatesmode
 ,checkdates
) where

import Hledger
import Hledger.Cli.CliOptions
import System.Console.CmdArgs.Explicit
import System.Exit
import Text.Printf

checkdatesmode :: Mode RawOpts
checkdatesmode :: Mode RawOpts
checkdatesmode = CommandDoc
-> [Flag RawOpts]
-> [(CommandDoc, [Flag RawOpts])]
-> [Flag RawOpts]
-> ([Arg RawOpts], Maybe (Arg RawOpts))
-> Mode RawOpts
hledgerCommandMode
  $(embedFileRelative "Hledger/Cli/Commands/Checkdates.txt")
  [[CommandDoc] -> (RawOpts -> RawOpts) -> CommandDoc -> Flag RawOpts
forall a. [CommandDoc] -> (a -> a) -> CommandDoc -> Flag a
flagNone [CommandDoc
"unique"] (CommandDoc -> RawOpts -> RawOpts
setboolopt CommandDoc
"unique") CommandDoc
"require that dates are unique"]
  [(CommandDoc, [Flag RawOpts])
generalflagsgroup1]
  [Flag RawOpts]
hiddenflags
  ([], Arg RawOpts -> Maybe (Arg RawOpts)
forall a. a -> Maybe a
Just (Arg RawOpts -> Maybe (Arg RawOpts))
-> Arg RawOpts -> Maybe (Arg RawOpts)
forall a b. (a -> b) -> a -> b
$ CommandDoc -> Arg RawOpts
argsFlag CommandDoc
"[QUERY]")

checkdates :: CliOpts -> Journal -> IO ()
checkdates :: CliOpts -> Journal -> IO ()
checkdates CliOpts{rawopts_ :: CliOpts -> RawOpts
rawopts_=RawOpts
rawopts,reportspec_ :: CliOpts -> ReportSpec
reportspec_=ReportSpec
rspec} Journal
j = do
  let ropts :: ReportOpts
ropts = (ReportSpec -> ReportOpts
rsOpts ReportSpec
rspec){accountlistmode_ :: AccountListMode
accountlistmode_=AccountListMode
ALFlat}
  let ts :: [Transaction]
ts = (Transaction -> Bool) -> [Transaction] -> [Transaction]
forall a. (a -> Bool) -> [a] -> [a]
filter (ReportSpec -> Query
rsQuery ReportSpec
rspec Query -> Transaction -> Bool
`matchesTransaction`) ([Transaction] -> [Transaction]) -> [Transaction] -> [Transaction]
forall a b. (a -> b) -> a -> b
$
           Journal -> [Transaction]
jtxns (Journal -> [Transaction]) -> Journal -> [Transaction]
forall a b. (a -> b) -> a -> b
$ ReportOpts -> Journal -> Journal
journalSelectingAmountFromOpts ReportOpts
ropts Journal
j
  -- pprint rawopts
  let unique :: Bool
unique = CommandDoc -> RawOpts -> Bool
boolopt CommandDoc
"--unique" RawOpts
rawopts  -- TEMP: it's this for hledger check dates
            Bool -> Bool -> Bool
|| CommandDoc -> RawOpts -> Bool
boolopt CommandDoc
"unique" RawOpts
rawopts    -- and this for hledger check-dates (for some reason)
  let date :: Transaction -> Day
date = ReportOpts -> Transaction -> Day
transactionDateFn ReportOpts
ropts
  let compare :: Transaction -> Transaction -> Bool
compare Transaction
a Transaction
b =
        if Bool
unique
        then Transaction -> Day
date Transaction
a Day -> Day -> Bool
forall a. Ord a => a -> a -> Bool
<  Transaction -> Day
date Transaction
b
        else Transaction -> Day
date Transaction
a Day -> Day -> Bool
forall a. Ord a => a -> a -> Bool
<= Transaction -> Day
date Transaction
b
  case (Transaction -> Transaction -> Bool)
-> [Transaction] -> FoldAcc Transaction Transaction
checkTransactions Transaction -> Transaction -> Bool
compare [Transaction]
ts of
    FoldAcc{fa_previous :: forall a b. FoldAcc a b -> Maybe b
fa_previous=Maybe Transaction
Nothing} -> () -> IO ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
    FoldAcc{fa_error :: forall a b. FoldAcc a b -> Maybe a
fa_error=Maybe Transaction
Nothing}    -> () -> IO ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
    FoldAcc{fa_error :: forall a b. FoldAcc a b -> Maybe a
fa_error=Just Transaction
error, fa_previous :: forall a b. FoldAcc a b -> Maybe b
fa_previous=Just Transaction
previous} -> do
      CommandDoc -> IO ()
putStrLn (CommandDoc -> IO ()) -> CommandDoc -> IO ()
forall a b. (a -> b) -> a -> b
$ CommandDoc
-> CommandDoc
-> CommandDoc
-> CommandDoc
-> CommandDoc
-> CommandDoc
forall r. PrintfType r => CommandDoc -> r
printf 
          (CommandDoc
"Error: transaction's date is not in date order%s,\n"
        CommandDoc -> CommandDoc -> CommandDoc
forall a. [a] -> [a] -> [a]
++ CommandDoc
"at %s:\n\n%sPrevious transaction's date was: %s")
        (if Bool
unique then CommandDoc
" and/or not unique" else CommandDoc
"")
        (GenericSourcePos -> CommandDoc
showGenericSourcePos (GenericSourcePos -> CommandDoc) -> GenericSourcePos -> CommandDoc
forall a b. (a -> b) -> a -> b
$ Transaction -> GenericSourcePos
tsourcepos Transaction
error)
        (Transaction -> CommandDoc
showTransaction Transaction
error)
        (Day -> CommandDoc
forall a. Show a => a -> CommandDoc
show (Day -> CommandDoc) -> Day -> CommandDoc
forall a b. (a -> b) -> a -> b
$ Transaction -> Day
date Transaction
previous)
      IO ()
forall a. IO a
exitFailure

data FoldAcc a b = FoldAcc
 { FoldAcc a b -> Maybe a
fa_error    :: Maybe a
 , FoldAcc a b -> Maybe b
fa_previous :: Maybe b
 }

foldWhile :: (a -> FoldAcc a b -> FoldAcc a b) -> FoldAcc a b -> [a] -> FoldAcc a b
foldWhile :: (a -> FoldAcc a b -> FoldAcc a b)
-> FoldAcc a b -> [a] -> FoldAcc a b
foldWhile a -> FoldAcc a b -> FoldAcc a b
_ FoldAcc a b
acc [] = FoldAcc a b
acc
foldWhile a -> FoldAcc a b -> FoldAcc a b
fold FoldAcc a b
acc (a
a:[a]
as) =
  case a -> FoldAcc a b -> FoldAcc a b
fold a
a FoldAcc a b
acc of
   acc :: FoldAcc a b
acc@FoldAcc{fa_error :: forall a b. FoldAcc a b -> Maybe a
fa_error=Just a
_} -> FoldAcc a b
acc
   FoldAcc a b
acc -> (a -> FoldAcc a b -> FoldAcc a b)
-> FoldAcc a b -> [a] -> FoldAcc a b
forall a b.
(a -> FoldAcc a b -> FoldAcc a b)
-> FoldAcc a b -> [a] -> FoldAcc a b
foldWhile a -> FoldAcc a b -> FoldAcc a b
fold FoldAcc a b
acc [a]
as

checkTransactions :: (Transaction -> Transaction -> Bool)
 -> [Transaction] -> FoldAcc Transaction Transaction
checkTransactions :: (Transaction -> Transaction -> Bool)
-> [Transaction] -> FoldAcc Transaction Transaction
checkTransactions Transaction -> Transaction -> Bool
compare = (Transaction
 -> FoldAcc Transaction Transaction
 -> FoldAcc Transaction Transaction)
-> FoldAcc Transaction Transaction
-> [Transaction]
-> FoldAcc Transaction Transaction
forall a b.
(a -> FoldAcc a b -> FoldAcc a b)
-> FoldAcc a b -> [a] -> FoldAcc a b
foldWhile Transaction
-> FoldAcc Transaction Transaction
-> FoldAcc Transaction Transaction
f FoldAcc :: forall a b. Maybe a -> Maybe b -> FoldAcc a b
FoldAcc{fa_error :: Maybe Transaction
fa_error=Maybe Transaction
forall a. Maybe a
Nothing, fa_previous :: Maybe Transaction
fa_previous=Maybe Transaction
forall a. Maybe a
Nothing}
  where
    f :: Transaction
-> FoldAcc Transaction Transaction
-> FoldAcc Transaction Transaction
f Transaction
current acc :: FoldAcc Transaction Transaction
acc@FoldAcc{fa_previous :: forall a b. FoldAcc a b -> Maybe b
fa_previous=Maybe Transaction
Nothing} = FoldAcc Transaction Transaction
acc{fa_previous :: Maybe Transaction
fa_previous=Transaction -> Maybe Transaction
forall a. a -> Maybe a
Just Transaction
current}
    f Transaction
current acc :: FoldAcc Transaction Transaction
acc@FoldAcc{fa_previous :: forall a b. FoldAcc a b -> Maybe b
fa_previous=Just Transaction
previous} =
      if Transaction -> Transaction -> Bool
compare Transaction
previous Transaction
current
      then FoldAcc Transaction Transaction
acc{fa_previous :: Maybe Transaction
fa_previous=Transaction -> Maybe Transaction
forall a. a -> Maybe a
Just Transaction
current}
      else FoldAcc Transaction Transaction
acc{fa_error :: Maybe Transaction
fa_error=Transaction -> Maybe Transaction
forall a. a -> Maybe a
Just Transaction
current}