Support using proxy dial-up connection to GitHub. (#46)

proxy/chunkreq.go

@@ -22,9 +22,9 @@ var (
 	BufferPool *sync.Pool
 )
 
-func InitReq() {
-	initChunkedHTTPClient()
-	initGitHTTPClient()
+func InitReq(cfg *config.Config) {
+	initChunkedHTTPClient(cfg)
+	initGitHTTPClient(cfg)
 
 	// 初始化固定大小的缓存池
 	BufferPool = &sync.Pool{
@@ -34,7 +34,7 @@ func InitReq() {
 	}
 }
 
-func initChunkedHTTPClient() {
+func initChunkedHTTPClient(cfg *config.Config) {
 	ctr = &http.Transport{
 		MaxIdleConns:          100,
 		MaxConnsPerHost:       60,
@@ -47,6 +47,7 @@ func initChunkedHTTPClient() {
 			KeepAlive: 30 * time.Second,
 		}).DialContext,
 	}
+	initTransport(cfg, ctr)
 	cclient = &http.Client{
 		Transport: ctr,
 	}

proxy/dial.go

@@ -0,0 +1,61 @@
+package proxy
+
+import (
+	"ghproxy/config"
+	"net/http"
+	"net/url"
+	"strings"
+
+	"golang.org/x/net/proxy"
+)
+
+func newProxyDial(proxyUrls string) proxy.Dialer {
+	var proxyDialer proxy.Dialer = proxy.Direct
+	for _, proxyUrl := range strings.Split(proxyUrls, ",") {
+		urlInfo, err := url.Parse(proxyUrl)
+		if err != nil {
+			continue
+		}
+		if urlInfo.Scheme != "socks5" {
+			continue
+		}
+		var auth *proxy.Auth = nil
+		if urlInfo.User != nil {
+			pwd, ok := urlInfo.User.Password()
+			if !ok {
+				continue
+			}
+			auth = &proxy.Auth{
+				User:     urlInfo.User.Username(),
+				Password: pwd,
+			}
+		}
+
+		dialer, err := proxy.SOCKS5("tcp", urlInfo.Host, auth, proxyDialer)
+		if err == nil {
+			proxyDialer = dialer
+		}
+	}
+	return proxyDialer
+}
+
+func initTransport(cfg *config.Config, transport *http.Transport) {
+	if !cfg.Proxy.Enabled {
+		return
+	}
+	if cfg.Proxy.Url == "" {
+		transport.Proxy = http.ProxyFromEnvironment
+		return
+	}
+
+	proxyInfo, err := url.Parse(cfg.Proxy.Url)
+	if err == nil {
+		if strings.HasPrefix(cfg.Proxy.Url, "http") {
+			transport.Proxy = http.ProxyURL(proxyInfo)
+		} else {
+			proxyDialer := newProxyDial(cfg.Proxy.Url)
+			transport.Dial = proxyDialer.Dial
+			transport.DialContext = proxyDialer.(proxy.ContextDialer).DialContext
+		}
+	}
+}

proxy/gitreq.go

@@ -17,12 +17,13 @@ var (
 	gtr     *http.Transport
 )
 
-func initGitHTTPClient() {
+func initGitHTTPClient(cfg *config.Config) {
 	gtr = &http.Transport{
 		MaxIdleConns:    30,
 		MaxConnsPerHost: 30,
 		IdleConnTimeout: 30 * time.Second,
 	}
+	initTransport(cfg, gtr)
 	gclient = &http.Client{
 		Transport: gtr,
 	}