You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
// Copyright 2018 Felix Kronlage <fkr@hazardous.org>
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/jessevdk/go-flags"
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
type CheckCommand struct {
|
|
Servers ServersCommand `command:"servers" description:"checks on servers"`
|
|
Snapshots SnapshotsCommand `command:"snapshots" description:"checks on snapshots"`
|
|
Storages StoragesCommand `command:"storages" description:"checks on storages"`
|
|
}
|
|
|
|
var Checker CheckCommand
|
|
|
|
var Config CheckConfig
|
|
|
|
type CheckConfig struct {
|
|
UserId string
|
|
Token string
|
|
}
|
|
|
|
func main() {
|
|
|
|
var parser = flags.NewParser(&Checker, flags.Default)
|
|
|
|
if (os.Getenv("GRIDSCALE_USER") != "") && (os.Getenv("GRIDSCALE_TOKEN") != "") {
|
|
Config.UserId = os.Getenv("GRIDSCALE_USER")
|
|
Config.Token = os.Getenv("GRIDSCALE_TOKEN")
|
|
} else {
|
|
// look in check.toml for the config
|
|
viper.SetConfigName("check")
|
|
viper.AddConfigPath("config")
|
|
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
log.Fatalf("Config file not found or error parsing: %v", err)
|
|
}
|
|
|
|
Config.UserId = viper.GetString("gridscale.userid")
|
|
Config.Token = viper.GetString("gridscale.token")
|
|
}
|
|
|
|
if _, err := parser.Parse(); err != nil {
|
|
if flagsErr, ok := err.(*flags.Error); ok && flagsErr.Type == flags.ErrHelp {
|
|
os.Exit(0)
|
|
} else {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
Unknown("no subcommands specificed.")
|
|
}
|
|
|
|
func Ok(msg string) {
|
|
fmt.Println("OK - " + msg)
|
|
os.Exit(0)
|
|
}
|
|
|
|
func Warning(msg string) {
|
|
fmt.Println("WARNING - " + msg)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func Critical(msg string) {
|
|
fmt.Println("CRITICAL - " + msg)
|
|
os.Exit(2)
|
|
}
|
|
|
|
func Unknown(msg string) {
|
|
fmt.Println("UNKNOWN - " + msg)
|
|
os.Exit(3)
|
|
}
|