ghproxy/config/config.go

package config

import (
	"github.com/BurntSushi/toml"
)

/*type Config struct {
	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"`

	Whitelist struct {
		Enabled       bool   `yaml:"enabled"`
		WhitelistFile string `yaml:"whitelistfile"`
	} `yaml:"whitelist"`
}

// LoadConfig 从 YAML 配置文件加载配置
func LoadConfig(filePath string) (*Config, error) {
	var config Config
	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)
}*/

type Config struct {
	Server    ServerConfig
	Log       LogConfig
	CORS      CORSConfig
	Auth      AuthConfig
	Blacklist BlacklistConfig
	Whitelist WhitelistConfig
}

type ServerConfig struct {
	Port      int    `toml:"port"`
	Host      string `toml:"host"`
	SizeLimit int    `toml:"sizelimit"`
}

type LogConfig struct {
	LogFilePath string `toml:"logfilepath"`
	MaxLogSize  int    `toml:"maxlogsize"`
}

type CORSConfig struct {
	Enabled bool `toml:"enabled"`
}

type AuthConfig struct {
	Enabled   bool   `toml:"enabled"`
	AuthToken string `toml:"authtoken"`
}

type BlacklistConfig struct {
	Enabled       bool   `toml:"enabled"`
	BlacklistFile string `toml:"blacklistfile"`
}

type WhitelistConfig struct {
	Enabled       bool   `toml:"enabled"`
	WhitelistFile string `toml:"whitelistfile"`
}

// LoadConfig 从 TOML 配置文件加载配置
func LoadConfig(filePath string) (*Config, error) {
	var config Config
	if _, err := toml.DecodeFile(filePath, &config); err != nil {
		return nil, err
	}
	return &config, nil
}