增加支持走代理

src/http_client.go

@@ -1,9 +1,14 @@
 package main
 
 import (
+	"context"
+	"log"
 	"net"
 	"net/http"
+	"net/url"
 	"time"
+
+	"golang.org/x/net/proxy"
 )
 
 var (
@@ -15,13 +20,65 @@ var (
 
 // initHTTPClients 初始化HTTP客户端
 func initHTTPClients() {
+	cfg := GetConfig()
+	
+	// 创建DialContext函数，支持SOCKS5代理
+	createDialContext := func(timeout time.Duration) func(ctx context.Context, network, addr string) (net.Conn, error) {
+		if cfg.Proxy.Socks5 == "" {
+			// 没有配置代理，使用直连
+			dialer := &net.Dialer{
+				Timeout:   timeout,
+				KeepAlive: 30 * time.Second,
+			}
+			return dialer.DialContext
+		}
+		
+		// 解析SOCKS5代理URL
+		proxyURL, err := url.Parse(cfg.Proxy.Socks5)
+		if err != nil {
+			log.Printf("SOCKS5代理配置错误，使用直连: %v", err)
+			dialer := &net.Dialer{
+				Timeout:   timeout,
+				KeepAlive: 30 * time.Second,
+			}
+			return dialer.DialContext
+		}
+		
+		// 创建基础dialer
+		baseDialer := &net.Dialer{
+			Timeout:   timeout,
+			KeepAlive: 30 * time.Second,
+		}
+		
+		// 创建SOCKS5代理dialer
+		var auth *proxy.Auth
+		if proxyURL.User != nil {
+			if password, ok := proxyURL.User.Password(); ok {
+				auth = &proxy.Auth{
+					User:     proxyURL.User.Username(),
+					Password: password,
+				}
+			}
+		}
+		
+		socks5Dialer, err := proxy.SOCKS5("tcp", proxyURL.Host, auth, baseDialer)
+		if err != nil {
+			log.Printf("创建SOCKS5代理失败，使用直连: %v", err)
+			return baseDialer.DialContext
+		}
+		
+		log.Printf("使用SOCKS5代理: %s", proxyURL.Host)
+		
+		// 返回带上下文的dial函数
+		return func(ctx context.Context, network, addr string) (net.Conn, error) {
+			return socks5Dialer.Dial(network, addr)
+		}
+	}
+	
 	// 代理客户端配置 - 适用于大文件传输
 	globalHTTPClient = &http.Client{
 		Transport: &http.Transport{
-			DialContext: (&net.Dialer{
-				Timeout:   30 * time.Second,
-				KeepAlive: 30 * time.Second,
-			}).DialContext,
+			DialContext:           createDialContext(30 * time.Second),
 			MaxIdleConns:          1000,
 			MaxIdleConnsPerHost:   1000,
 			IdleConnTimeout:       90 * time.Second,
@@ -35,10 +92,7 @@ func initHTTPClients() {
 	searchHTTPClient = &http.Client{
 		Timeout: 10 * time.Second,
 		Transport: &http.Transport{
-			DialContext: (&net.Dialer{
-				Timeout:   5 * time.Second,
-				KeepAlive: 30 * time.Second,
-			}).DialContext,
+			DialContext:         createDialContext(5 * time.Second),
 			MaxIdleConns:        100,
 			MaxIdleConnsPerHost: 10,
 			IdleConnTimeout:     90 * time.Second,