package conf

// Config 配置文件不能有下划线或横线，否则不能解析
type Config struct {
	Http *Http `yaml:"http"`

	Grpc *Grpc `yaml:"grpc"`

	Debug *Debug `yaml:"debug"`

	Database *Database `yaml:"database"`

	Redis *Redis `yaml:"redis"`

	JWKS *JWKS `yaml:"jwks"`

	Metrics *Metrics `yaml:"metrics"`

	S3 *S3 `yaml:"s3"`

	Kafka *Kafka `yaml:"kafka"`

	OpenAI *OpenAI `yaml:"openai"`

	Milvus *Milvus `yaml:"milvus"`

	//ThirdParty *ThirdParty `yaml:"third_party" mapstructure:"third_party"`
}

type ThirdParty struct {
	OpenAIApiKey string `yaml:"openai_api_key" mapstructure:"openai_api_key"`
}

type Http struct {
	Host string `yaml:"host"`
	Port int    `yaml:"port"`
	Url  string `yaml:"url"`
}

type Debug struct {
	Enabled bool `yaml:"enabled"`
}

type Database struct {
	Host     string `yaml:"host"`
	Port     int    `yaml:"port"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Name     string `yaml:"name"`
}

type Redis struct {
	Host     string `yaml:"host"`
	Port     int    `yaml:"port"`
	Password string `yaml:"password"`
	DB       int    `yaml:"db"`
}

type JWKS struct {
	Url string `yaml:"url"`
}

type Metrics struct {
	Enabled bool   `yaml:"enabled"`
	Port    int    `yaml:"port"`
	Host    string `yaml:"host"`
}

type S3 struct {
	Endpoint         string `yaml:"endpoint" mapstructure:"endpoint"`
	ExternalEndpoint string `yaml:"external_endpoint" mapstructure:"external_endpoint"`
	AccessKey        string `yaml:"access_key" mapstructure:"access_key"`
	SecretKey        string `yaml:"secret_key" mapstructure:"secret_key"`
	Bucket           string `yaml:"bucket" mapstructure:"bucket"`
	UseSSL           bool   `yaml:"use_ssl" mapstructure:"use_ssl"`
	Region           string `yaml:"region" mapstructure:"region"`
}

type Kafka struct {
	BootstrapServers []string `yaml:"bootstrap_servers" mapstructure:"bootstrap_servers"`
	Topic            string   `yaml:"topic" mapstructure:"topic"`
	GroupId          string   `yaml:"group_id" mapstructure:"group_id"`
	Username         string   `yaml:"username" mapstructure:"username"`
	Password         string   `yaml:"password" mapstructure:"password"`
}

type Grpc struct {
	Address string `yaml:"address" mapstructure:"address"`
}

type OpenAI struct {
	ApiKey            string `yaml:"api_key" mapstructure:"api_key"`
	InternalBaseUrl   string `yaml:"internal_base_url" mapstructure:"internal_base_url"`
	LongContextModel  string `yaml:"long_context_model" mapstructure:"long_context_model"`
	EmbeddingModel    string `yaml:"embedding_model" mapstructure:"embedding_model"`
	EmbeddingDim      int    `yaml:"embedding_dim" mapstructure:"embedding_dim"`
	EmbeddingMaxToken int    `yaml:"embedding_max_token"  mapstructure:"embedding_max_token"`
}

type Milvus struct {
	Host               string `yaml:"host" mapstructure:"host"`
	Port               int    `yaml:"port" mapstructure:"port"`
	DBName             string `yaml:"db_name" mapstructure:"db_name"`
	DocumentCollection string `yaml:"document_collection" mapstructure:"document_collection"`
	User               string `yaml:"user" mapstructure:"user"`
	Password           string `yaml:"password" mapstructure:"password"`
}