package config

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

var Conf *Config

func Init() {
	workDir, _ := os.Getwd()
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(workDir + "/agent/config")
	err := viper.ReadInConfig()
	if err != nil {
		panic(err)
	}
	err = viper.Unmarshal(&Conf)
	if err != nil {
		panic(err)
	}
}

type Config struct {
	Server *Server `yaml:"server"`
	Etcd   *Etcd   `yaml:"etcd"`
	Kafka  *Kafka  `yaml:"kafka"`
	Es     *Es     `yaml:"es"`
	MySQL  *MySQL  `yaml:"mysql"`
	Node   *Node   `yaml:"node"`
}

type Server struct {
	Producer string `yaml:"producer"`
	Consumer string `yaml:"consumer"`
	Host     string `yaml:"host"`
	Version  string `yaml:"version"`
}

type Etcd struct {
	Endpoints []string `yaml:"endpoints"`
	Ttl       int64    `yaml:"ttl"`
}

type Kafka struct {
	Brokers  []string `yaml:"brokers"`
	Topic    string   `yaml:"topic"`
	RetryNum int      `yaml:"retryNum"`
}

type Es struct {
	Address     []string `yaml:"address"`
	MaxChanSize int      `yaml:"maxChanSize"`
	GoNum       int      `yaml:"goNum"`
	Index       Index    `yaml:"index"`
	Username    string   `yaml:"username"`
	Password    string   `yaml:"password"`
}

type Index struct {
	Name       string `yaml:"name"`
	CutByApp   bool   `yaml:"cutByApp"`
	CutByMonth bool   `yaml:"cutByMonth"`
}

type MySQL struct {
	DriverName string `yaml:"driver"`
	Host       string `yaml:"host"`
	Port       string `yaml:"port"`
	Database   string `yaml:"database"`
	UserName   string `yaml:"username"`
	Password   string `yaml:"password"`
	Charset    string `yaml:"charset"`
}

type Node struct {
	ServerAddr string `yaml:"serverAddr"`
	ServerName string `yaml:"serverName"`
	ClientName string `yaml:"clientName"`
}