@pheem49/mint 1.3.0 → 1.4.1

package/src/CLI/chat_router.js

@@ -1,8 +1,8 @@
 const fs = require('fs');
 const path = require('path');
 const { GoogleGenAI } = require('@google/genai');
-const { executeCodeTask } = require('./code_agent');
-const { readConfig } = require('../System/config_manager');
+const { executeCodeTask, _helpers: codeAgentHelpers } = require('./code_agent');
+const { readConfig, getAvailableProviders } = require('../System/config_manager');
 
 const DEFAULT_GEMINI_MODEL = 'gemini-2.5-flash';
 
@@ -26,7 +26,19 @@ Return JSON only:
 }
 
 Choose "code" when the user is asking to inspect, edit, review, debug, explain, refactor, verify, or otherwise operate on the current project/workspace/codebase/files.
-Choose "chat" for general conversation, factual Q&A, or non-code assistant tasks.`;
+Choose "chat" for general conversation, factual Q&A, small/simple requests, non-code assistant tasks, or direct file-system actions like finding/opening a folder or file by name.
+Only choose "code" for substantial coding work that likely needs multiple steps, workspace inspection, edits, verification, or project-wide reasoning.`;
+
+function isDirectFilesystemActionRequest(text) {
+    const input = (text || '').trim().toLowerCase();
+    if (!input) return false;
+
+    const filesystemActionPattern = /(open|find|locate|search for|look for|หา|ค้นหา|เปิด)/;
+    const filesystemTargetPattern = /(folder|directory|dir|file|โฟลเดอร์|ไฟล์|ไดเรกทอรี)/;
+    const codeOperationPattern = /(inspect|review|refactor|debug|implement|edit|change|fix|explain|analyze|สำรวจ|รีวิว|รีแฟกเตอร์|แก้|อธิบาย|วิเคราะห์)/;
+
+    return filesystemActionPattern.test(input) && filesystemTargetPattern.test(input) && !codeOperationPattern.test(input);
+}
 
 function workspaceLooksLikeCodebase(workspaceRoot) {
     const markers = [
@@ -45,13 +57,27 @@ function detectCodeIntentHeuristic(text, workspaceRoot = process.cwd()) {
     const input = (text || '').trim().toLowerCase();
     if (!input) return false;
     if (input.startsWith('/code ')) return true;
+    if (isDirectFilesystemActionRequest(input)) return false;
+
+    return isLargeCodeTaskRequest(input, workspaceRoot);
+}
+
+function isLargeCodeTaskRequest(text, workspaceRoot = process.cwd()) {
+    const input = (text || '').trim().toLowerCase();
+    if (!input) return false;
+    if (!workspaceLooksLikeCodebase(workspaceRoot)) return false;
 
     const hasCodeKeyword = CODE_KEYWORDS.some(keyword => input.includes(keyword));
     const hasThaiCodeKeyword = THAI_CODE_KEYWORDS.some(keyword => input.includes(keyword));
     const referencesProject = /โปรเจคนี้|โปรเจ็กต์นี้|this project|this repo|this repository|codebase|workspace/.test(input);
     const asksForAction = /สำรวจ|ดู|แก้|เพิ่ม|ลบ|ปรับ|ตรวจ|วิเคราะห์|implement|inspect|explore|fix|update|change|refactor|review|explain|debug/.test(input);
+    const strongTaskSignal = /failing tests?|run tests?|verify|verification|bug|issue|error|refactor|implement|feature|patch|edit|modify|analyze the project|แก้บั๊ก|รันเทสต์|ทดสอบ|ตรวจสอบ|ยืนยันผล|รีแฟกเตอร์|เพิ่มฟีเจอร์|แก้โค้ด|วิเคราะห์โปรเจค/.test(input);
+    const multiStepSignal = /and|then|พร้อม|แล้ว|จากนั้น|ทั้ง|ทั่วทั้ง|ทั้งโปรเจค|project-wide|entire project|whole project/.test(input);
 
-    return workspaceLooksLikeCodebase(workspaceRoot) && (referencesProject || ((hasCodeKeyword || hasThaiCodeKeyword) && asksForAction));
+    if (referencesProject && strongTaskSignal) return true;
+    if ((hasCodeKeyword || hasThaiCodeKeyword) && asksForAction && strongTaskSignal) return true;
+    if ((hasCodeKeyword || hasThaiCodeKeyword) && multiStepSignal && asksForAction) return true;
+    return false;
 }
 
 function getRouterClient() {
@@ -71,7 +97,7 @@ function summarizeWorkspace(workspaceRoot) {
         .join(', ') || '(no obvious code markers)';
 }
 
-async function detectCodeIntent(text, workspaceRoot = process.cwd()) {
+async function detectCodeIntent(text, workspaceRoot = process.cwd(), history = []) {
     const input = (text || '').trim();
     if (!input) {
         return { route: 'chat', reason: 'Empty input.' };
@@ -81,6 +107,10 @@ async function detectCodeIntent(text, workspaceRoot = process.cwd()) {
         return { route: 'code', reason: 'Explicit /code command.' };
     }
 
+    if (isDirectFilesystemActionRequest(input)) {
+        return { route: 'chat', reason: 'Direct file-system action request.' };
+    }
+
     const heuristicRoute = detectCodeIntentHeuristic(input, workspaceRoot);
     const routerClient = getRouterClient();
     if (!routerClient) {
@@ -103,7 +133,8 @@ async function detectCodeIntent(text, workspaceRoot = process.cwd()) {
                     text: [
                         `Workspace: ${workspaceRoot}`,
                         `Workspace markers: ${summarizeWorkspace(workspaceRoot)}`,
-                        `Message: ${input}`
+                        `Context (Last 5 turns): ${history.slice(-10).map(m => `${m.sender}: ${m.text}`).join('\n')}`,
+                        `Current Message: ${input}`
                     ].join('\n')
                 }]
             }]
@@ -112,6 +143,12 @@ async function detectCodeIntent(text, workspaceRoot = process.cwd()) {
         const textOutput = typeof response.text === 'function' ? response.text() : response.text;
         const parsed = JSON.parse(textOutput);
         const route = parsed.route === 'code' ? 'code' : 'chat';
+        if (route === 'code' && !isLargeCodeTaskRequest(input, workspaceRoot)) {
+            return {
+                route: 'chat',
+                reason: 'Request looks small enough for normal chat.'
+            };
+        }
         return {
             route,
             reason: parsed.reason || (route === 'code' ? 'Model classified as code.' : 'Model classified as chat.')
@@ -126,9 +163,13 @@ async function detectCodeIntent(text, workspaceRoot = process.cwd()) {
 
 async function runChatRoutedTask(input, context) {
     const text = input.startsWith('/code ') ? input.slice('/code '.length).trim() : input;
-    const { appendMessage, setThinking, requestApproval, setMode } = context;
+    const { appendMessage, setThinking, requestApproval, setMode, history } = context;
+
+    const config = readConfig();
+    const availableProviders = getAvailableProviders(config);
+    const preferredProvider = codeAgentHelpers.selectSupportedCodeProvider(config, availableProviders);
 
-    appendMessage('system', `Routing this request to Code Mode for workspace: ${process.cwd()}`);
+    appendMessage('system', `Routing this request to Code Mode for workspace: ${process.cwd()} using [${preferredProvider}]`);
     if (setMode) setMode('Code');
 
     let seconds = 0;
@@ -142,6 +183,8 @@ async function runChatRoutedTask(input, context) {
         const result = await executeCodeTask(text, {
             cwd: process.cwd(),
             requestApproval,
+            provider: preferredProvider,
+            history: history,
             onProgress: (message) => appendMessage('system', `[Code] ${message}`)
         });
         clearInterval(timer);
@@ -162,5 +205,10 @@ async function runChatRoutedTask(input, context) {
 
 module.exports = {
     detectCodeIntent,
-    runChatRoutedTask
+    runChatRoutedTask,
+    _helpers: {
+        detectCodeIntentHeuristic,
+        isDirectFilesystemActionRequest,
+        isLargeCodeTaskRequest
+    }
 };

package/src/CLI/chat_ui.js

@@ -14,6 +14,10 @@ const SLASH_COMMANDS = [
     { name: '/copy',   desc: 'Copy last response to clipboard' },
     { name: '/clear',  desc: 'Clear conversation history' },
     { name: '/reset',  desc: 'Reset conversation history' },
+    { name: '/agent',  desc: 'Switch AI personas (coder, researcher, etc)' },
+    { name: '/workspace', desc: 'Manage project-specific contexts' },
+    { name: '/review', desc: 'Request a second-pass review of the last response' },
+    { name: '/stats',  desc: 'Show system health stats (CPU/RAM/Disk)' },
     { name: '/help',   desc: 'Show help information' },
     { name: '/exit',   desc: 'Exit Mint' }
 ];
@@ -86,11 +90,10 @@ function createChatUI({ onSubmit, onExit }) {
         style: { bg: 'default' }
     });
 
-    // ─── Input area ───────────────────────────────────────────────────────────
     const inputBox = blessed.textbox({
         bottom: 3, left: 1, width: '100%-2', height: 3,
         tags: false,
-        inputOnFocus: true,
+        inputOnFocus: false, // We'll manage this manually for stability
         keys: true,
         style: {
             bg: INPUT_BG,
@@ -107,6 +110,16 @@ function createChatUI({ onSubmit, onExit }) {
         label: ' Message '
     });
 
+    // --- SAFETY PATCH ---
+    // Prevent "TypeError: done is not a function" if a listener survives a blur/focus cycle.
+    const originalListener = inputBox._listener;
+    inputBox._listener = function(ch, key) {
+        if (typeof this._done !== 'function') return;
+        return originalListener.call(this, ch, key);
+    };
+
+
+
     // ─── Placeholder (SIBLING widget floating over input content area) ─────────
     // inputBox: bottom=3, height=3, border=1 → content row at bottom=4, left=2
     const placeholderWidget = blessed.text({
@@ -232,6 +245,7 @@ function createChatUI({ onSubmit, onExit }) {
 
     /** Update model name in status bar (called after /models switch) */
     function updateStatusModel(newModel) {
+        if (!newModel) return;
         statusRight.setContent(`{#88e0b0-fg}${newModel}{/}`);
         screen.render();
     }
@@ -280,7 +294,7 @@ function createChatUI({ onSubmit, onExit }) {
             border: { fg: '#88e0b0' }
         },
         width: '80%',
-        height: 'shrink',
+        height: 12, // Fixed height to avoid 'shrink' miscalculation with buttons
         top: 'center',
         left: 'center',
         label: ' Approval ',
@@ -377,26 +391,28 @@ function createChatUI({ onSubmit, onExit }) {
 
 
     // Submit or Select Suggestion on Enter
-    inputBox.key(['enter'], () => {
+    inputBox.on('submit', (value) => {
         if (!commandList.hidden) {
             const selected = activeSuggestions[commandList.selected];
             if (selected) {
                 inputBox.setValue(selected.name + ' ');
                 commandList.hide();
                 hidePlaceholder();
-                inputBox.focus();
+                inputBox.focus(); 
+                inputBox.readInput(); // Re-focus to continue typing
                 refreshInputStyles();
                 screen.render();
                 return; // Don't submit yet, let user add args or press enter again
             }
         }
 
-        const raw = (inputBox.getValue ? inputBox.getValue() : inputBox.value) || '';
+        const raw = value || '';
         const text = raw.trim();
         if (!text) {
             inputBox.clearValue();
             showPlaceholder();
-            inputBox.focus();
+            inputBox.focus(); 
+            inputBox.readInput(); // Re-focus to continue typing
             refreshInputStyles();
             screen.render();
             return;
@@ -405,7 +421,8 @@ function createChatUI({ onSubmit, onExit }) {
         // Clear input and restore placeholder
         inputBox.clearValue();
         showPlaceholder();
-        inputBox.focus();
+        inputBox.focus(); 
+        inputBox.readInput(); // Explicitly restart reading
         refreshInputStyles();
         screen.render();
 
@@ -482,6 +499,7 @@ function createChatUI({ onSubmit, onExit }) {
 
     // ─── Initial render ───────────────────────────────────────────────────────
     inputBox.focus();
+    inputBox.readInput(); // Initial start
     refreshInputStyles();
     screen.render();
 
@@ -588,6 +606,84 @@ function createChatUI({ onSubmit, onExit }) {
         screen.render();
     }
 
+    /**
+     * Opens a streaming message bubble for the assistant.
+     * Returns { appendChunk(text), finalize(timestamp) } for typewriter rendering.
+     * Usage:
+     *   const stream = streamMessage('assistant');
+     *   stream.appendChunk('Hello'); stream.appendChunk(' World');
+     *   stream.finalize(timestamp);
+     */
+    function streamMessage(role = 'assistant') {
+        const now = new Date();
+        const timeStr = now.toLocaleTimeString('th-TH', { hour: '2-digit', minute: '2-digit', hour12: false });
+        const maxLineWidth = Math.max(screen.width - 20, 36);
+
+        // Print the header bubble once
+        chatBox.log('');
+        if (role === 'assistant') {
+            chatBox.log(` {bold}{#d4a8ff-fg}Mint{/} {gray-fg}${timeStr}{/}`);
+        }
+
+        let buffer = '';           // accumulates the full response text
+        let lineBuffer = '';       // current partial line being built
+        let lineRendered = false;  // whether we already pushed the first line prefix
+
+        function flushLine(force = false) {
+            // Flush content that fits on one line-width or when forced
+            if (!lineBuffer && !force) return;
+            if (!lineRendered) {
+                chatBox.log(` {#5a456d-fg}▏{/} {#ffffff-fg}${lineBuffer}{/}`);
+                lineRendered = true;
+            } else {
+                // Overwrite the last line by popping + re-pushing (blessed.log limitation)
+                // We can't truly overwrite, so we just keep appending new lines for each chunk.
+                // For large chunks, split on newline and emit per-line.
+                chatBox.log(` {#5a456d-fg}▏{/} {#ffffff-fg}${lineBuffer}{/}`);
+            }
+            screen.render();
+        }
+
+        function appendChunk(text) {
+            if (!text) return;
+            buffer += text;
+            const segments = text.split('\n');
+            for (let i = 0; i < segments.length; i++) {
+                lineBuffer += segments[i];
+                if (i < segments.length - 1) {
+                    // Newline boundary — emit current line
+                    const lines = wrapLineSmart(lineBuffer, maxLineWidth);
+                    lines.forEach(l => chatBox.log(` {#5a456d-fg}▏{/} {#ffffff-fg}${l}{/}`));
+                    lineBuffer = '';
+                    lineRendered = true;
+                    screen.render();
+                } else if (lineBuffer.length >= maxLineWidth) {
+                    // Line overflow — auto-wrap
+                    const lines = wrapLineSmart(lineBuffer, maxLineWidth);
+                    lines.slice(0, -1).forEach(l => chatBox.log(` {#5a456d-fg}▏{/} {#ffffff-fg}${l}{/}`));
+                    lineBuffer = lines[lines.length - 1] || '';
+                    lineRendered = true;
+                    screen.render();
+                }
+                // Otherwise keep buffering the partial line
+            }
+        }
+
+        function finalize(timestamp = null) {
+            // Flush remaining buffer
+            if (lineBuffer) {
+                const lines = wrapLineSmart(lineBuffer, maxLineWidth);
+                lines.forEach(l => chatBox.log(` {#5a456d-fg}▏{/} {#ffffff-fg}${l}{/}`));
+                lineBuffer = '';
+            }
+            // Track last response for clipboard
+            lastAssistantResponse = buffer;
+            screen.render();
+        }
+
+        return { appendChunk, finalize };
+    }
+
     /** Show/hide thinking indicator in status bar */
     function setThinking(active, secondsElapsed = 0) {
         if (active) {
@@ -609,18 +705,28 @@ function createChatUI({ onSubmit, onExit }) {
                 ? 'Shell Command'
                 : request.type === 'patch'
                     ? 'Patch Edit'
-                    : 'File Write';
+                    : request.type === 'code_mode'
+                        ? 'Enter Code Mode'
+                        : 'File Write';
             const preview = request.preview || request.label || '';
             const message = [
                 `{bold}${typeLabel}{/bold}`,
                 '',
                 preview,
                 '',
-                'Approve this action?'
+                'Approve this action?',
+                '', // Extra lines to push buttons down and avoid overlapping
+                ''
             ].join('\n');
 
+            // Temporarily stop reading input so the dialog can receive keys
+            if (inputBox._reading) {
+                inputBox.cancel();
+            }
+
             approvalDialog.ask(message, (approved) => {
                 inputBox.focus();
+                inputBox.readInput(); // Ensure we resume reading after dialog
                 refreshInputStyles();
                 screen.render();
                 resolve(Boolean(approved));
@@ -628,7 +734,7 @@ function createChatUI({ onSubmit, onExit }) {
         });
     }
 
-    return { screen, appendMessage, setThinking, updateStatusModel, copyLastResponse, requestApproval, setMode };
+    return { screen, appendMessage, streamMessage, setThinking, updateStatusModel, copyLastResponse, requestApproval, setMode };
 }
 
 module.exports = { createChatUI };