feat: add authentication status updates via MCP logging (#172)

* feat: add authentication status updates via MCP logging

* Remove a commented out line.

Author: allenhutchison

workspace-server/src/auth/AuthManager.ts

@@ -32,11 +32,16 @@ interface OauthWebLogin {
 export class AuthManager {
     private client: Auth.OAuth2Client | null = null;
     private scopes: string[];
+    private onStatusUpdate: ((message: string) => void) | null = null;
 
     constructor(scopes: string[]) {
         this.scopes = scopes;
     }
 
+    public setOnStatusUpdate(callback: (message: string) => void) {
+        this.onStatusUpdate = callback;
+    }
+
     private isTokenExpiringSoon(credentials: Auth.Credentials): boolean {
         return !!(credentials.expiry_date && 
             credentials.expiry_date < Date.now() + TOKEN_EXPIRY_BUFFER_MS);
@@ -156,16 +161,20 @@ export class AuthManager {
 
         const webLogin = await this.authWithWeb(oAuth2Client);
         await open(webLogin.authUrl);
-        console.log('Waiting for authentication...');
+        const msg = 'Waiting for authentication... Check your browser.';
+        logToFile(msg);
+        if (this.onStatusUpdate) {
+            this.onStatusUpdate(msg);
+        }
 
         // Add timeout to prevent infinite waiting when browser tab gets stuck
         const authTimeout = 5 * 60 * 1000; // 5 minutes timeout
         const timeoutPromise = new Promise<never>((_, reject) => {
         setTimeout(() => {
             reject(
             new Error(
-                'Authentication timed out after 5 minutes. The browser tab may have gotten stuck in a loading state. ' +
-                'Please try again.',
+                'User is not authenticated. Authentication timed out after 5 minutes. The user did not complete the login process in the browser. ' +
+                'Please ask the user to check their browser and try again.',
             ),
             );
         }, authTimeout);

workspace-server/src/index.ts

@@ -58,8 +58,21 @@ async function main() {
     }
 
     const authManager = new AuthManager(SCOPES);
-    // Trigger auth flow immediately on startup
-    await authManager.getAuthenticatedClient();
+
+    // 2. Create the server instance
+    const server = new McpServer({
+        name: "google-workspace-server",
+        version,
+    });
+
+    authManager.setOnStatusUpdate((message) => {
+        server.sendLoggingMessage({
+            level: 'info',
+            data: message
+        }).catch(err => {
+            console.error('Failed to send logging message:', err);
+        });
+    });
 
     const driveService = new DriveService(authManager);
     const docsService = new DocsService(authManager, driveService);
@@ -71,12 +84,7 @@ async function main() {
     const slidesService = new SlidesService(authManager);
     const sheetsService = new SheetsService(authManager);
 
-    // 2. Create the server instance
-    const server = new McpServer({
-        name: "google-workspace-server",
-        version,
-    });
-
+    
     // 3. Register tools directly on the server
     server.registerTool(
         "auth.clear",