package config

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

const (
	LogLevelKey  = "LOG_LEVEL"  // 日志级别
	LogPathKey   = "LOG_PATH"   // 日志路径
	LogFormatKey = "LOG_FORMAT" // 日志格式
	PortKey      = "PORT"       // 端口
	AppEnvKey    = "APP_ENV"    // 环境

	DBUserNameKey = "DB_USERNAME" // 数据库用户名
	DBPasswordKey = "DB_PASSWORD" // 数据库密码
	DBHostKey     = "DB_HOST"     // 数据库地址
	DBPortKey     = "DB_PORT"     // 数据库端口
	DBNameKey     = "DB_NAME"     // 数据库名
)

func init() {
	// 设置默认值

	// 日志
	viper.SetDefault(LogLevelKey, "debug")
	viper.SetDefault(LogPathKey, "./data/logs/ecpm.log")
	viper.SetDefault(LogFormatKey, "json")

	// 端口
	viper.SetDefault(PortKey, "8080")

	// 环境
	viper.SetDefault(AppEnvKey, "")

	// 数据库
	viper.SetDefault(DBUserNameKey, "root")
	viper.SetDefault(DBPasswordKey, "youtu!0113")
	viper.SetDefault(DBHostKey, "localhost")
	viper.SetDefault(DBPortKey, "3306")
	viper.SetDefault(DBNameKey, "ecpm")

	viper.AutomaticEnv()
}

// GetLogLevel 获取日志级别
func GetLogLevel() string {
	return viper.GetString(LogLevelKey)
}

// GetLogPath 获取日志路径
func GetLogPath() string {
	return viper.GetString(LogPathKey)
}

// GetLogFormat 获取日志格式
func GetLogFormat() string {
	return viper.GetString(LogFormatKey)
}

// GetPort 获取端口
func GetPort() string {
	return viper.GetString(PortKey)
}

// GetAppEnv 获取环境
func GetAppEnv() string {
	return viper.GetString(AppEnvKey)
}

// GetDBUserName 获取数据库用户名
func GetDBUserName() string {
	return viper.GetString(DBUserNameKey)
}

// GetDBPassword 获取数据库密码
func GetDBPassword() string {
	return viper.GetString(DBPasswordKey)
}

// GetDBHost 获取数据库地址
func GetDBHost() string {
	return viper.GetString(DBHostKey)
}

// GetDBPort 获取数据库端口
func GetDBPort() int {
	return viper.GetInt(DBPortKey)
}

// GetDBName 获取数据库名
func GetDBName() string {
	return viper.GetString(DBNameKey)
}