X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FCommandLine.hs;h=ad60c9f0a7e1667037e49e7131d194adbea0a2a8;hb=4bce4546cbd70c8aab8ca81441010e0527ebbb07;hp=1088f2a906a2d2055143f1c0b2036f039fb0155c;hpb=a6d2e7470f17b44c0e8fe31c1268488a6788631f;p=email-validator.git

diff --git a/src/CommandLine.hs b/src/CommandLine.hs
index 1088f2a..ad60c9f 100644
--- a/src/CommandLine.hs
+++ b/src/CommandLine.hs
@@ -1,24 +1,34 @@
 {-# LANGUAGE DeriveDataTypeable #-}
 
-module CommandLine
+module CommandLine (
+  Args(..),
+  get_args )
 where
 
-import System.Console.CmdArgs
-import System.Console.CmdArgs.Explicit (process)
-import System.Environment (getArgs, withArgs)
-import System.Exit (ExitCode(..), exitWith)
-import System.IO (hPutStrLn, stderr)
+import System.Console.CmdArgs (
+  Data,
+  Typeable,
+  (&=),
+  cmdArgs,
+  def,
+  details,
+  help,
+  program,
+  summary,
+  typFile)
+
 
 -- Get the version from Cabal.
 import Paths_email_validator (version)
 import Data.Version (showVersion)
 
-import ExitCodes
 
 -- We optionally accept input/output files to use instead of
 -- stdin/stdout.
-data Args = Args { input_file :: Maybe FilePath,
-                   output_file :: Maybe FilePath }
+data Args = Args { accept_a :: Bool,
+                   input_file :: Maybe FilePath,
+                   output_file :: Maybe FilePath,
+                   rfc5322 :: Bool }
   deriving   (Show, Data, Typeable)
 
 description :: String
@@ -30,6 +40,10 @@ program_name = "email-validator"
 my_summary :: String
 my_summary = program_name ++ "-" ++ (showVersion version)
 
+accept_a_help :: String
+accept_a_help =
+  "Accept an 'A' record for the domain instead of requiring an MX record."
+
 input_file_help :: String
 input_file_help =
   "Path to the input file (default: stdin), one email address per line"
@@ -38,35 +52,21 @@ output_file_help :: String
 output_file_help =
   "Path to the output file (default: stdout)"
 
-arg_spec :: Mode (CmdArgs Args)
+rfc5322_help :: String
+rfc5322_help =
+  "Validate according to RFC 5322 (incredibly lenient)."
+
+arg_spec :: Args
 arg_spec =
-  cmdArgsMode $
-    Args { input_file  = def &= typFile &= help input_file_help,
-           output_file = def &= typFile &= help output_file_help }
+    Args { accept_a    = def &=            help accept_a_help,
+           input_file  = def &= typFile &= help input_file_help,
+           output_file = def &= typFile &= help output_file_help,
+           rfc5322     = def &=            help rfc5322_help }
       &= program program_name
       &= summary my_summary
       &= details [description]
 
-show_help :: IO (CmdArgs Args)
-show_help = withArgs ["--help"] parse_args
-
-
-
-parse_args :: IO (CmdArgs Args)
-parse_args = do
-  x <- getArgs
-  let y = process arg_spec x
-  case y of
-    Right result -> return result
-    Left err -> do
-      hPutStrLn stderr err
-      exitWith (ExitFailure exit_args_parse_failed)
 
 
--- | Really get the command-line arguments. This calls 'parse_args'
---   first to replace the default "wrong number of arguments" error,
---   and then runs 'cmdArgsApply' on the result to do what the
---   'cmdArgs' function usually does.
-apply_args :: IO Args
-apply_args =
-  parse_args >>= cmdArgsApply
+get_args :: IO Args
+get_args = cmdArgs arg_spec