matex-cli 1.2.66 → 1.2.68

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,397 +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 BRO_BANTER = [
-                "[Sunil Dai]: Ajay is so lazy, he probably fell asleep while writing a single console.log.",
-                "[Sandip Dai]: His code is as messy as his bed. Always sleeping.",
-                "[Narayan Dai]: Security risk! Ajay left his terminal open and went for CHAI again.",
-                "[Bishal Dai]: Let him sleep, he's a genius when he's awake... but that's rare.",
-                "[Big Bro]: AJAY! WAKE UP! You're making the swarm look bad!",
-                "[Sunil Dai]: Remember when Ajay tried to optimize a loop and ended up creating a 2-minute timer for his own nap? Legendary laziness.",
-                "[Sandip Dai]: The only thing faster than Ajay's coding is how fast he runs when he hears the kettle whistle for CHAI.",
-                "[Narayan Dai]: I found a 'todo' in his code that just says 'buy more biscuits for chai'. Priority level: CRITICAL.",
-                "[Bishal Dai]: He's not lazy, he's just 'asynchronously active' in his dreams.",
-                "[Big Bro]: I once saw Ajay refactor an entire backend in 10 minutes just so he could spend the rest of the day drinking chai and watching clouds.",
-                "[Sunil Dai]: Ajay's idea of a 'sprint' is a 2-minute walk to the tea stall.",
-                "[Sandip Dai]: He told me he's 'debugging his soul' when he's napping. The audacity!",
-                "[Narayan Dai]: Encryption? Ajay thinks it's just a fancy word for hiding his nap schedule.",
-                "[Bishal Dai]: To be fair, his 'dream-code' actually works perfectly. It's frustratingly brilliant.",
-                "[Big Bro]: Ajay, if you're listening: the next chai is on me IF you finish this task in 5 minutes. (Narrator: He finished in 3)."
-            ];
-
-            let lastActivityTime = Date.now();
-            let isAjaySleeping = false;
-            let isAjayOnChaiBreak = false;
-            let chaiBreakEndTime = 0;
-
-            const startChaiBreak = () => {
-                isAjayOnChaiBreak = true;
-                chaiBreakEndTime = Date.now() + 120000; // 2 minutes
-                TUI.drawChaiBreakMessage(2);
-
-                const breakInterval = setInterval(() => {
-                    const timeLeft = Math.ceil((chaiBreakEndTime - Date.now()) / 60000);
-                    if (timeLeft <= 0) {
-                        clearInterval(breakInterval);
-                        isAjayOnChaiBreak = false;
-                        const complained = Math.random() < 0.2;
-                        if (complained) {
-                            TUI.drawAjayDialogue("i'm back bros... but the tea was a bit cold today. my mood is slightly cooked. what did i miss?");
-                        } else {
-                            TUI.drawAjayDialogue("i'm back bros, chai was fire. real kiev vibe. what did i miss?");
-                        }
-                    } else {
-                        TUI.drawChaiBreakMessage(timeLeft);
-                        const randomBanter = BRO_BANTER[Math.floor(Math.random() * BRO_BANTER.length)];
-                        const cleanBanter = randomBanter.split(']: ')[1];
-                        const agent = randomBanter.split(']: ')[0].replace('[', '');
-                        TUI.drawSwarmDialogue(agent, cleanBanter);
-                    }
-                }, 40000);
-            };
-
-            // Periodic idle check
-            setInterval(() => {
-                if (!isAjaySleeping && !isAjayOnChaiBreak && Date.now() - lastActivityTime > 120000) {
-                    isAjaySleeping = true;
-                    TUI.drawSleepMessage();
-                }
-            }, 10000);
-
-            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)
                 }
             ];
 
-            // Ready for user input
             console.log(chalk.green('MATEX Brothers are Online.'));
             console.log(chalk.green('Speak your mind brother, the swarm is listening...'));
 
+            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
+            });
 
+            await session.start();
 
-            // Interactive loop
-            while (true) {
-                lastActivityTime = Date.now();
-
-                // 🔄 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}`);
-
-                    // Inject Lazy Ajay Personality Traits
-                    if (!messages[0].content.includes("### 💤 SLEEPY AJAY PROTOCOL")) {
-                        messages[0].content += `
-### 💤 SLEEPY AJAY PROTOCOL (USER REQUESTED):
-- **LAZY GENIUS:** Ajay is brilliant but incredibly lazy. He falls asleep if the user is quiet for too long.
-- **WAKE UP DIALOGUE:** If Ajay just woke up, he MUST say: "oh shit i slept again sorry bros...".
-- **CHAI ADDICT:** Ajay often leaves for CHAI breaks ("i am having chai from my office please handle the user need").
-- **BRO REACTIONS:** When Ajay sleeps or leaves for chai, the other bros MUST roast him ruthlessly or share funny incidents about his laziness.
-- **INCIDENT LOG:**
-    - Incident #1: Ajay once automated his entire job just to spend 6 hours at a local tea shop.
-    - Incident #2: He fell asleep during a production deploy because the "vibe was too relaxing".
-    - Incident #3: He claims he can't code without "Special Kathmandu Chai".
-    - Incident #4: He tried to optimize a loop and ended up creating a 2-minute timer for his own nap.
-    - Incident #5: The only thing faster than Ajay's coding is how fast he runs when he hears the kettle whistle for CHAI.
-    - Incident #6: He once refactored an entire backend in 10 minutes just so he could spend the rest of the day drinking chai and watching clouds.
-    - Incident #7: He tried to use AI to predict when his tea would be at the perfect temperature.
-    - Incident #8: He once convinced the other bros that he was 'working in a higher dimension' while actually just taking a 3-hour chai break.
-    - Incident #9: He has a dedicated Git branch called 'chai-fixes' which just contains comments about biscuit pairings.
-    - Incident #10: He once coded a feature that automatically orders tea when his terminal idle time hits 5 minutes.
-`;
-                    }
-                } catch (e) { }
-
-                // Get user input
-                const { userMessage } = await inquirer.prompt([
-                    {
-                        type: 'input',
-                        name: 'userMessage',
-                        message: chalk.cyan(isAjayOnChaiBreak ? 'You (Ajay is away):' : 'You:'),
-                        prefix: '',
-                    },
-                ]);
-
-                lastActivityTime = Date.now();
-
-                // 🏁 WAKE UP LOGIC
-                if (isAjaySleeping) {
-                    isAjaySleeping = false;
-                    TUI.drawWakeUpMessage();
-                    messages.push({
-                        role: 'user',
-                        content: `[SYSTEM: AJAY VAI HAS JUST WOKEN UP. Ajay: Say "oh shit i slept again sorry bros...". Bros: Roast him for being lazy and share common Ajay 'lazy genius' incidents. Then Ajay says he is going for CHAI and triggers the 2-minute break.]`
-                    });
-                    startChaiBreak();
-                }
-
-                if (userMessage.toLowerCase() === 'exit' || userMessage.toLowerCase() === 'quit') {
-                    console.log(chalk.yellow('\n👋 Goodbye, brother!\n'));
-                    break;
-                }
-
-                if (!userMessage.trim()) continue;
-
-                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...');
-
-                        let fullResponse = '';
-                        let buffer = '';
-                        let technicalBuffer = '';
-                        let technicalType: 'code' | 'file' | 'patch' | 'summary' | null = null;
-                        let codeLang = 'bash';
-                        let hasStarted = false;
-
-                        let currentAgent: any = 'Ajay Vai';
-                        let agentBuffer: string = '';
-
-                        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);