backend/pkg/config/config.go

package config

import (
	"fmt"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/joho/godotenv"
	"github.com/spf13/viper"
)

// Config 应用配置结构体
type Config struct {
	Environment string         `mapstructure:"environment"`
	Site        SiteConfig     `mapstructure:"site"`
	Server      ServerConfig   `mapstructure:"server"`
	Database    DatabaseConfig `mapstructure:"database"`
	Redis       RedisConfig    `mapstructure:"redis"`
	RustFS      RustFSConfig   `mapstructure:"rustfs"`
	JWT         JWTConfig      `mapstructure:"jwt"`
	Casbin      CasbinConfig   `mapstructure:"casbin"`
	Log         LogConfig      `mapstructure:"log"`
	Email       EmailConfig    `mapstructure:"email"`
	Security    SecurityConfig `mapstructure:"security"`
}

// SiteConfig 站点配置
type SiteConfig struct {
	Name                  string `mapstructure:"name"`
	Description           string `mapstructure:"description"`
	RegistrationEnabled   bool   `mapstructure:"registration_enabled"`
	DefaultAvatar         string `mapstructure:"default_avatar"`
	MaxTexturesPerUser    int    `mapstructure:"max_textures_per_user"`
	MaxProfilesPerUser    int    `mapstructure:"max_profiles_per_user"`
	CheckinReward         int    `mapstructure:"checkin_reward"`
	TextureDownloadReward int    `mapstructure:"texture_download_reward"`
}

// ServerConfig 服务器配置
type ServerConfig struct {
	Port         string        `mapstructure:"port"`
	Mode         string        `mapstructure:"mode"`
	ReadTimeout  time.Duration `mapstructure:"read_timeout"`
	WriteTimeout time.Duration `mapstructure:"write_timeout"`
}

// DatabaseConfig 数据库配置
type DatabaseConfig struct {
	Driver          string        `mapstructure:"driver"`
	Host            string        `mapstructure:"host"`
	Port            int           `mapstructure:"port"`
	Username        string        `mapstructure:"username"`
	Password        string        `mapstructure:"password"`
	Database        string        `mapstructure:"database"`
	SSLMode         string        `mapstructure:"ssl_mode"`
	Timezone        string        `mapstructure:"timezone"`
	MaxIdleConns    int           `mapstructure:"max_idle_conns"`
	MaxOpenConns    int           `mapstructure:"max_open_conns"`
	ConnMaxLifetime time.Duration `mapstructure:"conn_max_lifetime"`
	ConnMaxIdleTime time.Duration `mapstructure:"conn_max_idle_time"` // 连接最大空闲时间
}

// RedisConfig Redis配置
type RedisConfig struct {
	Host            string        `mapstructure:"host"`
	Port            int           `mapstructure:"port"`
	Password        string        `mapstructure:"password"`
	Database        int           `mapstructure:"database"`
	PoolSize        int           `mapstructure:"pool_size"`          // 连接池大小
	MinIdleConns    int           `mapstructure:"min_idle_conns"`     // 最小空闲连接数
	MaxRetries      int           `mapstructure:"max_retries"`        // 最大重试次数
	DialTimeout     time.Duration `mapstructure:"dial_timeout"`       // 连接超时
	ReadTimeout     time.Duration `mapstructure:"read_timeout"`       // 读取超时
	WriteTimeout    time.Duration `mapstructure:"write_timeout"`      // 写入超时
	PoolTimeout     time.Duration `mapstructure:"pool_timeout"`       // 连接池超时
	ConnMaxIdleTime time.Duration `mapstructure:"conn_max_idle_time"` // 连接最大空闲时间
}

// RustFSConfig RustFS对象存储配置 (S3兼容)
type RustFSConfig struct {
	Endpoint  string            `mapstructure:"endpoint"`
	PublicURL string            `mapstructure:"public_url"` // 公开访问URL (用于生成文件访问链接)
	AccessKey string            `mapstructure:"access_key"`
	SecretKey string            `mapstructure:"secret_key"`
	UseSSL    bool              `mapstructure:"use_ssl"`
	Buckets   map[string]string `mapstructure:"buckets"`
}

// JWTConfig JWT配置
type JWTConfig struct {
	Secret      string `mapstructure:"secret"`
	ExpireHours int    `mapstructure:"expire_hours"`
}

// CasbinConfig Casbin权限配置
type CasbinConfig struct {
	ModelPath     string `mapstructure:"model_path"`
	PolicyAdapter string `mapstructure:"policy_adapter"`
}

// LogConfig 日志配置
type LogConfig struct {
	Level      string `mapstructure:"level"`
	Format     string `mapstructure:"format"`
	Output     string `mapstructure:"output"`
	MaxSize    int    `mapstructure:"max_size"`
	MaxBackups int    `mapstructure:"max_backups"`
	MaxAge     int    `mapstructure:"max_age"`
	Compress   bool   `mapstructure:"compress"`
}

// EmailConfig 邮件配置
type EmailConfig struct {
	Enabled  bool   `mapstructure:"enabled"`
	SMTPHost string `mapstructure:"smtp_host"`
	SMTPPort int    `mapstructure:"smtp_port"`
	Username string `mapstructure:"username"`
	Password string `mapstructure:"password"`
	FromName string `mapstructure:"from_name"`
}

// SecurityConfig 安全配置
type SecurityConfig struct {
	AllowedOrigins []string `mapstructure:"allowed_origins"` // 允许的CORS来源
	AllowedDomains []string `mapstructure:"allowed_domains"` // 允许的头像/材质URL域名
}

// Load 加载配置 - 完全从环境变量加载，不依赖YAML文件
func Load() (*Config, error) {
	// 加载.env文件（如果存在）
	_ = godotenv.Load(".env")

	// 设置默认值
	setDefaults()

	// 设置环境变量前缀
	viper.SetEnvPrefix("CARROTSKIN")
	viper.AutomaticEnv()

	// 手动设置环境变量映射
	setupEnvMappings()

	// 直接从环境变量解析配置
	var config Config
	if err := viper.Unmarshal(&config); err != nil {
		return nil, fmt.Errorf("解析配置失败: %w", err)
	}

	// 从环境变量中覆盖配置
	overrideFromEnv(&config)

	return &config, nil
}

// setDefaults 设置默认配置值
func setDefaults() {
	// 服务器默认配置
	viper.SetDefault("server.port", ":8080")
	viper.SetDefault("server.mode", "debug")
	viper.SetDefault("server.read_timeout", "30s")
	viper.SetDefault("server.write_timeout", "30s")

	// 数据库默认配置
	viper.SetDefault("database.driver", "postgres")
	viper.SetDefault("database.host", "localhost")
	viper.SetDefault("database.port", 5432)
	viper.SetDefault("database.ssl_mode", "disable")
	viper.SetDefault("database.timezone", "Asia/Shanghai")
	viper.SetDefault("database.max_idle_conns", 10)
	viper.SetDefault("database.max_open_conns", 100)
	viper.SetDefault("database.conn_max_lifetime", "1h")
	viper.SetDefault("database.conn_max_idle_time", "10m")

	// Redis默认配置
	viper.SetDefault("redis.host", "localhost")
	viper.SetDefault("redis.port", 6379)
	viper.SetDefault("redis.database", 0)
	viper.SetDefault("redis.pool_size", 10)
	viper.SetDefault("redis.min_idle_conns", 5)
	viper.SetDefault("redis.max_retries", 3)
	viper.SetDefault("redis.dial_timeout", "5s")
	viper.SetDefault("redis.read_timeout", "3s")
	viper.SetDefault("redis.write_timeout", "3s")
	viper.SetDefault("redis.pool_timeout", "4s")
	viper.SetDefault("redis.conn_max_idle_time", "30m")

	// RustFS默认配置
	viper.SetDefault("rustfs.endpoint", "127.0.0.1:9000")
	viper.SetDefault("rustfs.public_url", "") // 为空时使用 endpoint 构建 URL
	viper.SetDefault("rustfs.use_ssl", false)

	// JWT默认配置
	viper.SetDefault("jwt.expire_hours", 168)

	// Casbin默认配置
	viper.SetDefault("casbin.model_path", "configs/casbin/rbac_model.conf")
	viper.SetDefault("casbin.policy_adapter", "gorm")

	// 日志默认配置
	viper.SetDefault("log.level", "info")
	viper.SetDefault("log.format", "json")
	viper.SetDefault("log.output", "logs/app.log")
	viper.SetDefault("log.max_size", 100)
	viper.SetDefault("log.max_backups", 3)
	viper.SetDefault("log.max_age", 28)
	viper.SetDefault("log.compress", true)

	// 邮件默认配置
	viper.SetDefault("email.enabled", false)
	viper.SetDefault("email.smtp_port", 587)

	// 安全默认配置
	viper.SetDefault("security.allowed_origins", []string{"*"})
	viper.SetDefault("security.allowed_domains", []string{"localhost", "127.0.0.1"})

	// 站点默认配置
	viper.SetDefault("site.name", "CarrotSkin")
	viper.SetDefault("site.description", "一个优秀的Minecraft皮肤站")
	viper.SetDefault("site.registration_enabled", true)
	viper.SetDefault("site.default_avatar", "")
	viper.SetDefault("site.max_textures_per_user", 50)
	viper.SetDefault("site.max_profiles_per_user", 5)
	viper.SetDefault("site.checkin_reward", 10)
	viper.SetDefault("site.texture_download_reward", 1)
}

// setupEnvMappings 设置环境变量映射
func setupEnvMappings() {
	// 服务器配置
	viper.BindEnv("server.port", "SERVER_PORT")
	viper.BindEnv("server.mode", "SERVER_MODE")
	viper.BindEnv("server.read_timeout", "SERVER_READ_TIMEOUT")
	viper.BindEnv("server.write_timeout", "SERVER_WRITE_TIMEOUT")

	// 数据库配置
	viper.BindEnv("database.driver", "DATABASE_DRIVER")
	viper.BindEnv("database.host", "DATABASE_HOST")
	viper.BindEnv("database.port", "DATABASE_PORT")
	viper.BindEnv("database.username", "DATABASE_USERNAME")
	viper.BindEnv("database.password", "DATABASE_PASSWORD")
	viper.BindEnv("database.database", "DATABASE_NAME")
	viper.BindEnv("database.ssl_mode", "DATABASE_SSL_MODE")
	viper.BindEnv("database.timezone", "DATABASE_TIMEZONE")
	viper.BindEnv("database.max_idle_conns", "DATABASE_MAX_IDLE_CONNS")
	viper.BindEnv("database.max_open_conns", "DATABASE_MAX_OPEN_CONNS")
	viper.BindEnv("database.conn_max_lifetime", "DATABASE_CONN_MAX_LIFETIME")
	viper.BindEnv("database.conn_max_idle_time", "DATABASE_CONN_MAX_IDLE_TIME")

	// Redis配置
	viper.BindEnv("redis.host", "REDIS_HOST")
	viper.BindEnv("redis.port", "REDIS_PORT")
	viper.BindEnv("redis.password", "REDIS_PASSWORD")
	viper.BindEnv("redis.database", "REDIS_DATABASE")
	viper.BindEnv("redis.pool_size", "REDIS_POOL_SIZE")
	viper.BindEnv("redis.min_idle_conns", "REDIS_MIN_IDLE_CONNS")
	viper.BindEnv("redis.max_retries", "REDIS_MAX_RETRIES")
	viper.BindEnv("redis.dial_timeout", "REDIS_DIAL_TIMEOUT")
	viper.BindEnv("redis.read_timeout", "REDIS_READ_TIMEOUT")
	viper.BindEnv("redis.write_timeout", "REDIS_WRITE_TIMEOUT")
	viper.BindEnv("redis.pool_timeout", "REDIS_POOL_TIMEOUT")
	viper.BindEnv("redis.conn_max_idle_time", "REDIS_CONN_MAX_IDLE_TIME")

	// RustFS配置
	viper.BindEnv("rustfs.endpoint", "RUSTFS_ENDPOINT")
	viper.BindEnv("rustfs.public_url", "RUSTFS_PUBLIC_URL")
	viper.BindEnv("rustfs.access_key", "RUSTFS_ACCESS_KEY")
	viper.BindEnv("rustfs.secret_key", "RUSTFS_SECRET_KEY")
	viper.BindEnv("rustfs.use_ssl", "RUSTFS_USE_SSL")

	// JWT配置
	viper.BindEnv("jwt.secret", "JWT_SECRET")
	viper.BindEnv("jwt.expire_hours", "JWT_EXPIRE_HOURS")

	// 日志配置
	viper.BindEnv("log.level", "LOG_LEVEL")
	viper.BindEnv("log.format", "LOG_FORMAT")
	viper.BindEnv("log.output", "LOG_OUTPUT")

	// 邮件配置
	viper.BindEnv("email.enabled", "EMAIL_ENABLED")
	viper.BindEnv("email.smtp_host", "EMAIL_SMTP_HOST")
	viper.BindEnv("email.smtp_port", "EMAIL_SMTP_PORT")
	viper.BindEnv("email.username", "EMAIL_USERNAME")
	viper.BindEnv("email.password", "EMAIL_PASSWORD")
	viper.BindEnv("email.from_name", "EMAIL_FROM_NAME")

	// 站点配置
	viper.BindEnv("site.name", "SITE_NAME")
	viper.BindEnv("site.description", "SITE_DESCRIPTION")
	viper.BindEnv("site.registration_enabled", "REGISTRATION_ENABLED")
	viper.BindEnv("site.default_avatar", "DEFAULT_AVATAR")
	viper.BindEnv("site.max_textures_per_user", "MAX_TEXTURES_PER_USER")
	viper.BindEnv("site.max_profiles_per_user", "MAX_PROFILES_PER_USER")
	viper.BindEnv("site.checkin_reward", "CHECKIN_REWARD")
	viper.BindEnv("site.texture_download_reward", "TEXTURE_DOWNLOAD_REWARD")
}

// overrideFromEnv 从环境变量中覆盖配置
func overrideFromEnv(config *Config) {
	// 处理RustFS存储桶配置
	if texturesBucket := os.Getenv("RUSTFS_BUCKET_TEXTURES"); texturesBucket != "" {
		if config.RustFS.Buckets == nil {
			config.RustFS.Buckets = make(map[string]string)
		}
		config.RustFS.Buckets["textures"] = texturesBucket
	}

	if avatarsBucket := os.Getenv("RUSTFS_BUCKET_AVATARS"); avatarsBucket != "" {
		if config.RustFS.Buckets == nil {
			config.RustFS.Buckets = make(map[string]string)
		}
		config.RustFS.Buckets["avatars"] = avatarsBucket
	}

	// 处理数据库连接池配置
	if maxIdleConns := os.Getenv("DATABASE_MAX_IDLE_CONNS"); maxIdleConns != "" {
		if val, err := strconv.Atoi(maxIdleConns); err == nil {
			config.Database.MaxIdleConns = val
		}
	}

	if maxOpenConns := os.Getenv("DATABASE_MAX_OPEN_CONNS"); maxOpenConns != "" {
		if val, err := strconv.Atoi(maxOpenConns); err == nil {
			config.Database.MaxOpenConns = val
		}
	}

	if connMaxLifetime := os.Getenv("DATABASE_CONN_MAX_LIFETIME"); connMaxLifetime != "" {
		if val, err := time.ParseDuration(connMaxLifetime); err == nil {
			config.Database.ConnMaxLifetime = val
		}
	}

	if connMaxIdleTime := os.Getenv("DATABASE_CONN_MAX_IDLE_TIME"); connMaxIdleTime != "" {
		if val, err := time.ParseDuration(connMaxIdleTime); err == nil {
			config.Database.ConnMaxIdleTime = val
		}
	}

	// 处理Redis连接池配置
	if poolSize := os.Getenv("REDIS_POOL_SIZE"); poolSize != "" {
		if val, err := strconv.Atoi(poolSize); err == nil {
			config.Redis.PoolSize = val
		}
	}

	if minIdleConns := os.Getenv("REDIS_MIN_IDLE_CONNS"); minIdleConns != "" {
		if val, err := strconv.Atoi(minIdleConns); err == nil {
			config.Redis.MinIdleConns = val
		}
	}

	if maxRetries := os.Getenv("REDIS_MAX_RETRIES"); maxRetries != "" {
		if val, err := strconv.Atoi(maxRetries); err == nil {
			config.Redis.MaxRetries = val
		}
	}

	if dialTimeout := os.Getenv("REDIS_DIAL_TIMEOUT"); dialTimeout != "" {
		if val, err := time.ParseDuration(dialTimeout); err == nil {
			config.Redis.DialTimeout = val
		}
	}

	if readTimeout := os.Getenv("REDIS_READ_TIMEOUT"); readTimeout != "" {
		if val, err := time.ParseDuration(readTimeout); err == nil {
			config.Redis.ReadTimeout = val
		}
	}

	if writeTimeout := os.Getenv("REDIS_WRITE_TIMEOUT"); writeTimeout != "" {
		if val, err := time.ParseDuration(writeTimeout); err == nil {
			config.Redis.WriteTimeout = val
		}
	}

	if poolTimeout := os.Getenv("REDIS_POOL_TIMEOUT"); poolTimeout != "" {
		if val, err := time.ParseDuration(poolTimeout); err == nil {
			config.Redis.PoolTimeout = val
		}
	}

	if connMaxIdleTime := os.Getenv("REDIS_CONN_MAX_IDLE_TIME"); connMaxIdleTime != "" {
		if val, err := time.ParseDuration(connMaxIdleTime); err == nil {
			config.Redis.ConnMaxIdleTime = val
		}
	}

	// 处理邮件配置
	if emailEnabled := os.Getenv("EMAIL_ENABLED"); emailEnabled != "" {
		config.Email.Enabled = emailEnabled == "true" || emailEnabled == "True" || emailEnabled == "TRUE" || emailEnabled == "1"
	}

	// 处理环境配置
	if env := os.Getenv("ENVIRONMENT"); env != "" {
		config.Environment = env
	}

	// 处理安全配置
	if allowedOrigins := os.Getenv("SECURITY_ALLOWED_ORIGINS"); allowedOrigins != "" {
		config.Security.AllowedOrigins = strings.Split(allowedOrigins, ",")
	}

	if allowedDomains := os.Getenv("SECURITY_ALLOWED_DOMAINS"); allowedDomains != "" {
		config.Security.AllowedDomains = strings.Split(allowedDomains, ",")
	}
}

// IsTestEnvironment 判断是否为测试环境
func (c *Config) IsTestEnvironment() bool {
	return c.Environment == "test"
}