dygou_go/internal/config/config.go

package config

import (
	"encoding/json"
	"errors"
	"os"
	"path/filepath"
	"strconv"

	"douyu_go/internal/activity"
	"douyu_go/internal/constants"
	imgpkg "douyu_go/internal/image"
	"douyu_go/internal/logger"
	"douyu_go/internal/types"
)

// DefaultAppPaths 收集运行时路径，支持环境变量覆盖。
// 所有相对路径会基于当前工作目录解析为绝对路径，避免打包后工作目录不同导致路径错位。
func DefaultAppPaths() types.AppPaths {
	return types.AppPaths{
		ConfigFile:  resolvePath(envOrDefault("DOUYU_CONFIG", constants.DefaultConfigFile)),
		AccountFile: resolvePath(envOrDefault("DOUYU_USERS", constants.DefaultAccountFile)),
		LogDir:      resolvePath(envOrDefault("DOUYU_LOG_DIR", constants.DefaultLogDir)),
		SuccessDir:  resolvePath(envOrDefault("DOUYU_SUCCESS_DIR", constants.DefaultSuccessDir)),
		FontFiles:   imgpkg.CollectFontCandidates(),
	}
}

// resolvePath 将相对路径解析为绝对路径。
func resolvePath(path string) string {
	if filepath.IsAbs(path) {
		return path
	}
	if abs, err := filepath.Abs(path); err == nil {
		return abs
	}
	return path
}

func envOrDefault(key, fallback string) string {
	if value := os.Getenv(key); value != "" {
		return filepath.Clean(value)
	}
	return fallback
}

// LoadTraceConfig 读取 trace 开关；默认关闭，避免保存敏感调试数据。
func LoadTraceConfig() types.TraceConfig {
	return types.TraceConfig{
		Enabled:      isTruthyEnv("DOUYU_TRACE"),
		MaxBodyBytes: envInt("DOUYU_TRACE_MAX_BODY", 0),
	}
}

func isTruthyEnv(key string) bool {
	switch os.Getenv(key) {
	case "1", "true", "TRUE", "yes", "YES", "on", "ON":
		return true
	default:
		return false
	}
}

func envInt(key string, fallback int) int {
	raw := os.Getenv(key)
	if raw == "" {
		return fallback
	}
	value, err := strconv.Atoi(raw)
	if err != nil || value <= 0 {
		return fallback
	}
	return value
}

// defaultConfig 返回内置默认配置。
func defaultConfig() types.Config {
	return types.Config{
		Activity:     activity.DefaultActivityID,
		TimeAPIURL:   constants.DefaultTimeAPI,
		TraceEnabled: false,
		TraceMaxBody: 0,
		LogLevel:     "info",
		Concurrency:  constants.DefaultConcurrency,
		Attempts:     constants.DefaultAttempts,
		RetryStages:  defaultRetryStages(),
		SeckillMode:  types.SeckillModeAggressive,
	}
}

// defaultRetryStages 返回默认重试间隔阶段配置。
func defaultRetryStages() []types.RetryStage {
	return []types.RetryStage{
		{MaxElapsed: 4, Interval: 120},
		{MaxElapsed: 12, Interval: 200},
		{MaxElapsed: 30, Interval: 500},
		{MaxElapsed: 75, Interval: 1000},
	}
}

// LoadConfig 读取配置文件；缺失时安静使用内置默认配置。
func LoadConfig(path string) types.Config {
	cfg := defaultConfig()
	if path == "" {
		path = constants.DefaultConfigFile
	}

	f, err := os.Open(path)
	if err != nil {
		if errors.Is(err, os.ErrNotExist) {
			logger.LogEvent("配置", "info", "配置加载完成: 内置默认配置")
		} else {
			logger.LogEvent("配置", "warn", "读取 %s 出错，使用内置默认配置: %v", path, err)
		}
		return cfg
	}
	defer f.Close()

	if err := json.NewDecoder(f).Decode(&cfg); err != nil {
		logger.LogEvent("配置", "warn", "解析 %s 失败，使用内置默认配置: %v", path, err)
		return defaultConfig()
	}
	if cfg.Activity == "" {
		cfg.Activity = activity.DefaultActivityID
	}
	if cfg.TimeAPIURL == "" {
		cfg.TimeAPIURL = constants.DefaultTimeAPI
	}
	if cfg.Concurrency <= 0 {
		cfg.Concurrency = constants.DefaultConcurrency
	}
	if cfg.Attempts <= 0 {
		cfg.Attempts = constants.DefaultAttempts
	}
	if cfg.LogLevel == "" {
		cfg.LogLevel = "info"
	}
	if len(cfg.RetryStages) == 0 {
		cfg.RetryStages = defaultRetryStages()
	}
	logger.LogEvent("配置", "success", "配置加载完成: %s", path)
	return cfg
}

// SaveConfig 将配置写入文件。
func SaveConfig(path string, cfg types.Config) error {
	if path == "" {
		path = constants.DefaultConfigFile
	}
	data, err := json.MarshalIndent(cfg, "", "  ")
	if err != nil {
		return err
	}
	return os.WriteFile(path, data, 0644)
}