refactor: improve auth sync function placement and removed logs

OyinloluB · OyinloluB · commit f512b455d740 · 2025-07-29T16:50:58.000-04:00
Move syncAuth function and event listener to better location in code
Cleaned up logs
diff --git a/src/extension.ts b/src/extension.ts
@@ -66,6 +66,60 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
 		storage,
 	);
 
+	/**
+	 * Synchronize authentication state across all VS Code windows.
+	 * Fixes Issue #498 by ensuring consistent logout behavior when the session token changes.
+	 */
+	async function syncAuth() {
+		const url = storage.getUrl();
+		const token = await storage.getSessionToken();
+
+		// Update the REST client with current credentials
+		restClient.setHost(url || "");
+		restClient.setSessionToken(token || "");
+
+		// Determine authentication state
+		const isAuthenticated = !!(url && token);
+
+		// Update VS Code contexts to reflect current auth state
+		await vscode.commands.executeCommand(
+			"setContext",
+			"coder.authenticated",
+			isAuthenticated,
+		);
+
+		if (!isAuthenticated) {
+			// Clear owner context since user is not authenticated
+			await vscode.commands.executeCommand(
+				"setContext",
+				"coder.isOwner",
+				false,
+			);
+
+			// Show consistent logout notification across all windows
+			vscode.window
+				.showInformationMessage("You've been logged out of Coder!", "Login")
+				.then((action) => {
+					if (action === "Login") {
+						vscode.commands.executeCommand("coder.login");
+					}
+				});
+
+			vscode.commands.executeCommand("coder.refreshWorkspaces");
+		} else {
+			vscode.commands.executeCommand("coder.refreshWorkspaces");
+		}
+	}
+
+	// Listen for session token changes to sync auth state across windows
+	ctx.subscriptions.push(
+		ctx.secrets.onDidChange((e) => {
+			if (e.key === "sessionToken") {
+				syncAuth();
+			}
+		}),
+	);
+
 	const myWorkspacesProvider = new WorkspaceProvider(
 		WorkspaceQuery.Mine,
 		restClient,
@@ -412,64 +466,4 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
 			}
 		}
 	}
-
-	/**
-	 * Synchronize authentication state across all VS Code windows.
-	 * Fixes Issue #498 by ensuring consistent logout behavior when the session token changes.
-	 */
-	async function syncAuth() {
-		const url = storage.getUrl();
-		const token = await storage.getSessionToken();
-
-		storage.output.info("Auth sync called!");
-		output.info(
-			`Auth sync triggered: url=${url ? "present" : "none"}, token=${token ? "present" : "none"}`,
-		);
-
-		// Update the REST client with current credentials
-		restClient.setHost(url || "");
-		restClient.setSessionToken(token || "");
-
-		// Determine authentication state
-		const isAuthenticated = !!(url && token);
-
-		// Update VS Code contexts to reflect current auth state
-		await vscode.commands.executeCommand(
-			"setContext",
-			"coder.authenticated",
-			isAuthenticated,
-		);
-
-		if (!isAuthenticated) {
-			// Clear owner context since user is not authenticated
-			await vscode.commands.executeCommand(
-				"setContext",
-				"coder.isOwner",
-				false,
-			);
-
-			// Show consistent logout notification across all windows
-			vscode.window
-				.showInformationMessage("You've been logged out of Coder!", "Login")
-				.then((action) => {
-					if (action === "Login") {
-						vscode.commands.executeCommand("coder.login");
-					}
-				});
-
-			vscode.commands.executeCommand("coder.refreshWorkspaces");
-		} else {
-			vscode.commands.executeCommand("coder.refreshWorkspaces");
-		}
-	}
-
-	// Listen for session token changes to sync auth state across windows
-	ctx.subscriptions.push(
-		ctx.secrets.onDidChange((e) => {
-			if (e.key === "sessionToken") {
-				output.info("Session token changed, syncing auth state");
-				syncAuth();
-			}
-		}),
-	);
 }
