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, XiaodianPageSize: 200, } } // 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() } normalizeXiaodianConfig(&cfg) logger.LogEvent("配置", "success", "配置加载完成: %s", path) return cfg } // normalizeXiaodianConfig 约束和平小店分页参数,接口单页最大为 200。 func normalizeXiaodianConfig(cfg *types.Config) { if cfg.XiaodianPageSize <= 0 || cfg.XiaodianPageSize > 200 { cfg.XiaodianPageSize = 200 } } // 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) }