24w08b

config/blacklist.yaml

@@ -0,0 +1,9 @@
+blacklist:
+  username1:
+    - repo1
+    - repo2
+  username2:
+    - repo3
+    - repo4
+  username3:
+    - repo5

config/config.go

@@ -7,26 +7,59 @@ import (
 )
 
 type Config struct {
-	Port        int    `yaml:"port"`
-	Host        string `yaml:"host"`
-	SizeLimit   int    `yaml:"sizelimit"`
-	LogFilePath string `yaml:"logfilepath"`
-	MaxLogSize  int    `yaml:"maxlogsize"`
-	CORSOrigin  bool   `yaml:"CorsAllowOrigins"`
-	Auth        bool   `yaml:"auth"`
-	AuthToken   string `yaml:"authtoken"`
+	Server struct {
+		Port      int    `yaml:"port"`
+		Host      string `yaml:"host"`
+		SizeLimit int    `yaml:"sizelimit"`
+	} `yaml:"server"`
+
+	Log struct {
+		LogFilePath string `yaml:"logfilepath"`
+		MaxLogSize  int    `yaml:"maxlogsize"`
+	} `yaml:"logger"`
+
+	CORS struct {
+		Enabled bool `yaml:"enabled"`
+	} `yaml:"cors"`
+
+	Auth struct {
+		Enabled   bool   `yaml:"enabled"`
+		AuthToken string `yaml:"authtoken"`
+	} `yaml:"auth"`
+
+	Blacklist struct {
+		Enabled       bool   `yaml:"enabled"`
+		BlacklistFile string `yaml:"blacklistfile"`
+	} `yaml:"blacklist"`
+}
+
+type Blacklist struct {
+	Blacklist map[string][]string `yaml:"blacklist"`
 }
 
 // LoadConfig 从 YAML 配置文件加载配置
 func LoadConfig(filePath string) (*Config, error) {
 	var config Config
-	data, err := os.ReadFile(filePath)
-	if err != nil {
-		return nil, err
-	}
-	err = yaml.Unmarshal(data, &config)
-	if err != nil {
+	if err := loadYAML(filePath, &config); err != nil {
 		return nil, err
 	}
 	return &config, nil
 }
+
+// LoadBlacklistConfig 从 YAML 配置文件加载黑名单配置
+func LoadBlacklistConfig(filePath string) (*Blacklist, error) {
+	var config Blacklist
+	if err := loadYAML(filePath, &config); err != nil {
+		return nil, err
+	}
+	return &config, nil
+}
+
+// LoadyamlConfig 从 YAML 配置文件加载配置
+func loadYAML(filePath string, out interface{}) error {
+	data, err := os.ReadFile(filePath)
+	if err != nil {
+		return err
+	}
+	return yaml.Unmarshal(data, out)
+}

config/config.yaml

@@ -1,8 +1,24 @@
-port: 8080
-host: "127.0.0.1"
-sizelimit: 131072000 # 125MB
-logfilepath: "/data/ghproxy/log/ghproxy.log"
-maxlogsize: 25 #MB
-CorsAllowOrigins: true
-auth: false
-authtoken: "test"
+# Server Configuration
+server:
+  port: 8080
+  host: "127.0.0.1"
+  sizelimit: 131072000 # 125MB
+
+# Logging Configuration
+logger:
+  logfilepath: "/data/ghproxy/log/ghproxy.log"
+  maxlogsize: 25 # MB
+
+# CORS Configuration
+cors:
+  enabled: true
+
+# Authentication Configuration
+auth:
+  enabled: false
+  authtoken: "test"
+
+# Blacklist Configuration
+blacklist:
+  enabled: true
+  blacklistfile: "/data/ghproxy/config/blacklist.yaml"