pentesting 0.1.8 → 0.1.9

package/dist/index.js

@@ -2610,6 +2610,52 @@ var ASCII_BANNER = `
  \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u255D
 `;
 
+// src/config/agent-constants.ts
+var AGENT_EVENT = {
+  // Lifecycle
+  PLUGINS_LOADED: "plugins_loaded",
+  HOOKS_LOADED: "hooks_loaded",
+  COMMANDS_LOADED: "commands_loaded",
+  MCP_SERVER_ADDED: "mcp_server_added",
+  // Execution
+  ITERATION: "iteration",
+  THOUGHT: "thought",
+  RESPONSE: "response",
+  TOOL_CALL: "tool_call",
+  TOOL_RESULT: "tool_result",
+  COMMAND_EXECUTE: "command_execute",
+  // State changes
+  TARGET_SET: "target_set",
+  PHASE_CHANGE: "phase_change",
+  AGENT_SWITCH: "agent_switch",
+  // Discoveries
+  FINDING: "finding",
+  CREDENTIAL: "credential",
+  COMPROMISED: "compromised",
+  // Completion
+  COMPLETE: "complete",
+  REPORT: "report",
+  ERROR: "error"
+};
+var CLI_COMMAND = {
+  HELP: "help",
+  TARGET: "target",
+  START: "start",
+  STOP: "stop",
+  FINDINGS: "findings",
+  CLEAR: "clear",
+  EXIT: "exit"
+};
+var MESSAGE_TYPE = {
+  USER: "user",
+  ASSISTANT: "assistant",
+  TOOL: "tool",
+  THINKING: "thinking",
+  ERROR: "error",
+  SYSTEM: "system",
+  RESULT: "result"
+};
+
 // src/cli/app.tsx
 import { jsx, jsxs } from "react/jsx-runtime";
 var App = ({ autoApprove = false, target }) => {
@@ -2647,39 +2693,39 @@ var App = ({ autoApprove = false, target }) => {
     return duration;
   }, []);
   useEffect(() => {
-    addMessage("system", "Pentesting Agent initialized. Type /help for commands.");
+    addMessage(MESSAGE_TYPE.SYSTEM, "Pentesting Agent initialized. Type /help for commands.");
     if (target) {
       agent.setTarget(target);
-      addMessage("system", `Target: ${target}`);
+      addMessage(MESSAGE_TYPE.SYSTEM, `Target: ${target}`);
     }
-    agent.on("thought", (type, content) => {
-      setCurrentStatus(content.slice(0, 60));
+    agent.on(AGENT_EVENT.THOUGHT, (thought) => {
+      setCurrentStatus(thought.content.slice(0, 60));
     });
-    agent.on("tool_call", (data) => {
+    agent.on(AGENT_EVENT.TOOL_CALL, (data) => {
       const args = Object.entries(data.input).slice(0, 2).map(([k, v]) => `${k}=${typeof v === "string" ? v.slice(0, 30) : "..."}`).join(" ");
       setCurrentStatus(`Running ${data.name}...`);
-      addMessage("tool", `\u25B6 ${data.name} ${args}`);
+      addMessage(MESSAGE_TYPE.TOOL, `\u25B6 ${data.name} ${args}`);
     });
-    agent.on("tool_result", (data) => {
+    agent.on(AGENT_EVENT.TOOL_RESULT, (data) => {
       const icon = data.result.success ? "\u2713" : "\u2717";
       const preview = data.result.output?.slice(0, 100).replace(/\n/g, " ") || "";
-      addMessage("result", `${icon} ${preview}`);
+      addMessage(MESSAGE_TYPE.RESULT, `${icon} ${preview}`);
     });
-    agent.on("iteration", (data) => {
-      setCurrentStatus(`Phase: ${data.phase} (iteration ${data.iteration})`);
+    agent.on(AGENT_EVENT.ITERATION, (data) => {
+      setCurrentStatus(`Phase: ${data.phase} (iteration ${data.current})`);
     });
-    agent.on("finding", (finding) => {
-      addMessage("system", `\u{1F3AF} [${finding.severity.toUpperCase()}] ${finding.title}`);
+    agent.on(AGENT_EVENT.FINDING, (finding) => {
+      addMessage(MESSAGE_TYPE.SYSTEM, `\u{1F3AF} [${finding.severity.toUpperCase()}] ${finding.title}`);
     });
-    agent.on("complete", () => {
+    agent.on(AGENT_EVENT.COMPLETE, () => {
       const duration = stopTimer();
-      addMessage("system", `\u2713 Complete (${duration}s)`);
+      addMessage(MESSAGE_TYPE.SYSTEM, `\u2713 Complete (${duration}s)`);
       setIsProcessing(false);
       setCurrentStatus("");
     });
-    agent.on("error", (error) => {
+    agent.on(AGENT_EVENT.ERROR, (error) => {
       stopTimer();
-      addMessage("error", error.message);
+      addMessage(MESSAGE_TYPE.ERROR, error.message);
       setIsProcessing(false);
     });
     return () => {
@@ -2690,14 +2736,14 @@ var App = ({ autoApprove = false, target }) => {
     const trimmed = value.trim();
     if (!trimmed || isProcessing) return;
     setInput("");
-    addMessage("user", trimmed);
+    addMessage(MESSAGE_TYPE.USER, trimmed);
     if (trimmed.startsWith("/")) {
       const [cmd, ...args] = trimmed.slice(1).split(" ");
       switch (cmd) {
-        case "help":
+        case CLI_COMMAND.HELP:
         case "h":
           addMessage(
-            "system",
+            MESSAGE_TYPE.SYSTEM,
             `/target <ip>     Set target
 /start [goal]    Start pentest  
 /stop            Stop operation
@@ -2706,16 +2752,16 @@ var App = ({ autoApprove = false, target }) => {
 /exit            Exit`
           );
           return;
-        case "target":
+        case CLI_COMMAND.TARGET:
         case "t":
           if (args[0]) {
             agent.setTarget(args[0]);
-            addMessage("system", `Target \u2192 ${args[0]}`);
+            addMessage(MESSAGE_TYPE.SYSTEM, `Target \u2192 ${args[0]}`);
           } else {
-            addMessage("error", "Usage: /target <ip>");
+            addMessage(MESSAGE_TYPE.ERROR, "Usage: /target <ip>");
           }
           return;
-        case "start":
+        case CLI_COMMAND.START:
         case "s":
           setIsProcessing(true);
           startTimer();
@@ -2724,39 +2770,39 @@ var App = ({ autoApprove = false, target }) => {
           try {
             await agent.runAutonomous(objective);
           } catch (e) {
-            addMessage("error", e instanceof Error ? e.message : String(e));
+            addMessage(MESSAGE_TYPE.ERROR, e instanceof Error ? e.message : String(e));
           }
           stopTimer();
           setIsProcessing(false);
           setCurrentStatus("");
           return;
-        case "stop":
+        case CLI_COMMAND.STOP:
           agent.pause();
           stopTimer();
-          addMessage("system", "Stopped.");
+          addMessage(MESSAGE_TYPE.SYSTEM, "Stopped.");
           setIsProcessing(false);
           setCurrentStatus("");
           return;
-        case "findings":
+        case CLI_COMMAND.FINDINGS:
         case "f":
           const findings = agent.getState().findings;
           if (findings.length === 0) {
-            addMessage("system", "No findings.");
+            addMessage(MESSAGE_TYPE.SYSTEM, "No findings.");
           } else {
-            findings.forEach((f) => addMessage("system", `[${f.severity}] ${f.title}`));
+            findings.forEach((f) => addMessage(MESSAGE_TYPE.SYSTEM, `[${f.severity}] ${f.title}`));
           }
           return;
-        case "clear":
+        case CLI_COMMAND.CLEAR:
         case "c":
           setMessages([]);
           return;
-        case "exit":
+        case CLI_COMMAND.EXIT:
         case "quit":
         case "q":
           exit();
           return;
         default:
-          addMessage("error", `Unknown: ${cmd}`);
+          addMessage(MESSAGE_TYPE.ERROR, `Unknown: ${cmd}`);
           return;
       }
     }
@@ -2765,9 +2811,9 @@ var App = ({ autoApprove = false, target }) => {
     setCurrentStatus("Thinking...");
     try {
       await agent.processUserHint(trimmed);
-      addMessage("system", "Hint received.");
+      addMessage(MESSAGE_TYPE.SYSTEM, "Hint received.");
     } catch (e) {
-      addMessage("error", e instanceof Error ? e.message : String(e));
+      addMessage(MESSAGE_TYPE.ERROR, e instanceof Error ? e.message : String(e));
     }
     stopTimer();
     setIsProcessing(false);
@@ -2780,7 +2826,7 @@ var App = ({ autoApprove = false, target }) => {
         stopTimer();
         setIsProcessing(false);
         setCurrentStatus("");
-        addMessage("system", "Interrupted.");
+        addMessage(MESSAGE_TYPE.SYSTEM, "Interrupted.");
       } else {
         exit();
       }
@@ -2788,15 +2834,15 @@ var App = ({ autoApprove = false, target }) => {
   });
   const getStyle = (type) => {
     const styles = {
-      user: { color: THEME.text.accent, prefix: "\u276F" },
-      assistant: { color: THEME.text.primary, prefix: "\u25C8" },
-      tool: { color: THEME.text.muted, prefix: " ", dim: true },
-      result: { color: THEME.text.muted, prefix: " ", dim: true },
-      thinking: { color: THEME.status.running, prefix: "\u25CB" },
-      error: { color: THEME.status.error, prefix: "\u2717" },
-      system: { color: THEME.text.muted, prefix: "\u2022" }
+      [MESSAGE_TYPE.USER]: { color: THEME.text.accent, prefix: "\u276F" },
+      [MESSAGE_TYPE.ASSISTANT]: { color: THEME.text.primary, prefix: "\u25C8" },
+      [MESSAGE_TYPE.TOOL]: { color: THEME.text.muted, prefix: " ", dim: true },
+      [MESSAGE_TYPE.RESULT]: { color: THEME.text.muted, prefix: " ", dim: true },
+      [MESSAGE_TYPE.THINKING]: { color: THEME.status.running, prefix: "\u25CB" },
+      [MESSAGE_TYPE.ERROR]: { color: THEME.status.error, prefix: "\u2717" },
+      [MESSAGE_TYPE.SYSTEM]: { color: THEME.text.muted, prefix: "\u2022" }
     };
-    return styles[type] || styles.system;
+    return styles[type] || styles[MESSAGE_TYPE.SYSTEM];
   };
   const state = agent.getState();
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", paddingX: 1, children: [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pentesting",
-  "version": "0.1.8",
+  "version": "0.1.9",
   "description": "Autonomous Penetration Testing AI Agent",
   "type": "module",
   "main": "dist/index.js",