예제 #1
0
func LoadEmptyConfig() {
	var logWriter io.Writer
	logWriter = os.Stdout
	Config = CachetConfig{}
	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")
	}
	Config.SystemName = system.GetHostname()
	flags := log.Llongfile | log.Ldate | log.Ltime
	if len(os.Getenv("DEVELOPMENT")) > 0 {
		flags = 0
	}

	Logger = log.New(logWriter, "", flags)
}
예제 #2
0
func LoadCachetConfigClassic() {
	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)
}