| package internal
|
|
|
| import ( |
| "fmt" |
| "os" |
| "strings" |
| "sync" |
| "time" |
|
|
| fhttp "github.com/bogdanfinn/fhttp"
|
| tls_client "github.com/bogdanfinn/tls-client"
|
| "github.com/bogdanfinn/tls-client/profiles"
|
| )
|
|
|
|
|
| var DefaultBrowserProfile = profiles.Chrome_133
|
|
|
|
|
| const BrowserUserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36"
|
|
|
| var (
|
| tlsClientMu sync.Mutex
|
| tlsBySec = map[int]tls_client.HttpClient{}
|
| )
|
|
|
|
|
| func TLSHTTPClient(timeout time.Duration) (tls_client.HttpClient, error) { |
| sec := int(timeout.Round(time.Second) / time.Second) |
| if sec < 1 { |
| sec = 1 |
| } |
| proxyURL := resolveProxyURL() |
| tlsClientMu.Lock() |
| defer tlsClientMu.Unlock() |
| if c, ok := tlsBySec[sec]; ok { |
| return c, nil |
| } |
| options := []tls_client.HttpClientOption{ |
| tls_client.WithTimeoutSeconds(sec), |
| tls_client.WithClientProfile(DefaultBrowserProfile), |
| tls_client.WithRandomTLSExtensionOrder(), |
| } |
| if proxyURL != "" { |
| options = append(options, tls_client.WithProxyUrl(proxyURL)) |
| } |
| c, err := tls_client.NewHttpClient(tls_client.NewNoopLogger(), options...) |
| if err != nil { |
| return nil, fmt.Errorf("tls-client: %w", err) |
| } |
| tlsBySec[sec] = c |
| return c, nil |
| } |
|
|
| func resolveProxyURL() string { |
| for _, name := range []string{"PROXY_SERVER", "ZAI_PROXY", "HTTPS_PROXY", "HTTP_PROXY"} { |
| if value := strings.TrimSpace(os.Getenv(name)); value != "" { |
| return value |
| } |
| } |
| return "" |
| } |
|
|
|
|
| func ApplyBrowserFingerprintHeaders(h fhttp.Header) {
|
| h.Set("User-Agent", BrowserUserAgent)
|
| h.Set("sec-ch-ua", `"Google Chrome";v="133", "Chromium";v="133", "Not A(Brand";v="24"`)
|
| h.Set("sec-ch-ua-mobile", "?0")
|
| h.Set("sec-ch-ua-platform", `"Windows"`)
|
| }
|
|
|
