X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fhalcyon.git;a=blobdiff_plain;f=src%2FTwitter%2FStatus.hs;h=ba27d527d44f14ceeb3c6766e326d24d8beae501;hp=ef4e1037b9ca54b212f5666d60271ff832468301;hb=1b72ed45ef890ed1329a32457b4d7f3a7fb37788;hpb=9b6d95a82745ced2a58d9bc4ded555ee36b36673 diff --git a/src/Twitter/Status.hs b/src/Twitter/Status.hs index ef4e103..ba27d52 100644 --- a/src/Twitter/Status.hs +++ b/src/Twitter/Status.hs @@ -1,24 +1,34 @@ +{-# LANGUAGE NoMonomorphismRestriction #-} + -- | Functions and data for working with Twitter statuses. -module Twitter.Status +module Twitter.Status ( + Status(..), + Timeline, + get_max_status_id, + pretty_print, + status_tests, + utc_time_to_rfc822 ) where -import Control.Applicative ((<$>), (<*>)) -import Control.Monad (liftM) -import Data.Aeson ((.:), FromJSON(..), Value(Object)) -import Data.Maybe (catMaybes, isJust) -import Data.Monoid (mempty) -import Data.String.Utils (join, splitWs) -import Data.Text (pack) -import Data.Time (formatTime) -import Data.Time.Clock (UTCTime) -import Data.Time.Format (parseTime) -import Data.Time.LocalTime (TimeZone, utcToZonedTime) -import System.Locale (defaultTimeLocale, rfc822DateFormat) -import Test.HUnit -import Text.Regex (matchRegex, mkRegex) - -import StringUtils (listify) -import Twitter.User +import Control.Applicative ( (<$>), (<*>) ) +import Control.Monad ( liftM ) +import Data.Aeson ( (.:), FromJSON(..), Value(Object) ) +import Data.Maybe ( mapMaybe, isJust ) +import Data.Monoid ( mempty ) +import Data.String.Utils ( join, splitWs ) +import Data.Text ( pack ) +import Data.Time ( formatTime ) +import Data.Time.Clock ( UTCTime ) +import Data.Time.Format ( parseTime ) +import Data.Time.LocalTime ( TimeZone, utcToZonedTime ) +import System.Locale ( defaultTimeLocale, rfc822DateFormat ) +import Test.Tasty ( TestTree, testGroup ) +import Test.Tasty.HUnit ( (@?=), testCase ) +import Text.Regex ( matchRegex, mkRegex ) + +import Html ( replace_entities ) +import StringUtils ( listify ) +import Twitter.User ( User(..), screen_name_to_timeline_url ) data Status = Status { created_at :: Maybe UTCTime, @@ -26,8 +36,8 @@ data Status = Status { reply :: Bool, retweeted :: Bool, text :: String, - user :: User - } deriving (Show, Eq) + user :: User } + deriving (Eq, Show) type Timeline = [Status] @@ -38,7 +48,7 @@ instance FromJSON Status where (t .: id_field) <*> liftM isJustInt (t .: in_reply_to_status_id_field) <*> (t .: retweeted_field) <*> - (t .: text_field) <*> + liftM replace_entities (t .: text_field) <*> (t .: user_field) where -- The typechecker flips out without this. @@ -55,6 +65,8 @@ instance FromJSON Status where -- Do whatever. parseJSON _ = mempty +-- | Parse a timestamp from a status into a UTCTime (or Nothing). +-- parse_status_time :: String -> Maybe UTCTime parse_status_time = parseTime defaultTimeLocale status_format @@ -63,6 +75,10 @@ parse_status_time = status_format :: String status_format = "%a %b %d %H:%M:%S %z %Y" + +-- | Given a 'TimeZone', convert a 'UTCTime' into an RFC822-format +-- time string. If no 'TimeZone' is given, assume UTC. +-- utc_time_to_rfc822 :: Maybe TimeZone -> UTCTime -> String utc_time_to_rfc822 mtz utc = case mtz of @@ -72,12 +88,18 @@ utc_time_to_rfc822 mtz utc = foo = formatTime defaultTimeLocale rfc822DateFormat +-- | Get the 'created_at' time out of a 'Status' and display it as an +-- RFC822-format time string. If there's no created-at time in the +-- status, you'll get an empty string instead. +-- show_created_at :: Maybe TimeZone -> Status -> String show_created_at mtz = - (maybe "" id) . (fmap $ utc_time_to_rfc822 mtz) . created_at + (maybe "" (utc_time_to_rfc822 mtz)) . created_at + -- | Returns a nicely-formatted String representing the given 'Status' -- object. +-- pretty_print :: Maybe TimeZone -> Status -> String pretty_print mtz status = concat [ name, @@ -99,6 +121,7 @@ pretty_print mtz status = -- | Given a list of statuses, returns the greatest status_id -- belonging to one of the statuses in the list. +-- get_max_status_id :: Timeline -> Integer get_max_status_id statuses = maximum status_ids @@ -107,6 +130,7 @@ get_max_status_id statuses = -- | Parse one username from a word. +-- parse_username :: String -> Maybe String parse_username word = case matches of @@ -119,13 +143,16 @@ parse_username word = -- | Parse all usernames of the form \@username from a status. +-- parse_usernames_from_status :: Status -> [String] parse_usernames_from_status status = - catMaybes (map parse_username status_words) + mapMaybe parse_username status_words where status_words = splitWs (text status) + -- | Get all referenced users' timeline URLs. +-- make_user_timeline_urls :: Status -> [String] make_user_timeline_urls status = map screen_name_to_timeline_url usernames @@ -133,26 +160,27 @@ make_user_timeline_urls status = usernames = parse_usernames_from_status status -status_tests :: [Test] -status_tests = [ test_parse_usernames ] +status_tests :: TestTree +status_tests = + testGroup "Status Tests" [ test_parse_usernames ] -test_parse_usernames :: Test +test_parse_usernames :: TestTree test_parse_usernames = - TestCase $ - assertEqual - "All usernames are parsed." - expected_usernames - actual_usernames - where - dummy_user = User { screen_name = "nobody" } - dummy_status = Status { status_id = 1, - created_at = Nothing, - text = "Hypothesis: @donsbot and @bonus500 are two personalities belonging to the same person.", - user = dummy_user, - reply = False, - retweeted = False - } - - actual_usernames = parse_usernames_from_status dummy_status - expected_usernames = ["donsbot", "bonus500"] + testCase description $ actual @?= expected + where + description = "all usernames are parsed" + + dummy_user = User { screen_name = "nobody" } + dummy_text = "Hypothesis: @donsbot and @bonus500 are two " ++ + "personalities belonging to the same person." + dummy_status = Status { status_id = 1, + created_at = Nothing, + text = dummy_text, + user = dummy_user, + reply = False, + retweeted = False + } + + actual = parse_usernames_from_status dummy_status + expected = ["donsbot", "bonus500"]