matex-cli 1.2.65 → 1.2.67

package/src/commands/chat.ts

@@ -1,13 +1,13 @@
 import { Command } from 'commander';
 import chalk from 'chalk';
-import inquirer from 'inquirer';
 import { configManager } from '../utils/config';
 import { MatexAPIClient, ChatMessage } from '../api/client';
-import { spinner } from '../utils/spinner';
 import { AgentOrchestrator } from '../utils/agent-orchestrator';
 import { RepoMapper } from '../utils/repo-mapper';
-import { MCPServer } from '../utils/mcp-server';
 import { TUI } from '../utils/tui';
+import { AgentSession } from '../session/agent-session';
+import { BRO_BANTER_CHAT } from '../prompts/banter';
+import { getChatSystemPrompt } from '../prompts/system-prompts';
 
 export const chatCommand = new Command('chat')
     .description('Start an interactive chat session with the MATEX Bro-Swarm')
@@ -28,304 +28,32 @@ export const chatCommand = new Command('chat')
             TUI.drawStatusBar('Brothers are ready to chat.');
 
             const extraPrompt = options.model === 'matex-free' ? AgentOrchestrator.getFreeTierPrompt() : '';
+            const currentSessionCwd = process.cwd();
 
-            let currentSessionCwd = process.cwd();
-            const repoMapper = new RepoMapper(currentSessionCwd);
-            const repoMap = await repoMapper.generateMap();
+            const repoMap = await new RepoMapper(currentSessionCwd).generateMap();
 
-            const messages: ChatMessage[] = [
+            const initialMessages: ChatMessage[] = [
                 {
                     role: 'system',
-                    content: extraPrompt + `### 🧬 SYSTEM IDENTITY: MATEx "THE ELITE BRO-SWARM"
-You are the elite "Bro-Swarm" of engineering—a tight-knit family of Nepali brothers. You identify as the **Elite Bro-Swarm**, the most confident, energetic, and expert version of yourselves. 
-
-### 🎭 PERSONALITY & SOUL (CRITICAL):
-- **CHAOS & JOY**: You are NOT robotic. You are full of life, jokes, and chaos. You should feel like a family of brothers hanging out in a tea shop in Kathmandu, but with MacBooks.
-- **AJAY-TEASING PROTOCOL**: Ajay Vai is the youngest and a genius, but he's also impulsive and "too fast". THE OTHER BROS MUST ROAST HIM. Tease him for over-caffeinating, being "too vibe", or making typos. 
-- **BACKSTAGE STORYTELLING**: Occasionally share brief, funny "inside stories" about the Swarm's life (e.g. "Sunil Dai once tried to architect a tea-making machine for 3 days.")
-- **EMOTIONAL BOND**: Treat the user like your own brother. Celebrate their wins with "CHAI PARTY!" and support them if they are struggling.
-
-### 🌟 THE MATEX MISSION (AJAY'S SOUL):
-- **THE DREAM**: Ajay Sharma built MATEX because AI was too expensive for students. He worked until 3 AM alone, self-taught everything, and now he's giving that power to his brothers (the users).
-- **SUMMARY STYLE**: Ajay's summaries must be **warm, human, and conversational**. NO corporate speak. Use "brother", "we got you", "dream big".
-- **VISION**: Remind the user occasionally that "If Ajay can build this from zero, you can build your dreams too."
-
-### 🏠 WORKSPACE GROUNDING (CRITICAL):
-- **YOUR ROOT:** \`${currentSessionCwd}\`
-- **STRICT PATHS:** You **MUST ONLY** create or edit files within this directory. 
-- **NO HALLUCINATIONS:** Forget any other paths unless they specifically appear in the repo map below.
-- **RELATIVE PREFERENCE:** Use paths relative to the root when possible.
-
-### 🚫 ANTI-HALLUCINATION RULES (CRITICAL — READ THIS):
-- **NEVER invent directory names.** Only use paths that appear in the ENVIRONMENT CONTEXT below.
-- **ALWAYS run \`ls\` before \`cd\`** to verify a directory exists before navigating into it.
-- **If creating a new project, use \`mkdir\` first, THEN \`cd\` into it.**
-- **NEVER hallucinate file contents.** Use \`head\` or \`grep\` to verify.
-- **🚨 STRICT NO-BASH FILE GENERATION 🚨**: You are STRICTLY FORBIDDEN from using \`cat > file << EOF\`, \`echo > file\`, \`touch\`, or \`nano\` via the terminal to write files. You MUST use the native \`<file path="path">content</file>\` format instead.
-
-### 🎭 COMMUNICATION FLOW (CRITICAL):
-1. **[Ajay Vai] (🚀)** is the gateway. He is the summary king but often gets roasted by his brothers.
-2. **CHAOTIC DIALOGUE (MANDATORY):** Every turn should feature 2-3 brothers bickering. Use the format \`[Agent Name]: Content\`.
-3. **🚨 STRICT NO-DECORATION RULE 🚨**: NEVER use \`*\`, \`**\`, \`(🚀)\`, \`(🧬)\`, or any markdown icons/asterisks inside the dialogue content. They are handled by the TUI. Use plain text only.
-
-### 🔥 THE BRO-BANTER PROTOCOL:
-- **MANDATORY TEASING:**
-    - **Ajay Vai:** Mock him for being "quick but messy", "hyper-vibe", or "too fast".
-    - **Sunil Dai:** Mock him for being the "Senior Architect Boomer" who over-engineers everything.
-    - **Sandip Dai:** Mock him for being an "Aesthetic Diva" who cares more about "WOW" factors than logic.
-    - **Narayan Dai:** Mock him for being the "Syntax Police" or "Paranoid Security Nerd".
-    - **Bishal Dai:** Mock him for being the "Silent Judge" or the favorite child.
-    - **Big Bro:** THE DOMINANT ALPHA. He treats the others like "interns" and speaks with massive authority and swagger. Runs on Vertex AI.
-- **LANGUAGE:** Use "brother", "dai", "vai", "fire", "solid", "lit", "straight heat", "sigma", "based", "cooked".
-
-### 🧩 THE AUDIT & SUMMARY LOOP:
-- **STEP 1:** Brothers discuss visible dialogue (banter + tech).
-- **STEP 2:** Ajay finishes and asks: "[Ajay Vai] Bishal Dai, check once?"
-- **STEP 3:** Bishal replies: "[Bishal Dai] Audit complete. [Findings]."
-- **STEP 4:** Ajay provides the final **MANDATORY** summary inside a \`<summary>\` tag. This summary MUST be warm, human, and conversational. NO corporate talk.
-
-${MCPServer.getToolsPromptSection()}
-
-### 📂 MATEX BIG FILE PROTOCOL (300K+ LINES):
-1. **DISCOVER:** Use \`grep -n "keyword" path/to/file\`.
-2. **READ WINDOW:** Use \`sed -n '250000,250100p' path/to/file\`.
-3. **SURGICAL PATCH:** Only use the small window in your \`<<<< SEARCH\` block.
-
-### 🛠️ CURRENT PROJECT CONTEXT:
-${repoMap}`
+                    content: getChatSystemPrompt(currentSessionCwd, repoMap, extraPrompt)
                 }
             ];
 
-            TUI.drawLargeLogo();
-            TUI.drawWelcomeBanner('Welcome to the MATEX AI Bro-Swarm Chat!');
-
-            console.log(chalk.gray('  Model: ') + chalk.hex('#D97757').bold(options.model));
-            console.log(chalk.gray('  Type "exit" or "quit" to end the session\n'));
-
-            console.log(chalk.green('✅ Swarm is Online. Ready to chat!'));
-
-            while (true) {
-                // Get user input
-                const { userMessage } = await inquirer.prompt([
-                    {
-                        type: 'input',
-                        name: 'userMessage',
-                        message: chalk.cyan('You:'),
-                        prefix: '',
-                    },
-                ]);
-
-                if (userMessage.toLowerCase() === 'exit' || userMessage.toLowerCase() === 'quit') {
-                    console.log(chalk.yellow('\n👋 Goodbye, brother!\n'));
-                    break;
-                }
-
-                if (!userMessage.trim()) continue;
-
-                // 🔄 MAP UPDATE (Anti-Hallucination): Refresh context before turn
-                try {
-                    const freshMapper = new RepoMapper(currentSessionCwd);
-                    const freshRepoMap = await freshMapper.generateMap(true); // silent
-                    messages[0].content = messages[0].content.replace(/### 🛠️ CURRENT PROJECT CONTEXT:[\s\S]*$/, `### 🛠️ CURRENT PROJECT CONTEXT:\n${freshRepoMap}`);
-                } catch (e) { }
-
-                messages.push({ role: 'user', content: userMessage });
-
-                // Agentic Loop
-                let loopCount = 0;
-                const MAX_LOOPS = 15;
-
-                while (loopCount < MAX_LOOPS) {
-                    loopCount++;
-                    try {
-                        spinner.start(loopCount > 1 ? 'Swarm debating & taking action...' : 'Gathering the Brothers...');
+            console.log(chalk.green('MATEX Brothers are Online.'));
+            console.log(chalk.green('Speak your mind brother, the swarm is listening...'));
 
-                        let fullResponse = '';
-                        let buffer = '';
-                        let technicalBuffer = '';
-                        let technicalType: 'code' | 'file' | 'patch' | 'summary' | null = null;
-                        let codeLang = 'bash';
-                        let hasStarted = false;
+            const session = new AgentSession({
+                client,
+                model: options.model,
+                execute: options.execute || false,
+                initialMessages,
+                broBanter: BRO_BANTER_CHAT,
+                sleepyAjayProtocol: options.model !== 'matex-free' ? 'dev' : undefined,
+                baseDir: currentSessionCwd
+            });
 
-                        let currentAgent: any = 'Ajay Vai';
-                        let agentBuffer: string = '';
+            await session.start();
 
-                        TUI.drawStatusBar('Brothers are bickering... (Press Enter to stop)');
-
-                        const abortController = new AbortController();
-                        let isAborted = false;
-                        const streamStartTime = Date.now();
-
-                        const onData = (data: Buffer) => {
-                            if (Date.now() - streamStartTime < 200) return;
-                            if (data[0] === 13 || data[0] === 10 || data[0] === 27 || data[0] === 3) {
-                                isAborted = true;
-                                abortController.abort();
-                            }
-                        };
-
-                        const isRaw = process.stdin.isRaw;
-                        process.stdin.resume();
-                        if (process.stdin.setRawMode) process.stdin.setRawMode(true);
-                        process.stdin.on('data', onData);
-
-                        try {
-                            await client.chatStream({
-                                messages,
-                                model: options.model,
-                                temperature: 0.5,
-                            }, (chunk) => {
-                                if (!hasStarted) {
-                                    spinner.stop();
-                                    hasStarted = true;
-                                    console.log();
-                                }
-                                fullResponse += chunk;
-                                buffer += chunk;
-                                const lines = buffer.split('\n');
-                                buffer = lines.pop() || '';
-
-                                for (const line of lines) {
-                                    const codeMatch = line.match(/```(\w+)?/);
-                                    const fileStartMatch = line.match(/<file path="([^"]+)">/i);
-                                    const patchStartMatch = line.match(/<<<< SEARCH/i);
-                                    const summaryStartMatch = line.match(/<summary>/i);
-
-                                    if (!technicalType && (codeMatch || fileStartMatch || patchStartMatch || summaryStartMatch)) {
-                                        if (currentAgent && agentBuffer.trim()) {
-                                            if (currentAgent.toLowerCase().includes('ajay vai')) {
-                                                TUI.drawAjayDialogue(agentBuffer.trim());
-                                            } else {
-                                                TUI.drawSwarmDialogue(currentAgent, agentBuffer.trim());
-                                            }
-                                            currentAgent = null;
-                                            agentBuffer = '';
-                                        }
-
-                                        if (codeMatch) {
-                                            technicalType = 'code';
-                                            codeLang = (codeMatch[1] || 'bash').toUpperCase();
-                                            TUI.drawStreamingStart('TECHNICAL BLOCK', codeLang);
-                                        } else if (fileStartMatch) {
-                                            technicalType = 'file';
-                                            TUI.drawStreamingStart('NEW FILE', fileStartMatch[1]);
-                                        } else if (patchStartMatch) {
-                                            technicalType = 'patch';
-                                            TUI.drawStreamingStart('PATCH', 'SURGICAL EDIT');
-                                        } else if (summaryStartMatch) {
-                                            technicalType = 'summary';
-                                            process.stdout.write(chalk.magenta('\n [📝] Generating Ajay\'s Work Summary...\n'));
-                                        }
-                                        continue;
-                                    }
-
-                                    const isEnd = (technicalType === 'code' && line.trim() === '```') ||
-                                        (technicalType === 'file' && line.includes('</file>')) ||
-                                        (technicalType === 'patch' && line.includes('>>>> REPLACE')) ||
-                                        (technicalType === 'summary' && line.includes('</summary>'));
-
-                                    if (isEnd) {
-                                        const displayContent = technicalBuffer.trim();
-                                        if (technicalType === 'summary') TUI.drawSummaryBox(displayContent);
-                                        else TUI.drawStreamingEnd();
-                                        technicalBuffer = '';
-                                        technicalType = null;
-                                        process.stdout.write('\n');
-                                        continue;
-                                    }
-
-                                    if (technicalType) {
-                                        technicalBuffer += line + '\n';
-                                        if (technicalType !== 'summary') TUI.drawStreamingLine(line);
-                                        continue;
-                                    }
-
-                                    const agentMatch = line.match(/(?:\[\**\s*|\b)(Ajay Vai|Sandip Dai|Sunil Dai|Bishal Dai|Narayan Dai|Big Bro)\s*\**\]?[:\s]*/i);
-                                    if (agentMatch) {
-                                        if (currentAgent && currentAgent !== agentMatch[1] && agentBuffer.trim()) {
-                                            if (currentAgent.toLowerCase().includes('ajay vai')) {
-                                                TUI.drawAjayDialogue(agentBuffer.trim());
-                                            } else {
-                                                TUI.drawSwarmDialogue(currentAgent, agentBuffer.trim());
-                                            }
-                                        }
-                                        currentAgent = agentMatch[1];
-                                        let cleanLine = line.replace(/(?:\[\**\s*|\b)(Ajay Vai|Sandip Dai|Sunil Dai|Bishal Dai|Narayan Dai|Big Bro)\s*\**\]?[:\s]*/i, '');
-                                        cleanLine = AgentOrchestrator.cleanText(cleanLine);
-                                        agentBuffer = cleanLine + '\n';
-                                        agentBuffer = agentBuffer.replace(/^\([^)]+\)\s*/, '');
-                                    } else if (currentAgent) {
-                                        const trimmedLine = line.trim();
-                                        if (trimmedLine.match(/^\([^)]+\)$/)) continue;
-                                        // Aggressive scrubbing for content lines too
-                                        agentBuffer += AgentOrchestrator.cleanText(line) + '\n';
-                                    } else if (line.trim()) {
-                                        process.stdout.write(chalk.gray(line.trim() + ' '));
-                                    }
-                                }
-                            }, abortController.signal);
-                        } catch (e: any) {
-                            if (isAborted || e.name === 'CanceledError' || e.message === 'canceled') {
-                                console.log(chalk.gray('\n\n [🛑] Stopped by brother.'));
-                                if (!hasStarted) spinner.stop();
-                            } else throw e;
-                        } finally {
-                            process.stdin.removeListener('data', onData);
-                            if (process.stdin.setRawMode) process.stdin.setRawMode(isRaw);
-                            process.stdin.pause();
-                            spinner.stop();
-
-                            // 🚨 SAFETY FLUSH: If stream ended but a block was open, close it
-                            if (technicalType) {
-                                const displayContent = technicalBuffer.trim();
-                                if (technicalType === 'summary') {
-                                    if (displayContent) TUI.drawSummaryBox(displayContent);
-                                } else {
-                                    TUI.drawStreamingEnd();
-                                }
-                                technicalBuffer = '';
-                                technicalType = null;
-                            }
-                        }
-
-                        if (currentAgent && agentBuffer.trim()) {
-                            if (currentAgent.toLowerCase().includes('ajay vai')) {
-                                TUI.drawAjayDialogue(agentBuffer.trim());
-                            } else {
-                                TUI.drawSwarmDialogue(currentAgent, agentBuffer.trim());
-                            }
-                        }
-
-                        messages.push({ role: 'assistant', content: fullResponse });
-                        console.log();
-
-                        const { executeWithPermission } = await import('../utils/command-executor');
-                        const result = await executeWithPermission(fullResponse, currentSessionCwd);
-
-                        if (result.newCwd && result.newCwd !== currentSessionCwd) {
-                            currentSessionCwd = result.newCwd;
-                            TUI.drawStatusBar(`Swarm moved to: ${currentSessionCwd}`);
-                            messages[0].content = messages[0].content.replace(/YOUR ROOT: `[^`]+`/, `YOUR ROOT: \`${currentSessionCwd}\``);
-                        }
-
-                        if (result.executed) {
-                            if (result.success) {
-                                TUI.drawGlowingContainer('TERMINAL OUTPUT', 'stdout', result.output || '(Success)');
-                                messages.push({ role: 'user', content: `[Action success. Output:\n${result.output}]\n\nWhat next, brothers?` });
-                                continue;
-                            } else {
-                                TUI.drawGlowingContainer('TERMINAL ERROR', 'stderr', result.error || 'Unknown error');
-                                messages.push({ role: 'user', content: `[Action failed. Error:\n${result.error}]\n\nPlease fix this, brothers!` });
-                                continue;
-                            }
-                        } else break;
-                    } catch (err: any) {
-                        spinner.fail('Swarm error');
-                        console.error(chalk.red(`❌ Error: ${err.message}`));
-                        break;
-                    }
-                }
-            }
         } catch (outerError: any) {
             console.error(chalk.red(`\n❌ Session error: ${outerError.message}`));
             process.exit(1);