module Attoparsec.Time.ByteString
(
timeOfDayInISO8601,
dayInISO8601,
yearAndMonthInISO8601,
timeZoneInISO8601,
utcTimeInISO8601,
diffTime,
nominalDiffTime,
)
where
import Attoparsec.Time.Prelude hiding (take, takeWhile)
import Data.Attoparsec.ByteString
import qualified Attoparsec.Time.Pure as A
import qualified Attoparsec.Time.Validation as B
import qualified Data.ByteString as C
import qualified Data.Attoparsec.ByteString.Char8 as D
validated :: Show a => B.Validator a -> Parser a -> Parser a
validated validator parser =
parser >>= \x -> B.run validator (pure x) fail x
sign :: Parser Bool
sign =
anyWord8 >>= \case
43 -> return True
45 -> return False
_ -> empty
decimalOfLength :: Integral a => Int -> Parser a
decimalOfLength length =
do
bytes <- take length
if C.all A.word8IsAsciiDigit bytes
then return (A.decimalFromBytes bytes)
else fail "Not all chars are valid decimals"
picoWithBasisOfLength :: Int -> Parser Pico
picoWithBasisOfLength basisLength =
MkFixed <$> ((+) <$> beforePoint <*> ((word8 46 *> afterPoint) <|> pure 0))
where
beforePoint =
(* (10 ^ 12)) <$> decimalOfLength basisLength
afterPoint =
fmap (updater . C.take 12) (takeWhile1 A.word8IsAsciiDigit)
where
updater bytes =
let
afterPoint =
A.decimalFromBytes bytes
afterPointLength =
C.length bytes
paddedAfterPoint =
if afterPointLength < 12
then afterPoint * (10 ^ (12 - afterPointLength))
else afterPoint
in paddedAfterPoint
{-# INLINE hour #-}
hour :: Parser Int
hour =
validated B.hour (decimalOfLength 2) <?> "hour"
{-# INLINE minute #-}
minute :: Parser Int
minute =
validated B.minute (decimalOfLength 2) <?> "minute"
{-# INLINE second #-}
second :: Parser Pico
second =
validated B.second (picoWithBasisOfLength 2) <?> "second"
{-# INLINE timeOfDayInISO8601 #-}
timeOfDayInISO8601 :: Parser TimeOfDay
timeOfDayInISO8601 =
unnamedParser <?> "timeOfDayInISO8601"
where
unnamedParser =
A.timeOfDay <$>
(hour <* word8 58) <*>
(minute <* word8 58) <*>
(second)
{-# INLINE dayInISO8601 #-}
dayInISO8601 :: Parser Day
dayInISO8601 =
unnamedParser <?> "dayInISO8601"
where
unnamedParser =
do
year <- decimalOfLength 4
word8 45
month <- decimalOfLength 2
word8 45
day <- decimalOfLength 2
case fromGregorianValid year month day of
Just day -> return day
Nothing -> fail (error year month day)
where
error year month day =
showString "Invalid combination of year month and day: " $
show (year, month, day)
yearAndMonthInISO8601 :: Parser (Word, Word)
yearAndMonthInISO8601 =
unnamedParser <?> "yearAndMonthInISO8601"
where
unnamedParser =
do
year <- decimalOfLength 4
word8 45
month <- decimalOfLength 2
return (year, month)
timeZoneInISO8601 :: Parser TimeZone
timeZoneInISO8601 =
unnamedParser <?> "timeZoneInISO8601"
where
unnamedParser =
z <|> offset
where
z =
word8 90 $> utc
offset =
A.timeZone <$> sign <*> decimalOfLength 2 <*> (word8 58 *> decimalOfLength 2 <|> decimalOfLength 2 <|> pure 0)
utcTimeInISO8601 :: Parser UTCTime
utcTimeInISO8601 =
unnamedParser <?> "utcTimeInISO8601"
where
unnamedParser =
do
day <- dayInISO8601
word8 84
time <- timeOfDayInISO8601
zone <- timeZoneInISO8601
return (A.utcTimeFromDayAndTimeOfDay day time zone)
diffTime :: Parser DiffTime
diffTime =
unnamedParser <?> "diffTime"
where
unnamedParser =
do
amount <- D.scientific
factor <- timeUnitFactor
return (factor (realToFrac amount))
nominalDiffTime :: Parser NominalDiffTime
nominalDiffTime =
unnamedParser <?> "nominalDiffTime"
where
unnamedParser =
do
amount <- D.scientific
factor <- timeUnitFactor
return (factor (realToFrac amount))
timeUnitFactor :: Fractional a => Parser (a -> a)
timeUnitFactor =
takeWhile A.word8IsAsciiAlpha >>= \case
"" -> return id
"s" -> return id
"ms" -> return (/ 1000)
"μs" -> return (/ 1000000)
"us" -> return (/ 1000000)
"ns" -> return (/ 1000000000)
"ps" -> return (/ 1000000000000)
"m" -> return (* 60)
"h" -> return (* 3600)
"d" -> return (* 86400)
unit -> fail ("Unsupported unit: " <> show unit)