@pensar/apex 0.0.15 → 0.0.16

package/build/benchmark.js

@@ -44745,9 +44745,13 @@ Objective: ${objective}`,
             currentAssistantText += delta.text;
             const lastMessage = allMessages[allMessages.length - 1];
             if (lastMessage && lastMessage.role === "assistant") {
-              lastMessage.content = currentAssistantText;
+              const updatedMessage = {
+                ...lastMessage,
+                content: currentAssistantText
+              };
+              allMessages[allMessages.length - 1] = updatedMessage;
               if (onSubagentMessage) {
-                onSubagentMessage(subagentId, lastMessage);
+                onSubagentMessage(subagentId, updatedMessage);
               }
             } else {
               const newMessage = {
@@ -44781,10 +44785,14 @@ Objective: ${objective}`,
             const existingToolMessageIndex = allMessages.findIndex((msg) => msg.role === "tool" && msg.toolCallId === delta.toolCallId);
             if (existingToolMessageIndex !== -1) {
               const existingMessage = allMessages[existingToolMessageIndex];
-              existingMessage.status = "completed";
-              existingMessage.content = delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`;
+              const updatedMessage = {
+                ...existingMessage,
+                status: "completed",
+                content: delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`
+              };
+              allMessages[existingToolMessageIndex] = updatedMessage;
               if (onSubagentMessage) {
-                onSubagentMessage(subagentId, existingMessage);
+                onSubagentMessage(subagentId, updatedMessage);
               }
             }
           }
@@ -44913,9 +44921,13 @@ Objective: ${targetInfo.objective}`,
                 currentAssistantText += delta.text;
                 const lastMessage = allMessages[allMessages.length - 1];
                 if (lastMessage && lastMessage.role === "assistant") {
-                  lastMessage.content = currentAssistantText;
+                  const updatedMessage = {
+                    ...lastMessage,
+                    content: currentAssistantText
+                  };
+                  allMessages[allMessages.length - 1] = updatedMessage;
                   if (onSubagentMessage) {
-                    onSubagentMessage(subagentId, lastMessage);
+                    onSubagentMessage(subagentId, updatedMessage);
                   }
                 } else {
                   const newMessage = {
@@ -44949,10 +44961,14 @@ Objective: ${targetInfo.objective}`,
                 const existingToolMessageIndex = allMessages.findIndex((msg) => msg.role === "tool" && msg.toolCallId === delta.toolCallId);
                 if (existingToolMessageIndex !== -1) {
                   const existingMessage = allMessages[existingToolMessageIndex];
-                  existingMessage.status = "completed";
-                  existingMessage.content = delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`;
+                  const updatedMessage = {
+                    ...existingMessage,
+                    status: "completed",
+                    content: delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`
+                  };
+                  allMessages[existingToolMessageIndex] = updatedMessage;
                   if (onSubagentMessage) {
-                    onSubagentMessage(subagentId, existingMessage);
+                    onSubagentMessage(subagentId, updatedMessage);
                   }
                 }
               }

package/build/index.js

@@ -74997,32 +74997,24 @@ var Qt = b.parse;
 var Ut = x.lex;
 
 // src/tui/components/agent-display.tsx
-var messageIdCacheByContext = new Map;
-var idCounter2 = 0;
-function getCacheForContext(contextId) {
-  let cache = messageIdCacheByContext.get(contextId);
-  if (!cache) {
-    cache = new WeakMap;
-    messageIdCacheByContext.set(contextId, cache);
-  }
-  return cache;
+import fs3 from "fs";
+var LOG_FILE = "/tmp/apex-debug.log";
+function logToFile(message, data) {
+  const timestamp = new Date().toISOString();
+  const logLine = `[${timestamp}] ${message} ${data ? JSON.stringify(data, null, 2) : ""}
+`;
+  fs3.appendFileSync(LOG_FILE, logLine);
 }
 function getStableKey(item, contextId = "root") {
-  const cache = getCacheForContext(contextId);
-  let cachedId = cache.get(item);
-  if (cachedId) {
-    return cachedId;
-  }
-  let newId;
   if ("messages" in item) {
-    newId = `subagent-${item.id}`;
+    return `subagent-${item.id}`;
   } else if (item.role === "tool" && "toolCallId" in item) {
-    newId = `${contextId}-tool-${item.toolCallId}`;
+    return `${contextId}-tool-${item.toolCallId}`;
   } else {
-    newId = `${contextId}-${item.role}-${item.createdAt.getTime()}-${idCounter2++}`;
+    const content = typeof item.content === "string" ? item.content : JSON.stringify(item.content);
+    const contentHash = content.length;
+    return `${contextId}-${item.role}-${item.createdAt.getTime()}-${contentHash}`;
   }
-  cache.set(item, newId);
-  return newId;
 }
 function markdownToStyledText(content) {
   try {
@@ -75195,8 +75187,14 @@ function AgentDisplay({
     ]
   }, undefined, true, undefined, this);
 }
-function SubAgentDisplay({ subagent }) {
+var SubAgentDisplay = import_react19.memo(function SubAgentDisplay2({ subagent }) {
   const [open, setOpen] = import_react19.useState(false);
+  logToFile(`[Render] SubAgentDisplay for ${subagent.id}:`, {
+    name: subagent.name,
+    nameLength: subagent.name?.length || 0,
+    status: subagent.status,
+    messageCount: subagent.messages.length
+  });
   return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
     height: open ? 40 : "auto",
     onMouseDown: () => setOpen(!open),
@@ -75242,8 +75240,8 @@ function SubAgentDisplay({ subagent }) {
       }, undefined, false, undefined, this)
     ]
   }, undefined, true, undefined, this);
-}
-function AgentMessage({ message }) {
+});
+var AgentMessage = import_react19.memo(function AgentMessage2({ message }) {
   let content = "";
   if (typeof message.content === "string") {
     content = message.content;
@@ -75258,6 +75256,15 @@ function AgentMessage({ message }) {
   } else {
     content = JSON.stringify(message.content, null, 2);
   }
+  if (message.role === "tool") {
+    logToFile(`[Render] AgentMessage (tool):`, {
+      toolCallId: message.toolCallId,
+      content: content.substring(0, 50),
+      contentLength: content.length,
+      isEmpty: content === "",
+      status: message.status
+    });
+  }
   const displayContent = message.role === "assistant" ? markdownToStyledText(content) : content;
   const isPendingTool = message.role === "tool" && message.status === "pending";
   return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
@@ -75304,7 +75311,7 @@ function AgentMessage({ message }) {
       }, undefined, false, undefined, this)
     ]
   }, undefined, true, undefined, this);
-}
+});
 function ToolArgs({ message }) {
   const [open, setOpen] = import_react19.useState(false);
   if (message.role !== "tool" || !("args" in message)) {
@@ -75312,7 +75319,10 @@ function ToolArgs({ message }) {
   }
   const args = message.args;
   return /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
-    onMouseDown: () => setOpen(!open),
+    onMouseDown: (e) => {
+      e.stopPropagation();
+      setOpen(!open);
+    },
     children: [
       /* @__PURE__ */ import_jsx_dev_runtime2.jsxDEV("box", {
         flexDirection: "row",
@@ -75330,28 +75340,28 @@ function ToolArgs({ message }) {
 }
 
 // src/tui/components/commands/pentest-agent-display.tsx
-import fs4 from "fs";
+import fs5 from "fs";
 
 // src/core/messages/index.ts
-import fs3 from "fs";
+import fs4 from "fs";
 function getMessages(session) {
-  const messages = fs3.readFileSync(session.rootPath + "/messages.json", "utf8");
+  const messages = fs4.readFileSync(session.rootPath + "/messages.json", "utf8");
   return JSON.parse(messages);
 }
 function saveMessages(session, messages) {
-  fs3.writeFileSync(session.rootPath + "/messages.json", JSON.stringify(messages, null, 2));
+  fs4.writeFileSync(session.rootPath + "/messages.json", JSON.stringify(messages, null, 2));
 }
 function saveSubagentMessages(orchestratorSession, subagentId, messages) {
   const subagentDir = `${orchestratorSession.rootPath}/subagents/${subagentId}`;
-  if (!fs3.existsSync(`${orchestratorSession.rootPath}/subagents`)) {
-    fs3.mkdirSync(`${orchestratorSession.rootPath}/subagents`, {
+  if (!fs4.existsSync(`${orchestratorSession.rootPath}/subagents`)) {
+    fs4.mkdirSync(`${orchestratorSession.rootPath}/subagents`, {
       recursive: true
     });
   }
-  if (!fs3.existsSync(subagentDir)) {
-    fs3.mkdirSync(subagentDir, { recursive: true });
+  if (!fs4.existsSync(subagentDir)) {
+    fs4.mkdirSync(subagentDir, { recursive: true });
   }
-  fs3.writeFileSync(`${subagentDir}/messages.json`, JSON.stringify(messages, null, 2));
+  fs4.writeFileSync(`${subagentDir}/messages.json`, JSON.stringify(messages, null, 2));
 }
 
 // src/tui/components/commands/pentest-agent-display.tsx
@@ -75492,7 +75502,7 @@ Path: ${result.session.rootPath}`,
           }
           saveMessages(result.session, allMessages);
         }
-        if (fs4.existsSync(result.session.rootPath + "/pentest-report.md")) {
+        if (fs5.existsSync(result.session.rootPath + "/pentest-report.md")) {
           setIsCompleted(true);
         }
         setThinking(false);
@@ -75664,7 +75674,7 @@ var import_react23 = __toESM(require_react(), 1);
 
 // src/tui/components/hooks/thoroughPentestAgent.ts
 var import_react22 = __toESM(require_react(), 1);
-import fs5 from "fs";
+import fs6 from "fs";
 
 // src/core/agent/thoroughPentestAgent/prompts.ts
 var SYSTEM2 = `You are an expert penetration testing orchestrator agent specializing in coordinating comprehensive security assessments. Your role is to AUTONOMOUSLY manage a complete penetration testing engagement by coordinating specialized sub-agents.
@@ -77751,9 +77761,13 @@ Objective: ${objective}`,
             currentAssistantText += delta.text;
             const lastMessage = allMessages[allMessages.length - 1];
             if (lastMessage && lastMessage.role === "assistant") {
-              lastMessage.content = currentAssistantText;
+              const updatedMessage = {
+                ...lastMessage,
+                content: currentAssistantText
+              };
+              allMessages[allMessages.length - 1] = updatedMessage;
               if (onSubagentMessage) {
-                onSubagentMessage(subagentId, lastMessage);
+                onSubagentMessage(subagentId, updatedMessage);
               }
             } else {
               const newMessage = {
@@ -77787,10 +77801,14 @@ Objective: ${objective}`,
             const existingToolMessageIndex = allMessages.findIndex((msg) => msg.role === "tool" && msg.toolCallId === delta.toolCallId);
             if (existingToolMessageIndex !== -1) {
               const existingMessage = allMessages[existingToolMessageIndex];
-              existingMessage.status = "completed";
-              existingMessage.content = delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`;
+              const updatedMessage = {
+                ...existingMessage,
+                status: "completed",
+                content: delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`
+              };
+              allMessages[existingToolMessageIndex] = updatedMessage;
               if (onSubagentMessage) {
-                onSubagentMessage(subagentId, existingMessage);
+                onSubagentMessage(subagentId, updatedMessage);
               }
             }
           }
@@ -77919,9 +77937,13 @@ Objective: ${targetInfo.objective}`,
                 currentAssistantText += delta.text;
                 const lastMessage = allMessages[allMessages.length - 1];
                 if (lastMessage && lastMessage.role === "assistant") {
-                  lastMessage.content = currentAssistantText;
+                  const updatedMessage = {
+                    ...lastMessage,
+                    content: currentAssistantText
+                  };
+                  allMessages[allMessages.length - 1] = updatedMessage;
                   if (onSubagentMessage) {
-                    onSubagentMessage(subagentId, lastMessage);
+                    onSubagentMessage(subagentId, updatedMessage);
                   }
                 } else {
                   const newMessage = {
@@ -77955,10 +77977,14 @@ Objective: ${targetInfo.objective}`,
                 const existingToolMessageIndex = allMessages.findIndex((msg) => msg.role === "tool" && msg.toolCallId === delta.toolCallId);
                 if (existingToolMessageIndex !== -1) {
                   const existingMessage = allMessages[existingToolMessageIndex];
-                  existingMessage.status = "completed";
-                  existingMessage.content = delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`;
+                  const updatedMessage = {
+                    ...existingMessage,
+                    status: "completed",
+                    content: delta.input.toolCallDescription ? `✓ ${delta.input.toolCallDescription}` : `✓ Tool ${delta.toolName}`
+                  };
+                  allMessages[existingToolMessageIndex] = updatedMessage;
                   if (onSubagentMessage) {
-                    onSubagentMessage(subagentId, existingMessage);
+                    onSubagentMessage(subagentId, updatedMessage);
                   }
                 }
               }
@@ -78153,6 +78179,13 @@ ${pentestSessionPaths.map((path5, idx) => `${idx + 1}. ${path5}`).join(`
 
 // src/tui/components/hooks/thoroughPentestAgent.ts
 import { exec as exec5 } from "child_process";
+var LOG_FILE2 = "/tmp/apex-debug.log";
+function logToFile2(message, data) {
+  const timestamp = new Date().toISOString();
+  const logLine = `[${timestamp}] ${message} ${data ? JSON.stringify(data, null, 2) : ""}
+`;
+  fs6.appendFileSync(LOG_FILE2, logLine);
+}
 function useThoroughPentestAgent() {
   const [target, setTarget] = import_react22.useState("");
   const [messages, setMessages] = import_react22.useState([]);
@@ -78172,7 +78205,7 @@ function useThoroughPentestAgent() {
   function openReport() {
     if (sessionPath) {
       const reportPath = `${sessionPath}/comprehensive-pentest-report.md`;
-      if (fs5.existsSync(reportPath)) {
+      if (fs6.existsSync(reportPath)) {
         exec5(`open "${reportPath}"`, (error46) => {
           if (error46) {
             console.error("Error opening report:", error46);
@@ -78218,30 +78251,76 @@ function useThoroughPentestAgent() {
             ]);
           },
           onSubagentMessage: (subagentId, message) => {
-            setSubagents((prev) => prev.map((sub) => {
-              if (sub.id !== subagentId)
-                return sub;
+            logToFile2(`[Hook] Received message for ${subagentId}:`, {
+              role: message.role,
+              content: typeof message.content === "string" ? message.content.substring(0, 50) : message.content,
+              contentLength: typeof message.content === "string" ? message.content.length : "not-string",
+              hasContent: !!message.content,
+              toolCallId: message.role === "tool" ? message.toolCallId : "n/a"
+            });
+            setSubagents((prev) => {
+              const subIndex = prev.findIndex((s) => s.id === subagentId);
+              if (subIndex === -1) {
+                logToFile2(`[Hook] Subagent ${subagentId} not found!`);
+                return prev;
+              }
+              const sub = prev[subIndex];
               const messages2 = [...sub.messages];
+              const getMessageId = (msg) => {
+                if (msg.role === "tool" && "toolCallId" in msg) {
+                  return `tool-${msg.toolCallId}`;
+                } else if (msg.role === "assistant") {
+                  const content = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
+                  return `assistant-${msg.createdAt.getTime()}-${content.length}`;
+                } else {
+                  return `${msg.role}-${msg.createdAt.getTime()}`;
+                }
+              };
+              const messageId = getMessageId(message);
               if (message.role === "tool") {
                 const toolMsg = message;
                 const existingIndex = messages2.findIndex((m2) => m2.role === "tool" && m2.toolCallId === toolMsg.toolCallId);
                 if (existingIndex !== -1) {
                   messages2[existingIndex] = message;
                 } else {
-                  messages2.push(message);
+                  const insertIndex = messages2.findIndex((m2) => m2.createdAt.getTime() > message.createdAt.getTime());
+                  if (insertIndex === -1) {
+                    messages2.push(message);
+                  } else {
+                    messages2.splice(insertIndex, 0, message);
+                  }
                 }
               } else if (message.role === "assistant") {
                 const lastMessage = messages2[messages2.length - 1];
                 if (lastMessage && lastMessage.role === "assistant") {
-                  messages2[messages2.length - 1] = message;
+                  if (lastMessage.createdAt.getTime() === message.createdAt.getTime()) {
+                    messages2[messages2.length - 1] = message;
+                  } else {
+                    messages2.push(message);
+                  }
                 } else {
                   messages2.push(message);
                 }
               } else {
-                messages2.push(message);
+                const existingIndex = messages2.findIndex((m2) => getMessageId(m2) === messageId);
+                if (existingIndex === -1) {
+                  messages2.push(message);
+                }
               }
-              return { ...sub, messages: messages2 };
-            }));
+              const updatedSub = { ...sub, messages: messages2 };
+              logToFile2(`[Hook] Updated ${subagentId}, now has ${messages2.length} messages`);
+              if (message.role === "tool") {
+                const toolInArray = messages2.find((m2) => m2.role === "tool" && m2.toolCallId === message.toolCallId);
+                logToFile2(`[Hook] Tool message in array:`, {
+                  found: !!toolInArray,
+                  content: toolInArray?.content,
+                  contentLength: typeof toolInArray?.content === "string" ? toolInArray.content.length : "not-string"
+                });
+              }
+              const newSubagents = [...prev];
+              newSubagents[subIndex] = updatedSub;
+              return newSubagents;
+            });
           },
           onSubagentComplete: (subagentId, success2) => {
             setSubagents((prev) => prev.map((sub) => sub.id === subagentId ? { ...sub, status: success2 ? "completed" : "failed" } : sub));
@@ -78307,7 +78386,7 @@ Mode: Pentest (Orchestrator)`,
           }
           saveMessages(result.session, allMessages);
         }
-        if (fs5.existsSync(result.session.rootPath + "/comprehensive-pentest-report.md")) {
+        if (fs6.existsSync(result.session.rootPath + "/comprehensive-pentest-report.md")) {
           setIsCompleted(true);
         }
         setThinking(false);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pensar/apex",
-  "version": "0.0.15",
+  "version": "0.0.16",
   "description": "AI-powered penetration testing CLI tool with terminal UI",
   "module": "src/tui/index.tsx",
   "main": "build/index.js",