X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fhalcyon.git;a=blobdiff_plain;f=src%2FMain.hs;h=53c3f905e863e9c98537e4b62d5307335b573939;hp=ac33b2bdc654ffe143e7aeaa548fc93413d64f63;hb=76af720585a173aa47e7858964b581fdbf7a01e6;hpb=53c26134670775854677470e402b8b41f19246d1

diff --git a/src/Main.hs b/src/Main.hs
index ac33b2b..53c3f90 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -4,6 +4,7 @@ where
 import Control.Concurrent (forkIO, threadDelay)
 import Control.Monad (forever, when)
 import System.Exit (ExitCode(..), exitWith)
+import System.Time (getClockTime)
 import System.IO (hPutStrLn, stderr)
 
 import CommandLine
@@ -14,21 +15,34 @@ import Twitter.Status
 import Twitter.User
 
 
--- The length of all calls to sleep, in seconds.
+-- |The length of all calls to sleep (or threadDelay), in seconds.
 heartbeat :: Int
 heartbeat = 600
 
+-- |A wrapper around threadDelay which takes seconds instead of
+-- microseconds as its argument.
 thread_sleep :: Int -> IO ()
-thread_sleep microseconds = do
-  let seconds = microseconds * (10 ^ (6 :: Int))
-  threadDelay seconds
+thread_sleep seconds = do
+  let microseconds = seconds * (10 ^ (6 :: Int))
+  threadDelay microseconds
 
 
-message_from_status :: Message -> Status -> Message
-message_from_status message status =
+-- |Given a 'Message', 'Status', and date, update that message's body
+-- and subject with the information contained in the status. Adds a
+-- /Date: / header, and returns the updated message.
+message_from_status :: Message -> String -> Status -> Message
+message_from_status message date status =
     message { subject = "Twat: " ++ (screen_name (user status)),
-              body    = (pretty_print status) }
+              body    = (pretty_print status),
+              headers = ((headers message) ++ ["Date: " ++ date])}
 
+
+-- |This is the main recursive loop. It takes a username, a
+-- latest_status_id, and optionally a 'Message' as arguments. The
+-- latest_status_id is the last status (that we know of) to be posted
+-- to username's Twitter account. If we find any newer statuses when
+-- we check, they are printed and optionally emailed (if a 'Message'
+-- was supplied). Then, the process repeats.
 recurse :: String -> Integer -> (Maybe Message) -> IO ()
 recurse username latest_status_id maybe_message = do
     thread_sleep heartbeat
@@ -52,13 +66,16 @@ recurse username latest_status_id maybe_message = do
              recurse username new_latest_status_id maybe_message
              return ()
           Just message -> do
-             let messages = map (message_from_status message) new_statuses
+             date <- getClockTime
+             let messages = map (message_from_status message (show date)) new_statuses
              sendmail_results <- mapM sendmail messages
              mapM print_sendmail_result sendmail_results
              recurse username new_latest_status_id maybe_message
              return ()
 
 
+-- |Try continually to download username's timeline, and determine the
+-- latest status id to be posted once we have done so.
 get_latest_status_id :: String -> IO Integer
 get_latest_status_id username = do
   xmldata <- get_user_timeline username
@@ -77,6 +94,11 @@ get_latest_status_id username = do
 
 
 
+-- |This function wraps two steps. First, we need to find the latest
+-- status id posted by username. Once we have that, we can begin the
+-- 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 :: Maybe Message -> String -> IO ()
 run_twat message username = do
   latest_status_id <- get_latest_status_id username
@@ -84,6 +106,9 @@ run_twat message username = do
   return ()
 
 
+-- |The main function just parses the command-line arguments and then
+-- forks off calls to 'run_twat' for each supplied username. After
+-- forking, main loops forever.
 main :: IO ()
 main = do
   errors <- parse_errors