Add simple OAuth command

Author: EhabY

package.json

@@ -255,6 +255,11 @@
 				"title": "Search",
 				"category": "Coder",
 				"icon": "$(search)"
+			},
+			{
+				"command": "coder.oauth.testAuth",
+				"title": "Test OAuth Auth",
+				"category": "Coder"
 			}
 		],
 		"menus": {

src/core/secretsManager.ts

@@ -4,6 +4,8 @@ const SESSION_TOKEN_KEY = "sessionToken";
 
 const LOGIN_STATE_KEY = "loginState";
 
+const OAUTH_CLIENT_REGISTRATION_KEY = "oauthClientRegistration";
+
 export enum AuthAction {
 	LOGIN,
 	LOGOUT,
@@ -70,4 +72,28 @@ export class SecretsManager {
 			}
 		});
 	}
+
+	/**
+	 * Store OAuth client registration data.
+	 */
+	public async setOAuthClientRegistration(
+		registration: string | undefined,
+	): Promise<void> {
+		if (registration) {
+			await this.secrets.store(OAUTH_CLIENT_REGISTRATION_KEY, registration);
+		} else {
+			await this.secrets.delete(OAUTH_CLIENT_REGISTRATION_KEY);
+		}
+	}
+
+	/**
+	 * Get OAuth client registration data.
+	 */
+	public async getOAuthClientRegistration(): Promise<string | undefined> {
+		try {
+			return await this.secrets.get(OAUTH_CLIENT_REGISTRATION_KEY);
+		} catch {
+			return undefined;
+		}
+	}
 }

src/extension.ts

@@ -2,7 +2,7 @@
 
 import axios, { isAxiosError } from "axios";
 import { getErrorMessage } from "coder/site/src/api/errors";
-import * as module from "module";
+import * as module from "node:module";
 import * as vscode from "vscode";
 
 import { errToStr } from "./api/api-helper";
@@ -12,6 +12,7 @@ import { Commands } from "./commands";
 import { ServiceContainer } from "./core/container";
 import { AuthAction } from "./core/secretsManager";
 import { CertificateError, getErrorDetail } from "./error";
+import { activateCoderOAuth, CALLBACK_PATH } from "./oauth";
 import { maybeAskUrl } from "./promptUtils";
 import { Remote } from "./remote/remote";
 import { toSafeHost } from "./util";
@@ -122,11 +123,22 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
 		ctx.subscriptions,
 	);
 
+	const oauthHelper = activateCoderOAuth(client, secretsManager, output, ctx);
+
 	// Handle vscode:// URIs.
 	const uriHandler = vscode.window.registerUriHandler({
 		handleUri: async (uri) => {
-			const cliManager = serviceContainer.getCliManager();
 			const params = new URLSearchParams(uri.query);
+
+			if (uri.path === CALLBACK_PATH) {
+				const code = params.get("code");
+				const state = params.get("state");
+				const error = params.get("error");
+				oauthHelper.handleCallback(code, state, error);
+				return;
+			}
+
+			const cliManager = serviceContainer.getCliManager();
 			if (uri.path === "/open") {
 				const owner = params.get("owner");
 				const workspace = params.get("workspace");