Update workspace network client helpers

Author: janekbaraniewski

pkg/daemon/workspace/network/client.go

@@ -1,26 +1,40 @@
 package network
 
 import (
+	"context"
 	"net"
 	"net/http"
 	"path/filepath"
 	"time"
 )
 
-// GetClient returns a new HTTP client that uses a DevPod network socket for communication.
-func GetClient() *http.Client {
-	// Set up HTTP transport that uses our network socket.
+// Dial returns a net.Conn to the network proxy socket.
+func Dial() (net.Conn, error) {
 	socketPath := filepath.Join(RootDir, NetworkProxySocket)
-	transport := &http.Transport{
+	return net.Dial("unix", socketPath)
+}
+
+// GetCOntextDialer returns ContextDialer interface function that uses our network socket.
+func GetContextDialer() func(ctx context.Context, addr string) (net.Conn, error) {
+	return func(ctx context.Context, addr string) (net.Conn, error) {
+		return Dial()
+	}
+}
+
+// GetHTTPTransport returns http.Transport that uses our network socket for HTTP requests.
+func GetHTTPTransport() *http.Transport {
+	// Set up HTTP transport that uses our network socket.
+	return &http.Transport{
 		Dial: func(network, addr string) (net.Conn, error) {
-			return net.Dial("unix", socketPath)
+			return Dial()
 		},
 	}
+}
 
-	client := &http.Client{
-		Transport: transport,
-		Timeout:   30 * time.Second, // TODO: extract this to config
+// GetClient returns a new HTTP client that uses our network socket for transport.
+func GetHTTPClient() *http.Client {
+	return &http.Client{
+		Transport: GetHTTPTransport(),
+		Timeout:   30 * time.Second,
 	}
-
-	return client
 }