midas-mcp 2.9.0 → 3.2.0

package/dist/tui.js

@@ -375,6 +375,14 @@ export async function runInteractive() {
         console.error('Uncaught exception:', err);
         process.exit(1);
     });
+    process.on('unhandledRejection', (reason) => {
+        cleanup();
+        console.error('Unhandled rejection:', reason);
+        process.exit(1);
+    });
+    // Debounce tracking for rapid key presses
+    let lastKeyTime = 0;
+    const DEBOUNCE_MS = 50;
     // Start a new session for metrics tracking
     const currentPhase = loadState(projectPath).current;
     const sessionId = startSession(projectPath, currentPhase);
@@ -398,14 +406,30 @@ export async function runInteractive() {
         suggestionAcceptanceRate: getSuggestionAcceptanceRate(projectPath),
     };
     const render = () => {
-        // Hide cursor, clear, draw, show cursor - prevents flicker
-        process.stdout.write(hideCursor + clearScreen);
-        process.stdout.write(drawUI(tuiState, projectPath));
-        if (tuiState.message) {
-            process.stdout.write(`\n  ${tuiState.message}`);
-            tuiState.message = '';
+        try {
+            // Hide cursor, clear, draw, show cursor - prevents flicker
+            process.stdout.write(hideCursor + clearScreen);
+            process.stdout.write(drawUI(tuiState, projectPath));
+            if (tuiState.message) {
+                process.stdout.write(`\n  ${tuiState.message}`);
+                tuiState.message = '';
+            }
+            process.stdout.write(showCursor);
+        }
+        catch (error) {
+            // Attempt recovery
+            process.stdout.write(clearScreen + showCursor);
+            console.error('Render error:', error);
         }
-        process.stdout.write(showCursor);
+    };
+    // Reset state to recover from bad state
+    const resetState = () => {
+        tuiState.analysis = null;
+        tuiState.isAnalyzing = false;
+        tuiState.showingSessionStart = false;
+        tuiState.showingRejectionInput = false;
+        tuiState.message = `${yellow}!${reset} State reset. Press [r] to re-analyze.`;
+        tuiState.filesChanged = false;
     };
     const runAnalysis = async () => {
         if (!tuiState.hasApiKey) {
@@ -445,6 +469,7 @@ export async function runInteractive() {
         render();
     };
     const promptForRejectionReason = async () => {
+        const TIMEOUT_MS = 60000; // 60 second timeout
         return new Promise((resolve) => {
             console.log(`\n  ${yellow}Why are you declining this suggestion?${reset}`);
             console.log(`  ${dim}(This helps Midas learn. Press Enter to skip.)${reset}\n`);
@@ -455,7 +480,16 @@ export async function runInteractive() {
                 input: process.stdin,
                 output: process.stdout,
             });
+            // Timeout protection
+            const timeout = setTimeout(() => {
+                rl.close();
+                if (process.stdin.isTTY) {
+                    process.stdin.setRawMode(true);
+                }
+                resolve(''); // Timeout = no reason given
+            }, TIMEOUT_MS);
             rl.question('  > ', (answer) => {
+                clearTimeout(timeout);
                 rl.close();
                 if (process.stdin.isTTY) {
                     process.stdin.setRawMode(true);
@@ -466,12 +500,12 @@ export async function runInteractive() {
     };
     // Multi-line input for pasting AI responses
     const promptForResponse = async () => {
+        const TIMEOUT_MS = 120000; // 2 minute timeout for pasting
         return new Promise((resolve) => {
             // Exit alternate screen temporarily for input
             process.stdout.write(exitAltScreen);
             console.log(`\n  ${cyan}━━━ Paste AI Response ━━━${reset}`);
-            console.log(`  ${dim}Paste the exchange below. When done, type ${bold}END${reset}${dim} on a new line and press Enter.${reset}\n`);
-            console.log(`  ${dim}Format: First paste your prompt, then the AI response.${reset}\n`);
+            console.log(`  ${dim}Paste the response, then press Enter twice to submit.${reset}\n`);
             if (process.stdin.isTTY) {
                 process.stdin.setRawMode(false);
             }
@@ -480,48 +514,76 @@ export async function runInteractive() {
                 output: process.stdout,
             });
             const lines = [];
-            console.log(`  ${dim}--- Start pasting (type END when done) ---${reset}\n`);
-            rl.on('line', (line) => {
-                if (line.trim().toUpperCase() === 'END') {
-                    rl.close();
-                    if (process.stdin.isTTY) {
-                        process.stdin.setRawMode(true);
+            let emptyLineCount = 0;
+            let timeoutId = null;
+            // Timeout protection - auto-cancel after 2 minutes
+            timeoutId = setTimeout(() => {
+                rl.close();
+                if (process.stdin.isTTY) {
+                    process.stdin.setRawMode(true);
+                }
+                process.stdout.write(enterAltScreen);
+                resolve(null);
+            }, TIMEOUT_MS);
+            const finishInput = () => {
+                if (timeoutId)
+                    clearTimeout(timeoutId);
+                rl.close();
+                if (process.stdin.isTTY) {
+                    process.stdin.setRawMode(true);
+                }
+                process.stdout.write(enterAltScreen);
+                // Remove trailing empty lines
+                while (lines.length > 0 && lines[lines.length - 1].trim() === '') {
+                    lines.pop();
+                }
+                const fullText = lines.join('\n');
+                if (fullText.trim().length < 10) {
+                    resolve(null);
+                    return;
+                }
+                // Try to split into user prompt and AI response
+                const splitPatterns = [
+                    /\n(?:assistant|ai|claude|cursor):\s*/i,
+                    /\n(?:response|answer):\s*/i,
+                    /\n---+\n/,
+                ];
+                let userPrompt = '';
+                let aiResponse = fullText;
+                for (const pattern of splitPatterns) {
+                    const match = fullText.match(pattern);
+                    if (match && match.index) {
+                        userPrompt = fullText.slice(0, match.index).trim();
+                        aiResponse = fullText.slice(match.index + match[0].length).trim();
+                        break;
                     }
-                    // Re-enter alternate screen
-                    process.stdout.write(enterAltScreen);
-                    const fullText = lines.join('\n');
-                    if (fullText.trim().length < 10) {
-                        resolve(null);
+                }
+                // If no split found, treat first 20% as prompt
+                if (!userPrompt && fullText.length > 100) {
+                    const splitPoint = Math.min(500, Math.floor(fullText.length * 0.2));
+                    userPrompt = fullText.slice(0, splitPoint).trim();
+                    aiResponse = fullText.slice(splitPoint).trim();
+                }
+                resolve({ userPrompt, aiResponse });
+            };
+            rl.on('line', (line) => {
+                // Check for double Enter (two empty lines in a row)
+                if (line.trim() === '') {
+                    emptyLineCount++;
+                    if (emptyLineCount >= 2 && lines.length > 0) {
+                        finishInput();
                         return;
                     }
-                    // Try to split into user prompt and AI response
-                    // Look for common patterns
-                    const splitPatterns = [
-                        /\n(?:assistant|ai|claude|cursor):\s*/i,
-                        /\n(?:response|answer):\s*/i,
-                        /\n---+\n/,
-                    ];
-                    let userPrompt = '';
-                    let aiResponse = fullText;
-                    for (const pattern of splitPatterns) {
-                        const match = fullText.match(pattern);
-                        if (match && match.index) {
-                            userPrompt = fullText.slice(0, match.index).trim();
-                            aiResponse = fullText.slice(match.index + match[0].length).trim();
-                            break;
-                        }
-                    }
-                    // If no split found, treat first 20% as prompt
-                    if (!userPrompt && fullText.length > 100) {
-                        const splitPoint = Math.min(500, Math.floor(fullText.length * 0.2));
-                        userPrompt = fullText.slice(0, splitPoint).trim();
-                        aiResponse = fullText.slice(splitPoint).trim();
-                    }
-                    resolve({ userPrompt, aiResponse });
                 }
                 else {
-                    lines.push(line);
+                    emptyLineCount = 0;
+                }
+                // Also support "END" for those who prefer explicit termination
+                if (line.trim().toUpperCase() === 'END' && lines.length > 0) {
+                    finishInput();
+                    return;
                 }
+                lines.push(line);
             });
             rl.on('close', () => {
                 if (process.stdin.isTTY) {
@@ -573,203 +635,224 @@ export async function runInteractive() {
         }
     });
     process.stdin.on('data', async (key) => {
-        if (key === 'q' || key === '\u0003') {
-            // End session and save metrics
-            const endPhase = tuiState.analysis?.currentPhase || { phase: 'IDLE' };
-            endSession(projectPath, tuiState.sessionId, endPhase);
-            clearInterval(activityInterval);
-            stopWatchingEvents();
-            cleanup(); // Restore terminal state
-            console.log(`\n  ${cyan}Midas${reset} signing off. Session saved. Happy vibecoding!\n`);
-            process.exit(0);
-        }
-        // Session start screen handling
-        if (tuiState.showingSessionStart) {
-            if (key === 'c') {
-                try {
-                    await copyToClipboard(tuiState.sessionStarterPrompt);
-                    tuiState.message = `${green}OK${reset} Session starter copied! Paste it in your new Cursor chat.`;
-                }
-                catch {
-                    tuiState.message = `${yellow}!${reset} Could not copy.`;
-                }
+        // Debounce rapid key presses
+        const now = Date.now();
+        if (now - lastKeyTime < DEBOUNCE_MS)
+            return;
+        lastKeyTime = now;
+        try {
+            // Shift+R for hard reset (recovery from bad state)
+            if (key === 'R') {
+                resetState();
                 render();
                 return;
             }
-            if (key === 'p') {
-                tuiState.showingSessionStart = false;
-                render();
-                if (tuiState.hasApiKey) {
-                    await runAnalysis();
-                }
-                return;
+            if (key === 'q' || key === '\u0003') {
+                // End session and save metrics
+                const endPhase = tuiState.analysis?.currentPhase || { phase: 'IDLE' };
+                endSession(projectPath, tuiState.sessionId, endPhase);
+                clearInterval(activityInterval);
+                stopWatchingEvents();
+                cleanup(); // Restore terminal state
+                console.log(`\n  ${cyan}Midas${reset} signing off. Session saved. Happy vibecoding!\n`);
+                process.exit(0);
             }
-            if (key === 'u') {
-                try {
-                    await copyUserRules();
-                    tuiState.message = `${green}OK${reset} User Rules copied! Paste in Cursor Settings -> Rules for AI`;
-                }
-                catch {
-                    tuiState.message = `${yellow}!${reset} Could not copy.`;
+            // Session start screen handling
+            if (tuiState.showingSessionStart) {
+                if (key === 'c') {
+                    try {
+                        await copyToClipboard(tuiState.sessionStarterPrompt);
+                        tuiState.message = `${green}OK${reset} Session starter copied! Paste it in your new Cursor chat.`;
+                    }
+                    catch {
+                        tuiState.message = `${yellow}!${reset} Could not copy.`;
+                    }
+                    render();
+                    return;
                 }
-                render();
-                return;
-            }
-            return; // Ignore other keys on session start screen
-        }
-        if (key === 'c') {
-            if (tuiState.analysis?.suggestedPrompt) {
-                try {
-                    await copyToClipboard(tuiState.analysis.suggestedPrompt);
-                    tuiState.message = `${green}OK${reset} Prompt copied to clipboard!`;
-                    logEvent(projectPath, { type: 'prompt_copied', message: tuiState.analysis.suggestedPrompt.slice(0, 100) });
-                    recordPromptCopied(projectPath, tuiState.sessionId);
+                if (key === 'p') {
+                    tuiState.showingSessionStart = false;
+                    render();
+                    if (tuiState.hasApiKey) {
+                        await runAnalysis();
+                    }
+                    return;
                 }
-                catch {
-                    tuiState.message = `${yellow}!${reset} Could not copy.`;
+                if (key === 'u') {
+                    try {
+                        await copyUserRules();
+                        tuiState.message = `${green}OK${reset} User Rules copied! Paste in Cursor Settings -> Rules for AI`;
+                    }
+                    catch {
+                        tuiState.message = `${yellow}!${reset} Could not copy.`;
+                    }
+                    render();
+                    return;
                 }
+                return; // Ignore other keys on session start screen
             }
-            else {
-                tuiState.message = `${yellow}!${reset} No prompt to copy.`;
-            }
-            render();
-        }
-        if (key === 'r') {
-            await runAnalysis();
-        }
-        if (key === 'x') {
-            // Decline/reject the current suggestion
-            if (tuiState.analysis?.suggestedPrompt) {
-                tuiState.showingRejectionInput = true;
-                const reason = await promptForRejectionReason();
-                tuiState.showingRejectionInput = false;
-                recordSuggestionOutcome(projectPath, false, undefined, reason || undefined);
-                tuiState.suggestionAcceptanceRate = getSuggestionAcceptanceRate(projectPath);
-                if (reason) {
-                    tuiState.message = `${yellow}OK${reset} Noted: "${truncate(reason, 40)}" - will learn from this.`;
+            if (key === 'c') {
+                if (tuiState.analysis?.suggestedPrompt) {
+                    try {
+                        await copyToClipboard(tuiState.analysis.suggestedPrompt);
+                        tuiState.message = `${green}OK${reset} Prompt copied to clipboard!`;
+                        logEvent(projectPath, { type: 'prompt_copied', message: tuiState.analysis.suggestedPrompt.slice(0, 100) });
+                        recordPromptCopied(projectPath, tuiState.sessionId);
+                    }
+                    catch {
+                        tuiState.message = `${yellow}!${reset} Could not copy.`;
+                    }
                 }
                 else {
-                    tuiState.message = `${yellow}OK${reset} Suggestion declined.`;
+                    tuiState.message = `${yellow}!${reset} No prompt to copy.`;
                 }
-                logEvent(projectPath, {
-                    type: 'prompt_copied',
-                    message: 'Suggestion declined',
-                    data: { reason: reason || 'No reason given' }
-                });
                 render();
             }
-            else {
-                tuiState.message = `${yellow}!${reset} No suggestion to decline.`;
-                render();
+            if (key === 'r') {
+                await runAnalysis();
             }
-        }
-        if (key === 'v') {
-            // Run verification gates
-            tuiState.message = `${magenta}...${reset} Running verification gates (build, test, lint)...`;
-            render();
-            try {
-                const { verify } = await import('./tools/verify.js');
-                const result = verify({ projectPath });
-                tuiState.gatesStatus = getGatesStatus(projectPath);
-                if (result.allPass) {
-                    tuiState.message = `${green}OK${reset} All gates pass!${result.autoAdvanced ? ` Auto-advanced to ${result.autoAdvanced.to}` : ''}`;
-                    // Re-analyze to get new suggestions
-                    await runAnalysis();
+            if (key === 'x') {
+                // Decline/reject the current suggestion
+                if (tuiState.analysis?.suggestedPrompt) {
+                    tuiState.showingRejectionInput = true;
+                    const reason = await promptForRejectionReason();
+                    tuiState.showingRejectionInput = false;
+                    recordSuggestionOutcome(projectPath, false, undefined, reason || undefined);
+                    tuiState.suggestionAcceptanceRate = getSuggestionAcceptanceRate(projectPath);
+                    if (reason) {
+                        tuiState.message = `${yellow}OK${reset} Noted: "${truncate(reason, 40)}" - will learn from this.`;
+                    }
+                    else {
+                        tuiState.message = `${yellow}OK${reset} Suggestion declined.`;
+                    }
+                    logEvent(projectPath, {
+                        type: 'prompt_copied',
+                        message: 'Suggestion declined',
+                        data: { reason: reason || 'No reason given' }
+                    });
+                    render();
                 }
                 else {
-                    tuiState.message = `${red}!${reset} Failing: ${result.failing.join(', ')}`;
+                    tuiState.message = `${yellow}!${reset} No suggestion to decline.`;
                     render();
                 }
             }
-            catch (error) {
-                tuiState.message = `${red}!${reset} Verification failed.`;
+            if (key === 'v') {
+                // Run verification gates
+                tuiState.message = `${magenta}...${reset} Running verification gates (build, test, lint)...`;
                 render();
-            }
-        }
-        if (key === 'a') {
-            // Add .cursorrules to project
-            try {
-                const { writeCursorRules, detectTechStack } = await import('./techstack.js');
-                const projectName = projectPath.split('/').pop() || 'project';
-                const result = writeCursorRules(projectPath, projectName);
-                if (result.success) {
-                    const stack = detectTechStack(projectPath);
-                    tuiState.message = `${green}OK${reset} Created .cursorrules for ${stack.language}${stack.framework ? `/${stack.framework}` : ''}`;
+                try {
+                    const { verify } = await import('./tools/verify.js');
+                    const result = verify({ projectPath });
+                    tuiState.gatesStatus = getGatesStatus(projectPath);
+                    if (result.allPass) {
+                        tuiState.message = `${green}OK${reset} All gates pass!${result.autoAdvanced ? ` Auto-advanced to ${result.autoAdvanced.to}` : ''}`;
+                        // Re-analyze to get new suggestions
+                        await runAnalysis();
+                    }
+                    else {
+                        tuiState.message = `${red}!${reset} Failing: ${result.failing.join(', ')}`;
+                        render();
+                    }
                 }
-                else {
-                    tuiState.message = `${red}!${reset} Failed to create .cursorrules`;
+                catch (error) {
+                    tuiState.message = `${red}!${reset} Verification failed.`;
+                    render();
                 }
             }
-            catch (error) {
-                tuiState.message = `${red}!${reset} Error creating .cursorrules`;
-            }
-            render();
-        }
-        if (key === 'i') {
-            // Input/paste AI response for analysis
-            const exchange = await promptForResponse();
-            if (!exchange) {
-                tuiState.message = `${yellow}!${reset} No response pasted.`;
-                render();
-                return;
-            }
-            tuiState.message = `${magenta}...${reset} Analyzing response...`;
-            render();
-            try {
-                // Analyze the pasted response
-                const analysis = await analyzeResponse(projectPath, exchange.userPrompt, exchange.aiResponse);
-                // Auto-save to journal
-                const journalTitle = analysis.summary.slice(0, 80);
-                saveToJournal({
-                    projectPath,
-                    title: journalTitle,
-                    conversation: `USER:\n${exchange.userPrompt}\n\nAI:\n${exchange.aiResponse}`,
-                    tags: analysis.taskComplete ? ['completed'] : ['in-progress'],
-                });
-                // Record any errors to error memory
-                if (analysis.errors.length > 0) {
-                    const { recordError } = await import('./tracker.js');
-                    for (const err of analysis.errors) {
-                        recordError(projectPath, err, undefined, undefined);
+            if (key === 'a') {
+                // Add .cursorrules to project
+                try {
+                    const { writeCursorRules, detectTechStack } = await import('./techstack.js');
+                    const projectName = projectPath.split('/').pop() || 'project';
+                    const result = writeCursorRules(projectPath, projectName);
+                    if (result.success) {
+                        const stack = detectTechStack(projectPath);
+                        tuiState.message = `${green}OK${reset} Created .cursorrules for ${stack.language}${stack.framework ? `/${stack.framework}` : ''}`;
                     }
+                    else {
+                        tuiState.message = `${red}!${reset} Failed to create .cursorrules`;
+                    }
+                }
+                catch (error) {
+                    tuiState.message = `${red}!${reset} Error creating .cursorrules`;
                 }
-                // Log event
-                logEvent(projectPath, {
-                    type: 'ai_suggestion',
-                    message: `Analyzed: ${analysis.summary}`,
-                    data: {
-                        accomplished: analysis.accomplished.length,
-                        errors: analysis.errors.length,
-                        taskComplete: analysis.taskComplete,
-                    },
-                });
-                // Update analysis with new suggested prompt
-                if (tuiState.analysis) {
-                    tuiState.analysis.suggestedPrompt = analysis.suggestedNextPrompt;
-                    tuiState.analysis.whatsNext = analysis.suggestedNextPrompt;
+                render();
+            }
+            if (key === 'i') {
+                // Input/paste AI response for analysis
+                const exchange = await promptForResponse();
+                if (!exchange) {
+                    tuiState.message = `${yellow}!${reset} No response pasted.`;
+                    render();
+                    return;
                 }
-                // Show summary
-                const accomplishedCount = analysis.accomplished.length;
-                const errorsCount = analysis.errors.length;
-                let statusMsg = `${green}OK${reset} Analyzed: ${analysis.summary.slice(0, 40)}`;
-                if (accomplishedCount > 0)
-                    statusMsg += ` | ${green}${accomplishedCount} done${reset}`;
-                if (errorsCount > 0)
-                    statusMsg += ` | ${red}${errorsCount} errors${reset}`;
-                tuiState.message = statusMsg;
-                tuiState.filesChanged = true; // Trigger refresh prompt
-                // Re-analyze if task complete or errors found
-                if (analysis.taskComplete || analysis.shouldAdvancePhase) {
-                    await runAnalysis();
+                tuiState.message = `${magenta}...${reset} Analyzing response...`;
+                render();
+                try {
+                    // Analyze the pasted response
+                    const analysis = await analyzeResponse(projectPath, exchange.userPrompt, exchange.aiResponse);
+                    // Auto-save to journal
+                    const journalTitle = analysis.summary.slice(0, 80);
+                    saveToJournal({
+                        projectPath,
+                        title: journalTitle,
+                        conversation: `USER:\n${exchange.userPrompt}\n\nAI:\n${exchange.aiResponse}`,
+                        tags: analysis.taskComplete ? ['completed'] : ['in-progress'],
+                    });
+                    // Record any errors to error memory
+                    if (analysis.errors.length > 0) {
+                        const { recordError } = await import('./tracker.js');
+                        for (const err of analysis.errors) {
+                            recordError(projectPath, err, undefined, undefined);
+                        }
+                    }
+                    // Log event
+                    logEvent(projectPath, {
+                        type: 'ai_suggestion',
+                        message: `Analyzed: ${analysis.summary}`,
+                        data: {
+                            accomplished: analysis.accomplished.length,
+                            errors: analysis.errors.length,
+                            taskComplete: analysis.taskComplete,
+                        },
+                    });
+                    // Update analysis with new suggested prompt
+                    if (tuiState.analysis) {
+                        tuiState.analysis.suggestedPrompt = analysis.suggestedNextPrompt;
+                        tuiState.analysis.whatsNext = analysis.suggestedNextPrompt;
+                    }
+                    // Show summary
+                    const accomplishedCount = analysis.accomplished.length;
+                    const errorsCount = analysis.errors.length;
+                    let statusMsg = `${green}OK${reset} Analyzed: ${analysis.summary.slice(0, 40)}`;
+                    if (accomplishedCount > 0)
+                        statusMsg += ` | ${green}${accomplishedCount} done${reset}`;
+                    if (errorsCount > 0)
+                        statusMsg += ` | ${red}${errorsCount} errors${reset}`;
+                    tuiState.message = statusMsg;
+                    tuiState.filesChanged = true; // Trigger refresh prompt
+                    // Re-analyze if task complete or errors found
+                    if (analysis.taskComplete || analysis.shouldAdvancePhase) {
+                        await runAnalysis();
+                    }
+                    else {
+                        render();
+                    }
                 }
-                else {
+                catch (error) {
+                    tuiState.message = `${red}!${reset} Failed to analyze response.`;
                     render();
                 }
             }
-            catch (error) {
-                tuiState.message = `${red}!${reset} Failed to analyze response.`;
+        }
+        catch (error) {
+            // Global error handler for key processing
+            tuiState.message = `${red}!${reset} Error processing key. Press Shift+R to reset.`;
+            try {
                 render();
             }
+            catch { /* ignore render errors */ }
         }
     });
 }