tyapi-server/internal/config/config.go

package config

import (
	"time"
)

// Config 应用程序总配置
type Config struct {
	Server      ServerConfig      `mapstructure:"server"`
	Database    DatabaseConfig    `mapstructure:"database"`
	Redis       RedisConfig       `mapstructure:"redis"`
	Cache       CacheConfig       `mapstructure:"cache"`
	Logger      LoggerConfig      `mapstructure:"logger"`
	JWT         JWTConfig         `mapstructure:"jwt"`
	API         APIConfig         `mapstructure:"api"`
	SMS         SMSConfig         `mapstructure:"sms"`
	Email       EmailConfig       `mapstructure:"email"`
	Storage     StorageConfig     `mapstructure:"storage"`
	OCR         OCRConfig         `mapstructure:"ocr"`
	RateLimit   RateLimitConfig   `mapstructure:"ratelimit"`
	Monitoring  MonitoringConfig  `mapstructure:"monitoring"`
	Health      HealthConfig      `mapstructure:"health"`
	Resilience  ResilienceConfig  `mapstructure:"resilience"`
	Development DevelopmentConfig `mapstructure:"development"`
	App         AppConfig         `mapstructure:"app"`
	WechatWork  WechatWorkConfig  `mapstructure:"wechat_work"`
	Esign       EsignConfig       `mapstructure:"esign"`
	Wallet      WalletConfig      `mapstructure:"wallet"`
	WestDex     WestDexConfig     `mapstructure:"westdex"`
	AliPay      AliPayConfig      `mapstructure:"alipay"`
	Yushan      YushanConfig      `mapstructure:"yushan"`
	TianYanCha  TianYanChaConfig  `mapstructure:"tianyancha"`
}

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

// DatabaseConfig 数据库配置
type DatabaseConfig struct {
	Host            string        `mapstructure:"host"`
	Port            string        `mapstructure:"port"`
	User            string        `mapstructure:"user"`
	Password        string        `mapstructure:"password"`
	Name            string        `mapstructure:"name"`
	SSLMode         string        `mapstructure:"sslmode"`
	Timezone        string        `mapstructure:"timezone"`
	MaxOpenConns    int           `mapstructure:"max_open_conns"`
	MaxIdleConns    int           `mapstructure:"max_idle_conns"`
	ConnMaxLifetime time.Duration `mapstructure:"conn_max_lifetime"`
	AutoMigrate     bool          `mapstructure:"auto_migrate"`
}

// RedisConfig Redis配置
type RedisConfig struct {
	Host         string        `mapstructure:"host"`
	Port         string        `mapstructure:"port"`
	Password     string        `mapstructure:"password"`
	DB           int           `mapstructure:"db"`
	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"`
}

// CacheConfig 缓存配置
type CacheConfig struct {
	DefaultTTL      time.Duration `mapstructure:"default_ttl"`
	CleanupInterval time.Duration `mapstructure:"cleanup_interval"`
	MaxSize         int           `mapstructure:"max_size"`
}

// LoggerConfig 日志配置
type LoggerConfig struct {
	Level      string `mapstructure:"level"`
	Format     string `mapstructure:"format"`
	Output     string `mapstructure:"output"`
	LogDir     string `mapstructure:"log_dir"`     // 日志目录
	MaxSize    int    `mapstructure:"max_size"`    // 单个文件最大大小(MB)
	MaxBackups int    `mapstructure:"max_backups"` // 最大备份文件数
	MaxAge     int    `mapstructure:"max_age"`     // 最大保留天数
	Compress   bool   `mapstructure:"compress"`    // 是否压缩
	UseColor   bool   `mapstructure:"use_color"`   // 是否使用彩色输出
	UseDaily   bool   `mapstructure:"use_daily"`   // 是否按日分包
	// 按级别分文件配置
	EnableLevelSeparation bool                       `mapstructure:"enable_level_separation"` // 是否启用按级别分文件
	LevelConfigs          map[string]LevelFileConfig `mapstructure:"level_configs"`           // 各级别配置
}

// LevelFileConfig 单个级别文件配置
type LevelFileConfig struct {
	MaxSize    int  `mapstructure:"max_size"`    // 单个文件最大大小(MB)
	MaxBackups int  `mapstructure:"max_backups"` // 最大备份文件数
	MaxAge     int  `mapstructure:"max_age"`     // 最大保留天数
	Compress   bool `mapstructure:"compress"`    // 是否压缩
}

// JWTConfig JWT配置
type JWTConfig struct {
	Secret           string        `mapstructure:"secret"`
	ExpiresIn        time.Duration `mapstructure:"expires_in"`
	RefreshExpiresIn time.Duration `mapstructure:"refresh_expires_in"`
}

// RateLimitConfig 限流配置
type RateLimitConfig struct {
	Requests int           `mapstructure:"requests"`
	Window   time.Duration `mapstructure:"window"`
	Burst    int           `mapstructure:"burst"`
}

// MonitoringConfig 监控配置
type MonitoringConfig struct {
	MetricsEnabled  bool    `mapstructure:"metrics_enabled"`
	MetricsPort     string  `mapstructure:"metrics_port"`
	TracingEnabled  bool    `mapstructure:"tracing_enabled"`
	TracingEndpoint string  `mapstructure:"tracing_endpoint"`
	SampleRate      float64 `mapstructure:"sample_rate"`
}

// HealthConfig 健康检查配置
type HealthConfig struct {
	Enabled  bool          `mapstructure:"enabled"`
	Interval time.Duration `mapstructure:"interval"`
	Timeout  time.Duration `mapstructure:"timeout"`
}

// ResilienceConfig 容错配置
type ResilienceConfig struct {
	CircuitBreakerEnabled   bool          `mapstructure:"circuit_breaker_enabled"`
	CircuitBreakerThreshold int           `mapstructure:"circuit_breaker_threshold"`
	CircuitBreakerTimeout   time.Duration `mapstructure:"circuit_breaker_timeout"`
	RetryMaxAttempts        int           `mapstructure:"retry_max_attempts"`
	RetryInitialDelay       time.Duration `mapstructure:"retry_initial_delay"`
	RetryMaxDelay           time.Duration `mapstructure:"retry_max_delay"`
}

// DevelopmentConfig 开发配置
type DevelopmentConfig struct {
	Debug          bool   `mapstructure:"debug"`
	EnableProfiler bool   `mapstructure:"enable_profiler"`
	EnableCors     bool   `mapstructure:"enable_cors"`
	CorsOrigins    string `mapstructure:"cors_allowed_origins"`
	CorsMethods    string `mapstructure:"cors_allowed_methods"`
	CorsHeaders    string `mapstructure:"cors_allowed_headers"`
}

// AppConfig 应用程序配置
type AppConfig struct {
	Name    string `mapstructure:"name"`
	Version string `mapstructure:"version"`
	Env     string `mapstructure:"env"`
}

// APIConfig API配置
type APIConfig struct {
	Domain string `mapstructure:"domain"`
}

// SMSConfig 短信配置
type SMSConfig struct {
	AccessKeyID     string        `mapstructure:"access_key_id"`
	AccessKeySecret string        `mapstructure:"access_key_secret"`
	EndpointURL     string        `mapstructure:"endpoint_url"`
	SignName        string        `mapstructure:"sign_name"`
	TemplateCode    string        `mapstructure:"template_code"`
	CodeLength      int           `mapstructure:"code_length"`
	ExpireTime      time.Duration `mapstructure:"expire_time"`
	RateLimit       SMSRateLimit  `mapstructure:"rate_limit"`
	MockEnabled     bool          `mapstructure:"mock_enabled"` // 是否启用模拟短信服务
}

// SMSRateLimit 短信限流配置
type SMSRateLimit struct {
	DailyLimit  int           `mapstructure:"daily_limit"`  // 每日发送限制
	HourlyLimit int           `mapstructure:"hourly_limit"` // 每小时发送限制
	MinInterval time.Duration `mapstructure:"min_interval"` // 最小发送间隔
}

// EmailConfig 邮件服务配置
type EmailConfig struct {
	Host     string        `mapstructure:"host"`      // SMTP服务器地址
	Port     int           `mapstructure:"port"`      // SMTP服务器端口
	Username string        `mapstructure:"username"`  // 邮箱用户名
	Password string        `mapstructure:"password"`  // 邮箱密码/授权码
	FromEmail string       `mapstructure:"from_email"` // 发件人邮箱
	UseSSL   bool          `mapstructure:"use_ssl"`   // 是否使用SSL
	Timeout  time.Duration `mapstructure:"timeout"`   // 超时时间
	Domain   string        `mapstructure:"domain"`    // 控制台域名
}

// GetDSN 获取数据库DSN连接字符串
func (d DatabaseConfig) GetDSN() string {
	return "host=" + d.Host +
		" user=" + d.User +
		" password=" + d.Password +
		" dbname=" + d.Name +
		" port=" + d.Port +
		" sslmode=" + d.SSLMode +
		" TimeZone=" + d.Timezone
}

// GetRedisAddr 获取Redis地址
func (r RedisConfig) GetRedisAddr() string {
	return r.Host + ":" + r.Port
}

// IsProduction 检查是否为生产环境
func (a AppConfig) IsProduction() bool {
	return a.Env == "production"
}

// IsDevelopment 检查是否为开发环境
func (a AppConfig) IsDevelopment() bool {
	return a.Env == "development"
}

// IsStaging 检查是否为测试环境
func (a AppConfig) IsStaging() bool {
	return a.Env == "staging"
}

// WechatWorkConfig 企业微信配置
type WechatWorkConfig struct {
	WebhookURL string `mapstructure:"webhook_url"`
	Secret     string `mapstructure:"secret"`
}

// StorageConfig 存储服务配置
type StorageConfig struct {
	AccessKey string `mapstructure:"access_key"`
	SecretKey string `mapstructure:"secret_key"`
	Bucket    string `mapstructure:"bucket"`
	Domain    string `mapstructure:"domain"`
}

// OCRConfig OCR服务配置
type OCRConfig struct {
	APIKey    string `mapstructure:"api_key"`
	SecretKey string `mapstructure:"secret_key"`
}

// EsignConfig e签宝配置
type EsignConfig struct {
	AppID      string `mapstructure:"app_id"`      // 应用ID
	AppSecret  string `mapstructure:"app_secret"`  // 应用密钥
	ServerURL  string `mapstructure:"server_url"`  // 服务器URL
	TemplateID string `mapstructure:"template_id"` // 模板ID

	Contract ContractConfig `mapstructure:"contract"` // 合同配置
	Auth     AuthConfig     `mapstructure:"auth"`     // 认证配置
	Sign     SignConfig     `mapstructure:"sign"`     // 签署配置
}

// ContractConfig 合同配置
type ContractConfig struct {
	Name       string `mapstructure:"name"`        // 合同名称
	ExpireDays int    `mapstructure:"expire_days"` // 签署链接过期天数
	RetryCount int    `mapstructure:"retry_count"` // 重试次数
}

// AuthConfig 认证配置
type AuthConfig struct {
	OrgAuthModes         []string `mapstructure:"org_auth_modes"`         // 机构可用认证模式
	DefaultAuthMode      string   `mapstructure:"default_auth_mode"`      // 默认认证模式
	PsnAuthModes         []string `mapstructure:"psn_auth_modes"`         // 个人可用认证模式
	WillingnessAuthModes []string `mapstructure:"willingness_auth_modes"` // 意愿认证模式
	RedirectURL          string   `mapstructure:"redirect_url"`           // 重定向URL
}

// SignConfig 签署配置
type SignConfig struct {
	AutoFinish     bool   `mapstructure:"auto_finish"`      // 是否自动完结
	SignFieldStyle int    `mapstructure:"sign_field_style"` // 签署区样式
	ClientType     string `mapstructure:"client_type"`      // 客户端类型
	RedirectURL    string `mapstructure:"redirect_url"`     // 重定向URL
}

// WalletConfig 钱包配置
type WalletConfig struct {
	DefaultCreditLimit  float64                   `mapstructure:"default_credit_limit"`
	MinAmount           string                    `mapstructure:"min_amount"` // 最低充值金额
	MaxAmount           string                    `mapstructure:"max_amount"` // 最高充值金额
	AliPayRechargeBonus []AliPayRechargeBonusRule `mapstructure:"alipay_recharge_bonus"`
}

// AliPayRechargeBonusRule 支付宝充值赠送规则
type AliPayRechargeBonusRule struct {
	RechargeAmount float64 `mapstructure:"recharge_amount"` // 充值金额
	BonusAmount    float64 `mapstructure:"bonus_amount"`    // 赠送金额
}

// WestDexConfig WestDex配置
type WestDexConfig struct {
	URL            string `mapstructure:"url"`
	Key            string `mapstructure:"key"`
	SecretId       string `mapstructure:"secret_id"`
	SecretSecondId string `mapstructure:"secret_second_id"`
}

// AliPayConfig 支付宝配置
type AliPayConfig struct {
	AppID           string `mapstructure:"app_id"`
	PrivateKey      string `mapstructure:"private_key"`
	AlipayPublicKey string `mapstructure:"alipay_public_key"`
	IsProduction    bool   `mapstructure:"is_production"`
	NotifyURL       string `mapstructure:"notify_url"`
	ReturnURL       string `mapstructure:"return_url"`
}

// YushanConfig 羽山配置
type YushanConfig struct {
	URL    string `mapstructure:"url"`
	APIKey string `mapstructure:"api_key"`
	AcctID string `mapstructure:"acct_id"`
}

// TianYanChaConfig 天眼查配置
type TianYanChaConfig struct {
	BaseURL string `mapstructure:"base_url"`
	APIKey  string `mapstructure:"api_key"`
}

// DomainConfig 域名配置
type DomainConfig struct {
	API string `mapstructure:"api"` // API域名
}