X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fhalcyon.git;a=blobdiff_plain;f=src%2FCommandLine.hs;h=919a3a2136f11a76a9da1441e36b7497084ad407;hp=da6a04b06b35e2f2cdf3472bb68d53311611e05c;hb=bfe49b970e83f1d422d57ece7cebdfe7a56b817c;hpb=69b8af30f49aaad0f5c051998d2556b9ec291df7 diff --git a/src/CommandLine.hs b/src/CommandLine.hs index da6a04b..919a3a2 100644 --- a/src/CommandLine.hs +++ b/src/CommandLine.hs @@ -2,8 +2,11 @@ -- It should more or less be a black box, providing Main with only the -- information it requires. module CommandLine -( help_set, +( heartbeat, + help_set, help_text, + ignore_replies_set, + ignore_retweets_set, from_email_address, to_email_address, parse_errors, @@ -17,7 +20,10 @@ import System.Environment (getArgs) -- |A record containing values for all available options. -data Options = Options { opt_help :: Bool, +data Options = Options { opt_heartbeat :: Maybe Int, + opt_help :: Bool, + opt_ignore_replies :: Bool, + opt_ignore_retweets :: Bool, opt_from :: Maybe String, opt_to :: Maybe String } @@ -25,7 +31,10 @@ data Options = Options { opt_help :: Bool, -- |Constructs an instance of Options, with each of its members set to -- default values. default_options :: Options -default_options = Options { opt_help = False, +default_options = Options { opt_heartbeat = Just 600, + opt_help = False, + opt_ignore_replies = False, + opt_ignore_retweets = False, opt_from = Nothing, opt_to = Nothing } @@ -40,16 +49,63 @@ default_options = Options { opt_help = False, -- options :: [OptDescr (Options -> IO Options)] options = - [ Option ['h'][] (NoArg set_help) "Prints this help message.", - Option ['t'][] (ReqArg set_to "email_address") "Send tweets TO email_address.", - Option ['f'][] (ReqArg set_from "email_address") "Send tweets FROM email_address." + [ Option + ['h']["help"] + (NoArg set_help) + "Prints this help message.", + + Option + ['n']["heartbeat"] + (ReqArg set_heartbeat "heartbeat") + "How many seconds to wait between polling.", + + Option + ['t']["to"] + (ReqArg set_to "email_address") + "Send tweets TO email_address.", + + Option + ['f']["from"] + (ReqArg set_from "email_address") + "Send tweets FROM email_address.", + + Option + ['i']["ignore-replies"] + (NoArg set_ignore_replies) + "Ignore replies.", + + Option + ['I']["ignore-retweets"] + (NoArg set_ignore_retweets) + "Ignore retweets." ] +-- | Attempt to parse an 'Int' from a 'String'. This is just a 'Maybe' +-- wrapper around 'reads'. +parse_int :: String -> Maybe Int +parse_int s = + case (reads s) of + [(n,_)] -> Just n + _ -> Nothing + +set_heartbeat :: String -> Options -> IO Options +set_heartbeat arg opts = do + let new_heartbeat = parse_int arg + return opts { opt_heartbeat = new_heartbeat } + set_help :: Options -> IO Options set_help opts = do return opts { opt_help = True } +set_ignore_retweets :: Options -> IO Options +set_ignore_retweets opts = + return opts { opt_ignore_retweets = True } + +set_ignore_replies :: Options -> IO Options +set_ignore_replies opts = + return opts { opt_ignore_replies = True } + set_to :: String -> Options -> IO Options set_to arg opts = do return opts { opt_to = Just arg } @@ -61,7 +117,7 @@ set_from arg opts = do -- The usage header. usage :: String -usage = "Usage: twat [-t to_address] [-f from_address] [username2, [username3]...]" +usage = "Usage: twat [-n heartbeat] [-t to_address] [-f from_address] [username2, [username3]...]" -- The usage header, and all available flags (as generated by GetOpt) @@ -84,6 +140,14 @@ parse_options = do return opts +-- | A list of parse errors relating to the heartbeat. +heartbeat_errors :: IO [String] +heartbeat_errors = do + hb <- heartbeat + if (isNothing hb) + then return ["\"heartbeat\" does not appear to be an integer."] + else return [] + -- |Parse errors relating to the list of usernames. username_errors :: IO [String] username_errors = do @@ -91,7 +155,7 @@ username_errors = do let (_, usernames, _) = getOpt Permute options argv if (null usernames) - then return ["No usernames provided."] + then return ["no usernames provided."] else return [] @@ -101,7 +165,7 @@ to_errors = do toaddr <- to_email_address fromaddr <- from_email_address if (isNothing toaddr) && (isJust fromaddr) - then return ["\"From\" address specified without \"To\" address."] + then return ["\"from\" address specified without \"to\" address."] else return [] @@ -111,7 +175,7 @@ from_errors = do toaddr <- to_email_address fromaddr <- from_email_address if (isJust toaddr) && (isNothing fromaddr) - then return ["\"To\" address specified without \"From\" address."] + then return ["\"to\" address specified without \"from\" address."] else return [] @@ -125,10 +189,15 @@ parse_errors :: IO [String] parse_errors = do argv <- getArgs let (_, _, errors) = getOpt Permute options argv + errs_heartbeat <- heartbeat_errors errs_username <- username_errors errs_to <- to_errors errs_from <- from_errors - return $ map format_error (errors ++ errs_username ++ errs_to ++ errs_from) + return $ map format_error (errors ++ + errs_heartbeat ++ + errs_username ++ + errs_to ++ + errs_from) -- |Was the "help" option passed on the command line? help_set :: IO Bool @@ -136,6 +205,23 @@ help_set = do opts <- parse_options return (opt_help opts) +-- | Was the "ignore-replies" option passes on the command line? +ignore_replies_set :: IO Bool +ignore_replies_set = do + opts <- parse_options + return (opt_ignore_replies opts) + +-- | Was the "ignore-retweets" option passes on the command line? +ignore_retweets_set :: IO Bool +ignore_retweets_set = do + opts <- parse_options + return (opt_ignore_retweets opts) + +-- |What's the heartbeat? +heartbeat :: IO (Maybe Int) +heartbeat = do + opts <- parse_options + return (opt_heartbeat opts) -- |What "To" address was given on the command line? to_email_address :: IO (Maybe String)