Statistics
| Branch: | Tag: | Revision:

root / htools / mon-collector.hs @ f127e585

History | View | Annotate | Download (1.9 kB)

1
{-| Main binary for all stand-alone data collectors
2

    
3
-}
4

    
5
{-
6

    
7
Copyright (C) 2012 Google Inc.
8

    
9
This program is free software; you can redistribute it and/or modify
10
it under the terms of the GNU General Public License as published by
11
the Free Software Foundation; either version 2 of the License, or
12
(at your option) any later version.
13

    
14
This program is distributed in the hope that it will be useful, but
15
WITHOUT ANY WARRANTY; without even the implied warranty of
16
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17
General Public License for more details.
18

    
19
You should have received a copy of the GNU General Public License
20
along with this program; if not, write to the Free Software
21
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
22
02110-1301, USA.
23

    
24
-}
25

    
26
module Main (main) where
27

    
28
import Data.Char (toLower)
29
import System.Environment
30
import System.IO
31

    
32
import Ganeti.Utils
33
import Ganeti.HTools.CLI (parseOpts, genericOpts)
34
import Ganeti.DataCollectors.Program (personalities)
35

    
36
-- | Display usage and exit.
37
usage :: String -> IO ()
38
usage name = do
39
  hPutStrLn stderr $ "Unrecognised personality '" ++ name ++ "'."
40
  hPutStrLn stderr "This program must be executed specifying one of the \
41
                    \following names as the first parameter:"
42
  mapM_ (hPutStrLn stderr . ("  - " ++) . fst) personalities
43
  exitErr "Please specify the desired role."
44

    
45
main :: IO ()
46
main = do
47
  cmd_args <- getArgs
48
  let binary =
49
        if null cmd_args
50
          then ""
51
          else head cmd_args
52
      name = map toLower binary
53
      boolnames = map (\(x, y) -> (x == name, Just y)) personalities
54
  case select Nothing boolnames of
55
    Nothing -> usage name
56
    Just (fn, options, arguments) -> do
57
         let actual_args = tail cmd_args
58
         real_options <- options
59
         (opts, args) <- parseOpts actual_args name (real_options ++
60
                           genericOpts) arguments
61
         fn opts args