From 693e5d94bc477d7f49945fa7ce5628a4138852db Mon Sep 17 00:00:00 2001 From: Kyle Carberry Date: Tue, 18 Apr 2023 08:07:10 -0500 Subject: [PATCH] fix: add global headers to `vscodessh` command (#7181) --- cli/vscodessh.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cli/vscodessh.go b/cli/vscodessh.go index 85ee1e18e4..363f0215a7 100644 --- a/cli/vscodessh.go +++ b/cli/vscodessh.go @@ -27,7 +27,7 @@ import ( // This command needs to remain stable for compatibility with // various VS Code versions, so it's kept separate from our // standard SSH command. -func (*RootCmd) vscodeSSH() *clibase.Cmd { +func (r *RootCmd) vscodeSSH() *clibase.Cmd { var ( sessionTokenFile string urlFile string @@ -82,6 +82,12 @@ func (*RootCmd) vscodeSSH() *clibase.Cmd { client := codersdk.New(serverURL) client.SetSessionToken(string(sessionToken)) + // This adds custom headers to the request! + err = r.setClient(client, serverURL) + if err != nil { + return xerrors.Errorf("set client: %w", err) + } + parts := strings.Split(inv.Args[0], "--") if len(parts) < 3 { return xerrors.Errorf("invalid argument format. must be: coder-vscode----")