120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package cachet
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
|
|
"github.com/castawaylabs/cachet-monitor/system"
|
|
)
|
|
|
|
// Static config
|
|
var Config CachetConfig
|
|
|
|
// Central logger
|
|
var Logger *log.Logger
|
|
|
|
// CachetConfig is the monitoring tool configuration
|
|
type CachetConfig struct {
|
|
APIUrl string `json:"api_url"`
|
|
APIToken string `json:"api_token"`
|
|
Interval int64 `json:"interval"`
|
|
Monitors []*Monitor `json:"monitors"`
|
|
SystemName string `json:"system_name"`
|
|
LogPath string `json:"log_path"`
|
|
InsecureAPI bool `json:"insecure_api"`
|
|
}
|
|
|
|
func init() {
|
|
var configPath string
|
|
var systemName string
|
|
var logPath string
|
|
flag.StringVar(&configPath, "c", "/etc/cachet-monitor.config.json", "Config path")
|
|
flag.StringVar(&systemName, "name", "", "System Name")
|
|
flag.StringVar(&logPath, "log", "", "Log path")
|
|
flag.Parse()
|
|
|
|
var data []byte
|
|
|
|
// test if its a url
|
|
url, err := url.ParseRequestURI(configPath)
|
|
if err == nil && len(url.Scheme) > 0 {
|
|
// download config
|
|
response, err := http.Get(configPath)
|
|
if err != nil {
|
|
fmt.Printf("Cannot download network config: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
data, _ = ioutil.ReadAll(response.Body)
|
|
|
|
fmt.Println("Downloaded network configuration.")
|
|
} else {
|
|
data, err = ioutil.ReadFile(configPath)
|
|
if err != nil {
|
|
fmt.Println("Config file '" + configPath + "' missing!")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
err = json.Unmarshal(data, &Config)
|
|
|
|
if err != nil {
|
|
fmt.Println("Cannot parse config!")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if len(systemName) > 0 {
|
|
Config.SystemName = systemName
|
|
}
|
|
if len(Config.SystemName) == 0 {
|
|
// get hostname
|
|
Config.SystemName = system.GetHostname()
|
|
}
|
|
|
|
if len(os.Getenv("CACHET_API")) > 0 {
|
|
Config.APIUrl = os.Getenv("CACHET_API")
|
|
}
|
|
if len(os.Getenv("CACHET_TOKEN")) > 0 {
|
|
Config.APIToken = os.Getenv("CACHET_TOKEN")
|
|
}
|
|
|
|
if len(Config.APIToken) == 0 || len(Config.APIUrl) == 0 {
|
|
fmt.Printf("API URL or API Token not set. cachet-monitor won't be able to report incidents.\n\nPlease set:\n CACHET_API and CACHET_TOKEN environment variable to override settings.\n\nGet help at https://github.com/CastawayLabs/cachet-monitor\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if len(Config.Monitors) == 0 {
|
|
fmt.Printf("No monitors defined!\nSee sample configuration: https://github.com/CastawayLabs/cachet-monitor/blob/master/example.config.json\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if len(logPath) > 0 {
|
|
Config.LogPath = logPath
|
|
}
|
|
|
|
var logWriter io.Writer
|
|
logWriter = os.Stdout
|
|
if len(Config.LogPath) > 0 {
|
|
logWriter, err = os.Create(Config.LogPath)
|
|
if err != nil {
|
|
fmt.Printf("Unable to open file '%v' for logging\n", Config.LogPath)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
flags := log.Llongfile | log.Ldate | log.Ltime
|
|
if len(os.Getenv("DEVELOPMENT")) > 0 {
|
|
flags = 0
|
|
}
|
|
|
|
Logger = log.New(logWriter, "", flags)
|
|
}
|