sdd

pkg/api/config.go

@@ -0,0 +1,90 @@
+package api
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+
+	"gopkg.in/yaml.v3"
+)
+
+// Config 保存应用程序配置
+type Config struct {
+	API struct {
+		BaseURL      string `yaml:"base_url"`
+		DisableProxy bool   `yaml:"disable_proxy"`
+	} `yaml:"api"`
+	User struct {
+		DefaultDomain   string `yaml:"default_domain"`
+		DefaultPassword string `yaml:"default_password"`
+		DefaultQuota    int64  `yaml:"default_quota"`
+	} `yaml:"user"`
+	Auth struct {
+		Basic struct {
+			AdminUsername string `yaml:"username"`
+			AdminPassword string `yaml:"password"`
+		} `yaml:"basic"`
+		APIKey string `yaml:"api_key"`
+	} `yaml:"auth"`
+}
+
+// LoadConfig 从指定路径加载配置文件
+func LoadConfig(path string) (*Config, error) {
+	// 如果未指定路径，使用默认路径
+	if path == "" {
+		// 尝试在当前目录和上一级目录查找配置文件
+		candidates := []string{
+			"config.yaml",
+			"config/config.yaml",
+			"../config/config.yaml",
+		}
+
+		for _, candidate := range candidates {
+			if _, err := os.Stat(candidate); err == nil {
+				path = candidate
+				break
+			}
+		}
+
+		if path == "" {
+			return nil, fmt.Errorf("未找到配置文件")
+		}
+	}
+
+	// 读取配置文件
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("读取配置文件失败: %w", err)
+	}
+
+	// 解析YAML
+	config := &Config{}
+	if err := yaml.Unmarshal(data, config); err != nil {
+		return nil, fmt.Errorf("解析配置文件失败: %w", err)
+	}
+
+	// 设置默认值
+	if config.API.BaseURL == "" {
+		config.API.BaseURL = "https://mail.evnmail.com/api"
+	}
+	if config.User.DefaultDomain == "" {
+		config.User.DefaultDomain = "evnmail.com"
+	}
+	if config.User.DefaultPassword == "" {
+		config.User.DefaultPassword = "123456"
+	}
+	if config.User.DefaultQuota == 0 {
+		config.User.DefaultQuota = 1073741824 // 1GB
+	}
+
+	fmt.Printf("已从 %s 加载配置\n", path)
+	return config, nil
+}
+
+// DisableProxy 禁用代理设置
+func DisableProxy() {
+	os.Setenv("HTTP_PROXY", "")
+	os.Setenv("HTTPS_PROXY", "")
+	os.Setenv("http_proxy", "")
+	os.Setenv("https_proxy", "")
+}