X-Git-Url: http://gitweb.michael.orlitzky.com/?p=haeredes.git;a=blobdiff_plain;f=src%2FCommandLine.hs;h=0ea89a0173c56456162c10ab882dc1dc35e84341;hp=699ea97aceea436ce6abecbace894d43542984eb;hb=7cc26ab4bfa90c4abe0c741d5a3eaf575544466a;hpb=825753e2731b8491d93d9f2d66b457fe31b1c763

diff --git a/src/CommandLine.hs b/src/CommandLine.hs
index 699ea97..0ea89a0 100644
--- a/src/CommandLine.hs
+++ b/src/CommandLine.hs
@@ -2,46 +2,70 @@
 
 module CommandLine (
   Args(..),
-  Delegates(..),
   get_args
   )
 where
 
-import System.Console.CmdArgs
+import System.Console.CmdArgs (
+  Data,
+  Typeable,
+  (&=),
+  args,
+  auto,
+  cmdArgs,
+  def,
+  details,
+  explicit,
+  groupname,
+  help,
+  helpArg,
+  modes,
+  name,
+  program,
+  summary,
+  typ,
+  versionArg )
 
 -- Get the version from Cabal.
-import Paths_haeres (version)
+import Paths_haeredes (version)
 import Data.Version (showVersion)
 
+-- | Description of the 'NS' mode.
 ns_description :: String
 ns_description =
   "Confirm delegation of NS records. " ++
   "This is the default mode."
 
+-- | Description of the 'MX' mode.
 mx_description :: String
 mx_description = "Confirm delegation of MX records."
 
 program_name :: String
-program_name = "haeres"
+program_name = "haeredes"
 
 my_summary :: String
 my_summary = program_name ++ "-" ++ (showVersion version)
 
+no_append_root_help :: String
+no_append_root_help =
+  "Don't append a trailing dot to DNS names."
+
+-- | Help string for the --server flag.
 server_help :: String
 server_help =
-  "IP address of server to query " ++
-  "(will use resolv.conf if not specified)"
-
-newtype Delegates =
-  Delegates { get_delegates :: [String] }
-  deriving (Data, Show, Typeable)
-
-instance Default Delegates where
-  def = Delegates []
+  "IP address or hostname of server to query " ++
+  "(will use resolv.conf if not specified)."
 
+-- | The Args type represents the possible command-line options. The
+--   duplication here seems necessary; CmdArgs' magic requires us to
+--   define some things explicitly.
 data Args =
-  NS { server :: Maybe String, delegates :: Delegates } |
-  MX { server :: Maybe String, delegates :: Delegates }
+  NS { no_append_root :: Bool,
+       server :: Maybe String,
+       delegates :: [String] } |
+  MX { no_append_root :: Bool,
+       server :: Maybe String,
+       delegates :: [String] }
   deriving (Data, Show, Typeable)
 
 arg_spec :: Args
@@ -58,23 +82,30 @@ arg_spec =
                    name "v",
                    groupname "Common flags"]
   where
-    -- The repetition here is necessary, some Template Haskell magic
-    -- going on.
+    -- The repetition here is necessary, some CmdArgs magic going on.
     ns :: Args
     ns = NS {
+           no_append_root = def
+                         &= groupname "Common flags"
+                         &= help no_append_root_help,
+
            server = def
-                  &= groupname "Common flags"
-                  &= typ "IP"
-                  &= help server_help,
+                 &= groupname "Common flags"
+                 &= typ "IP"
+                 &= help server_help,
 
            delegates = def
-                     &= args
-                     &= typ "DELEGATES" }
+                    &= args
+                    &= typ "DELEGATES" }
 
             &= details [ns_description]
 
     mx :: Args
     mx = MX {
+           no_append_root = def
+                         &= groupname "Common flags"
+                         &= help no_append_root_help,
+
            server = def
                  &= groupname "Common flags"
                  &= typ "IP"