package config

import (
	"basic.com/valib/logger.git"
	"github.com/fsnotify/fsnotify"
	"log"

	"github.com/spf13/viper"
)

type server struct {
	ServerName     string `mapstructure:"serverName"`
	Runmode        string `mapstructure:"runmode"`
	JwtSecret      string `mapstructure:"jwtSecret"`
	JwtExpire      string `mapstructure:"jwtExpire"`
	Url            string `mapstructure:"url"`
	AnalyServerId  string `mapstructure:"analyServerId"`
	NetworkAdapter string `mapstructure:"networkAdapter"`

	DeviceNum       string `mapstructure:"deviceNum"`       //设备编号
	DeviceType      string `mapstructure:"deviceType"`      //设备类型
	DeviceModel     string `mapstructure:"deviceModel"`     //设备型号
	DeviceSerialNum string `mapstructure:"deviceSerialNum"` //设备序列号
	DeviceDesc      string `mapstructure:"deviceDesc"`      //设备描述
	MasterVersion   string `mapstructure:"masterVersion"`   //主控版本
	WebVersion      string `mapstructure:"webVersion"`      //web版本
	ChannelCount    string `mapstructure:"channelCount"`    //通道个数
	DiskCount       string `mapstructure:"diskCount"`       //硬盘个数
	LicensePath     string `mapstructure:"licensePath"`     //license文件位置
	LKey            string `mapstructure:"lKey"`
	LPublicKeyPath  string `mapstructure:"lPublicKeyPath"`

	SudoPassword  string      `mapstructure:"sudoPassword"` //系统密码
	SysThresholds []threshold `mapstructure:"sysThresholds"`
	PTZSpeed      int         `mapstructure:"ptzSpeed"`  // 云台移动速度
	PatchPath     string      `mapstructure:"patchPath"` //系统更新包路径
	BakPath       string      `mapstructure:"bakPath"`   //系统更新包路径
	Voices        []Voice     `mapstructure:"voices"`
	GbWaitSec     int         `mapstructure:"gbWaitSec"`  //国标刷新后，请求等待时间
	GbCamCount    int         `mapstructure:"gbCamCount"` //国标可配置摄像机数量
	RtspUnique    bool        `mapstructure:"rtspUnique"` //false:允许单机rtsp重复，为true时不允许单机上rtsp重复

	Resolutions []resolution `mapstructure:"resolutions"` //数据栈可用的分辨率
	HeadPics    []string     `mapstructure:"headpics"`    //默认头像列表
	Langs       []lang       `mapstructure:"langs"`       //语言
}

type threshold struct {
	Value int    `mapstructure:"value"`
	Color string `mapstructure:"color"`
}

type Voice struct {
	Id   string `mapstructure:"id"`
	Name string `mapstructure:"name"`
	Path string `mapstructure:"path"`
}

type resolution struct {
	Width  int `mapstructure:"width"`
	Height int `mapstructure:"height"`
}

type lang struct {
	Name     string `mapstructure:"name"`
	Lang     string `mapstructure:"lang"`
	Language string `mapstructure:"language"`
}

var Server = &server{}

type database struct {
	Driver   string `mapstructure:"driver"`
	Name     string `mapstructure:"name"`
	FilePath string `mapstructure:"filepath"`
}

var DBconf = &database{}

type TraceConfig struct {
	UseBlockList int `mapstructure:"useBlockList"`
}

var Traceconf = &TraceConfig{}

type dbpersoncompare struct {
	Ip   string `mapstructure:"ip"`
	Port int    `mapstructure:"port"`
}

type cluster struct {
	PwdPre           string `mapstructure:"pwdpre"`
	SerfSnapShotPath string `mapstructure:"serfsnapshotpath"`
	Port             int    `mapstructure:"port"`
	HeartBeat        int    `mapstructure:"heartBeat"`
	SaasReportUrl    string `mapstructure:"saasReportUrl"`
}

var ClusterSet = &cluster{}

type esinfo struct {
	Port              int         `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"`
}
type index struct {
	IndexName string `mapstructure:"index"`
	IndexType string `mapstructure:"type"`
}

type weedfs struct {
	Port int `mapstructure:"port"`
}

type facedetect struct {
	Ip   string `mapstructure:"Ip"`
	Port int    `mapstructure:"port"`
}

type espersoncompare struct {
	Port int      `mapstructure:"port"`
	Ips  []string `mapstructure:"ips"`
}

type storageConfig struct {
	VolumePath []string `mapstructure:"volumePath"`
}

var StorageConf = &storageConfig{}

var DbPersonCompare = &dbpersoncompare{}

var EsCompServerInfo = &espersoncompare{}

var FaceDetectSet = &facedetect{}

var EsConfig = &esinfo{}

var WeedFs = &weedfs{}

var EsInfo = &esinfo{}

type elastic struct {
	IndexName string `mapstructure:"indexName"`
	IndexType string `mapstructure:"indexType"`
}

var BasicFS = &elastic{}

type logConfig struct {
	Path       string `mapstructure:"path"`       //日志存储路径
	Level      int    `mapstructure:"level"`      //日志等级
	MaxSize    int    `mapstructure:"maxSize"`    //日志文件大小上限
	MaxBackups int    `mapstructure:"maxBackups"` //日志压缩包个数
	MaxAge     int    `mapstructure:"maxAge"`     //保留压缩包天数
}

var LogConf = &logConfig{}

type shop struct {
	Url        string   `mapstructure:"url"`
	LocalShop  string   `mapstructure:"localShop"`  //本地商城地址
	SysDomains []string `mapstructure:"sysDomains"` //系统升级的域名列表
}

var ShopConf = &shop{}

type saas struct {
	Url string `mapstructure:"url"`
}

var SaasConf = &saas{}

type pushSetMenu struct {
	Id       string        `mapstructure:"id" json:"id"`
	Name     string        `mapstructure:"name" json:"name"`
	Checked  bool          `mapstructure:"checked" json:"checked"`
	Alias    string        `mapstructure:"alias" json:"alias"`
	Children []pushSetMenu `mapstructure:"children" json:"children"`
}

var PushSet = make([]pushSetMenu, 0)

// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
func Init(env string) {
	var err error
	v := viper.New()
	v.SetConfigType("yaml")
	v.SetConfigName(env)
	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)
	})
}

func read2Conf(v *viper.Viper) {
	v.UnmarshalKey("es", EsInfo)
	v.UnmarshalKey("server", Server)
	v.UnmarshalKey("database", DBconf)
	v.UnmarshalKey("trace", Traceconf)

	v.UnmarshalKey("facedetect", FaceDetectSet)
	v.UnmarshalKey("dbpersoncompare", DbPersonCompare)
	v.UnmarshalKey("espersoncompare", EsCompServerInfo)
	v.UnmarshalKey("elastic.basicFS", BasicFS)
	v.UnmarshalKey("shop", ShopConf)
	v.UnmarshalKey("saas", SaasConf)
	v.UnmarshalKey("thresholdTime", EsInfo.ThresholdTime)
	v.UnmarshalKey("thresholdTime", EsInfo.ThresholdStayTime)
	v.UnmarshalKey("cluster", ClusterSet)
	v.UnmarshalKey("es", EsConfig)
	v.UnmarshalKey("weedfs", WeedFs)
	v.UnmarshalKey("log", LogConf)
	v.UnmarshalKey("storage", StorageConf)
	v.UnmarshalKey("pushSet", &PushSet)
	logger.SetLevel(LogConf.Level)
}