refactor(api): rename applyProxy yo NewProxyTransport

- NewProxyTransport creates a cloned transport which does the required
  proxy handling in dial etc.
- Remove use of 'customTransport'

Author: burmudar

internal/api/api.go

@@ -87,39 +87,39 @@ type ClientOpts struct {
 	ProxyPath string
 }
 
-// NewClient creates a new API client.
-func NewClient(opts ClientOpts) Client {
-	if opts.Out == nil {
-		panic("unexpected nil out option")
-	}
-
-	flags := opts.Flags
-	if flags == nil {
-		flags = defaultFlags()
-	}
-
-	httpClient := http.DefaultClient
-
+func buildTransport(opts ClientOpts, flags *Flags) *http.Transport {
 	transport := http.DefaultTransport.(*http.Transport).Clone()
-	customTransport := false
 
 	if flags.insecureSkipVerify != nil && *flags.insecureSkipVerify {
-		customTransport = true
 		transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
 	}
 
 	if transport.TLSClientConfig == nil {
 		transport.TLSClientConfig = &tls.Config{}
 	}
 
-	if applyProxy(transport, opts.ProxyURL, opts.ProxyPath) {
-		customTransport = true
+	if opts.ProxyURL != nil || opts.ProxyPath != "" {
+		transport = NewProxyTransport(transport, opts.ProxyURL, opts.ProxyPath)
 	}
 
-	if customTransport {
-		httpClient = &http.Client{
-			Transport: transport,
-		}
+	return transport
+}
+
+// NewClient creates a new API client.
+func NewClient(opts ClientOpts) Client {
+	if opts.Out == nil {
+		panic("unexpected nil out option")
+	}
+
+	flags := opts.Flags
+	if flags == nil {
+		flags = defaultFlags()
+	}
+
+	transport := buildTransport(opts, flags)
+
+	httpClient := &http.Client{
+		Transport: transport,
 	}
 
 	return &client{

internal/api/proxy.go

@@ -11,10 +11,9 @@ import (
 	"net/url"
 )
 
-func applyProxy(transport *http.Transport, proxyURL *url.URL, proxyPath string) (applied bool) {
-	if proxyURL == nil && proxyPath == "" {
-		return false
-	}
+func NewProxyTransport(base *http.Transport, proxyURL *url.URL, proxyPath string) *http.Transport {
+	// Clone so that we don't change the original transport
+	transport := base.Clone()
 
 	handshakeTLS := func(ctx context.Context, conn net.Conn, addr string) (net.Conn, error) {
 		// Extract the hostname (without the port) for TLS SNI
@@ -34,8 +33,6 @@ func applyProxy(transport *http.Transport, proxyURL *url.URL, proxyPath string)
 		return tlsConn, nil
 	}
 
-	proxyApplied := false
-
 	if proxyPath != "" {
 		dial := func(ctx context.Context, _, _ string) (net.Conn, error) {
 			d := net.Dialer{}
@@ -52,13 +49,11 @@ func applyProxy(transport *http.Transport, proxyURL *url.URL, proxyPath string)
 		transport.DialTLSContext = dialTLS
 		// clear out any system proxy settings
 		transport.Proxy = nil
-		proxyApplied = true
 	} else if proxyURL != nil {
 		switch proxyURL.Scheme {
 		case "socks5", "socks5h":
 			// SOCKS proxies work out of the box - no need to manually dial
 			transport.Proxy = http.ProxyURL(proxyURL)
-			proxyApplied = true
 		case "http", "https":
 			dial := func(ctx context.Context, network, addr string) (net.Conn, error) {
 				// Dial the proxy
@@ -130,9 +125,8 @@ func applyProxy(transport *http.Transport, proxyURL *url.URL, proxyPath string)
 			transport.DialTLSContext = dialTLS
 			// clear out any system proxy settings
 			transport.Proxy = nil
-			proxyApplied = true
 		}
 	}
 
-	return proxyApplied
+	return transport
 }