From db2bdd1cab4e1c44503b6203ca6bbc5ba71a0e76 Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@coder.com>
Date: Wed, 8 Mar 2023 20:29:02 -0600
Subject: [PATCH] fix: pass sdk header transport to websocket connections
 (#6521)

Fixes #6520.
---
 codersdk/provisionerdaemons.go | 6 ++++--
 codersdk/workspaceagents.go    | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/codersdk/provisionerdaemons.go b/codersdk/provisionerdaemons.go
index d336b7c04a..0a751169bf 100644
--- a/codersdk/provisionerdaemons.go
+++ b/codersdk/provisionerdaemons.go
@@ -140,7 +140,8 @@ func (c *Client) provisionerJobLogsAfter(ctx context.Context, path string, after
 		Value: c.SessionToken(),
 	}})
 	httpClient := &http.Client{
-		Jar: jar,
+		Jar:       jar,
+		Transport: c.HTTPClient.Transport,
 	}
 	conn, res, err := websocket.Dial(ctx, followURL.String(), &websocket.DialOptions{
 		HTTPClient:      httpClient,
@@ -205,7 +206,8 @@ func (c *Client) ServeProvisionerDaemon(ctx context.Context, organization uuid.U
 		Value: c.SessionToken(),
 	}})
 	httpClient := &http.Client{
-		Jar: jar,
+		Jar:       jar,
+		Transport: c.HTTPClient.Transport,
 	}
 	conn, res, err := websocket.Dial(ctx, serverURL.String(), &websocket.DialOptions{
 		HTTPClient: httpClient,
diff --git a/codersdk/workspaceagents.go b/codersdk/workspaceagents.go
index f59b7ce923..4d953f8e05 100644
--- a/codersdk/workspaceagents.go
+++ b/codersdk/workspaceagents.go
@@ -284,7 +284,8 @@ func (c *Client) WorkspaceAgentReconnectingPTY(ctx context.Context, agentID, rec
 		Value: c.SessionToken(),
 	}})
 	httpClient := &http.Client{
-		Jar: jar,
+		Jar:       jar,
+		Transport: c.HTTPClient.Transport,
 	}
 	conn, res, err := websocket.Dial(ctx, serverURL.String(), &websocket.DialOptions{
 		HTTPClient: httpClient,