Add a CLI parameter for input files

For many data collectors it is useful (especially for testing) to have to
possibility to specify an input file.

This commit adds a generic option for doing that.

Signed-off-by: Michele Tartara <mtartara@google.com>
Reviewed-by: Klaus Aehlig <aehlig@google.com>

diff --git a/src/Ganeti/DataCollectors/CLI.hs b/src/Ganeti/DataCollectors/CLI.hs
index 3c58a3f..db30441 100644 (file)
--- a/src/Ganeti/DataCollectors/CLI.hs
+++ b/src/Ganeti/DataCollectors/CLI.hs
@@ -39,6 +39,7 @@ module Ganeti.DataCollectors.CLI
   , oNode
   , oConfdAddr
   , oConfdPort
+  , oInputFile
   , genericOptions
   ) where
 
@@ -64,6 +65,8 @@ data Options = Options
   , optConfdAddr   :: Maybe String   -- ^ IP address of the Confd server
   , optConfdPort   :: Maybe Int      -- ^ The port of the Confd server to
                                      -- connect to
+  , optInputFile   :: Maybe FilePath -- ^ Path to the file containing the
+                                     -- information to be parsed
   } deriving Show
 
 -- | Default values for the command line options.
@@ -77,6 +80,7 @@ defaultOptions  = Options
   , optNode        = Nothing
   , optConfdAddr   = Nothing
   , optConfdPort   = Nothing
+  , optInputFile   = Nothing
   }
 
 -- | Abbreviation for the option type.
@@ -127,6 +131,13 @@ oConfdPort =
     "Network port of the Confd server to connect to",
     OptComplInteger)
 
+oInputFile :: OptType
+oInputFile =
+  ( Option "f" ["file"]
+      (ReqArg (\ f o -> Ok o { optInputFile = Just f }) "FILE")
+      "the input FILE",
+    OptComplFile)
+
 -- | Generic options.
 genericOptions :: [GenericOptType Options]
 genericOptions =  [ oShowVer