Add speedtest command

package.json

@@ -313,6 +313,11 @@
 				"category": "Coder",
 				"icon": "$(refresh)"
 			},
+			{
+				"command": "coder.speedTest",
+				"title": "Run Speed Test",
+				"category": "Coder"
+			},
 			{
 				"command": "coder.viewLogs",
 				"title": "Coder: View Logs",
@@ -371,6 +376,10 @@
 					"command": "coder.createWorkspace",
 					"when": "coder.authenticated"
 				},
+				{
+					"command": "coder.speedTest",
+					"when": "coder.workspace.connected"
+				},
 				{
 					"command": "coder.navigateToWorkspace",
 					"when": "coder.workspace.connected"

src/commands.ts

@@ -146,6 +146,53 @@ export class Commands {
 		this.logger.debug("Login complete to deployment:", url);
 	}
 
+	/**
+	 * Run a speed test against the currently connected workspace and display the
+	 * results in a new editor document.
+	 */
+	public async speedTest(): Promise<void> {
+		if (!this.workspace) {
+			vscode.window.showInformationMessage("No workspace connected.");
+			return;
+		}
+
+		await withProgress(
+			{
+				location: vscode.ProgressLocation.Notification,
+				title: "Running speed test...",
+			},
+			async () => {
+				const baseUrl = this.requireExtensionBaseUrl();
+				const safeHost = toSafeHost(baseUrl);
+				const binary = await this.cliManager.fetchBinary(this.extensionClient);
+				const version = semver.parse(await cliUtils.version(binary));
+				const featureSet = featureSetForVersion(version);
+				const configDir = this.pathResolver.getGlobalConfigDir(safeHost);
+				const configs = vscode.workspace.getConfiguration();
+				const auth = resolveCliAuth(configs, featureSet, baseUrl, configDir);
+				const workspaceName = createWorkspaceIdentifier(this.workspace!);
+
+				try {
+					const stdout = await cliUtils.speedtest(
+						binary,
+						auth,
+						workspaceName,
+					);
+					const doc = await vscode.workspace.openTextDocument({
+						content: stdout,
+						language: "json",
+					});
+					await vscode.window.showTextDocument(doc);
+				} catch (error) {
+					this.logger.error("Speed test failed", error);
+					vscode.window.showErrorMessage(
+						`Speed test failed: ${error instanceof Error ? error.message : String(error)}`,
+					);
+				}
+			},
+		);
+	}
+
 	/**
 	 * View the logs for the currently connected workspace.
 	 */

src/core/cliUtils.ts

@@ -6,6 +6,8 @@ import os from "node:os";
 import path from "node:path";
 import { promisify } from "node:util";
 
+import type { CliAuth } from "../cliConfig";
+
 /**
  * Custom error thrown when a binary file is locked (typically on Windows).
  */
@@ -72,6 +74,34 @@ export async function version(binPath: string): Promise<string> {
 	return json.version;
 }
 
+/**
+ * Run a speed test against the specified workspace and return the JSON output.
+ * Throw if unable to execute the binary or parse the output.
+ */
+export async function speedtest(
+	binPath: string,
+	auth: CliAuth,
+	workspaceName: string,
+): Promise<string> {
+	const result = await promisify(execFile)(binPath, [
+		...authArgs(auth),
+		"speedtest",
+		workspaceName,
+		"--output",
+		"json",
+	]);
+	return result.stdout;
+}
+
+/**
+ * Build CLI auth flags for execFile (no shell escaping).
+ */
+function authArgs(auth: CliAuth): string[] {
+	return auth.mode === "url"
+		? ["--url", auth.url]
+		: ["--global-config", auth.configDir];
+}
+
 export interface RemovalResult {
 	fileName: string;
 	error: unknown;

src/extension.ts

@@ -281,6 +281,10 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
 			void myWorkspacesProvider.fetchAndRefresh();
 			void allWorkspacesProvider.fetchAndRefresh();
 		}),
+		vscode.commands.registerCommand(
+			"coder.speedTest",
+			commands.speedTest.bind(commands),
+		),
 		vscode.commands.registerCommand(
 			"coder.viewLogs",
 			commands.viewLogs.bind(commands),

test/fixtures/scripts/echo-args.bash

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+
+# Prints each argument on its own line, so tests can verify exact args.
+for arg in "$@"; do
+  echo "$arg"
+done

test/unit/core/cliUtils.test.ts

@@ -142,6 +142,63 @@ describe("CliUtils", () => {
 		]);
 	});
 
+	describe("speedtest", () => {
+		const echoArgsBin = path.join(tmp, "echo-args");
+
+		beforeAll(async () => {
+			const tmpl = await fs.readFile(
+				getFixturePath("scripts", "echo-args.bash"),
+				"utf8",
+			);
+			await fs.writeFile(echoArgsBin, tmpl);
+			await fs.chmod(echoArgsBin, "755");
+		});
+
+		it("passes global-config auth flags", async () => {
+			const result = await cliUtils.speedtest(
+				echoArgsBin,
+				{ mode: "global-config", configDir: "/tmp/test-config" },
+				"owner/workspace",
+			);
+			const args = result.trim().split("\n");
+			expect(args).toEqual([
+				"--global-config",
+				"/tmp/test-config",
+				"speedtest",
+				"owner/workspace",
+				"--output",
+				"json",
+			]);
+		});
+
+		it("passes url auth flags", async () => {
+			const result = await cliUtils.speedtest(
+				echoArgsBin,
+				{ mode: "url", url: "http://localhost:3000" },
+				"owner/workspace",
+			);
+			const args = result.trim().split("\n");
+			expect(args).toEqual([
+				"--url",
+				"http://localhost:3000",
+				"speedtest",
+				"owner/workspace",
+				"--output",
+				"json",
+			]);
+		});
+
+		it("throws when binary does not exist", async () => {
+			await expect(
+				cliUtils.speedtest(
+					"/nonexistent/binary",
+					{ mode: "global-config", configDir: "/tmp" },
+					"owner/workspace",
+				),
+			).rejects.toThrow("ENOENT");
+		});
+	});
+
 	it("ETag", async () => {
 		const binPath = path.join(tmp, "hash");