package setting

import (
	"fmt"
	"strings"

	"github.com/spf13/viper"
)

// import (
// 	"fmt"
// 	"strings"

// 	"github.com/spf13/viper"
// )

type Settings = viper.Viper

var s *Settings

// 导出函数, 如果需要使用其他函数. 参考: https://github.com/spf13/viper/blob/master/viper.go
var (
	GetString               = viper.GetString   // 获取字符串
	GetBool                 = viper.GetBool     // 获取 bool
	GetDuration             = viper.GetDuration // 获取时间长度
	GetInt                  = viper.GetInt      // 获取整形
	GetInt32                = viper.GetInt32
	GetInt64                = viper.GetInt64 // 获取64位整形
	GetUint                 = viper.GetUint
	GetUint16               = viper.GetUint16
	GetUint32               = viper.GetUint32
	GetUint64               = viper.GetUint64
	GetTime                 = viper.GetTime
	GetIntSlice             = viper.GetIntSlice
	GetFloat64              = viper.GetFloat64
	GetStringSlice          = viper.GetStringSlice
	GetStringMap            = viper.GetStringMap
	GetStringMapString      = viper.GetStringMapString
	GetStringMapStringSlice = viper.GetStringMapStringSlice
	GetSizeInBytes          = viper.GetSizeInBytes
)

func NewSettings(fileName string) *Settings {
	if fileName != "" {
		viper.SetConfigName(fileName) // name of config file (without extension)
		viper.AddConfigPath(".")
		err := viper.ReadInConfig() // Find and read the config file
		if err != nil {             // Handle errors reading the config file
			// panic(fmt.Errorf("配置文件不存在: %s", err))
			fmt.Printf("配置文件不存在: %s\n", err)
			// myLogger.Error("当前文件夹没有找到 配置文件")
		}
	}
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
	viper.AutomaticEnv()
	return viper.GetViper()
}

func Init(fileName string) *Settings {
	s = NewSettings(fileName)
	return s
}

func GetSettings() *Settings {
	if s == nil {
		panic("请先初始化settings")
	}
	return s
}