56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"nocternity.net/go-monitoring/cmd/sslcert"
|
||
|
|
"nocternity.net/go-monitoring/cmd/zoneserial"
|
||
|
|
"nocternity.net/go-monitoring/pkg/program"
|
||
|
|
"nocternity.net/go-monitoring/pkg/version"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
programs map[string]program.ProgramBuilder = map[string]program.ProgramBuilder{
|
||
|
|
"check_ssl_certificate": sslcert.NewProgram,
|
||
|
|
"check_zone_serial": zoneserial.NewProgram,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func exitError(msg string, args ...interface{}) {
|
||
|
|
fmt.Fprintf(os.Stderr, msg+"\n", args...)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
ownName := filepath.Base(os.Args[0])
|
||
|
|
|
||
|
|
var program program.Program
|
||
|
|
if builder, ok := programs[ownName]; ok {
|
||
|
|
program = builder()
|
||
|
|
} else if len(os.Args) < 2 {
|
||
|
|
exitError("Syntax: %s <program> [arguments]\n %s programs", os.Args[0], os.Args[0])
|
||
|
|
} else if os.Args[1] == "programs" {
|
||
|
|
fmt.Println("Available programs:")
|
||
|
|
for name := range programs {
|
||
|
|
fmt.Printf(" %s\n", name)
|
||
|
|
}
|
||
|
|
os.Exit(0)
|
||
|
|
} else if os.Args[1] == "--version" || os.Args[1] == "-v" {
|
||
|
|
fmt.Printf("%s %s\n", ownName, version.Version())
|
||
|
|
os.Exit(0)
|
||
|
|
|
||
|
|
} else if builder, ok := programs[os.Args[1]]; ok {
|
||
|
|
os.Args = os.Args[1:]
|
||
|
|
program = builder()
|
||
|
|
} else {
|
||
|
|
exitError("Unknown program: %s", os.Args[1])
|
||
|
|
}
|
||
|
|
|
||
|
|
defer program.Done()
|
||
|
|
if program.CheckArguments() {
|
||
|
|
program.RunCheck()
|
||
|
|
}
|
||
|
|
}
|