Recovers lost files and commits work from the past 5-6 days. Holy shit that was a close call.

mcp-server/src/core/direct-functions/expand-all-tasks.js

@@ -3,8 +3,11 @@
  */
 
 import { expandAllTasks } from '../../../../scripts/modules/task-manager.js';
-import { enableSilentMode, disableSilentMode } from '../../../../scripts/modules/utils.js';
+import { enableSilentMode, disableSilentMode, isSilentMode } from '../../../../scripts/modules/utils.js';
 import { findTasksJsonPath } from '../utils/path-utils.js';
+import { getAnthropicClientForMCP } from '../utils/ai-client-utils.js';
+import path from 'path';
+import fs from 'fs';
 
 /**
  * Expand all pending tasks with subtasks
@@ -16,43 +19,71 @@ import { findTasksJsonPath } from '../utils/path-utils.js';
  * @param {string} [args.file] - Path to the tasks file
  * @param {string} [args.projectRoot] - Project root directory
  * @param {Object} log - Logger object
+ * @param {Object} context - Context object containing session
  * @returns {Promise<{success: boolean, data?: Object, error?: {code: string, message: string}}>}
  */
-export async function expandAllTasksDirect(args, log) {
+export async function expandAllTasksDirect(args, log, context = {}) {
+  const { session } = context; // Only extract session, not reportProgress
+  
   try {
     log.info(`Expanding all tasks with args: ${JSON.stringify(args)}`);
     
-    // Find the tasks.json path
-    const tasksPath = findTasksJsonPath(args, log);
-    
-    // Parse parameters
-    const numSubtasks = args.num ? parseInt(args.num, 10) : undefined;
-    const useResearch = args.research === true;
-    const additionalContext = args.prompt || '';
-    const forceFlag = args.force === true;
-    
-    log.info(`Expanding all tasks with ${numSubtasks || 'default'} subtasks each...`);
-    if (useResearch) {
-      log.info('Using Perplexity AI for research-backed subtask generation');
-    }
-    if (additionalContext) {
-      log.info(`Additional context: "${additionalContext}"`);
-    }
-    if (forceFlag) {
-      log.info('Force regeneration of subtasks is enabled');
-    }
+    // Enable silent mode early to prevent any console output
+    enableSilentMode();
     
     try {
-      // Enable silent mode to prevent console logs from interfering with JSON response
-      enableSilentMode();
+      // Find the tasks.json path
+      const tasksPath = findTasksJsonPath(args, log);
       
-      // Call the core function
-      await expandAllTasks(numSubtasks, useResearch, additionalContext, forceFlag);
+      // Parse parameters
+      const numSubtasks = args.num ? parseInt(args.num, 10) : undefined;
+      const useResearch = args.research === true;
+      const additionalContext = args.prompt || '';
+      const forceFlag = args.force === true;
       
-      // Restore normal logging
-      disableSilentMode();
+      log.info(`Expanding all tasks with ${numSubtasks || 'default'} subtasks each...`);
       
-      // The expandAllTasks function doesn't have a return value, so we'll create our own success response
+      if (useResearch) {
+        log.info('Using Perplexity AI for research-backed subtask generation');
+        
+        // Initialize AI client for research-backed expansion
+        try {
+          await getAnthropicClientForMCP(session, log);
+        } catch (error) {
+          // Ensure silent mode is disabled before returning error
+          disableSilentMode();
+          
+          log.error(`Failed to initialize AI client: ${error.message}`);
+          return {
+            success: false,
+            error: {
+              code: 'AI_CLIENT_ERROR',
+              message: `Cannot initialize AI client: ${error.message}`
+            }
+          };
+        }
+      }
+      
+      if (additionalContext) {
+        log.info(`Additional context: "${additionalContext}"`);
+      }
+      if (forceFlag) {
+        log.info('Force regeneration of subtasks is enabled');
+      }
+      
+      // Call the core function with session context for AI operations
+      // and outputFormat as 'json' to prevent UI elements
+      const result = await expandAllTasks(
+        tasksPath,
+        numSubtasks, 
+        useResearch, 
+        additionalContext, 
+        forceFlag, 
+        { mcpLog: log, session },
+        'json' // Use JSON output format to prevent UI elements
+      );
+      
+      // The expandAllTasks function now returns a result object
       return {
         success: true,
         data: {
@@ -61,18 +92,21 @@ export async function expandAllTasksDirect(args, log) {
             numSubtasks: numSubtasks,
             research: useResearch,
             prompt: additionalContext,
-            force: forceFlag
+            force: forceFlag,
+            tasksExpanded: result.expandedCount,
+            totalEligibleTasks: result.tasksToExpand
           }
         }
       };
-    } catch (error) {
-      // Make sure to restore normal logging even if there's an error
+    } finally {
+      // Restore normal logging in finally block to ensure it runs even if there's an error
       disableSilentMode();
-      throw error; // Rethrow to be caught by outer catch block
     }
   } catch (error) {
-    // Ensure silent mode is disabled
-    disableSilentMode();
+    // Ensure silent mode is disabled if an error occurs
+    if (isSilentMode()) {
+      disableSilentMode();
+    }
     
     log.error(`Error in expandAllTasksDirect: ${error.message}`);
     return {