balki/tss
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mvp before debug

Browse Source
This commit is contained in:
Balakrishnan Balasubramanian
2022-05-01 23:28:54 -04:00
parent 5e55fea0fd
commit f5e4628459
6 changed files with 148 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
proxy/proxy.go
View File

@ -8,9 +8,9 @@ import (
"golang.org/x/net/proxy"
)
func GetClient(proxy string) (*http.Client, error) {
func GetTransport(proxy string) (http.RoundTripper, error) {
if proxy == "" {
return http.DefaultClient, nil
return http.DefaultTransport, nil
}
proxyUrl, err := url.Parse(proxy)
if err != nil {
@ -23,7 +23,7 @@ func GetClient(proxy string) (*http.Client, error) {
return proxyHttp(proxyUrl)
}
func unixSocks5Proxy(path string) (*http.Client, error) {
func unixSocks5Proxy(path string) (http.RoundTripper, error) {
// TODO: Auth?
dialer, err := proxy.SOCKS5("unix", path, nil /*auth*/, nil)
if err != nil {
@ -36,13 +36,13 @@ func unixSocks5Proxy(path string) (*http.Client, error) {
trans := defaultTransport()
trans.DialContext = ctxDialer.DialContext
trans.Proxy = nil
return &http.Client{Transport: trans}, nil
return trans, nil
}
func proxyHttp(proxyUrl *url.URL) (*http.Client, error) {
func proxyHttp(proxyUrl *url.URL) (http.RoundTripper, error) {
trans := defaultTransport()
trans.Proxy = http.ProxyURL(proxyUrl)
return &http.Client{Transport: trans}, nil
return trans, nil
}
func defaultTransport() *http.Transport {