pentesting 0.1.4 → 0.1.6

package/dist/index.js

@@ -6,7 +6,8 @@ import { Command } from "commander";
 
 // src/cli/app.tsx
 import { useState, useEffect, useCallback } from "react";
-import { Box, Text, useApp, useInput, Static } from "ink";
+import { Box, Text, useInput, useApp, Static } from "ink";
+import TextInput from "ink-text-input";
 import Spinner from "ink-spinner";
 
 // src/core/agent/autonomous-agent.ts
@@ -2608,397 +2609,191 @@ var ASCII_BANNER = `
  \u2551  \u25C8 PENTESTING - Autonomous Penetration Testing Agent      \u2551
  \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
 `;
-var ICONS = {
-  // Status
-  success: "\u2713",
-  error: "\u2717",
-  warning: "\u26A0",
-  info: "\u2139",
-  running: "\u25C9",
-  thinking: "\u25D0",
-  // Actions (pentesting-specific)
-  target: "\u25C8",
-  // Diamond for target
-  scan: "\u25CE",
-  exploit: "\u26A1",
-  shell: "\u276F",
-  // Progress
-  pending: "\u25CB",
-  completed: "\u25CF",
-  skipped: "\u25CC",
-  // Findings (severity)
-  critical: "\u25BC",
-  high: "\u25BD",
-  medium: "\u25C7",
-  low: "\u25E6",
-  // Security
-  lock: "\u{1F512}",
-  unlock: "\u{1F513}",
-  key: "\u{1F511}",
-  flag: "\u2691",
-  // Simple flag
-  pwned: "\u25C8"
-  // Compromised
-};
 
 // src/cli/app.tsx
 import { jsx, jsxs } from "react/jsx-runtime";
-var App = ({
-  apiKey,
-  target: initialTarget,
-  autoStart = false,
-  autoApprove = false,
-  objective: initialObjective,
-  maxIterations = 200
-}) => {
+var App = ({ autoApprove = false, target }) => {
   const { exit } = useApp();
-  const [agent] = useState(() => new AutonomousHackingAgent(apiKey, { maxIterations, autoApprove }));
-  const [appState, setAppState] = useState({
-    status: "idle",
-    target: initialTarget || "",
-    currentPhase: "recon",
-    iteration: 0,
-    maxIterations,
-    findings: 0,
-    compromised: [],
-    credentials: 0,
-    services: 0
-  });
-  const [phases, setPhases] = useState([]);
-  const [thoughts, setThoughts] = useState([]);
-  const [logs, setLogs] = useState([]);
+  const [messages, setMessages] = useState([]);
   const [input, setInput] = useState("");
-  const [showThoughts, setShowThoughts] = useState(true);
-  const [isThinking, setIsThinking] = useState(false);
-  const addLog = useCallback((type, message) => {
-    const entry = {
-      id: `log-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`,
+  const [isProcessing, setIsProcessing] = useState(false);
+  const [agent] = useState(() => new AutonomousHackingAgent(void 0, { autoApprove }));
+  const addMessage = useCallback((type, content) => {
+    setMessages((prev) => [...prev, {
+      id: `${Date.now()}-${Math.random().toString(36).slice(2)}`,
       type,
-      message,
+      content,
       timestamp: /* @__PURE__ */ new Date()
-    };
-    setLogs((prev) => [...prev.slice(-50), entry]);
+    }]);
   }, []);
   useEffect(() => {
-    agent.on("thought", (thought) => {
-      setThoughts((prev) => [...prev.slice(-30), thought]);
-      setIsThinking(false);
-    });
-    agent.on("iteration", ({ current, max, phase }) => {
-      setAppState((prev) => ({
-        ...prev,
-        iteration: current,
-        maxIterations: max,
-        currentPhase: phase,
-        status: "running"
-      }));
-      setIsThinking(true);
+    addMessage("system", "Pentesting Agent ready. Type a message or use /help for commands.");
+    if (target) {
+      agent.setTarget(target);
+      addMessage("system", `Target set: ${target}`);
+    }
+    agent.on("thought", (type, content) => {
+      addMessage("assistant", `[${type}] ${content}`);
     });
-    agent.on("phase_change", ({ phaseId, newStatus }) => {
-      setAppState((prev) => ({ ...prev, currentPhase: phaseId }));
-      addLog("info", `[phase] ${phaseId} -> ${newStatus}`);
+    agent.on("tool_call", (data) => {
+      const inputStr = Object.entries(data.input).map(([k, v]) => `${k}=${typeof v === "string" ? v : JSON.stringify(v)}`).join(" ");
+      addMessage("tool", `\u25B6 ${data.name} ${inputStr}`);
     });
-    agent.on("tool_call", ({ name, input: input2 }) => {
-      addLog("tool", `[tool] ${name}: ${JSON.stringify(input2).slice(0, 100)}...`);
-    });
-    agent.on("tool_result", ({ name, result }) => {
-      if (result.success) {
-        addLog("success", `[+] ${name} completed (${result.duration}ms)`);
-      } else {
-        addLog("error", `[-] ${name} failed: ${result.error}`);
-      }
+    agent.on("tool_result", (data) => {
+      const status = data.result.success ? "\u2713" : "\u2717";
+      const output = data.result.output?.slice(0, 200) || "";
+      addMessage("tool", `${status} ${data.name}: ${output}`);
     });
     agent.on("finding", (finding) => {
-      addLog("finding", `[!] [${finding.severity.toUpperCase()}] ${finding.title}`);
-      setAppState((prev) => ({ ...prev, findings: prev.findings + 1 }));
-    });
-    agent.on("credential", (cred) => {
-      addLog("success", `[cred] ${cred.type} - ${cred.username || "unknown"}`);
-      setAppState((prev) => ({ ...prev, credentials: prev.credentials + 1 }));
-    });
-    agent.on("compromised", (host) => {
-      addLog("success", `[pwned] ${host}`);
-      setAppState((prev) => ({
-        ...prev,
-        compromised: [...prev.compromised, host]
-      }));
-    });
-    agent.on("complete", (summary) => {
-      setAppState((prev) => ({ ...prev, status: "completed" }));
-      addLog("success", `[done] Assessment complete: ${JSON.stringify(summary)}`);
+      addMessage("system", `\u{1F3AF} Finding: [${finding.severity.toUpperCase()}] ${finding.title}`);
     });
     agent.on("error", (error) => {
-      addLog("error", `[error] ${error.message}`);
-    });
-    agent.on("report", (report) => {
-      addLog("info", `[report] Generated (${report.length} chars)`);
+      addMessage("error", `Error: ${error.message}`);
     });
-    const state = agent.getState();
-    setPhases(state.phases);
-    if (autoStart && initialTarget) {
-      agent.setTarget(initialTarget);
-      setAppState((prev) => ({ ...prev, target: initialTarget }));
-      agent.runAutonomous(initialObjective);
-    }
-    return () => {
-      agent.removeAllListeners();
-    };
-  }, [agent, addLog, autoStart, initialTarget, initialObjective]);
-  useEffect(() => {
-    const interval = setInterval(() => {
-      const state = agent.getState();
-      setPhases(state.phases);
-      setAppState((prev) => ({
-        ...prev,
-        services: state.target.services.length,
-        credentials: state.target.credentials.length,
-        compromised: state.target.compromised,
-        findings: state.findings.length
-      }));
-    }, 1e3);
-    return () => clearInterval(interval);
-  }, [agent]);
-  useInput((key, mods) => {
-    if (mods.ctrl && key === "c") {
-      exit();
-      return;
-    }
-    if (key === "t") {
-      setShowThoughts((prev) => !prev);
-    }
-    if (key === "p" && appState.status === "running") {
-      agent.pause();
-      setAppState((prev) => ({ ...prev, status: "paused" }));
+  }, [agent, target, addMessage]);
+  const handleSubmit = useCallback(async (value) => {
+    const trimmed = value.trim();
+    if (!trimmed || isProcessing) return;
+    setInput("");
+    addMessage("user", trimmed);
+    if (trimmed.startsWith("/")) {
+      const [cmd, ...args] = trimmed.slice(1).split(" ");
+      switch (cmd) {
+        case "help":
+          addMessage("system", `Commands:
+  /target <ip>    Set target
+  /start          Start autonomous mode
+  /stop           Stop current operation
+  /findings       Show findings
+  /clear          Clear messages
+  /exit           Exit`);
+          return;
+        case "target":
+          if (args[0]) {
+            agent.setTarget(args[0]);
+            addMessage("system", `Target set: ${args[0]}`);
+          } else {
+            addMessage("error", "Usage: /target <ip or domain>");
+          }
+          return;
+        case "start":
+          const objective = args.join(" ") || "Perform comprehensive penetration test";
+          setIsProcessing(true);
+          addMessage("system", `Starting: ${objective}`);
+          try {
+            await agent.run(objective);
+          } catch (e) {
+            addMessage("error", `Failed: ${e instanceof Error ? e.message : String(e)}`);
+          }
+          setIsProcessing(false);
+          return;
+        case "stop":
+          agent.stop();
+          addMessage("system", "Stopped.");
+          setIsProcessing(false);
+          return;
+        case "findings":
+          const findings = agent.getState().findings;
+          if (findings.length === 0) {
+            addMessage("system", "No findings yet.");
+          } else {
+            findings.forEach((f) => {
+              addMessage("system", `[${f.severity.toUpperCase()}] ${f.title}`);
+            });
+          }
+          return;
+        case "clear":
+          setMessages([]);
+          addMessage("system", "Cleared.");
+          return;
+        case "exit":
+        case "quit":
+          exit();
+          return;
+        default:
+          addMessage("error", `Unknown command: ${cmd}. Type /help`);
+          return;
+      }
     }
-    if (key === "r" && appState.status === "paused") {
-      agent.resume();
-      setAppState((prev) => ({ ...prev, status: "running" }));
+    setIsProcessing(true);
+    try {
+      await agent.chat(trimmed);
+    } catch (e) {
+      addMessage("error", `Error: ${e instanceof Error ? e.message : String(e)}`);
+    }
+    setIsProcessing(false);
+  }, [agent, isProcessing, addMessage, exit]);
+  useInput((input2, key) => {
+    if (key.ctrl && input2 === "c") {
+      if (isProcessing) {
+        agent.stop();
+        setIsProcessing(false);
+        addMessage("system", "Interrupted.");
+      } else {
+        exit();
+      }
     }
   });
-  const processCommand = useCallback((cmd) => {
-    const parts = cmd.trim().split(/\s+/);
-    const command = parts[0].toLowerCase();
-    const args = parts.slice(1).join(" ");
-    switch (command) {
-      case "/target":
-        if (args) {
-          agent.setTarget(args);
-          setAppState((prev) => ({ ...prev, target: args }));
-          addLog("info", `[target] Set: ${args}`);
-        }
-        break;
-      case "/start":
-      case "/auto":
-        if (appState.target) {
-          const objective = args || void 0;
-          addLog("info", `[*] Starting autonomous hacking...`);
-          agent.runAutonomous(objective);
-        } else {
-          addLog("error", "No target set. Use /target <ip> first.");
-        }
-        break;
-      case "/hint":
-        if (args && appState.status === "running") {
-          agent.processUserHint(args);
-          addLog("info", `[hint] Sent to agent`);
-        }
-        break;
-      case "/pause":
-        agent.pause();
-        setAppState((prev) => ({ ...prev, status: "paused" }));
-        break;
-      case "/resume":
-        agent.resume();
-        break;
-      case "/reset":
-        agent.reset();
-        setAppState({
-          status: "idle",
-          target: "",
-          currentPhase: "recon",
-          iteration: 0,
-          maxIterations,
-          findings: 0,
-          compromised: [],
-          credentials: 0,
-          services: 0
-        });
-        setThoughts([]);
-        setLogs([]);
-        addLog("info", "Session reset");
-        break;
-      case "/findings":
-        const state = agent.getState();
-        state.findings.forEach((f) => {
-          addLog("finding", `[${f.severity}] ${f.title}: ${f.description.slice(0, 100)}`);
-        });
-        break;
-      case "/help":
-        addLog("info", "Commands: /target <ip>, /start [objective], /hint <hint>, /pause, /resume, /reset, /findings, /help");
-        addLog("info", "Keys: [T] Toggle thoughts, [P] Pause, [R] Resume, Ctrl+C Exit");
-        break;
+  const getColor = (type) => {
+    switch (type) {
+      case "user":
+        return THEME.text.accent;
+      case "assistant":
+        return THEME.text.primary;
+      case "tool":
+        return THEME.status.info;
+      case "error":
+        return THEME.status.error;
+      case "system":
+        return THEME.text.muted;
       default:
-        if (appState.status === "running" && cmd) {
-          agent.processUserHint(cmd);
-          addLog("info", `[hint] ${cmd}`);
-        }
+        return THEME.text.primary;
     }
-    setInput("");
-  }, [agent, appState, addLog, maxIterations]);
-  const getThoughtStyle = (type) => {
-    const styles = {
-      observation: { prefix: "[observe]", color: THEME.text.accent },
-      hypothesis: { prefix: "[hypothesis]", color: THEME.status.warning },
-      plan: { prefix: "[plan]", color: THEME.status.running },
-      action: { prefix: "[action]", color: THEME.status.success },
-      result: { prefix: "[result]", color: THEME.text.primary },
-      reflection: { prefix: "[reflect]", color: THEME.semantic.high },
-      stuck: { prefix: "[stuck]", color: THEME.status.error },
-      breakthrough: { prefix: "[breakthrough]", color: "#00ff00" }
-    };
-    return styles[type] || { prefix: "[info]", color: THEME.text.muted };
   };
-  const getPhaseIcon = (status) => {
-    switch (status) {
-      case "completed":
-        return "[+]";
-      case "in_progress":
-        return "[*]";
-      case "failed":
-        return "[-]";
-      case "skipped":
-        return "[>]";
+  const getPrefix = (type) => {
+    switch (type) {
+      case "user":
+        return "\u276F";
+      case "assistant":
+        return "\u25C8";
+      case "tool":
+        return "\u2699";
+      case "error":
+        return "\u2717";
+      case "system":
+        return "\u2022";
       default:
-        return "[ ]";
+        return " ";
     }
   };
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", padding: 1, children: [
-    /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginBottom: 1, children: [
-      /* @__PURE__ */ jsx(Text, { color: THEME.status.success, children: ASCII_BANNER }),
-      /* @__PURE__ */ jsxs(Box, { marginTop: 1, justifyContent: "space-between", children: [
-        /* @__PURE__ */ jsxs(Text, { color: THEME.status.error, children: [
-          ICONS.target,
-          " Target: ",
-          /* @__PURE__ */ jsx(Text, { color: THEME.text.accent, bold: true, children: appState.target || "Not set" })
-        ] }),
-        /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, children: [
-          "Status: ",
-          /* @__PURE__ */ jsx(Text, { color: appState.status === "running" ? THEME.status.success : THEME.status.warning, children: appState.status.toUpperCase() })
-        ] }),
-        /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, children: [
-          "Iteration: ",
-          /* @__PURE__ */ jsxs(Text, { color: THEME.text.accent, children: [
-            appState.iteration,
-            "/",
-            appState.maxIterations
-          ] })
-        ] })
-      ] })
-    ] }),
-    /* @__PURE__ */ jsxs(Box, { borderStyle: "single", borderColor: THEME.status.running, paddingX: 1, children: [
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.muted, children: "Services: " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.status.success, children: appState.services }),
-      /* @__PURE__ */ jsx(Text, { children: " | " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.muted, children: "Creds: " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.status.warning, children: appState.credentials }),
-      /* @__PURE__ */ jsx(Text, { children: " | " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.muted, children: "Pwned: " }),
-      /* @__PURE__ */ jsx(Text, { color: appState.compromised.length > 0 ? THEME.status.success : THEME.text.muted, children: appState.compromised.length > 0 ? appState.compromised.join(", ") : "0" }),
-      /* @__PURE__ */ jsx(Text, { children: " | " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.muted, children: "Findings: " }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.semantic.high, children: appState.findings })
-    ] }),
-    /* @__PURE__ */ jsxs(Box, { borderStyle: "single", borderColor: THEME.text.accent, paddingX: 1, marginTop: 1, children: [
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.accent, bold: true, children: "[phases] " }),
-      phases.map((phase) => /* @__PURE__ */ jsx(Box, { marginRight: 1, children: /* @__PURE__ */ jsxs(
-        Text,
+    /* @__PURE__ */ jsx(Box, { flexDirection: "column", marginBottom: 1, children: /* @__PURE__ */ jsx(Static, { items: messages.slice(-30), children: (msg) => /* @__PURE__ */ jsx(Box, { children: /* @__PURE__ */ jsxs(Text, { color: getColor(msg.type), children: [
+      getPrefix(msg.type),
+      " ",
+      msg.content
+    ] }) }, msg.id) }) }),
+    /* @__PURE__ */ jsx(Box, { children: isProcessing ? /* @__PURE__ */ jsxs(Text, { color: THEME.status.running, children: [
+      /* @__PURE__ */ jsx(Spinner, { type: "dots" }),
+      " Processing..."
+    ] }) : /* @__PURE__ */ jsxs(Box, { children: [
+      /* @__PURE__ */ jsx(Text, { color: THEME.status.success, children: "\u276F " }),
+      /* @__PURE__ */ jsx(
+        TextInput,
         {
-          color: phase.id === appState.currentPhase ? THEME.status.warning : phase.status === "completed" ? THEME.status.success : THEME.text.muted,
-          bold: phase.id === appState.currentPhase,
-          children: [
-            getPhaseIcon(phase.status),
-            phase.name,
-            phase.attempts > 0 && /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
-              "(",
-              phase.attempts,
-              ")"
-            ] })
-          ]
+          value: input,
+          onChange: setInput,
+          onSubmit: handleSubmit,
+          placeholder: "Type a message or /help..."
         }
-      ) }, phase.id))
-    ] }),
-    showThoughts && /* @__PURE__ */ jsxs(
-      Box,
-      {
-        flexDirection: "column",
-        borderStyle: "single",
-        borderColor: THEME.status.running,
-        paddingX: 1,
-        height: 10,
-        marginTop: 1,
-        children: [
-          /* @__PURE__ */ jsxs(Box, { children: [
-            /* @__PURE__ */ jsx(Text, { color: THEME.status.running, bold: true, children: "[agent thoughts]" }),
-            isThinking && /* @__PURE__ */ jsxs(Box, { marginLeft: 2, children: [
-              /* @__PURE__ */ jsx(Text, { color: THEME.status.warning, children: /* @__PURE__ */ jsx(Spinner, { type: "dots" }) }),
-              /* @__PURE__ */ jsx(Text, { color: THEME.status.warning, children: " Thinking..." })
-            ] })
-          ] }),
-          /* @__PURE__ */ jsx(Static, { items: thoughts.slice(-6), children: (thought) => {
-            const style = getThoughtStyle(thought.type);
-            return /* @__PURE__ */ jsxs(Box, { children: [
-              /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, dimColor: true, children: [
-                "[",
-                thought.timestamp.toLocaleTimeString(),
-                "]"
-              ] }),
-              /* @__PURE__ */ jsxs(Text, { color: style.color, children: [
-                " ",
-                style.prefix,
-                " ",
-                thought.content.slice(0, 120),
-                thought.content.length > 120 && "..."
-              ] })
-            ] }, thought.id);
-          } })
-        ]
-      }
-    ),
-    /* @__PURE__ */ jsxs(
-      Box,
-      {
-        flexDirection: "column",
-        borderStyle: "single",
-        borderColor: THEME.status.success,
-        paddingX: 1,
-        height: 12,
-        marginTop: 1,
-        children: [
-          /* @__PURE__ */ jsx(Text, { color: THEME.status.success, bold: true, children: "[activity log]" }),
-          /* @__PURE__ */ jsx(Static, { items: logs.slice(-10), children: (log) => /* @__PURE__ */ jsxs(Box, { children: [
-            /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, dimColor: true, children: [
-              "[",
-              log.timestamp.toLocaleTimeString(),
-              "]"
-            ] }),
-            /* @__PURE__ */ jsxs(Text, { color: log.type === "success" ? THEME.status.success : log.type === "error" ? THEME.status.error : log.type === "warning" ? THEME.status.warning : log.type === "finding" ? THEME.semantic.high : log.type === "tool" ? THEME.text.accent : THEME.text.primary, children: [
-              " ",
-              log.message
-            ] })
-          ] }, log.id) })
-        ]
-      }
-    ),
-    /* @__PURE__ */ jsxs(Box, { marginTop: 1, children: [
-      /* @__PURE__ */ jsxs(Text, { color: THEME.status.success, bold: true, children: [
-        "pentesting ",
-        ">",
-        " "
-      ] }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.text.primary, children: input }),
-      /* @__PURE__ */ jsx(Text, { color: THEME.status.success, children: "_" })
-    ] }),
-    /* @__PURE__ */ jsx(Box, { marginTop: 1, children: /* @__PURE__ */ jsx(Text, { color: THEME.text.muted, dimColor: true, children: "[T] Toggle thoughts | [P] Pause | [R] Resume | Type command or hint" }) })
+      )
+    ] }) }),
+    /* @__PURE__ */ jsx(Box, { marginTop: 1, children: /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, dimColor: true, children: [
+      agent.getState().target ? `Target: ${agent.getState().target}` : "No target",
+      " | Findings: ",
+      agent.getState().findings.length,
+      " | Ctrl+C to ",
+      isProcessing ? "stop" : "exit"
+    ] }) })
   ] });
 };
 var app_default = App;

package/package.json

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