首次提交：初始化项目代码

2026-07-31 14:44:48 +08:00
commit a4fe393571
119 changed files with 29105 additions and 0 deletions
@@ -0,0 +1,220 @@
+// Package config 配置加载模块，基于 Viper 读取 YAML 配置并提供默认值
+package config
+
+import (
+	"errors"
+	"strings"
+
+	"github.com/spf13/viper"
+)
+
+const (
+	DefaultAdminPath        = "/admin"   // 默认后台管理路径
+	DefaultMaxLoginAttempts = 5          // 默认最大登录尝试次数
+	DefaultLockoutDuration  = 15         // 默认锁定时长（分钟）
+	DefaultServerHost       = "0.0.0.0"  // 默认服务监听地址
+	DefaultServerPort       = 8090       // 默认服务端口
+	DefaultServerMode       = "release"  // 默认运行模式
+	DefaultDatabaseType     = "sqlite"   // 默认数据库类型
+	DefaultDatabasePath     = "./data/resume.db" // 默认数据库文件路径
+	DefaultLogLevel         = "info"     // 默认日志级别
+	DefaultTemplateDefault  = "modern"   // 默认简历模板
+)
+
+// Config 全局配置结构体
+type Config struct {
+	Server   ServerConfig   `mapstructure:"server"`   // 服务配置
+	Database DatabaseConfig `mapstructure:"database"` // 数据库配置
+	Auth     AuthConfig     `mapstructure:"auth"`     // 认证配置
+	Template TemplateConfig `mapstructure:"template"` // 模板配置
+	Log      LogConfig      `mapstructure:"log"`      // 日志配置
+	AI       AIConfig       `mapstructure:"ai"`       // AI配置
+}
+
+// ServerConfig 服务配置
+type ServerConfig struct {
+	Host  string `mapstructure:"host"`  // 监听地址
+	Port  int    `mapstructure:"port"`  // 监听端口
+	Mode  string `mapstructure:"mode"`  // 运行模式（debug/release）
+	Debug bool   `mapstructure:"debug"` // 是否开启调试
+}
+
+// DatabaseConfig 数据库配置
+type DatabaseConfig struct {
+	Type     string `mapstructure:"type"`     // 数据库类型
+	Path     string `mapstructure:"path"`     // 数据库文件路径
+	Password string `mapstructure:"password"` // 数据库加密密码
+}
+
+// AuthConfig 认证配置
+type AuthConfig struct {
+	Enabled          bool   `mapstructure:"enabled"`            // 是否启用认证
+	Username         string `mapstructure:"username"`           // 管理员用户名
+	Password         string `mapstructure:"password"`           // 管理员密码
+	AdminPath        string `mapstructure:"admin_path"`         // 后台路径
+	MaxLoginAttempts int    `mapstructure:"max_login_attempts"` // 最大登录尝试
+	LockoutDuration  int    `mapstructure:"lockout_duration"`   // 锁定时长（分钟）
+}
+
+// TemplateConfig 模板配置
+type TemplateConfig struct {
+	Default string `mapstructure:"default"` // 默认模板名称
+}
+
+// LogConfig 日志配置
+type LogConfig struct {
+	Level  string `mapstructure:"level"`  // 日志级别
+	Output string `mapstructure:"output"` // 输出文件路径
+}
+
+// AIConfig AI 大模型配置
+type AIConfig struct {
+	Provider string         `mapstructure:"provider"`  // 默认供应商
+	Enabled  bool           `mapstructure:"enabled"`   // 是否启用
+	Spark    SparkConfig    `mapstructure:"spark"`     // 讯飞星火配置
+	Tongyi   TongyiConfig   `mapstructure:"tongyi"`    // 通义千问配置
+	Doubao   DoubaoConfig   `mapstructure:"doubao"`    // 豆包配置
+	DeepSeek DeepSeekConfig `mapstructure:"deepseek"`  // DeepSeek配置
+}
+
+// SparkConfig 讯飞星火配置
+type SparkConfig struct {
+	APPID        string `mapstructure:"app_id"`        // APP ID
+	APIKey       string `mapstructure:"api_key"`       // API Key
+	APISecret    string `mapstructure:"api_secret"`    // API Secret
+	APIPassword  string `mapstructure:"api_password"`  // API 密码（HTTP接口使用）
+	BaseURL      string `mapstructure:"base_url"`      // 接口地址
+	Model        string `mapstructure:"model"`         // 模型版本
+	MaxTokens    int    `mapstructure:"max_tokens"`    // 最大token数
+	Timeout      int    `mapstructure:"timeout"`       // 超时时间（秒）
+	WebSearch    bool   `mapstructure:"web_search"`    // 是否启用联网搜索
+	ShowRefLabel bool   `mapstructure:"show_ref_label"` // 是否显示引用标签
+}
+
+// TongyiConfig 通义千问配置
+type TongyiConfig struct {
+	APIKey    string `mapstructure:"api_key"`    // API Key
+	BaseURL   string `mapstructure:"base_url"`   // 接口地址
+	Model     string `mapstructure:"model"`      // 模型名称
+	MaxTokens int    `mapstructure:"max_tokens"` // 最大token数
+	Timeout   int    `mapstructure:"timeout"`    // 超时时间（秒）
+}
+
+// DoubaoConfig 豆包配置
+type DoubaoConfig struct {
+	APIKey    string `mapstructure:"api_key"`    // API Key
+	BaseURL   string `mapstructure:"base_url"`   // 接口地址
+	Model     string `mapstructure:"model"`      // 模型名称
+	MaxTokens int    `mapstructure:"max_tokens"` // 最大token数
+	Timeout   int    `mapstructure:"timeout"`    // 超时时间（秒）
+}
+
+// DeepSeekConfig DeepSeek 配置
+type DeepSeekConfig struct {
+	APIKey    string `mapstructure:"api_key"`    // API Key
+	BaseURL   string `mapstructure:"base_url"`   // 接口地址
+	Model     string `mapstructure:"model"`      // 模型名称
+	MaxTokens int    `mapstructure:"max_tokens"` // 最大token数
+	Timeout   int    `mapstructure:"timeout"`    // 超时时间（秒）
+}
+
+// setDefaults 设置 Viper 默认值
+// @author sunct
+func setDefaults() {
+	viper.SetDefault("server.host", DefaultServerHost)
+	viper.SetDefault("server.port", DefaultServerPort)
+	viper.SetDefault("server.mode", DefaultServerMode)
+	viper.SetDefault("server.debug", false)
+
+	viper.SetDefault("database.type", DefaultDatabaseType)
+	viper.SetDefault("database.path", DefaultDatabasePath)
+
+	viper.SetDefault("auth.enabled", true)
+	viper.SetDefault("auth.admin_path", DefaultAdminPath)
+	viper.SetDefault("auth.max_login_attempts", DefaultMaxLoginAttempts)
+	viper.SetDefault("auth.lockout_duration", DefaultLockoutDuration)
+
+	viper.SetDefault("template.default", DefaultTemplateDefault)
+
+	viper.SetDefault("log.level", DefaultLogLevel)
+}
+
+// LoadConfig 加载并解析配置文件，支持多路径查找
+// @return *Config 配置实例
+// @return error 加载错误
+// @author sunct
+func LoadConfig() (*Config, error) {
+	setDefaults()
+
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+
+	viper.AddConfigPath("./config")
+	viper.AddConfigPath(".")
+	viper.AddConfigPath("../config")
+	viper.AddConfigPath("../")
+
+	if err := viper.ReadInConfig(); err != nil {
+		var configFileNotFoundError viper.ConfigFileNotFoundError
+		if !errors.As(err, &configFileNotFoundError) {
+			return nil, err
+		}
+	}
+
+	var cfg Config
+	if err := viper.Unmarshal(&cfg); err != nil {
+		return nil, err
+	}
+
+	if err := validateAndNormalize(&cfg); err != nil {
+		return nil, err
+	}
+
+	return &cfg, nil
+}
+
+// validateAndNormalize 校验并规范化配置，补全默认值
+// @param cfg 配置实例
+// @return error 校验错误
+// @author sunct
+func validateAndNormalize(cfg *Config) error {
+	if cfg.Server.Port <= 0 || cfg.Server.Port > 65535 {
+		cfg.Server.Port = DefaultServerPort
+	}
+	if cfg.Server.Host == "" {
+		cfg.Server.Host = DefaultServerHost
+	}
+	if cfg.Server.Mode == "" {
+		cfg.Server.Mode = DefaultServerMode
+	}
+
+	if cfg.Database.Type == "" {
+		cfg.Database.Type = DefaultDatabaseType
+	}
+	if cfg.Database.Path == "" {
+		cfg.Database.Path = DefaultDatabasePath
+	}
+
+	if !strings.HasPrefix(cfg.Auth.AdminPath, "/") {
+		cfg.Auth.AdminPath = "/" + cfg.Auth.AdminPath
+	}
+	if cfg.Auth.AdminPath == "/" {
+		cfg.Auth.AdminPath = DefaultAdminPath
+	}
+	if cfg.Auth.MaxLoginAttempts <= 0 {
+		cfg.Auth.MaxLoginAttempts = DefaultMaxLoginAttempts
+	}
+	if cfg.Auth.LockoutDuration <= 0 {
+		cfg.Auth.LockoutDuration = DefaultLockoutDuration
+	}
+
+	if cfg.Template.Default == "" {
+		cfg.Template.Default = DefaultTemplateDefault
+	}
+
+	if cfg.Log.Level == "" {
+		cfg.Log.Level = DefaultLogLevel
+	}
+
+	return nil
+}