mirror of
https://github.com/router-for-me/CLIProxyAPIPlus.git
synced 2026-03-22 00:50:26 +00:00
feat(proxy): centralize proxy handling with `proxyutil` package and enhance test coverage - Added `proxyutil` package to simplify proxy handling across the codebase. - Refactored various components (`executor`, `cliproxy`, `auth`, etc.) to use `proxyutil` for consistent and reusable proxy logic. - Introduced support for "direct" proxy mode to explicitly bypass all proxies. - Updated tests to validate proxy behavior (e.g., `direct`, HTTP/HTTPS, and SOCKS5). - Enhanced YAML configuration documentation for proxy options.
31 lines
1003 B
Go
31 lines
1003 B
Go
// Package util provides utility functions for the CLI Proxy API server.
|
|
// It includes helper functions for proxy configuration, HTTP client setup,
|
|
// log level management, and other common operations used across the application.
|
|
package util
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/sdk/config"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/sdk/proxyutil"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// SetProxy configures the provided HTTP client with proxy settings from the configuration.
|
|
// It supports SOCKS5, HTTP, and HTTPS proxies. The function modifies the client's transport
|
|
// to route requests through the configured proxy server.
|
|
func SetProxy(cfg *config.SDKConfig, httpClient *http.Client) *http.Client {
|
|
if cfg == nil || httpClient == nil {
|
|
return httpClient
|
|
}
|
|
|
|
transport, _, errBuild := proxyutil.BuildHTTPTransport(cfg.ProxyURL)
|
|
if errBuild != nil {
|
|
log.Errorf("%v", errBuild)
|
|
}
|
|
if transport != nil {
|
|
httpClient.Transport = transport
|
|
}
|
|
return httpClient
|
|
}
|