X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=c104874c9c7b026d45d0de5f71c817385eb145c8;hb=f519b55ffe72acd791bdc91b16918603afce1995;hp=40703d7ea9170c10110efe3787ce070c47786256;hpb=7bb00e04c15781d889f950d00babf3f183047bff;p=dead%2Fhalcyon.git diff --git a/src/Main.hs b/src/Main.hs index 40703d7..c104874 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -24,7 +24,7 @@ import Mail ( import Twitter.Http import Twitter.Status import Twitter.User - +import Usernames (Usernames(..)) -- | A wrapper around threadDelay which takes seconds instead of -- microseconds as its argument. @@ -39,7 +39,7 @@ thread_sleep seconds = do -- /Date: / header, and returns the updated message. message_from_status :: Maybe TimeZone -> Message -> String -> Status -> Message message_from_status mtz message default_date status = - message { subject = "Twat: " ++ (screen_name (user status)), + message { subject = "Halcyon: " ++ (screen_name (user status)), body = (pretty_print mtz status), headers = ((headers message) ++ ["Date: " ++ date])} where @@ -166,8 +166,8 @@ get_latest_status_id cfg username = do -- recursive loop that checks for updates forever. The message -- argument is optional and is passed to recurse in case the updates -- should be emailed. -run_twat :: Cfg -> Maybe Message -> String -> IO () -run_twat cfg msg username = do +run :: Cfg -> Maybe Message -> String -> IO () +run cfg msg username = do latest_status_id <- get_latest_status_id cfg username recurse cfg username latest_status_id msg return () @@ -189,7 +189,7 @@ construct_message cfg = do from = f } -- |The main function just parses the command-line arguments and then --- forks off calls to 'run_twat' for each supplied username. After +-- forks off calls to 'run' for each supplied username. After -- forking, main loops forever. main :: IO () main = do @@ -205,7 +205,7 @@ main = do -- set in either the config file or on the command-line. let cfg = merge_optional default_config opt_config - when (null $ usernames cfg) $ do + when (null $ get_usernames (usernames cfg)) $ do hPutStrLn stderr "ERROR: no usernames supplied." _ <- show_help exitWith (ExitFailure exit_no_usernames) @@ -214,9 +214,9 @@ main = do -- message object to be passed to all of our threads. let message = construct_message cfg - -- Execute run_twat on each username in a new thread. - let run_twat_curried = run_twat cfg message - _ <- mapM (forkIO . run_twat_curried) (usernames cfg) + -- Execute run on each username in a new thread. + let run_curried = run cfg message + _ <- mapM (forkIO . run_curried) (get_usernames (usernames cfg)) _ <- forever $ -- This thread (the one executing main) doesn't do anything,