package config

import (
	"fmt"
	"os"
	"time"

	"github.com/goccy/go-yaml"
)

type Config struct {
	Server struct {
		Address string `yaml:"address"`
		Mode    string `yaml:"mode"` // dev, prod
	} `yaml:"server"`
	Database struct {
		DSN string `yaml:"dsn"`
	} `yaml:"database"`
	Storage struct {
		Type     string `yaml:"type"`     // local, oss, s3
		BasePath string `yaml:"basePath"` // 本地存储的基础路径
		BaseURL  string `yaml:"baseUrl"`  // 文件访问的 URL 前缀
	} `yaml:"storage"`
	JWT struct {
		Secret               string `yaml:"secret"`
		AccessTokenDuration  int    `yaml:"accessTokenDuration"`  // 秒
		RefreshTokenDuration int    `yaml:"refreshTokenDuration"` // 秒
	} `yaml:"jwt"`
	Frontend struct {
		Enabled bool   `yaml:"enabled"` // 是否启用前端静态资源服务
		Path    string `yaml:"path"`    // 前端静态资源路径
	} `yaml:"frontend"`
	Admin struct {
		Username string `yaml:"username"` // 管理员用户名
		Password string `yaml:"password"` // 管理员密码
	} `yaml:"admin"`
}

func Load() *Config {
	// 确定配置文件路径
	configFile := getConfigFile()

	// 读取配置文件
	data, err := os.ReadFile(configFile)
	if err != nil {
		fmt.Printf("Warning: failed to read config file %s: %v, using defaults\n", configFile, err)
		return loadDefaults()
	}

	// 解析 YAML
	config := &Config{}
	if err := yaml.Unmarshal(data, config); err != nil {
		fmt.Printf("Warning: failed to parse config file %s: %v, using defaults\n", configFile, err)
		return loadDefaults()
	}

	// 环境变量覆盖
	applyEnvOverrides(config)

	// 自动设置 frontend.enabled（生产模式才启用）
	if config.Server.Mode == "prod" && !config.Frontend.Enabled {
		config.Frontend.Enabled = true
	}

	return config
}

func getConfigFile() string {
	// 1. 优先使用环境变量指定的配置文件
	if configFile := os.Getenv("CONFIG_FILE"); configFile != "" {
		return configFile
	}

	// 2. 根据 SERVER_MODE 选择配置文件
	mode := os.Getenv("SERVER_MODE")
	if mode == "" {
		mode = "dev"
	}

	// 检查 config.{mode}.yaml
	modeConfig := fmt.Sprintf("config.%s.yaml", mode)
	if _, err := os.Stat(modeConfig); err == nil {
		return modeConfig
	}

	// 3. 使用默认 config.yaml
	if _, err := os.Stat("config.yaml"); err == nil {
		return "config.yaml"
	}

	// 4. 都不存在，使用默认配置
	return ""
}

func loadDefaults() *Config {
	config := &Config{}
	config.Server.Address = ":9050"
	config.Server.Mode = "dev"
	config.Database.DSN = "nebula.db"
	config.Storage.Type = "local"
	config.Storage.BasePath = "./uploads"
	config.Storage.BaseURL = "http://localhost:9050/files"
	config.JWT.Secret = "dev-secret-key-change-in-production"
	config.JWT.AccessTokenDuration = 7200    // 2 hours
	config.JWT.RefreshTokenDuration = 604800 // 7 days
	config.Frontend.Enabled = false
	config.Frontend.Path = "./web/dist"
	config.Admin.Username = "admin"
	config.Admin.Password = "admin123" // 默认密码，生产环境必须修改
	return config
}

func applyEnvOverrides(config *Config) {
	if val := os.Getenv("SERVER_ADDRESS"); val != "" {
		config.Server.Address = val
	}
	if val := os.Getenv("SERVER_MODE"); val != "" {
		config.Server.Mode = val
	}
	if val := os.Getenv("DATABASE_DSN"); val != "" {
		config.Database.DSN = val
	}
	if val := os.Getenv("STORAGE_TYPE"); val != "" {
		config.Storage.Type = val
	}
	if val := os.Getenv("STORAGE_BASE_PATH"); val != "" {
		config.Storage.BasePath = val
	}
	if val := os.Getenv("STORAGE_BASE_URL"); val != "" {
		config.Storage.BaseURL = val
	}
	if val := os.Getenv("JWT_SECRET"); val != "" {
		config.JWT.Secret = val
	}
	if val := os.Getenv("FRONTEND_PATH"); val != "" {
		config.Frontend.Path = val
	}
	if val := os.Getenv("ADMIN_USERNAME"); val != "" {
		config.Admin.Username = val
	}
	if val := os.Getenv("ADMIN_PASSWORD"); val != "" {
		config.Admin.Password = val
	}
}

// GetAccessTokenDuration 返回 Access Token 有效期
func (c *Config) GetAccessTokenDuration() time.Duration {
	return time.Duration(c.JWT.AccessTokenDuration) * time.Second
}

// GetRefreshTokenDuration 返回 Refresh Token 有效期
func (c *Config) GetRefreshTokenDuration() time.Duration {
	return time.Duration(c.JWT.RefreshTokenDuration) * time.Second
}