package conf

import (
	"log"
	"os"
	"wms/pkg/logx"
	"wms/pkg/mysqlx"

	"github.com/spf13/viper"
)

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

type (
	webConf struct {
		Host        string // 本机ip地址
		Port        string // 端口号
		NodeId      string // 主账户用户名
		OssType     string // 对象存储类型
		JWTSecret   string
		FileServer  string //文件服务器地址
		ServerId    string //服务ID
		GrpcPort    string //grpc端口号
		CompanyName string //公司名
	}

	localConf struct {
		StorePath string // 本地文件存储路径
	}

	grpcServerConf struct {
		ApsAddr string //aps服务grpc地址
		CrmAddr string //crm服务grpc地址
		SrmAddr string //srm服务grpc地址
	}

	fileTemplateConf struct {
		InputSelfmadeAddr string //入库自制单模版地址
	}

	dingTalkConf struct {
		AlarmKey string
		AlarmUrl string
	}
)

var (
	WebConf        = &webConf{}
	LogConf        = &logx.Conf{}
	DbConf         = &mysqlx.Conf{}
	LocalConf      = &localConf{}
	Viper          *viper.Viper
	GrpcServerConf = &grpcServerConf{}
	DingTalkConf   = &dingTalkConf{}
)

func Init() error {
	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)
	}
	read2Conf(Viper)

	nodeId := os.Getenv("NODE_ID")    // 主账户用户名
	host := os.Getenv("HOST")         // 本机IP地址
	GrpcPort := os.Getenv("WMS_GRPC") // 只对外提供grpc服务，本服务不用
	apsAddr := os.Getenv("APS_GRPC")
	crmAddr := os.Getenv("CRM_GRPC")
	srmAddr := os.Getenv("SRM_GRPC")
	companyName := os.Getenv("COMPANY_NAME")
	if len(GrpcPort) > 0 {
		WebConf.GrpcPort = GrpcPort
	}
	if len(nodeId) > 0 {
		WebConf.NodeId = nodeId
	}
	if len(host) > 0 {
		WebConf.Host = host
	}
	if len(apsAddr) > 0 {
		GrpcServerConf.ApsAddr = apsAddr
	}
	if len(crmAddr) > 0 {
		GrpcServerConf.CrmAddr = crmAddr
	}
	if len(srmAddr) > 0 {
		GrpcServerConf.SrmAddr = srmAddr
	}
	if len(companyName) > 0 {
		WebConf.CompanyName = companyName
	}

	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 {
		DbConf.Dsn = DBUser + ":" + DBPasswd + "@tcp(" + DBHost + ":" + DBPort + ")/" + DBName + "?charset=utf8&parseTime=True&loc=Local"
	}

	return nil
}

func read2Conf(v *viper.Viper) {
	_ = v.UnmarshalKey("web", WebConf)
	_ = v.UnmarshalKey("log", LogConf)
	_ = v.UnmarshalKey("db", DbConf)
	_ = v.UnmarshalKey("local", LocalConf)
	_ = v.UnmarshalKey("grpcServer", GrpcServerConf)
	_ = v.UnmarshalKey("dingTalk", DingTalkConf)
	showConfig()
}

func showConfig() {
	log.Println("......................................................")
	log.Printf("   WebConf:                %+v", WebConf)
	log.Printf("   LogConf:                %+v", LogConf)
	log.Printf("   DbConf:                 %+v", DbConf)
	log.Printf("   LocalConf:               %+v", LocalConf)
	log.Printf("   GrpcServerConf:               %+v", GrpcServerConf)
	log.Printf("   DingTalkConf:               %+v", DingTalkConf)
	log.Println("......................................................")
}