Add the format_commas function and use it to implement the earnings pickler.

[dead/htsn-import.git] / src / TSN / Picklers.hs

diff --git a/src/TSN/Picklers.hs b/src/TSN/Picklers.hs
index dc3ed9a05cb452fbba1ad884593fc5ace02967c9..ff7b0d1e677e9cff2c9cd469c77e226be272a2a6 100644 (file)
--- a/src/TSN/Picklers.hs
+++ b/src/TSN/Picklers.hs
@@ -4,6 +4,7 @@
 module TSN.Picklers (
   xp_date,
   xp_date_padded,
+  xp_earnings,
   xp_gamedate,
   xp_racedate,
   xp_tba_time,
@@ -12,6 +13,9 @@ module TSN.Picklers (
 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 )
@@ -76,6 +80,55 @@ xp_date_padded =
     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 = (read . strip_commas . show) 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.
 --