package config

import (
	"os"
	"sync"

	"gopkg.in/yaml.v2"
)

// Config 结构体定义了应用程序的所有配置项
type Config struct {
	DB struct {
		Host     string `yaml:"host"`
		Port     string `yaml:"port"`
		User     string `yaml:"user"`
		Password string `yaml:"password"`
		Name     string `yaml:"name"`
	} `yaml:"db"`

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

	Server struct {
		Port string `yaml:"port"`
		Mode string `yaml:"mode"`
	} `yaml:"server"`

	Jwt struct {
		Secret string `yaml:"secret"`
		Expire string `yaml:"expire"`
	} `yaml:"jwt"`

	Mmdb struct {
		Path string `yaml:"path"`
	} `yaml:"mmdb"`

	Log struct {
		Path string `yaml:"path"`
	} `yaml:"log"`
}

var (
	configInstance *Config
	configOnce     sync.Once
)

// LoadConfig 加载并返回应用程序配置（单例模式，只加载一次）
func LoadConfig() *Config {
	configOnce.Do(func() {
		if cfg := loadConfigFromFile(); cfg != nil {
			configInstance = cfg
			return
		}

		cfg := &Config{}

		cfg.DB.Host = getEnv("DB_HOST", "localhost")
		cfg.DB.Port = getEnv("DB_PORT", "3306")
		cfg.DB.User = getEnv("DB_USER", "root")
		cfg.DB.Password = getEnv("DB_PASSWORD", "123456")
		cfg.DB.Name = getEnv("DB_NAME", "quincy")

		cfg.Redis.Host = getEnv("REDIS_HOST", "localhost")
		cfg.Redis.Port = getEnv("REDIS_PORT", "6379")
		cfg.Redis.Password = getEnv("REDIS_PASSWORD", "")
		cfg.Redis.DB = getEnv("REDIS_DB", "0")

		cfg.Server.Port = getEnv("SERVER_PORT", "8080")
		cfg.Server.Mode = getEnv("SERVER_MODE", "debug")

		cfg.Jwt.Secret = getEnv("JWT_SECRET", "your-secret-key")
		cfg.Jwt.Expire = getEnv("JWT_EXPIRE", "24")

		cfg.Mmdb.Path = getEnv("MMDB_PATH", "./GeoLite2-City.mmdb")

		cfg.Log.Path = getEnv("LOG_DIR", "./")

		configInstance = cfg
	})

	return configInstance
}

func loadConfigFromFile() *Config {
	if data, err := os.ReadFile("/home/app/quincy/default.yaml"); err == nil {
		cfg := &Config{}
		if err := yaml.Unmarshal(data, cfg); err == nil {
			return cfg
		}
	}
	return nil
}

func getEnv(key, defaultValue string) string {
	value := os.Getenv(key)
	if value == "" {
		return defaultValue
	}
	return value
}