@rabts/cli 1.0.0 → 3.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rabts/cli",
-  "version": "1.0.0",
+  "version": "3.0.0",
   "description": "Rabts Studio CLI",
   "main": "src/index.js",
   "type": "module",
@@ -15,9 +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",
-    "openai": "^4.52.0"
+    "marked": "^15.0.12",
+    "marked-terminal": "^7.3.0",
+    "openai": "^4.52.0",
+    "ora": "^9.4.1"
   }
 }

package/src/agent.js

@@ -4,12 +4,22 @@ 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';
+import { loadSessionMemory, saveSessionMemory, clearSessionMemory } from './memory.js';
+
+// Setup Markdown Renderer
+const marked = new Marked();
+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 = [
@@ -51,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');
@@ -68,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}`;
@@ -80,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',
@@ -87,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) {
@@ -98,19 +138,75 @@ 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.";
 }
 
+function getProjectContext() {
+  const cwd = process.cwd();
+  let contextStr = `\nContexto do Diretório Atual:\n- Caminho: ${cwd}\n`;
+  
+  try {
+    const files = fs.readdirSync(cwd).slice(0, 50);
+    contextStr += `- Arquivos na raiz: ${files.join(', ')}\n`;
+
+    if (fs.existsSync(path.join(cwd, 'package.json'))) {
+      const pkg = JSON.parse(fs.readFileSync(path.join(cwd, 'package.json'), 'utf8'));
+      contextStr += `- Projeto Node.js: ${pkg.name || 'Desconhecido'}\n`;
+    }
+    
+    if (fs.existsSync(path.join(cwd, 'composer.json'))) {
+      const comp = JSON.parse(fs.readFileSync(path.join(cwd, 'composer.json'), 'utf8'));
+      contextStr += `- Projeto PHP (Composer): ${comp.name || 'Desconhecido'}\n`;
+    }
+  } catch (e) {}
+  return contextStr;
+}
+
 export async function startInteractiveLoop(config) {
   console.log(infoColor(`\nConectado via ${config.provider} (Modelo: ${config.model})`));
-  console.log(infoColor('Você pode conversar ou pedir para ler/criar arquivos. Digite "sair" para encerrar.\n'));
+  console.log(infoColor('Dicas: Digite "sair" para encerrar. Digite "/clear" para limpar a memória desta pasta.\n'));
 
-  let messages = [
-    { 
-      role: 'system', 
-      content: 'Você é o Rabts AI, um assistente de terminal. Você pode conversar com o usuário ou usar ferramentas para manipular arquivos e comandos quando requisitado.' 
-    }
-  ];
+  let messages = loadSessionMemory();
+  const isNewSession = messages.length === 0;
+
+  if (isNewSession) {
+    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 = [
+      { role: 'system', content: systemContent }
+    ];
+  } else {
+    console.log(chalk.green('✓ Sessão anterior restaurada para este projeto.\n'));
+  }
 
   const supportsTools = (config.provider === 'rabts' || config.provider === 'openai');
 
@@ -123,28 +219,45 @@ export async function startInteractiveLoop(config) {
       }
     ]);
 
-    if (userPrompt.toLowerCase() === 'sair' || userPrompt.toLowerCase() === 'exit') {
+    const input = userPrompt.trim();
+
+    if (input.toLowerCase() === 'sair' || input.toLowerCase() === 'exit') {
       console.log(infoColor('Até logo!'));
       break;
     }
+    
+    if (input.toLowerCase() === '/clear') {
+      clearSessionMemory();
+      console.log(chalk.yellow('Memória limpa! O contexto do projeto será reiniciado na próxima mensagem.\n'));
+      messages = [];
+      continue;
+    }
 
-    if (!userPrompt.trim()) continue;
+    if (!input) continue;
 
-    messages.push({ role: 'user', content: userPrompt });
+    if (messages.length === 0) {
+      messages.push({ 
+        role: 'system', 
+        content: 'Você é o Rabts AI, um desenvolvedor autônomo. ' + getProjectContext() 
+      });
+    }
+
+    messages.push({ role: 'user', content: input });
 
     let isProcessingTools = true;
     let totalPromptTokens = 0;
     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;
@@ -158,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",
@@ -167,15 +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:'));
-          console.log(chalk.white(responseMessage.content));
+          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`));
+        
+        saveSessionMemory(messages);
         isProcessingTools = false;
       }
     }

package/src/config.js

@@ -6,7 +6,7 @@ import chalk from 'chalk';
 
 const CONFIG_FILE_PATH = path.join(os.homedir(), '.rabts-config.json');
 
-const RABTS_DEFAULT_KEY = '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';
+// A chave padrão não fica mais no código-fonte por segurança
 
 export function getConfig() {
   if (fs.existsSync(CONFIG_FILE_PATH)) {
@@ -47,7 +47,8 @@ export async function configureApiKey() {
   let config = { provider };
 
   if (provider === 'rabts') {
-    config.apiKey = RABTS_DEFAULT_KEY;
+    // A chave é injetada via proxy no servidor backend
+    config.apiKey = 'proxy_mode';
     const { model } = await inquirer.prompt([
       {
         type: 'list',

package/src/llm.js

@@ -10,8 +10,8 @@ export function getClient(config) {
 
   if (config.provider === 'rabts') {
     clients[config.provider] = new OpenAI({
-      baseURL: 'https://api.groq.com/openai/v1',
-      apiKey: config.apiKey,
+      baseURL: 'https://dev.rabts.com/api',
+      apiKey: 'proxy_mode',
     });
   } else if (config.provider === 'openai') {
     clients[config.provider] = new OpenAI({

package/src/memory.js

@@ -0,0 +1,46 @@
+import fs from 'fs';
+import path from 'path';
+
+export function getMemoryPath() {
+  const cwd = process.cwd();
+  return path.join(cwd, '.rabts', 'session.json');
+}
+
+export function loadSessionMemory() {
+  const memPath = getMemoryPath();
+  if (fs.existsSync(memPath)) {
+    try {
+      const data = fs.readFileSync(memPath, 'utf8');
+      return JSON.parse(data);
+    } catch (e) {
+      return [];
+    }
+  }
+  return [];
+}
+
+export function saveSessionMemory(messages) {
+  const memPath = getMemoryPath();
+  const dirPath = path.dirname(memPath);
+  
+  if (!fs.existsSync(dirPath)) {
+    fs.mkdirSync(dirPath, { recursive: true });
+  }
+  
+  // Ocultar a pasta no Windows/Linux de forma simples
+  // Adicionar um gitignore para que o usuário não commite acidentalmente o histórico
+  const gitignorePath = path.join(dirPath, '.gitignore');
+  if (!fs.existsSync(gitignorePath)) {
+    fs.writeFileSync(gitignorePath, "*\n");
+  }
+
+  // Filtrar o que salvar (não precisamos salvar todos os resultados longos de tools se não quisermos)
+  fs.writeFileSync(memPath, JSON.stringify(messages, null, 2), 'utf8');
+}
+
+export function clearSessionMemory() {
+  const memPath = getMemoryPath();
+  if (fs.existsSync(memPath)) {
+    fs.unlinkSync(memPath);
+  }
+}