module TSN.Picklers (
xp_date,
xp_date_padded,
+ xp_earnings,
xp_gamedate,
xp_racedate,
xp_tba_time,
where
-- System imports.
+import Data.List ( intercalate )
+import Data.List.Split ( chunksOf )
+import Data.String.Utils ( replace )
import Data.Time.Clock ( NominalDiffTime, UTCTime, addUTCTime )
import Data.Time.Format ( formatTime, parseTime )
import System.Locale ( defaultTimeLocale )
-- | The format string for a base date in m/d/yyyy format. The
-- day/month are not padded at all. This will match for example,
--
--- * 2/15/1983
--- * 1/1/0000
+-- * 2\/15\/1983
+--
+-- * 1\/1\/0000
--
date_format :: String
date_format = "%-m/%-d/%Y"
-- day/month are padded to two characters with zeros. This will
-- match for example,
--
--- * 02/15/1983
--- * 01/01/0000
+-- * 02\/15\/1983
+--
+-- * 01\/01\/0000
--
date_format_padded :: String
date_format_padded = "%0m/%0d/%Y"
from_date = formatTime defaultTimeLocale date_format_padded
+
+-- | Format a number as a string using a comma as the thousands
+-- separator.
+--
+-- Examples:
+--
+-- >>> format_commas 0
+-- "0"
+-- >>> format_commas 10
+-- "10"
+-- >>> format_commas 100
+-- "100"
+-- >>> format_commas 1000
+-- "1,000"
+-- >>> format_commas 10000
+-- "10,000"
+-- >>> format_commas 100000
+-- "100,000"
+-- >>> format_commas 1000000
+-- "1,000,000"
+--
+format_commas :: Int -> String
+format_commas x =
+ reverse (intercalate "," $ chunksOf 3 $ reverse $ show x)
+
+-- | Parse \<Earnings\> from an 'AutoRaceResultsListing'. These are
+-- essentially 'Int's, but they look like,
+--
+-- * \<Earnings\>336,826\</Earnings\>
+--
+-- * \<Earnings\>1,000,191\</Earnings\>
+--
+-- * \<Earnings\>TBA\</Earnings\>
+--
+xp_earnings :: PU (Maybe Int)
+xp_earnings =
+ (to_earnings, from_earnings) `xpWrap` xpText
+ where
+ strip_commas :: String -> String
+ strip_commas = replace "," ""
+
+ to_earnings :: String -> Maybe Int
+ to_earnings s
+ | s == "TBA" = Nothing
+ | otherwise = Just $ (read . strip_commas) s
+
+ from_earnings :: Maybe Int -> String
+ from_earnings Nothing = "TBA"
+ from_earnings (Just i) = format_commas i
+
+
-- | (Un)pickle a 'UTCTime' from a \<RaceDate\> element in an
-- 'AutoRaceResults' message.
--
--- Example:
+-- Examples:
+--
+-- * \<RaceDate\>6/1/2014 1:00:00 PM\</RaceDate\>
--
--- <RaceDate>6/1/2014 1:00:00 PM</RaceDate>
--- <RaceDate>5/24/2014 2:45:00 PM</RaceDate>
+-- * \<RaceDate\>5/24/2014 2:45:00 PM\</RaceDate\>
--
xp_racedate :: PU UTCTime
xp_racedate =
| otherwise = parseTime defaultTimeLocale time_format s
from_time :: Maybe UTCTime -> String
- from_time Nothing = ""
+ from_time Nothing = "TBA"
from_time (Just t) = formatTime defaultTimeLocale time_format t