package conf

import (
	"aps_crm/pkg/convertx"
	"aps_crm/pkg/logx"
	"aps_crm/pkg/mysqlx"
	"aps_crm/pkg/redisx"
	"flag"
	"github.com/spf13/viper"
	"log"
	"os"
)

var (
	// config file name
	configName = "aps-crm"
	// config file paths
	configPaths = []string{
		"./",
		"../",
		"./conf",
		"../conf",
	}
)

type (
	Captcha struct {
		KeyLong            int // 验证码长度
		ImgWidth           int // 验证码宽度
		ImgHeight          int // 验证码高度
		OpenCaptcha        int // 防爆破验证码开启此数，0代表每次登录都需要验证码，其他数字代表错误密码次数，如3代表错误三次后出现验证码
		OpenCaptchaTimeOut int // 防爆破验证码超时时间，单位：s(秒)
	}

	JWT struct {
		SigningKey  string // jwt签名
		ExpiresTime string // 过期时间
		BufferTime  string // 缓冲时间
		Issuer      string // 签发者
	}

	JWT2 struct {
		SigningKey  string // jwt签名
		ExpiresTime string // 过期时间
		BufferTime  string // 缓冲时间
		Issuer      string // 签发者
	}

	System struct {
		Env           string // 环境值 develop test public
		Port          int    // 端口
		DbType        string // 数据库类型
		UseMultipoint bool   // 多点登录拦截
		UseRedis      bool   // 使用redis
		LimitCountIP  int
		LimitTimeIP   int
		RouterPrefix  string // 路由前缀
		SudoPassword  string // sudo密码
	}

	GrpcServiceAddr struct {
		Aps   string // aps服务地址
		Admin string // admin服务地址
	}

	config struct {
		// 系统配置
		System System

		// 日志
		Log logx.Conf

		// mysql配置
		Mysql mysqlx.Conf

		// redis配置
		Redis redisx.Conf

		// 验证码
		Captcha Captcha

		// JWT配置
		JWT JWT

		GrpcServiceAddr GrpcServiceAddr
	}
)

var (
	Conf  config
	Viper *viper.Viper
)

func init() {
	var configFile string
	flag.StringVar(&configFile, "config", "", "config file in json")
	flag.Parse()

	if len(configFile) > 0 {
		configName = configFile
	}

	Viper = viper.New()
	Viper.SetConfigName(configName)
	for _, path := range configPaths {
		Viper.AddConfigPath(path)
	}

	if err := Viper.ReadInConfig(); err != nil {
		log.Fatalf("ReadInConfig err:%v", err)
	}

	if err := Viper.Unmarshal(&Conf); err != nil {
		log.Fatalf("Unmarshal err:%v", err)
	}

	DBHost := os.Getenv("DB_HOST")
	DBName := os.Getenv("DB_NAME")
	DBPort := os.Getenv("DB_PORT")
	DBUser := os.Getenv("DB_USER")
	DBPasswd := os.Getenv("DB_PASSWD")
	if len(DBHost) > 0 &&
		len(DBName) > 0 &&
		len(DBPort) > 0 &&
		len(DBUser) > 0 &&
		len(DBPasswd) > 0 {
		Conf.Mysql.Dsn = DBUser + ":" + DBPasswd + "@tcp(" + DBHost + ":" + DBPort + ")/" + DBName + "?charset=utf8&parseTime=True&loc=Local"
	}

	AdminGrpc := os.Getenv("ADMIN_GRPC")
	if len(AdminGrpc) > 0 {
		Conf.GrpcServiceAddr.Admin = AdminGrpc
	}

	ApsGrpc := os.Getenv("GRPC_PORT")
	Host := os.Getenv("HOST")

	if len(Host) > 0 && len(ApsGrpc) > 0 {
		Conf.GrpcServiceAddr.Aps = Host + ":" + ApsGrpc
		//Conf.GrpcServiceAddr.Aps = DBUser + DBUser +
	}

	ShowConfig()
	CheckConfig()
}

// WriteConfig 回写配置
func (c *config) WriteConfig() error {
	cs := convertx.StructToMap(Conf)
	for k, v := range cs {
		Viper.Set(k, v)
	}
	return Viper.WriteConfig()
}

func ShowConfig() {
	log.Println("......................................................")
	log.Printf("   System:                %+v", Conf.System)
	log.Printf("   Log:                   %+v", Conf.Log)
	log.Printf("   Mysql:                 %+v", Conf.Mysql)
	log.Printf("   Captcha:               %+v", Conf.Captcha)
	log.Printf("   JWT:                   %+v", Conf.JWT)
	log.Printf("   GrpcServiceAddr:       %+v", Conf.GrpcServiceAddr)
	log.Println("......................................................")
}

func CheckConfig() {
	if Conf.GrpcServiceAddr.Aps == "" {
		log.Fatalf("%v 尚未配置", "GrpcServiceAddr.Aps")
	}
}