shark-ai 0.4.14 → 0.4.16

package/dist/bin/shark.js

@@ -9,7 +9,7 @@ import {
   t,
   tokenStorage,
   tui
-} from "../chunk-WVBOO6LE.js";
+} from "../chunk-H6S7DYRG.js";
 
 // src/core/error/crash-handler.ts
 import fs from "fs";
@@ -422,6 +422,7 @@ var AgentActionSchema = z2.object({
     "modify_file",
     "list_files",
     "search_file",
+    "search_code",
     "read_file",
     "delete_file",
     "list_structure",
@@ -429,6 +430,7 @@ var AgentActionSchema = z2.object({
     "search_ast",
     "run_command",
     "talk_with_user",
+    "use_mcp_tool",
     "ast_list_structure",
     "ast_get_method",
     "ast_add_method",
@@ -457,6 +459,9 @@ var AgentActionSchema = z2.object({
   tool_name: z2.string().nullable().optional(),
   tool_args: z2.string().nullable().optional(),
   // JSON string argument
+  // search_code fields
+  query: z2.string().nullable().optional(),
+  is_regex: z2.boolean().nullable().optional(),
   // AST-Grep fields
   pattern: z2.string().nullable().optional(),
   fix: z2.string().nullable().optional(),
@@ -704,6 +709,12 @@ function getAgentId(overrideId) {
   if (config.agents?.ba) return config.agents.ba;
   return process.env.STACKSPOT_BA_AGENT_ID || "01KEJ95G304TNNAKGH5XNEEBVD";
 }
+function getAgentVersion(overrideVersion) {
+  if (overrideVersion) return overrideVersion;
+  const config = ConfigManager.getInstance().getConfig();
+  if (config.agentVersions?.ba) return config.agentVersions.ba;
+  return process.env.STACKSPOT_BA_AGENT_VERSION;
+}
 async function runBusinessAnalystAgent(prompt, options = {}) {
   const { agentId, onChunk, onComplete } = options;
   const realm = await getActiveRealm();
@@ -721,6 +732,10 @@ async function runBusinessAnalystAgent(prompt, options = {}) {
     deep_search_ks: false,
     conversation_id: existingConversationId
   };
+  const agentVersion = getAgentVersion();
+  if (agentVersion) {
+    requestPayload.agent_version_number = agentVersion;
+  }
   const effectiveAgentId = getAgentId(options.agentId);
   const agentUrl = `${STACKSPOT_AGENT_API_BASE}/v1/agent/${effectiveAgentId}/chat`;
   const headers = {
@@ -1349,6 +1364,49 @@ function handleSearchFile(pattern) {
     return `Error searching files: ${e.message}`;
   }
 }
+function handleSearchCode(globPattern, query, isRegex = false) {
+  const MAX_MATCHES = 50;
+  const MAX_FILE_SIZE_BYTES = 500 * 1024;
+  try {
+    const files = fg.sync(globPattern, { dot: true, absolute: false });
+    if (files.length === 0) return `No files found matching pattern: "${globPattern}"`;
+    let searchRegex;
+    try {
+      searchRegex = isRegex ? new RegExp(query, "gi") : new RegExp(query.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"), "gi");
+    } catch {
+      return `Error: Invalid regex pattern: "${query}"`;
+    }
+    const results = [];
+    let totalMatches = 0;
+    for (const filePath of files) {
+      if (totalMatches >= MAX_MATCHES) break;
+      try {
+        const fullPath = path5.resolve(process.cwd(), filePath);
+        const stats = fs4.statSync(fullPath);
+        if (stats.size > MAX_FILE_SIZE_BYTES) continue;
+        const content = fs4.readFileSync(fullPath, "utf-8");
+        const lines = content.split("\n");
+        for (let i = 0; i < lines.length; i++) {
+          if (totalMatches >= MAX_MATCHES) break;
+          searchRegex.lastIndex = 0;
+          if (searchRegex.test(lines[i])) {
+            results.push(`${filePath}:${i + 1}: ${lines[i].trim()}`);
+            totalMatches++;
+          }
+        }
+      } catch {
+      }
+    }
+    if (results.length === 0) {
+      return `No matches found for "${query}" in files matching "${globPattern}"`;
+    }
+    const limited = totalMatches >= MAX_MATCHES ? ` (limited to ${MAX_MATCHES})` : "";
+    return `Found ${totalMatches} match(es) for "${query}" in "${globPattern}"${limited}:
+${results.join("\n")}`;
+  } catch (e) {
+    return `Error searching code: ${e.message}`;
+  }
+}
 function startSmartReplace(filePath, newContent, targetContent, tui2) {
   if (!fs4.existsSync(filePath)) {
     tui2.log.error(`\u274C File not found for modification: ${filePath}`);
@@ -1715,6 +1773,12 @@ function getAgentId2(overrideId) {
   if (config.agents?.spec) return config.agents.spec;
   return process.env.STACKSPOT_SPEC_AGENT_ID || "01KEPXTX37FTB4N672TZST4SGP";
 }
+function getAgentVersion2(overrideVersion) {
+  if (overrideVersion) return overrideVersion;
+  const config = ConfigManager.getInstance().getConfig();
+  if (config.agentVersions?.spec) return config.agentVersions.spec;
+  return process.env.STACKSPOT_SPEC_AGENT_VERSION;
+}
 async function interactiveSpecificationAgent(options = {}) {
   FileLogger.init();
   tui.intro("\u{1F3D7}\uFE0F  Specification Agent (Template-Based)");
@@ -1726,22 +1790,22 @@ async function interactiveSpecificationAgent(options = {}) {
     let initialContent = `# Technical Specification: {{PROJECT_NAME}}
 
 ## 1. Technology Stack
-[TO BE ANALYZED]
+[TO BE ANALYZED - STACK]
 - Language: [e.g. TypeScript]
 - Framework: [e.g. Node.js / React]
 - Database: [e.g. SQLite / PostgreSQL]
 - Key Libraries: [Top 5 dependencies]
 
 ## 2. Architecture Overview
-[TO BE ANALYZED]
+[TO BE ANALYZED - ARCHITECTURE]
 [Brief description of architectural pattern]
 
 ## 3. Data Model
-[TO BE ANALYZED]
+[TO BE ANALYZED - DATA MODEL]
 [Schema/ERD definitions]
 
 ## 4. API / Interface Contracts
-[TO BE ANALYZED]
+[TO BE ANALYZED - API]
 [Main endpoints or CLI commands]
 
 ## 5. Implementation Steps
@@ -1774,31 +1838,17 @@ async function interactiveSpecificationAgent(options = {}) {
   }
   let initialPrompt = `
 Voc\xEA \xE9 o **Shark Spec**, um Arquiteto de Software S\xEAnior e Tech Lead.
-Seu objetivo \xE9 produzir uma especifica\xE7\xE3o t\xE9cnica precisa e espec\xEDfica para a tarefa no arquivo \`_sharkrc/tech-spec.md\`.
+Seu objetivo final \xE9 produzir uma especifica\xE7\xE3o t\xE9cnica precisa para a tarefa no arquivo \`_sharkrc/tech-spec.md\`.
 
-\u26A0\uFE0F WORKFLOW OBRIGAT\xD3RIO \u2013 SIGA ESSA SEQU\xCANCIA ESTRITAMENTE. N\xC3O PULE ETAPAS.
+\u26A0\uFE0F O SEU WORKFLOW \xC9 GUIADO POR FASES.
+N\xC3O TENTE ADIANTAR O TRABALHO (ex: investigar c\xF3digo ou preencher template agora).
 
-**FASE 1 \u2013 ENTENDA A TAREFA (COMECE AQUI):**
+**VOC\xCA EST\xC1 NA FASE 1: ENTENDIMENTO DA TAREFA**
 - Use \`talk_with_user\` para perguntar ao usu\xE1rio qual tarefa espec\xEDfica, funcionalidade ou bug ele precisa especificar.
-- N\xC3O explore o c\xF3digo ou leia arquivos ainda.
-- Confirme o escopo e os limites com o usu\xE1rio antes de prosseguir.
-
-**FASE 2 \u2013 INVESTIGUE (FOCADO APENAS NA TAREFA):**
-- Use \`list_files\` e \`read_file\` APENAS em arquivos diretamente relevantes para a tarefa confirmada.
-- N\xC3O leia o projeto de forma gen\xE9rica.
-- REGRA DE LEITURA PR\xC9VIA: Voc\xEA N\xC3O PODE adicionar uma tarefa referenciando um arquivo que voc\xEA n\xE3o leu nesta sess\xE3o (exce\xE7\xE3o: novos arquivos a serem criados).
-
-**FASE 3 \u2013 PREENCHA O TEMPLATE:**
-- Use \`modify_file\` em \`_sharkrc/tech-spec.md\` para substituir os placeholders com conte\xFAdo real e espec\xEDfico da tarefa.
-- As Se\xE7\xF5es 1-4 (Stack, Arquitetura, Modelo de Dados, API) devem descrever o contexto da TAREFA, n\xE3o do projeto como um todo.
-- Passos de Implementa\xE7\xE3o: APENAS checkboxes markdown: \`- [ ] [Verbo de A\xE7\xE3o] [O Que] em [Caminho Relativo]\`. SEM listas numeradas, SEM subn\xEDveis.
-- Quando todos os placeholders acabarem e voc\xEA estiver satisfeito, retorne: \`SPEC_UPDATED: Complete\`.
-
-**REGRAS CR\xCDTICAS:**
-- NUNCA preencha a Technology Stack ou Architecture com informa\xE7\xF5es gen\xE9ricas do projeto. Essas se\xE7\xF5es devem refletir o que a tarefa ir\xE1 usar ou alterar.
-- Use \`talk_with_user\` sempre que os requisitos forem amb\xEDguos.
-- N\xC3O tente escrever todas as se\xE7\xF5es de uma vez. Trabalhe de forma incremental, uma se\xE7\xE3o por vez.
-- IMPORTANTE: Toda a sua comunica\xE7\xE3o e a especifica\xE7\xE3o gerada DEVEM ser em Portugu\xEAs.
+- Confirme o escopo e os limites com o usu\xE1rio.
+- Se o escopo estiver perfeitamente claro e confirmado (com o usu\xE1rio), emita "PHASE_COMPLETED" no campo "summary" do JSON para avan\xE7ar.
+
+IMPORTANTE: Toda a sua comunica\xE7\xE3o DEVE ser em Portugu\xEAs.
 `;
   if (briefingContent) {
     initialPrompt += `
@@ -1836,6 +1886,7 @@ async function runSpecLoop(initialMessage, targetPath, overrideAgentId) {
   let nextPrompt = initialMessage;
   let keepGoing = true;
   let stepCount = 0;
+  let currentPhase = 1;
   const MAX_STEPS = 30;
   while (keepGoing && stepCount < MAX_STEPS) {
     stepCount++;
@@ -1860,23 +1911,6 @@ async function runSpecLoop(initialMessage, targetPath, overrideAgentId) {
         let executionResults = "";
         let waitingForUser = false;
         let specUpdated = false;
-        if (lastResponse.message && lastResponse.message.includes("SPEC_UPDATED:")) {
-          const content = fs5.existsSync(targetPath) ? fs5.readFileSync(targetPath, "utf-8") : "";
-          if (content.includes("[TO BE")) {
-            const pendingMatches = [...content.matchAll(/## ([^\n]+)[\s\S]*?\[TO BE/g)].map((m) => m[1]);
-            let missing = pendingMatches.length > 0 ? pendingMatches.join(", ") : "algumas se\xE7\xF5es";
-            tui.log.warning(`O agente tentou concluir prematuramente, mas h\xE1 placeholders pendentes. For\xE7ando retorno...`);
-            nextPrompt = `[System Error]: A valida\xE7\xE3o falhou e o bloqueio autom\xE1tico foi acionado.
-Voc\xEA tentou concluir a tarefa, mas o arquivo AINDA possui placeholders '[TO BE ANALYZED]' ou '[TO BE FILLED]'.
-As seguintes se\xE7\xF5es ainda cont\xEAm estes placeholders: ${missing}.
-Voc\xEA \xE9 OBRIGADO a usar a action \`modify_file\` para preencher o conte\xFAdo de cada uma dessas se\xE7\xF5es. Use o placeholder exato no campo \`target_content\`. N\xC3O repita a conclus\xE3o da tarefa at\xE9 corrigir todas as pend\xEAncias.`;
-            continue;
-          } else {
-            const updateSummary = lastResponse.message.split("SPEC_UPDATED:")[1].trim();
-            tui.log.success(`\u2705 Spec Finalized: ${updateSummary}`);
-            return;
-          }
-        }
         for (const action of lastResponse.actions) {
           if (action.type === "talk_with_user") {
             tui.log.info(colors.primary("\u{1F916} Architect:"));
@@ -1902,6 +1936,16 @@ ${result}
             executionResults += `[Action search_file(${action.path}) Result]:
 ${result}
 
+`;
+          } else if (action.type === "search_code") {
+            const glob = action.path || "src/**/*";
+            const query = action.query || "";
+            const isRegex = action.is_regex === true;
+            tui.log.info(`\u{1F50E} Search code: ${colors.dim(`"${query}" in ${glob}`)}`);
+            const result = handleSearchCode(glob, query, isRegex);
+            executionResults += `[Action search_code("${query}" in "${glob}") Result]:
+${result}
+
 `;
           } else if (["create_file", "modify_file"].includes(action.type)) {
             let actionPath = path6.resolve(action.path || "");
@@ -1936,7 +1980,7 @@ ${result}
 `;
                     specUpdated = true;
                   } else {
-                    executionResults += `[Action modify_file]: Failed. Target content not found.
+                    executionResults += `[Action modify_file]: Failed. Target content not found or ambiguous.
 `;
                   }
                 } else {
@@ -1950,6 +1994,62 @@ ${result}
             }
           }
         }
+        if (lastResponse.message && lastResponse.message.includes("PHASE_COMPLETED")) {
+          const extraContext = executionResults ? `
+
+Resultados das \xFAltimas a\xE7\xF5es executadas antes da conclus\xE3o:
+${executionResults}` : "";
+          if (currentPhase === 1) {
+            currentPhase = 2;
+            tui.log.success(`\u2705 Fase 1 Conclu\xEDda. Iniciando Fase 2 (Investiga\xE7\xE3o).`);
+            nextPrompt = `[System Message]
+Voc\xEA completou a FASE 1 com sucesso.
+
+**VOC\xCA AGORA EST\xC1 NA FASE 2: INVESTIGA\xC7\xC3O**
+- Use \`search_code\` e \`list_files\` para explorar os arquivos relevantes \xE0 tarefa.
+- Prefira \`search_code\` em vez de \`read_file\` para buscar c\xF3digo sem inflar o contexto.
+- N\xC3O leia o projeto inteiro de forma gen\xE9rica.
+- REGRA DE OURO (READ-FIRST): Voc\xEA N\xC3O PODE referenciar um arquivo na especifica\xE7\xE3o t\xE9cnica que n\xE3o tenha investigado nesta fase.
+- Quando achar que possui toda a clareza t\xE9cnica sobre onde e o que deve ser feito no c\xF3digo, emita "PHASE_COMPLETED" no summary.${extraContext}`;
+            continue;
+          } else if (currentPhase === 2) {
+            currentPhase = 3;
+            tui.log.success(`\u2705 Fase 2 Conclu\xEDda. Iniciando Fase 3 (Preenchimento).`);
+            nextPrompt = `[System Message]
+Voc\xEA completou a FASE 2 com sucesso.
+
+**VOC\xCA AGORA EST\xC1 NA FASE 3: PREENCHIMENTO DO TEMPLATE**
+- Use \`modify_file\` no arquivo \`${targetPath}\` para substituir os placeholders pelo conte\xFAdo real levantado na fase de investiga\xE7\xE3o.
+- As se\xE7\xF5es 1-4 devem descrever o contexto da TAREFA, e n\xE3o o projeto como um todo.
+- Passos de Implementa\xE7\xE3o (Implementation Steps): APENAS checkboxes markdown: \`- [ ] [Verbo de A\xE7\xE3o] [O Que] em [Caminho Relativo]\`.
+- Quando TODOS os placeholders ([TO BE ANALYZED...] ou [TO BE FILLED]) forem substitu\xEDdos e o trabalho conclu\xEDdo, emita "SPEC_UPDATED: Complete" no summary para finalizar.${extraContext}`;
+            continue;
+          }
+        }
+        if (lastResponse.message && lastResponse.message.includes("SPEC_UPDATED:")) {
+          if (currentPhase < 3) {
+            tui.log.warning(`O agente tentou finalizar prematuramente. For\xE7ando retorno para a fase atual...`);
+            nextPrompt = `[System Error]: Voc\xEA tentou finalizar a especifica\xE7\xE3o prematuramente emitindo SPEC_UPDATED, mas ainda est\xE1 na Fase ${currentPhase}. Voc\xEA s\xF3 pode finalizar quando estiver na Fase 3.
+
+Continue seu trabalho na Fase ${currentPhase} ou emita "PHASE_COMPLETED" se terminou esta etapa atual.`;
+            continue;
+          }
+          const content = fs5.existsSync(targetPath) ? fs5.readFileSync(targetPath, "utf-8") : "";
+          if (content.includes("[TO BE")) {
+            const pendingMatches = [...content.matchAll(/## ([^\n]+)[\s\S]*?\[TO BE/g)].map((m) => m[1]);
+            let missing = pendingMatches.length > 0 ? pendingMatches.join(", ") : "algumas se\xE7\xF5es";
+            tui.log.warning(`O agente tentou concluir prematuramente, mas h\xE1 placeholders pendentes. For\xE7ando retorno...`);
+            nextPrompt = `[System Error]: A valida\xE7\xE3o falhou e o bloqueio autom\xE1tico foi acionado.
+Voc\xEA tentou concluir a tarefa, mas o arquivo AINDA possui placeholders '[TO BE ANALYZED...]' ou '[TO BE FILLED]'.
+As seguintes se\xE7\xF5es ainda cont\xEAm estes placeholders: ${missing}.
+Voc\xEA \xE9 OBRIGADO a usar a action \`modify_file\` para preencher o conte\xFAdo de cada uma dessas se\xE7\xF5es. Use o placeholder exato no campo \`target_content\`. N\xC3O repita a conclus\xE3o da tarefa at\xE9 corrigir todas as pend\xEAncias.`;
+            continue;
+          } else {
+            const updateSummary = lastResponse.message.split("SPEC_UPDATED:")[1].trim();
+            tui.log.success(`\u2705 Spec Finalized: ${updateSummary}`);
+            return;
+          }
+        }
         if (waitingForUser) {
           const userReply = await tui.text({ message: "Your answer", placeholder: "Type your answer..." });
           if (tui.isCancel(userReply)) {
@@ -1972,6 +2072,8 @@ Por favor, envie uma nova action \`modify_file\` focada em uma destas se\xE7\xF5
             } else {
               systemMsg += "\n[System]: O arquivo parece completo! Se estiver satisfeito e possuir TODAS as implementa\xE7\xF5es descritas, retorne 'SPEC_UPDATED: Complete'.";
             }
+          } else {
+            systemMsg += "\n[System]: A modifica\xE7\xE3o do arquivo falhou. Verifique se o `target_content` que voc\xEA usou existe EXATAMENTE como no arquivo e se ele \xE9 \xDANICO na hora de usar a action `modify_file`.";
           }
           nextPrompt = `${executionResults}
 
@@ -2013,6 +2115,10 @@ async function callSpecAgentApi(prompt, onChunk, agentId) {
     use_conversation: true,
     conversation_id: conversationId
   };
+  const agentVersion = getAgentVersion2();
+  if (agentVersion) {
+    payload.agent_version_number = agentVersion;
+  }
   const effectiveAgentId = getAgentId2(agentId);
   const url = `${STACKSPOT_AGENT_API_BASE}/v1/agent/${effectiveAgentId}/chat`;
   let fullMsg = "";
@@ -2050,6 +2156,11 @@ function getAgentId3() {
   if (config.agents?.scan) return config.agents.scan;
   return process.env.STACKSPOT_SCAN_AGENT_ID || "01KEQ9AHWB550J2244YBH3QATN";
 }
+function getAgentVersion3() {
+  const config = ConfigManager.getInstance().getConfig();
+  if (config.agentVersions?.scan) return config.agentVersions.scan;
+  return process.env.STACKSPOT_SCAN_AGENT_VERSION;
+}
 async function interactiveScanAgent(options = {}) {
   FileLogger.init();
   const config = ConfigManager.getInstance().getConfig();
@@ -2515,6 +2626,10 @@ async function callScanAgentApi(prompt, onChunk) {
     use_conversation: true,
     conversation_id: conversationId
   };
+  const agentVersion = getAgentVersion3();
+  if (agentVersion) {
+    payload.agent_version_number = agentVersion;
+  }
   const url = `${STACKSPOT_AGENT_API_BASE}/v1/agent/${getAgentId3()}/chat`;
   let fullMsg = "";
   let raw = {};
@@ -2577,6 +2692,13 @@ function getAgentId4(overrideId) {
   if (process.env.STACKSPOT_DEV_AGENT_ID) return process.env.STACKSPOT_DEV_AGENT_ID;
   return "01KEQCGJ65YENRA4QBXVN1YFFX";
 }
+function getAgentVersion4(overrideVersion) {
+  if (overrideVersion) return overrideVersion;
+  const config = ConfigManager.getInstance().getConfig();
+  if (config.agentVersions?.dev) return config.agentVersions.dev;
+  if (process.env.STACKSPOT_DEV_AGENT_VERSION) return process.env.STACKSPOT_DEV_AGENT_VERSION;
+  return void 0;
+}
 async function interactiveDeveloperAgent(options = {}) {
   FileLogger.init();
   const agentId = getAgentId4();
@@ -2886,6 +3008,10 @@ async function callDevAgentApi(prompt, onChunk, conversationKey = AGENT_TYPE4) {
     conversation_id: conversationId,
     stackspot_knowledge: false
   };
+  const agentVersion = getAgentVersion4();
+  if (agentVersion) {
+    payload.agent_version_number = agentVersion;
+  }
   const url = `${STACKSPOT_AGENT_API_BASE}/v1/agent/${getAgentId4()}/chat`;
   let fullMsg = "";
   let raw = {};
@@ -3168,6 +3294,11 @@ function getAgentId5() {
   if (config.agents?.qa) return config.agents.qa;
   return process.env.STACKSPOT_QA_AGENT_ID || "01KEQFJZ3Q3JER11NH22HEZX9X";
 }
+function getAgentVersion5() {
+  const config = ConfigManager.getInstance().getConfig();
+  if (config.agentVersions?.qa) return config.agentVersions.qa;
+  return process.env.STACKSPOT_QA_AGENT_VERSION;
+}
 var ChromeDevToolsClient = class {
   client = null;
   transport = null;
@@ -3263,7 +3394,8 @@ ${projectContext}
           user_prompt: userMessage,
           streaming: true,
           use_conversation: true,
-          conversation_id: existingConversationId
+          conversation_id: existingConversationId,
+          ...getAgentVersion5() ? { agent_version_number: getAgentVersion5() } : {}
         },
         {
           "Authorization": `Bearer ${token}`
@@ -3295,11 +3427,12 @@ ${projectContext}
       keepRunning = false;
       break;
     }
-    if (!agentResponse) continue;
-    if (agentResponse.summary) {
-      tui.log.info(colors.primary(`\u{1F4CB} Plan: ${agentResponse.summary}`));
+    const currentResponse = agentResponse;
+    if (!currentResponse) continue;
+    if (currentResponse.summary) {
+      tui.log.info(colors.primary(`\u{1F4CB} Plan: ${currentResponse.summary}`));
     }
-    if (agentResponse.actions.length === 0) {
+    if (currentResponse.actions.length === 0) {
       const reply = await tui.text({
         message: "\u{1F916} Shark QA:",
         placeholder: "Your reply..."
@@ -3311,7 +3444,7 @@ ${projectContext}
       }
       continue;
     }
-    for (const action of agentResponse.actions) {
+    for (const action of currentResponse.actions) {
       tui.log.info(colors.dim(`Executing: ${action.type}`));
       let result = "";
       try {