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/parse-prd.js

@@ -8,19 +8,39 @@ import fs from 'fs';
 import { parsePRD } from '../../../../scripts/modules/task-manager.js';
 import { findTasksJsonPath } from '../utils/path-utils.js';
 import { enableSilentMode, disableSilentMode } from '../../../../scripts/modules/utils.js';
+import { getAnthropicClientForMCP, getModelConfig } from '../utils/ai-client-utils.js';
 
 /**
  * Direct function wrapper for parsing PRD documents and generating tasks.
  * 
  * @param {Object} args - Command arguments containing input, numTasks or tasks, and output options.
  * @param {Object} log - Logger object.
+ * @param {Object} context - Context object containing session data.
  * @returns {Promise<Object>} - Result object with success status and data/error information.
  */
-export async function parsePRDDirect(args, log) {
+export async function parsePRDDirect(args, log, context = {}) {
+  const { session } = context; // Only extract session, not reportProgress
+  
   try {
     log.info(`Parsing PRD document with args: ${JSON.stringify(args)}`);
     
-    // Check required parameters
+    // Initialize AI client for PRD parsing
+    let aiClient;
+    try {
+      aiClient = getAnthropicClientForMCP(session, log);
+    } catch (error) {
+      log.error(`Failed to initialize AI client: ${error.message}`);
+      return {
+        success: false,
+        error: {
+          code: 'AI_CLIENT_ERROR',
+          message: `Cannot initialize AI client: ${error.message}`
+        },
+        fromCache: false
+      };
+    }
+    
+    // Parameter validation and path resolution
     if (!args.input) {
       const errorMessage = 'No input file specified. Please provide an input PRD document path.';
       log.error(errorMessage);
@@ -67,38 +87,54 @@ export async function parsePRDDirect(args, log) {
     
     log.info(`Preparing to parse PRD from ${inputPath} and output to ${outputPath} with ${numTasks} tasks`);
     
+    // Create the logger wrapper for proper logging in the core function
+    const logWrapper = {
+      info: (message, ...args) => log.info(message, ...args),
+      warn: (message, ...args) => log.warn(message, ...args),
+      error: (message, ...args) => log.error(message, ...args),
+      debug: (message, ...args) => log.debug && log.debug(message, ...args),
+      success: (message, ...args) => log.info(message, ...args) // Map success to info
+    };
+
+    // Get model config from session
+    const modelConfig = getModelConfig(session);
+    
     // Enable silent mode to prevent console logs from interfering with JSON response
     enableSilentMode();
-    
-    // Execute core parsePRD function (which is not async but we'll await it to maintain consistency)
-    await parsePRD(inputPath, outputPath, numTasks);
-    
-    // Restore normal logging
-    disableSilentMode();
-    
-    // Since parsePRD doesn't return a value but writes to a file, we'll read the result
-    // to return it to the caller
-    if (fs.existsSync(outputPath)) {
-      const tasksData = JSON.parse(fs.readFileSync(outputPath, 'utf8'));
-      log.info(`Successfully parsed PRD and generated ${tasksData.tasks?.length || 0} tasks`);
+    try {
+      // Execute core parsePRD function with AI client
+      await parsePRD(inputPath, outputPath, numTasks, { 
+        mcpLog: logWrapper, 
+        session 
+      }, aiClient, modelConfig);
       
-      return {
-        success: true,
-        data: {
-          message: `Successfully generated ${tasksData.tasks?.length || 0} tasks from PRD`,
-          taskCount: tasksData.tasks?.length || 0,
-          outputPath
-        },
-        fromCache: false // This operation always modifies state and should never be cached
-      };
-    } else {
-      const errorMessage = `Tasks file was not created at ${outputPath}`;
-      log.error(errorMessage);
-      return { 
-        success: false, 
-        error: { code: 'OUTPUT_FILE_NOT_CREATED', message: errorMessage },
-        fromCache: false 
-      };
+      // Since parsePRD doesn't return a value but writes to a file, we'll read the result
+      // to return it to the caller
+      if (fs.existsSync(outputPath)) {
+        const tasksData = JSON.parse(fs.readFileSync(outputPath, 'utf8'));
+        log.info(`Successfully parsed PRD and generated ${tasksData.tasks?.length || 0} tasks`);
+        
+        return {
+          success: true,
+          data: {
+            message: `Successfully generated ${tasksData.tasks?.length || 0} tasks from PRD`,
+            taskCount: tasksData.tasks?.length || 0,
+            outputPath
+          },
+          fromCache: false // This operation always modifies state and should never be cached
+        };
+      } else {
+        const errorMessage = `Tasks file was not created at ${outputPath}`;
+        log.error(errorMessage);
+        return { 
+          success: false, 
+          error: { code: 'OUTPUT_FILE_NOT_CREATED', message: errorMessage },
+          fromCache: false 
+        };
+      }
+    } finally {
+      // Always restore normal logging
+      disableSilentMode();
     }
   } catch (error) {
     // Make sure to restore normal logging even if there's an error