package cmd

import (
	"os"
	"strings"

	log "github.com/sirupsen/logrus"
	"github.com/spf13/viper"

	"github.com/noerw/osem_notify/core"
	"github.com/noerw/osem_notify/utils"
)

// initConfig reads in config file and ENV variables if set.
func initConfig() {
	theConfig := cfgFile
	if cfgFile == "" {
		theConfig = utils.GetConfigFile("osem_notify")
	}

	viper.SetConfigType("yaml")
	viper.SetConfigFile(theConfig)
	viper.SetEnvPrefix("OSEM_NOTIFY") // keys only work in upper case
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
	viper.AutomaticEnv() // WARNING: OSEM_NOTIFIY_CONFIG will not be considered this way. but why should it..

	// If a config file is found, read it in.
	if _, err := os.Stat(theConfig); err == nil {
		err := viper.ReadInConfig()
		if err != nil {
			log.Error("Error when reading config file:", err)
		}
	} else if cfgFile != "" {
		log.Error("Specified config file not found!")
		os.Exit(1)
	}

	validateConfig()
}

func validateConfig() {
	if viper.GetString("notify") != "" {
		if len(viper.GetStringSlice("healthchecks.default.notifications.options.recipients")) == 0 {
			log.Warn("No default recipients set up for notifications!")
		}

		var conf = &core.TransportConfig{}
		if err := viper.UnmarshalKey("healthchecks.default.notifications", conf); err != nil {
			log.Error("invalid default notification configuration: ", err)
			os.Exit(1)
		}

		// creating a notifier validates its configuration
		_, err := core.GetNotifier(conf)
		if err != nil {
			log.Error(err)
			os.Exit(1)
		}
	}
}

func getNotifyConf(boxID string) (*core.NotifyConfig, error) {
	// config used when no configuration is present at all
	conf := &core.NotifyConfig{
		Events: []core.NotifyEvent{
			core.NotifyEvent{
				Type:      "measurement_age",
				Target:    "all",
				Threshold: "15m",
			},
			core.NotifyEvent{
				Type:      "measurement_faulty",
				Target:    "all",
				Threshold: "",
			},
		},
	}

	// override with default configuration from file
	// considering the case that .events may be defined but empty
	// to allow to define no events, and don't leak shorter lists into
	// previous longer ones
	if keyDefined("healthchecks.default.events") {
		conf.Events = []core.NotifyEvent{}
	}
	if err := viper.UnmarshalKey("healthchecks.default", conf); err != nil {
		return nil, err
	}

	// override with per box configuration from file
	if keyDefined("healthchecks." + boxID + ".events") {
		conf.Events = []core.NotifyEvent{}
	}
	if err := viper.UnmarshalKey("healthchecks."+boxID, conf); err != nil {
		return nil, err
	}

	return conf, nil
}

// implement our own keyCheck, as viper.InConfig() does not work
func keyDefined(key string) bool {
	allConfKeys := viper.AllKeys()
	for _, k := range allConfKeys {
		if k == key {
			return true
		}
	}
	return false
}