功能更新

This commit is contained in:
李琦
2026-08-15 17:04:47 +08:00
parent d9475ac9da
commit b71b99a60d
49 changed files with 4976 additions and 438 deletions

View File

@@ -16,11 +16,21 @@ type Config struct {
Listen string `yaml:"listen"`
// BaseURL 拼接文件访问 URL 的前缀(客户端可达的地址);留空则按请求 Host 推断。
BaseURL string `yaml:"base_url"`
// APIKey 是客户端上传凭证:请求需带 Authorization: Bearer <api_key>。
APIKey string `yaml:"api_key"`
StorageDir string `yaml:"storage_dir"`
MaxUploadMB int64 `yaml:"max_upload_mb"`
MySQL struct {
// BasePath 反代保留的 URL 前缀(如 /pms-api)。1Panel 等不剥前缀时必填;留空挂在根路径。
BasePath string `yaml:"base_path"`
// JWTSecret 签发/校验 access、refresh token 的密钥,不允许留空。
JWTSecret string `yaml:"jwt_secret"`
// AccessTTLHours access token 有效小时数,默认 2。
AccessTTLHours int `yaml:"access_ttl_hours"`
// RefreshTTLDays refresh token 有效天数,默认 30。
RefreshTTLDays int `yaml:"refresh_ttl_days"`
StorageDir string `yaml:"storage_dir"`
MaxUploadMB int64 `yaml:"max_upload_mb"`
// MaxReleaseMB 发版安装包上传上限(MB),默认 200。
MaxReleaseMB int64 `yaml:"max_release_mb"`
// TrustedProxies 可信反代 CIDR/IP,用于正确解析 ClientIP;空则不信任 X-Forwarded-For。
TrustedProxies []string `yaml:"trusted_proxies"`
MySQL struct {
DSN string `yaml:"dsn"`
} `yaml:"mysql"`
}
@@ -28,7 +38,10 @@ type Config struct {
// MaxUploadBytes 返回单文件字节上限。
func (c *Config) MaxUploadBytes() int64 { return c.MaxUploadMB << 20 }
// Load 读取并校验配置,缺省值:prod / :8788 / ./uploads / 20MB。
// MaxReleaseBytes 返回发版包字节上限。
func (c *Config) MaxReleaseBytes() int64 { return c.MaxReleaseMB << 20 }
// Load 读取并校验配置,缺省值:prod / :8788 / ./uploads / 20MB / access 2h / refresh 30d。
func Load(path string) (*Config, error) {
b, err := os.ReadFile(path)
if err != nil {
@@ -50,10 +63,29 @@ func Load(path string) (*Config, error) {
if c.MaxUploadMB <= 0 {
c.MaxUploadMB = 20
}
if c.MaxReleaseMB <= 0 {
c.MaxReleaseMB = 200
}
if c.AccessTTLHours <= 0 {
c.AccessTTLHours = 2
}
if c.RefreshTTLDays <= 0 {
c.RefreshTTLDays = 30
}
c.BaseURL = strings.TrimRight(strings.TrimSpace(c.BaseURL), "/")
c.APIKey = strings.TrimSpace(c.APIKey)
if c.APIKey == "" {
return nil, errors.New("配置缺少 api_key(客户端上传凭证,不允许留空开放上传)")
c.BasePath = strings.TrimSpace(c.BasePath)
if c.BasePath != "" {
if !strings.HasPrefix(c.BasePath, "/") {
c.BasePath = "/" + c.BasePath
}
c.BasePath = strings.TrimRight(c.BasePath, "/")
if c.BasePath == "/" {
c.BasePath = ""
}
}
c.JWTSecret = strings.TrimSpace(c.JWTSecret)
if c.JWTSecret == "" {
return nil, errors.New("配置缺少 jwt_secret(JWT 签发密钥,不允许留空)")
}
if strings.TrimSpace(c.MySQL.DSN) == "" {
return nil, errors.New("配置缺少 mysql.dsn")