Initial work on hail

[ganeti-local] / Ganeti / HTools / CLI.hs

diff --git a/Ganeti/HTools/CLI.hs b/Ganeti/HTools/CLI.hs
index 1a30f58..3d14b4d 100644 (file)
--- a/Ganeti/HTools/CLI.hs
+++ b/Ganeti/HTools/CLI.hs
@@ -9,11 +9,13 @@ and this is more IO oriented.
 module Ganeti.HTools.CLI
     (
       parseOpts
+    , parseEnv
     , showVersion
     , shTemplate
     ) where
 
 import System.Console.GetOpt
+import System.Posix.Env
 import System.IO
 import System.Info
 import System
@@ -46,6 +48,13 @@ parseOpts argv progname options defaultOptions fn =
       where header = printf "%s %s\nUsage: %s [OPTION...]"
                      progname Version.version progname
 
+-- | Parse the environment and return the node/instance names.
+-- This also hardcodes here the default node/instance file names.
+parseEnv :: () -> IO (String, String)
+parseEnv () = do
+  a <- getEnvDefault "HTOOLS_NODES" "nodes"
+  b <- getEnvDefault "HTOOLS_INSTANCES" "instances"
+  return (a, b)
 
 -- | Return a version string for the program
 showVersion :: String -- ^ The program name