Add token refresh and revocation

Author: EhabY

package.json

@@ -257,8 +257,13 @@
 				"icon": "$(search)"
 			},
 			{
-				"command": "coder.oauth.testAuth",
-				"title": "Test OAuth Auth",
+				"command": "coder.oauth.login",
+				"title": "OAuth Login",
+				"category": "Coder"
+			},
+			{
+				"command": "coder.oauth.logout",
+				"title": "OAuth Logout",
 				"category": "Coder"
 			}
 		],

src/core/secretsManager.ts

@@ -1,4 +1,7 @@
-import { type ClientRegistrationResponse } from "../oauth/types";
+import {
+	type TokenResponse,
+	type ClientRegistrationResponse,
+} from "../oauth/types";
 
 import type { SecretStorage, Disposable } from "vscode";
 
@@ -8,6 +11,12 @@ const LOGIN_STATE_KEY = "loginState";
 
 const OAUTH_CLIENT_REGISTRATION_KEY = "oauthClientRegistration";
 
+const OAUTH_TOKENS_KEY = "oauthTokens";
+
+export type StoredOAuthTokens = Omit<TokenResponse, "expires_in"> & {
+	expiry_timestamp: number;
+};
+
 export enum AuthAction {
 	LOGIN,
 	LOGOUT,
@@ -109,4 +118,39 @@ export class SecretsManager {
 		}
 		return undefined;
 	}
+
+	/**
+	 * Store OAuth token data including expiry timestamp.
+	 */
+	public async setOAuthTokens(
+		tokens: StoredOAuthTokens | undefined,
+	): Promise<void> {
+		if (tokens) {
+			await this.secrets.store(OAUTH_TOKENS_KEY, JSON.stringify(tokens));
+		} else {
+			await this.secrets.delete(OAUTH_TOKENS_KEY);
+		}
+	}
+
+	/**
+	 * Get stored OAuth token data.
+	 */
+	public async getOAuthTokens(): Promise<StoredOAuthTokens | undefined> {
+		try {
+			const stringifiedTokens = await this.secrets.get(OAUTH_TOKENS_KEY);
+			if (stringifiedTokens) {
+				return JSON.parse(stringifiedTokens) as StoredOAuthTokens;
+			}
+		} catch {
+			// Do nothing
+		}
+		return undefined;
+	}
+
+	/**
+	 * Clear OAuth token data.
+	 */
+	public async clearOAuthTokens(): Promise<void> {
+		await this.secrets.delete(OAUTH_TOKENS_KEY);
+	}
 }