277 lines
13 KiB
Haskell
277 lines
13 KiB
Haskell
{-|
|
|
|
|
A 'TimeclockEntry' is a clock-in, clock-out, or other directive in a timeclock
|
|
file (see timeclock.el or the command-line version). These can be
|
|
converted to 'Transactions' and queried like a ledger.
|
|
|
|
-}
|
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Hledger.Data.Timeclock (
|
|
timeclockEntriesToTransactions
|
|
,timeclockEntriesToTransactionsSingle
|
|
,tests_Timeclock
|
|
)
|
|
where
|
|
|
|
import Data.List (partition, sortBy, uncons)
|
|
import Data.Maybe (fromMaybe)
|
|
import qualified Data.Text as T
|
|
import Data.Time.Calendar (addDays)
|
|
import Data.Time.Clock (addUTCTime, getCurrentTime)
|
|
import Data.Time.Format (defaultTimeLocale, formatTime, parseTimeM)
|
|
import Data.Time.LocalTime (LocalTime(..), TimeOfDay(..), getCurrentTimeZone,
|
|
localTimeToUTC, midnight, utc, utcToLocalTime)
|
|
import Text.Printf (printf)
|
|
|
|
import Hledger.Utils
|
|
import Hledger.Data.Types
|
|
import Hledger.Data.Dates
|
|
import Hledger.Data.Amount
|
|
import Hledger.Data.Posting
|
|
|
|
instance Show TimeclockEntry where
|
|
show t = printf "%s %s %s %s" (show $ tlcode t) (show $ tldatetime t) (tlaccount t) (tldescription t)
|
|
|
|
instance Show TimeclockCode where
|
|
show SetBalance = "b"
|
|
show SetRequiredHours = "h"
|
|
show In = "i"
|
|
show Out = "o"
|
|
show FinalOut = "O"
|
|
|
|
instance Read TimeclockCode where
|
|
readsPrec _ ('b' : xs) = [(SetBalance, xs)]
|
|
readsPrec _ ('h' : xs) = [(SetRequiredHours, xs)]
|
|
readsPrec _ ('i' : xs) = [(In, xs)]
|
|
readsPrec _ ('o' : xs) = [(Out, xs)]
|
|
readsPrec _ ('O' : xs) = [(FinalOut, xs)]
|
|
readsPrec _ _ = []
|
|
|
|
data Session = Session
|
|
{ in' :: TimeclockEntry,
|
|
out :: TimeclockEntry
|
|
}
|
|
|
|
data Sessions = Sessions
|
|
{ completed :: [Session],
|
|
active :: [TimeclockEntry]
|
|
}
|
|
|
|
-- | Find the relevant clockin in the actives list that should be paired with this clockout.
|
|
-- If there is a session that has the same account name, then use that.
|
|
-- Otherwise, if there is an active anonymous session, use that.
|
|
-- Otherwise, raise an error.
|
|
findInForOut :: TimeclockEntry -> ([TimeclockEntry], [TimeclockEntry]) -> (TimeclockEntry, [TimeclockEntry])
|
|
findInForOut _ (matchingout : othermatches, rest) = (matchingout, othermatches <> rest)
|
|
findInForOut o ([], activeins) =
|
|
if emptyname then (first, rest) else error' errmsg
|
|
where
|
|
l = show $ unPos $ sourceLine $ tlsourcepos o
|
|
c = unPos $ sourceColumn $ tlsourcepos o
|
|
emptyname = tlaccount o == ""
|
|
(first, rest) = case uncons activeins of
|
|
Just (hd, tl) -> (hd, tl)
|
|
Nothing -> error' errmsg
|
|
errmsg =
|
|
printf
|
|
"%s:\n%s\n%s\n\nCould not find previous clockin to match this clockout."
|
|
(sourcePosPretty $ tlsourcepos o)
|
|
(l ++ " | " ++ show o)
|
|
(replicate (length l) ' ' ++ " |" ++ replicate c ' ' ++ "^")
|
|
|
|
-- | Assuming that entries have been sorted, we go through each time log entry.
|
|
-- We collect all of the "i" in the list "actives," and each time we encounter
|
|
-- an "o," we look for the corresponding "i" in actives.
|
|
-- If we cannot find it, then it is an error (since the list is sorted).
|
|
-- If the "o" is recorded on a different day than the "i" then we close the
|
|
-- active entry at the end of its day, replace it in the active list
|
|
-- with a start at midnight on the next day, and try again.
|
|
-- This raises an error if any outs cannot be paired with an in.
|
|
pairClockEntries :: [TimeclockEntry] -> [TimeclockEntry] -> [Session] -> Sessions
|
|
pairClockEntries [] actives sessions = Sessions {completed = sessions, active = actives}
|
|
pairClockEntries (entry : rest) actives sessions
|
|
| tlcode entry == In = pairClockEntries rest inentries sessions
|
|
| tlcode entry == Out = pairClockEntries rest' actives' sessions'
|
|
| otherwise = pairClockEntries rest actives sessions
|
|
where
|
|
(inentry, newactive) = findInForOut entry (partition (\e -> tlaccount e == tlaccount entry) actives)
|
|
(itime, otime) = (tldatetime inentry, tldatetime entry)
|
|
(idate, odate) = (localDay itime, localDay otime)
|
|
omidnight = entry {tldatetime = itime {localDay = idate, localTimeOfDay = TimeOfDay 23 59 59}}
|
|
imidnight = inentry {tldatetime = itime {localDay = addDays 1 idate, localTimeOfDay = midnight}}
|
|
(sessions', actives', rest') = if odate > idate then
|
|
(Session {in' = inentry, out = omidnight} : sessions, imidnight : newactive, entry : rest)
|
|
else
|
|
(Session {in' = inentry, out = entry} : sessions, newactive, rest)
|
|
l = show $ unPos $ sourceLine $ tlsourcepos entry
|
|
c = unPos $ sourceColumn $ tlsourcepos entry
|
|
inentries =
|
|
if any (\e -> tlaccount e == tlaccount entry) actives
|
|
then error' $
|
|
printf
|
|
"%s:\n%s\n%s\n\nEncountered clockin entry for session \"%s\" that is already active."
|
|
(sourcePosPretty $ tlsourcepos entry)
|
|
(l ++ " | " ++ show entry)
|
|
(replicate (length l) ' ' ++ " |" ++ replicate c ' ' ++ "^")
|
|
(tlaccount entry)
|
|
else entry : actives
|
|
|
|
-- | Convert time log entries to journal transactions, allowing multiple clocked-in sessions at once.
|
|
-- When there is no clockout, one is added with the provided current time.
|
|
-- Sessions crossing midnight are split into days to give accurate per-day totals.
|
|
-- If any entries cannot be paired as expected, an error is raised.
|
|
-- This is the default behaviour.
|
|
timeclockEntriesToTransactions :: LocalTime -> [TimeclockEntry] -> [Transaction]
|
|
timeclockEntriesToTransactions now entries = transactions
|
|
where
|
|
sessions = pairClockEntries (sortBy (\e1 e2 -> compare (tldatetime e1) (tldatetime e2)) entries) [] []
|
|
transactionsFromSession s = entryFromTimeclockInOut (in' s) (out s)
|
|
-- If any "in" sessions are in the future, then set their out time to the initial time
|
|
outtime te = max now (tldatetime te)
|
|
createout te = TimeclockEntry (tlsourcepos te) Out (outtime te) (tlaccount te) "" "" []
|
|
outs = map createout (active sessions)
|
|
stillopen = pairClockEntries ((active sessions) <> outs) [] []
|
|
transactions = map transactionsFromSession $ sortBy (\s1 s2 -> compare (in' s1) (in' s2)) (completed sessions ++ completed stillopen)
|
|
|
|
-- | Convert time log entries to journal transactions, allowing only one clocked-in session at a time.
|
|
-- Entries must be a strict alternation of in and out, beginning with in.
|
|
-- When there is no clockout, one is added with the provided current time.
|
|
-- Sessions crossing midnight are split into days to give accurate per-day totals.
|
|
-- If entries are not in the expected in/out order, an error is raised.
|
|
-- This is the legacy behaviour, enabled by --old-timeclock.
|
|
timeclockEntriesToTransactionsSingle :: LocalTime -> [TimeclockEntry] -> [Transaction]
|
|
timeclockEntriesToTransactionsSingle _ [] = []
|
|
timeclockEntriesToTransactionsSingle now [i]
|
|
| tlcode i /= In = errorExpectedCodeButGot In i
|
|
| odate > idate = entryFromTimeclockInOut i o' : timeclockEntriesToTransactions now [i',o]
|
|
| otherwise = [entryFromTimeclockInOut i o]
|
|
where
|
|
o = TimeclockEntry (tlsourcepos i) Out end "" "" "" []
|
|
end = if itime > now then itime else now
|
|
(itime,otime) = (tldatetime i,tldatetime o)
|
|
(idate,odate) = (localDay itime,localDay otime)
|
|
o' = o{tldatetime=itime{localDay=idate, localTimeOfDay=TimeOfDay 23 59 59}}
|
|
i' = i{tldatetime=itime{localDay=addDays 1 idate, localTimeOfDay=midnight}}
|
|
timeclockEntriesToTransactionsSingle now (i:o:rest)
|
|
| tlcode i /= In = errorExpectedCodeButGot In i
|
|
| tlcode o /= Out = errorExpectedCodeButGot Out o
|
|
| odate > idate = entryFromTimeclockInOut i o' : timeclockEntriesToTransactions now (i':o:rest)
|
|
| otherwise = entryFromTimeclockInOut i o : timeclockEntriesToTransactions now rest
|
|
where
|
|
(itime,otime) = (tldatetime i,tldatetime o)
|
|
(idate,odate) = (localDay itime,localDay otime)
|
|
o' = o{tldatetime=itime{localDay=idate, localTimeOfDay=TimeOfDay 23 59 59}}
|
|
i' = i{tldatetime=itime{localDay=addDays 1 idate, localTimeOfDay=midnight}}
|
|
{- HLINT ignore timeclockEntriesToTransactions -}
|
|
|
|
errorExpectedCodeButGot :: TimeclockCode -> TimeclockEntry -> a
|
|
errorExpectedCodeButGot expected actual = error' $ printf
|
|
("%s:\n%s\n%s\n\nExpected a timeclock %s entry but got %s.\n"
|
|
++"Only one session may be clocked in at a time.\n"
|
|
++"Please alternate i and o, beginning with i.")
|
|
(sourcePosPretty $ tlsourcepos actual)
|
|
(l ++ " | " ++ show actual)
|
|
(replicate (length l) ' ' ++ " |" ++ replicate c ' ' ++ "^")
|
|
(show expected)
|
|
(show $ tlcode actual)
|
|
where
|
|
l = show $ unPos $ sourceLine $ tlsourcepos actual
|
|
c = unPos $ sourceColumn $ tlsourcepos actual
|
|
|
|
-- | Convert a timeclock clockin and clockout entry to an equivalent journal
|
|
-- transaction, representing the time expenditure. Note this entry is not balanced,
|
|
-- since we omit the \"assets:time\" transaction for simpler output.
|
|
entryFromTimeclockInOut :: TimeclockEntry -> TimeclockEntry -> Transaction
|
|
entryFromTimeclockInOut i o
|
|
| otime >= itime = t
|
|
| otherwise =
|
|
-- Clockout time earlier than clockin is an error.
|
|
-- (Clockin earlier than preceding clockin/clockout is allowed.)
|
|
-- We should never encounter this case now that we sort the entries,
|
|
-- but let's leave it in case of error.
|
|
error' $ printf
|
|
("%s:\n%s\nThis clockout time (%s) is earlier than the previous clockin.\n"
|
|
++"Please adjust it to be later than %s.")
|
|
(sourcePosPretty $ tlsourcepos o)
|
|
(unlines [
|
|
replicate (length l) ' '++ " | " ++ show i,
|
|
l ++ " | " ++ show o,
|
|
(replicate (length l) ' ' ++ " |" ++ replicate c ' ' ++ replicate 19 '^')
|
|
])
|
|
(show $ tldatetime o)
|
|
(show $ tldatetime i)
|
|
where
|
|
l = show $ unPos $ sourceLine $ tlsourcepos o
|
|
c = (unPos $ sourceColumn $ tlsourcepos o) + 2
|
|
t = Transaction {
|
|
tindex = 0,
|
|
tsourcepos = (tlsourcepos i, tlsourcepos i),
|
|
tdate = idate,
|
|
tdate2 = Nothing,
|
|
tstatus = Cleared,
|
|
tcode = "",
|
|
tdescription = desc,
|
|
tcomment = tlcomment i,
|
|
ttags = tltags i,
|
|
tpostings = ps,
|
|
tprecedingcomment=""
|
|
}
|
|
itime = tldatetime i
|
|
otime = tldatetime o
|
|
itod = localTimeOfDay itime
|
|
otod = localTimeOfDay otime
|
|
idate = localDay itime
|
|
desc | T.null (tldescription i) = T.pack $ showtime itod ++ "-" ++ showtime otod
|
|
| otherwise = tldescription i
|
|
showtime = take 5 . show
|
|
hours = elapsedSeconds (toutc otime) (toutc itime) / 3600 where toutc = localTimeToUTC utc
|
|
acctname = tlaccount i
|
|
-- Generate an hours amount. Unusually, we also round the internal Decimal value,
|
|
-- since otherwise it will often have large recurring decimal parts which (since 1.21)
|
|
-- print would display all 255 digits of. timeclock amounts have one second resolution,
|
|
-- so two decimal places is precise enough (#1527).
|
|
amt = mixedAmount $ setAmountInternalPrecision 2 $ hrs hours
|
|
ps = [posting{paccount=acctname, pamount=amt, ptype=VirtualPosting, ptransaction=Just t}]
|
|
|
|
|
|
-- tests
|
|
|
|
tests_Timeclock = testGroup "Timeclock" [
|
|
testCaseSteps "timeclockEntriesToTransactions tests" $ \step -> do
|
|
step "gathering data"
|
|
today <- getCurrentDay
|
|
now' <- getCurrentTime
|
|
tz <- getCurrentTimeZone
|
|
let now = utcToLocalTime tz now'
|
|
nowstr = showtime now
|
|
yesterday = prevday today
|
|
clockin = TimeclockEntry (initialPos "") In
|
|
clockout = TimeclockEntry (initialPos "") Out
|
|
mktime d = LocalTime d . fromMaybe midnight .
|
|
parseTimeM True defaultTimeLocale "%H:%M:%S"
|
|
showtime = formatTime defaultTimeLocale "%H:%M"
|
|
txndescs = map (T.unpack . tdescription) . timeclockEntriesToTransactions now
|
|
future = utcToLocalTime tz $ addUTCTime 100 now'
|
|
futurestr = showtime future
|
|
step "started yesterday, split session at midnight"
|
|
txndescs [clockin (mktime yesterday "23:00:00") "" "" "" []] @?= ["23:00-23:59","00:00-"++nowstr]
|
|
step "split multi-day sessions at each midnight"
|
|
txndescs [clockin (mktime (addDays (-2) today) "23:00:00") "" "" "" []] @?= ["23:00-23:59","00:00-23:59","00:00-"++nowstr]
|
|
step "auto-clock-out if needed"
|
|
txndescs [clockin (mktime today "00:00:00") "" "" "" []] @?= ["00:00-"++nowstr]
|
|
step "use the clockin time for auto-clockout if it's in the future"
|
|
txndescs [clockin future "" "" "" []] @?= [printf "%s-%s" futurestr futurestr]
|
|
step "multiple open sessions"
|
|
txndescs
|
|
[ clockin (mktime today "00:00:00") "a" "" "" [],
|
|
clockin (mktime today "01:00:00") "b" "" "" [],
|
|
clockin (mktime today "02:00:00") "c" "" "" [],
|
|
clockout (mktime today "03:00:00") "b" "" "" [],
|
|
clockout (mktime today "04:00:00") "a" "" "" [],
|
|
clockout (mktime today "05:00:00") "c" "" "" []
|
|
]
|
|
@?= ["00:00-04:00", "01:00-03:00", "02:00-05:00"]
|
|
]
|