diff --git a/config/config.go b/config/config.go index b87d819..584f794 100644 --- a/config/config.go +++ b/config/config.go @@ -34,7 +34,7 @@ type Config struct { } type Blacklist struct { - Blacklist map[string][]string `yaml:"blacklist"` + Blist map[string][]string `yaml:"blacklist"` } // LoadConfig 从 YAML 配置文件加载配置 @@ -48,11 +48,11 @@ func LoadConfig(filePath string) (*Config, error) { // LoadBlacklistConfig 从 YAML 配置文件加载黑名单配置 func LoadBlacklistConfig(filePath string) (*Blacklist, error) { - var config Blacklist - if err := loadYAML(filePath, &config); err != nil { + var blacklist Blacklist + if err := loadYAML(filePath, &blacklist); err != nil { return nil, err } - return &config, nil + return &blacklist, nil } // LoadyamlConfig 从 YAML 配置文件加载配置 diff --git a/proxy/proxy.go b/proxy/proxy.go index 8807146..badc504 100644 --- a/proxy/proxy.go +++ b/proxy/proxy.go @@ -48,14 +48,14 @@ func NoRouteHandler(cfg *config.Config, blacklist *config.Blacklist) gin.Handler repo := pathParts[2] logw("Blacklist Check > Username: %s, Repo: %s", username, repo) - if blacklist.Blacklist == nil { + if blacklist.Blist == nil { logw("Warning: Blacklist map is nil") // 根据需要初始化或处理 - blacklist.Blacklist = make(map[string][]string) + blacklist.Blist = make(map[string][]string) } // 检查仓库是否在黑名单中 - if auth.IsBlacklisted(username, repo, blacklist.Blacklist, cfg.Blacklist.Enabled) { + if auth.IsBlacklisted(username, repo, blacklist.Blist, cfg.Blacklist.Enabled) { c.String(http.StatusForbidden, "Access denied: repository is blacklisted.") logw("Blacklisted repository: %s/%s", username, repo) return