package config

import (
	"github.com/fsnotify/fsnotify"
	"log"
	"model-engine/pkg/logger"
	"model-engine/pkg/mysqlx"

	"github.com/spf13/viper"
)

var LogConf = &logger.LogConf{}

type esInfo struct {
	Ip                string      `mapstructure:"ip"`
	Port              string      `mapstructure:"port"`
	Shards            string      `mapstructure:"shards"`
	EsIndex           esIndexlist `mapstructure:"index"`
	StorePath         []string    `mapstructure:"storePath"`
	ThresholdTime     int         `mapstructure:"thresholdTime"`
	ThresholdStayTime int         `mapstructure:"thresholdStayTime"`
}

type esIndexlist struct {
	AiOcean index `mapstructure:"aiocean"`
	UserLog index `mapstructure:"userLog"`
}
type index struct {
	IndexName string `mapstructure:"index"`
	IndexType string `mapstructure:"type"`
}

var EsInfo = &esInfo{}

var MysqlConf = &mysqlx.Conf{}

// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
func Init() {
	var err error
	v := viper.New()
	v.SetConfigType("yaml")
	v.SetConfigName("model-engine")
	v.AddConfigPath("../config/")
	v.AddConfigPath("./config/")
	err = v.ReadInConfig()
	if err != nil {
		log.Fatal("error on parsing configuration file")
	}
	read2Conf(v)
	v.WatchConfig()
	v.OnConfigChange(func(in fsnotify.Event) {
		read2Conf(v)
	})
	showConfig()
}

func read2Conf(v *viper.Viper) {
	v.UnmarshalKey("log", LogConf)
	v.UnmarshalKey("mysqlConf", MysqlConf)
	v.UnmarshalKey("es", EsInfo)

}

func showConfig() {
	log.Println("......................................................")
	log.Printf("   log:                %+v", LogConf)
	log.Printf("   EsInfo:                 %+v", EsInfo)
	log.Printf("   mysqlConf:                 %+v", MysqlConf)
}