@rabts/cli 2.0.0 → 3.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rabts/cli",
-  "version": "2.0.0",
+  "version": "3.0.0",
   "description": "Rabts Studio CLI",
   "main": "src/index.js",
   "type": "module",
@@ -15,11 +15,13 @@
     "@google/genai": "^2.9.0",
     "axios": "^1.7.2",
     "chalk": "^5.3.0",
+    "cheerio": "^1.2.0",
     "commander": "^12.1.0",
     "dotenv": "^16.4.5",
     "inquirer": "^10.0.0",
     "marked": "^15.0.12",
     "marked-terminal": "^7.3.0",
-    "openai": "^4.52.0"
+    "openai": "^4.52.0",
+    "ora": "^9.4.1"
   }
 }

package/src/agent.js

@@ -4,6 +4,9 @@ import { exec } from 'child_process';
 import util from 'util';
 import chalk from 'chalk';
 import inquirer from 'inquirer';
+import ora from 'ora';
+import * as cheerio from 'cheerio';
+import axios from 'axios';
 import { Marked } from 'marked';
 import { markedTerminal } from 'marked-terminal';
 import { sendMessage } from './llm.js';
@@ -16,7 +19,7 @@ marked.use(markedTerminal());
 const execPromise = util.promisify(exec);
 
 // Tema Rabts
-const themeColor = chalk.cyan; // Azul claro para padronizar
+const themeColor = chalk.cyan; 
 const infoColor = chalk.gray;
 
 const tools = [
@@ -58,14 +61,39 @@ const tools = [
         required: ["command"]
       }
     }
+  },
+  {
+    type: "function",
+    function: {
+      name: "search_code",
+      description: "Faz uma busca global por texto (Grep) nos arquivos do projeto.",
+      parameters: {
+        type: "object",
+        properties: { query: { type: "string", description: "O que pesquisar" } },
+        required: ["query"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "read_url",
+      description: "Lê e extrai texto limpo de uma página web ou documentação.",
+      parameters: {
+        type: "object",
+        properties: { url: { type: "string" } },
+        required: ["url"]
+      }
+    }
   }
 ];
 
-async function executeToolCall(toolCall) {
+async function executeToolCall(toolCall, spinner) {
   const args = JSON.parse(toolCall.function.arguments);
   const cwd = process.cwd();
 
   if (toolCall.function.name === 'read_file') {
+    spinner.text = `Lendo arquivo: ${args.filepath}`;
     try {
       const fullPath = path.resolve(cwd, args.filepath);
       return fs.readFileSync(fullPath, 'utf8');
@@ -75,11 +103,11 @@ async function executeToolCall(toolCall) {
   }
 
   if (toolCall.function.name === 'write_file') {
+    spinner.text = `Salvando arquivo: ${args.filepath}`;
     try {
       const fullPath = path.resolve(cwd, args.filepath);
       fs.mkdirSync(path.dirname(fullPath), { recursive: true });
       fs.writeFileSync(fullPath, args.content, 'utf8');
-      console.log(themeColor(`  ↳ Arquivo salvo: `) + chalk.white(args.filepath));
       return `Arquivo salvo em ${fullPath}`;
     } catch (e) {
       return `Erro ao salvar arquivo: ${e.message}`;
@@ -87,6 +115,8 @@ async function executeToolCall(toolCall) {
   }
 
   if (toolCall.function.name === 'run_command') {
+    // Para perguntar ao usuário, precisamos pausar o spinner
+    spinner.stop();
     const { confirm } = await inquirer.prompt([{
       type: 'confirm',
       name: 'confirm',
@@ -94,10 +124,13 @@ async function executeToolCall(toolCall) {
       default: false
     }]);
 
-    if (!confirm) return `Usuário negou o comando: ${args.command}`;
+    if (!confirm) {
+      spinner.start('Retomando...');
+      return `Usuário negou o comando: ${args.command}`;
+    }
 
+    spinner.start(`Executando comando: ${args.command}`);
     try {
-      console.log(themeColor(`  ↳ Executando: `) + chalk.white(args.command));
       const { stdout, stderr } = await execPromise(args.command, { cwd });
       return stdout || stderr || "Sucesso sem output.";
     } catch (e) {
@@ -105,6 +138,34 @@ async function executeToolCall(toolCall) {
     }
   }
 
+  if (toolCall.function.name === 'search_code') {
+    spinner.text = `Buscando globalmente por: ${args.query}`;
+    try {
+      // Uso do grep recursivo ignorando node_modules
+      const cmd = `grep -rn --exclude-dir=node_modules --exclude-dir=.git "${args.query}" . || true`;
+      const { stdout } = await execPromise(cmd, { cwd });
+      if (!stdout.trim()) return "Nenhum resultado encontrado.";
+      // Limita resultados para não estourar tokens
+      return stdout.substring(0, 10000); 
+    } catch (e) {
+      return `Erro ao buscar: ${e.message}`;
+    }
+  }
+
+  if (toolCall.function.name === 'read_url') {
+    spinner.text = `Acessando URL: ${args.url}`;
+    try {
+      const response = await axios.get(args.url);
+      const $ = cheerio.load(response.data);
+      // Remove scripts, styles
+      $('script, style, noscript, svg, img, nav, footer').remove();
+      const text = $('body').text().replace(/\s+/g, ' ').trim();
+      return text.substring(0, 15000); // Retorna até 15k chars
+    } catch (e) {
+      return `Erro ao acessar URL: ${e.message}`;
+    }
+  }
+
   return "Função desconhecida.";
 }
 
@@ -113,7 +174,7 @@ function getProjectContext() {
   let contextStr = `\nContexto do Diretório Atual:\n- Caminho: ${cwd}\n`;
   
   try {
-    const files = fs.readdirSync(cwd).slice(0, 50); // Limita para não estourar tokens
+    const files = fs.readdirSync(cwd).slice(0, 50);
     contextStr += `- Arquivos na raiz: ${files.join(', ')}\n`;
 
     if (fs.existsSync(path.join(cwd, 'package.json'))) {
@@ -125,9 +186,7 @@ function getProjectContext() {
       const comp = JSON.parse(fs.readFileSync(path.join(cwd, 'composer.json'), 'utf8'));
       contextStr += `- Projeto PHP (Composer): ${comp.name || 'Desconhecido'}\n`;
     }
-  } catch (e) {
-    // Ignora erros de permissão ou parse
-  }
+  } catch (e) {}
   return contextStr;
 }
 
@@ -135,12 +194,11 @@ export async function startInteractiveLoop(config) {
   console.log(infoColor(`\nConectado via ${config.provider} (Modelo: ${config.model})`));
   console.log(infoColor('Dicas: Digite "sair" para encerrar. Digite "/clear" para limpar a memória desta pasta.\n'));
 
-  // Carrega memória anterior se existir
   let messages = loadSessionMemory();
   const isNewSession = messages.length === 0;
 
   if (isNewSession) {
-    let systemContent = 'Você é o Rabts AI, um assistente de terminal inteligente. Responda em Markdown limpo.';
+    let systemContent = 'Você é o Rabts AI, um desenvolvedor e assistente de terminal altamente inteligente capaz de navegar pela internet e gerenciar código autonomamente. Responda usando Markdown limpo.';
     systemContent += getProjectContext();
     
     messages = [
@@ -177,11 +235,10 @@ export async function startInteractiveLoop(config) {
 
     if (!input) continue;
 
-    // Se a memória foi limpa, recarrega o context inicial
     if (messages.length === 0) {
       messages.push({ 
         role: 'system', 
-        content: 'Você é o Rabts AI, um assistente de terminal inteligente.' + getProjectContext() 
+        content: 'Você é o Rabts AI, um desenvolvedor autônomo. ' + getProjectContext() 
       });
     }
 
@@ -192,14 +249,15 @@ export async function startInteractiveLoop(config) {
     let totalCompletionTokens = 0;
     let startTime = Date.now();
 
+    const spinner = ora({
+      text: 'Pensando...',
+      color: 'cyan',
+      spinner: 'dots'
+    }).start();
+
     while (isProcessingTools) {
-      process.stdout.write(infoColor('  Pensando...'));
-      
       const responseMessage = await sendMessage(config, messages, supportsTools ? tools : null);
       
-      process.stdout.clearLine(0);
-      process.stdout.cursorTo(0);
-
       if (responseMessage.usage) {
         totalPromptTokens += responseMessage.usage.prompt_tokens || 0;
         totalCompletionTokens += responseMessage.usage.completion_tokens || 0;
@@ -213,8 +271,9 @@ export async function startInteractiveLoop(config) {
 
       if (responseMessage.tool_calls && responseMessage.tool_calls.length > 0) {
         for (const toolCall of responseMessage.tool_calls) {
-          console.log(themeColor(`  ✦ Usando ferramenta: `) + chalk.white(toolCall.function.name));
-          const result = await executeToolCall(toolCall);
+          spinner.text = `Analisando ferramenta: ${toolCall.function.name}`;
+          const result = await executeToolCall(toolCall, spinner);
+          
           messages.push({
             tool_call_id: toolCall.id,
             role: "tool",
@@ -222,20 +281,19 @@ export async function startInteractiveLoop(config) {
             content: result,
           });
         }
+        spinner.text = 'Processando resultados...';
       } else {
+        spinner.stop();
         const duration = ((Date.now() - startTime) / 1000).toFixed(1);
         
         if (responseMessage.content) {
           console.log(themeColor('\nRabts:'));
-          // Renderiza a resposta visualmente com syntax highlighting
           console.log(marked.parse(responseMessage.content).trim());
         }
 
-        console.log(infoColor(`\n  ↳ Finalizado em ${duration}s • ${totalPromptTokens + totalCompletionTokens} tokens (${totalPromptTokens} in, ${totalCompletionTokens} out)\n`));
+        console.log(infoColor(`\n  ↳ Finalizado em ${duration}s • ${totalPromptTokens + totalCompletionTokens} tokens\n`));
         
-        // Salva memória a cada interação
         saveSessionMemory(messages);
-        
         isProcessingTools = false;
       }
     }