新增本机代理获取方法

internal/proxy.go

@@ -0,0 +1,173 @@
+package internal
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"golang.org/x/sys/windows/registry"
+)
+
+// ProxyInfo 保存当前系统代理信息
+type ProxyInfo struct {
+	// 是否启用了系统代理
+	Enabled bool
+	// 系统代理服务器地址 (例如 127.0.0.1:7890)
+	Server string
+	// 代理协议 (http, https, socks5)
+	Protocol string
+	// HTTP_PROXY 环境变量
+	HTTPProxy string
+	// HTTPS_PROXY 环境变量
+	HTTPSProxy string
+	// 代理绕过列表
+	Bypass string
+}
+
+// GetCurrentProxy 获取本机当前的代理设置
+// 优先读取 Windows 系统代理（注册表），同时也读取环境变量中的代理配置
+// 支持检测 Clash、V2Ray 等常见代理工具的配置
+func GetCurrentProxy() (*ProxyInfo, error) {
+	info := &ProxyInfo{}
+
+	// 1. 从环境变量读取代理
+	info.HTTPProxy = getEnvProxy("HTTP_PROXY", "http_proxy")
+	info.HTTPSProxy = getEnvProxy("HTTPS_PROXY", "https_proxy")
+
+	// 2. 从 Windows 注册表读取系统代理设置
+	if err := readWindowsSystemProxy(info); err != nil {
+		// 注册表读取失败不算致命错误，环境变量中可能仍有代理信息
+		fmt.Printf("读取系统代理设置时出错: %v\n", err)
+	}
+
+	// 3. 如果系统代理未启用，尝试从环境变量推断
+	if !info.Enabled && (info.HTTPProxy != "" || info.HTTPSProxy != "") {
+		info.Enabled = true
+		proxy := info.HTTPProxy
+		if proxy == "" {
+			proxy = info.HTTPSProxy
+		}
+		info.Server = extractServerFromURL(proxy)
+		info.Protocol = extractProtocolFromURL(proxy)
+	}
+
+	return info, nil
+}
+
+// readWindowsSystemProxy 从 Windows 注册表读取系统代理设置
+func readWindowsSystemProxy(info *ProxyInfo) error {
+	key, err := registry.OpenKey(
+		registry.CURRENT_USER,
+		`Software\Microsoft\Windows\CurrentVersion\Internet Settings`,
+		registry.QUERY_VALUE,
+	)
+	if err != nil {
+		return fmt.Errorf("无法打开注册表键: %w", err)
+	}
+	defer key.Close()
+
+	// 读取代理是否启用 (ProxyEnable: 0=关闭, 1=开启)
+	proxyEnable, _, err := key.GetIntegerValue("ProxyEnable")
+	if err != nil {
+		return fmt.Errorf("无法读取 ProxyEnable: %w", err)
+	}
+	info.Enabled = proxyEnable == 1
+
+	// 读取代理服务器地址
+	proxyServer, _, err := key.GetStringValue("ProxyServer")
+	if err == nil && proxyServer != "" {
+		info.Server = proxyServer
+		// 根据端口推断代理协议
+		info.Protocol = guessProtocol(proxyServer)
+	}
+
+	// 读取代理绕过列表
+	bypass, _, err := key.GetStringValue("ProxyOverride")
+	if err == nil {
+		info.Bypass = bypass
+	}
+
+	return nil
+}
+
+// getEnvProxy 依次检查多个环境变量名，返回第一个非空值
+func getEnvProxy(names ...string) string {
+	for _, name := range names {
+		if v := os.Getenv(name); v != "" {
+			return v
+		}
+	}
+	return ""
+}
+
+// extractServerFromURL 从代理 URL 中提取 host:port 部分
+// 例如 "http://127.0.0.1:7890" -> "127.0.0.1:7890"
+func extractServerFromURL(proxyURL string) string {
+	s := proxyURL
+	// 去掉协议前缀
+	if idx := strings.Index(s, "://"); idx != -1 {
+		s = s[idx+3:]
+	}
+	// 去掉尾部斜杠
+	s = strings.TrimRight(s, "/")
+	return s
+}
+
+// extractProtocolFromURL 从代理 URL 中提取协议
+// 例如 "socks5://127.0.0.1:7891" -> "socks5"
+func extractProtocolFromURL(proxyURL string) string {
+	if idx := strings.Index(proxyURL, "://"); idx != -1 {
+		return proxyURL[:idx]
+	}
+	return "http"
+}
+
+// guessProtocol 根据代理服务器地址猜测代理协议
+// Clash 默认: 7890(http), 7891(socks5)
+// V2Ray 默认: 10808(socks5), 10809(http)
+func guessProtocol(server string) string {
+	knownPorts := map[string]string{
+		"7890":  "http",   // Clash HTTP
+		"7891":  "socks5", // Clash SOCKS5
+		"7892":  "http",   // Clash mixed
+		"10808": "socks5", // V2Ray SOCKS5
+		"10809": "http",   // V2Ray HTTP
+		"1080":  "socks5", // 通用 SOCKS5
+		"1081":  "http",   // 通用 HTTP
+		"8080":  "http",   // 通用 HTTP
+	}
+
+	parts := strings.Split(server, ":")
+	if len(parts) >= 2 {
+		port := parts[len(parts)-1]
+		if protocol, ok := knownPorts[port]; ok {
+			return protocol
+		}
+	}
+	return "http"
+}
+
+// String 返回代理信息的可读字符串
+func (p *ProxyInfo) String() string {
+	if !p.Enabled {
+		return "当前未检测到系统代理"
+	}
+
+	var sb strings.Builder
+	sb.WriteString("当前系统代理信息:\n")
+	sb.WriteString("  状态:   已启用\n")
+	fmt.Fprintf(&sb, "  服务器: %s\n", p.Server)
+	fmt.Fprintf(&sb, "  协议:   %s\n", p.Protocol)
+
+	if p.Bypass != "" {
+		fmt.Fprintf(&sb, "  绕过:   %s\n", p.Bypass)
+	}
+	if p.HTTPProxy != "" {
+		fmt.Fprintf(&sb, "  HTTP_PROXY:  %s\n", p.HTTPProxy)
+	}
+	if p.HTTPSProxy != "" {
+		fmt.Fprintf(&sb, "  HTTPS_PROXY: %s\n", p.HTTPSProxy)
+	}
+
+	return sb.String()
+}