mirror of
https://github.com/black-ant/Ant-Browser.git
synced 2026-07-14 18:48:55 +08:00
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
package proxy
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"ant-chrome/backend/internal/config"
|
|
|
|
xproxy "golang.org/x/net/proxy"
|
|
)
|
|
|
|
// buildProxyHTTPClient 根据代理配置构建 HTTP 客户端，统一用于测速/健康检测场景。
|
|
func buildProxyHTTPClient(
|
|
src string,
|
|
proxyId string,
|
|
proxies []config.BrowserProxy,
|
|
xrayMgr *XrayManager,
|
|
singboxMgr *SingBoxManager,
|
|
timeout time.Duration,
|
|
) (*http.Client, error) {
|
|
src = resolveProxyConfig(src, proxies, proxyId)
|
|
l := strings.ToLower(strings.TrimSpace(src))
|
|
if l == "" || l == "direct://" {
|
|
return &http.Client{Timeout: timeout}, nil
|
|
}
|
|
|
|
if IsChainSocks5Proxy(src) {
|
|
if xrayMgr == nil {
|
|
return nil, fmt.Errorf("xray 管理器未初始化")
|
|
}
|
|
socks5Addr, err := xrayMgr.EnsureBridge(src, proxies, proxyId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("xray 桥接启动失败: %w", err)
|
|
}
|
|
return buildSocks5HTTPClient(strings.TrimPrefix(socks5Addr, "socks5://"), timeout)
|
|
}
|
|
|
|
if IsSingBoxProtocol(src) {
|
|
if singboxMgr == nil {
|
|
return nil, fmt.Errorf("sing-box 管理器未初始化")
|
|
}
|
|
socks5Addr, err := singboxMgr.EnsureBridge(src, proxies, proxyId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("sing-box 桥接启动失败: %w", err)
|
|
}
|
|
return buildSocks5HTTPClient(strings.TrimPrefix(socks5Addr, "socks5://"), timeout)
|
|
}
|
|
|
|
if RequiresBridge(src, proxies, proxyId) {
|
|
if xrayMgr == nil {
|
|
return nil, fmt.Errorf("xray 管理器未初始化")
|
|
}
|
|
socks5Addr, err := xrayMgr.EnsureBridge(src, proxies, proxyId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("xray 桥接启动失败: %w", err)
|
|
}
|
|
return buildSocks5HTTPClient(strings.TrimPrefix(socks5Addr, "socks5://"), timeout)
|
|
}
|
|
|
|
if strings.HasPrefix(l, "socks5://") {
|
|
u, err := url.Parse(src)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("SOCKS5 地址解析失败: %w", err)
|
|
}
|
|
var auth *xproxy.Auth
|
|
if u.User != nil {
|
|
pass, _ := u.User.Password()
|
|
auth = &xproxy.Auth{
|
|
User: u.User.Username(),
|
|
Password: pass,
|
|
}
|
|
}
|
|
dialer, err := xproxy.SOCKS5("tcp", u.Host, auth, xproxy.Direct)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("SOCKS5 dialer 创建失败: %w", err)
|
|
}
|
|
contextDialer, ok := dialer.(xproxy.ContextDialer)
|
|
if !ok {
|
|
return nil, fmt.Errorf("SOCKS5 dialer 不支持 ContextDialer")
|
|
}
|
|
transport := &http.Transport{DialContext: contextDialer.DialContext}
|
|
return &http.Client{Transport: transport, Timeout: timeout}, nil
|
|
}
|
|
|
|
proxyURL, err := url.Parse(src)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("代理地址解析失败: %w", err)
|
|
}
|
|
transport := &http.Transport{Proxy: http.ProxyURL(proxyURL)}
|
|
return &http.Client{Transport: transport, Timeout: timeout}, nil
|
|
}
|
|
|
|
func buildSocks5HTTPClient(socks5Host string, timeout time.Duration) (*http.Client, error) {
|
|
dialer, err := xproxy.SOCKS5("tcp", socks5Host, nil, xproxy.Direct)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("SOCKS5 dialer 创建失败: %w", err)
|
|
}
|
|
contextDialer, ok := dialer.(xproxy.ContextDialer)
|
|
if !ok {
|
|
return nil, fmt.Errorf("SOCKS5 dialer 不支持 ContextDialer")
|
|
}
|
|
transport := &http.Transport{DialContext: contextDialer.DialContext}
|
|
return &http.Client{Transport: transport, Timeout: timeout}, nil
|
|
}
|