Add the ability to twat a single status (as a debugging tool).

[dead/halcyon.git] / src / Main.hs

diff --git a/src/Main.hs b/src/Main.hs
index 53c3f905e863e9c98537e4b62d5307335b573939..fc7966f9f1df518283cae1ce2705b1e1effb058e 100644 (file)
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -4,7 +4,6 @@ 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
@@ -59,17 +58,17 @@ recurse username latest_status_id maybe_message = do
         recurse username latest_status_id maybe_message
       _ -> do
         let new_latest_status_id = get_max_status_id new_statuses
-        mapM (putStrLn . pretty_print) new_statuses
+        _ <- mapM (putStrLn . pretty_print) new_statuses
 
         case maybe_message of
           Nothing -> do
              recurse username new_latest_status_id maybe_message
              return ()
           Just message -> do
-             date <- getClockTime
-             let messages = map (message_from_status message (show date)) new_statuses
+             date_header <- construct_date_header
+             let messages = map (message_from_status message (date_header)) new_statuses
              sendmail_results <- mapM sendmail messages
-             mapM print_sendmail_result sendmail_results
+             _ <- mapM print_sendmail_result sendmail_results
              recurse username new_latest_status_id maybe_message
              return ()
 
@@ -139,19 +138,50 @@ main = do
                       case from_address of
                         Nothing -> Nothing
                         Just fromaddr ->
-                            Just (Message { headers = [],
+                            Just (Message { headers = default_headers,
                                             body = "",
                                             subject = "",
                                             to = toaddr,
                                             from = fromaddr })
 
   -- Execute run_twat on each username in a new thread.
-  mapM (forkIO . (run_twat message)) usernames
+  _ <- mapM (forkIO . (run_twat message)) usernames
 
-  forever $ do
+  _ <- forever $ do
     -- This thread (the one executing main) doesn't do anything,
     -- but when it terminates, so do all the threads we forked.
     -- As a result, we need to keep this thread on life support.
     thread_sleep heartbeat
 
   return ()
+
+
+-- |A debugging tool that will parse, print, and email a single status
+-- (given by its id).
+twat_single_status :: Integer -> (Maybe Message) -> IO ()
+twat_single_status the_status_id maybe_message = do
+    xmldata <- get_status the_status_id
+
+    -- Parsing an empty result can blow up. Just pretend there are
+    -- no new statuses in that case.
+    let statuses = case xmldata of
+                     Just xml -> parse_status xml
+                     Nothing  -> []
+
+    case (length statuses) of
+      0 -> do
+        putStrLn "No statuses returned."
+        return ()
+      _ -> do
+        _ <- mapM (putStrLn . pretty_print) statuses
+
+        case maybe_message of
+          Nothing -> do
+             putStrLn "No message object given."
+             return ()
+          Just message -> do
+             date_header <- construct_date_header
+             let messages = map (message_from_status message (date_header)) statuses
+             sendmail_results <- mapM sendmail messages
+             _ <- mapM print_sendmail_result sendmail_results
+             return ()