pentesting 0.2.5 → 0.2.7

package/dist/index.js

@@ -1376,7 +1376,7 @@ const { chromium } = require('playwright');
 }
 
 // src/config/constants.ts
-var APP_VERSION = "0.2.5";
+var APP_VERSION = "0.2.7";
 var APP_DESCRIPTION = "Autonomous Penetration Testing AI Agent";
 var LLM_API_KEY = process.env.PENTEST_API_KEY || process.env.ANTHROPIC_API_KEY || "";
 var LLM_BASE_URL = process.env.PENTEST_BASE_URL || void 0;
@@ -4326,7 +4326,10 @@ var App = ({ autoApprove = false, target }) => {
   const [currentStatus, setCurrentStatus] = useState("");
   const [elapsedTime, setElapsedTime] = useState(0);
   const [pendingApproval, setPendingApproval] = useState(null);
+  const [approvalSelectedIndex, setApprovalSelectedIndex] = useState(0);
   const [tokenUsage, setTokenUsage] = useState({ input: 0, output: 0, total: 0 });
+  const [showCommandHints, setShowCommandHints] = useState(false);
+  const [mode, setMode] = useState("agent");
   const [agent] = useState(() => new AutonomousHackingAgent(void 0, { autoApprove }));
   const sessionManager2 = getSessionManager();
   const approvalManager2 = getApprovalManager({ yoloMode: autoApprove });
@@ -4397,10 +4400,15 @@ var App = ({ autoApprove = false, target }) => {
       setPendingApproval({
         id: data.id,
         toolName: data.toolName,
+        toolInput: data.toolInput,
         riskLevel: data.riskLevel
       });
       addMessage(MESSAGE_TYPE.SYSTEM, `\u26A0\uFE0F  APPROVAL NEEDED: ${data.toolName} (${data.riskLevel} risk)`);
-      addMessage(MESSAGE_TYPE.SYSTEM, "   Type /approve or /deny");
+      const inputPreview = Object.entries(data.toolInput).slice(0, 2).map(([k, v]) => `${k}=${typeof v === "string" ? v.slice(0, 40) : JSON.stringify(v).slice(0, 40)}`).join(", ");
+      if (inputPreview) {
+        addMessage(MESSAGE_TYPE.SYSTEM, `   Args: ${inputPreview}`);
+      }
+      addMessage(MESSAGE_TYPE.SYSTEM, "   /y = approve, /n = deny, /ya = always approve");
     });
     agent.on(AGENT_EVENT.COMPLETE, () => {
       const duration = stopTimer();
@@ -4417,11 +4425,12 @@ var App = ({ autoApprove = false, target }) => {
       setPendingApproval({
         id: req.id,
         toolName: req.toolName,
+        toolInput: req.toolInput,
         riskLevel: req.riskLevel
       });
       addMessage(MESSAGE_TYPE.SYSTEM, `\u26A0\uFE0F  APPROVAL NEEDED: ${req.toolName} (${req.riskLevel} risk)`);
       addMessage(MESSAGE_TYPE.SYSTEM, `   ${req.reason}`);
-      addMessage(MESSAGE_TYPE.SYSTEM, "   Type /approve or /deny");
+      addMessage(MESSAGE_TYPE.SYSTEM, "   /y = approve, /n = deny, /ya = always approve");
     });
     return () => {
       if (timerRef.current) clearInterval(timerRef.current);
@@ -4430,13 +4439,29 @@ var App = ({ autoApprove = false, target }) => {
   const handleSubmit = useCallback(async (value) => {
     const trimmed = value.trim();
     if (!trimmed) return;
-    if (pendingApproval && (trimmed === "/approve" || trimmed === "/deny" || trimmed === "/y" || trimmed === "/n")) {
-      const decision = trimmed === "/approve" || trimmed === "/y" ? "approve" : "deny";
-      approvalManager2.respond(pendingApproval.id, decision);
-      addMessage(MESSAGE_TYPE.SYSTEM, decision === "approve" ? "\u2713 Approved" : "\u2717 Denied");
-      setPendingApproval(null);
-      setInput("");
-      return;
+    setShowCommandHints(false);
+    if (pendingApproval) {
+      if (trimmed === "/approve" || trimmed === "/y") {
+        approvalManager2.respond(pendingApproval.id, "approve");
+        addMessage(MESSAGE_TYPE.SYSTEM, "\u2713 Approved");
+        setPendingApproval(null);
+        setInput("");
+        return;
+      }
+      if (trimmed === "/deny" || trimmed === "/n") {
+        approvalManager2.respond(pendingApproval.id, "deny");
+        addMessage(MESSAGE_TYPE.SYSTEM, "\u2717 Denied");
+        setPendingApproval(null);
+        setInput("");
+        return;
+      }
+      if (trimmed === "/ya") {
+        approvalManager2.respond(pendingApproval.id, "approve_always");
+        addMessage(MESSAGE_TYPE.SYSTEM, `\u2713 Approved (always for ${pendingApproval.toolName})`);
+        setPendingApproval(null);
+        setInput("");
+        return;
+      }
     }
     if (isProcessing && !trimmed.startsWith("/")) return;
     setInput("");
@@ -4581,6 +4606,27 @@ var App = ({ autoApprove = false, target }) => {
           return;
       }
     }
+    if (mode === "shell") {
+      setIsProcessing(true);
+      startTimer();
+      setCurrentStatus(`Running: ${trimmed}`);
+      try {
+        const { execSync } = await import("child_process");
+        const output = execSync(trimmed, {
+          encoding: "utf-8",
+          timeout: 3e4,
+          maxBuffer: 1024 * 1024
+        }).trim();
+        addMessage(MESSAGE_TYPE.RESULT, output || "(no output)");
+      } catch (e) {
+        const error = e;
+        addMessage(MESSAGE_TYPE.ERROR, error.stderr?.toString() || error.message || "Command failed");
+      }
+      stopTimer();
+      setIsProcessing(false);
+      setCurrentStatus("");
+      return;
+    }
     setIsProcessing(true);
     startTimer();
     setCurrentStatus("Thinking...");
@@ -4595,8 +4641,31 @@ var App = ({ autoApprove = false, target }) => {
     stopTimer();
     setIsProcessing(false);
     setCurrentStatus("");
-  }, [agent, isProcessing, pendingApproval, addMessage, exit, startTimer, stopTimer, sessionManager2, approvalManager2]);
+  }, [agent, isProcessing, pendingApproval, addMessage, exit, startTimer, stopTimer, sessionManager2, approvalManager2, mode]);
+  const approvalOptions = [
+    { label: "\u2713 Approve once", decision: "approve" },
+    { label: "\u2713 Approve always (this session)", decision: "approve_always" },
+    { label: "\u2717 Deny", decision: "deny" }
+  ];
   useInput((input2, key) => {
+    if (pendingApproval) {
+      if (key.upArrow) {
+        setApprovalSelectedIndex((i) => (i - 1 + approvalOptions.length) % approvalOptions.length);
+        return;
+      }
+      if (key.downArrow) {
+        setApprovalSelectedIndex((i) => (i + 1) % approvalOptions.length);
+        return;
+      }
+      if (key.return) {
+        const selected = approvalOptions[approvalSelectedIndex];
+        approvalManager2.respond(pendingApproval.id, selected.decision);
+        addMessage(MESSAGE_TYPE.SYSTEM, selected.decision === "deny" ? "\u2717 Denied" : `\u2713 Approved${selected.decision === "approve_always" ? " (always)" : ""}`);
+        setPendingApproval(null);
+        setApprovalSelectedIndex(0);
+        return;
+      }
+    }
     if (key.ctrl && input2 === "c") {
       if (isProcessing) {
         agent.pause();
@@ -4608,6 +4677,11 @@ var App = ({ autoApprove = false, target }) => {
         exit();
       }
     }
+    if (key.ctrl && input2 === "x") {
+      const newMode = mode === "agent" ? "shell" : "agent";
+      setMode(newMode);
+      addMessage(MESSAGE_TYPE.SYSTEM, `Mode: ${newMode === "agent" ? "\u{1F916} Agent" : "$ Shell"}`);
+    }
   });
   const getStyle = (type) => {
     const styles = {
@@ -4636,13 +4710,21 @@ var App = ({ autoApprove = false, target }) => {
         ] })
       ] }) }, msg.id);
     } }) }),
-    pendingApproval && /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsxs(Text, { color: THEME.status.error, bold: true, children: [
-      "\u26A0\uFE0F  Awaiting approval for ",
-      pendingApproval.toolName,
-      " (",
-      pendingApproval.riskLevel,
-      ")"
-    ] }) }),
+    pendingApproval && /* @__PURE__ */ jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "yellow", paddingX: 1, marginBottom: 1, children: [
+      /* @__PURE__ */ jsxs(Text, { color: "yellow", bold: true, children: [
+        "\u26A0\uFE0F  APPROVAL NEEDED: ",
+        pendingApproval.toolName,
+        " (",
+        pendingApproval.riskLevel,
+        " risk)"
+      ] }),
+      /* @__PURE__ */ jsx(Text, { dimColor: true, children: Object.entries(pendingApproval.toolInput).slice(0, 2).map(([k, v]) => `${k}=${typeof v === "string" ? v.slice(0, 50) : JSON.stringify(v).slice(0, 50)}`).join(", ") }),
+      /* @__PURE__ */ jsx(Box, { flexDirection: "column", marginTop: 1, children: approvalOptions.map((opt, idx) => /* @__PURE__ */ jsxs(Text, { color: idx === approvalSelectedIndex ? "cyan" : "gray", children: [
+        idx === approvalSelectedIndex ? "\u2192 " : "  ",
+        opt.label
+      ] }, opt.decision)) }),
+      /* @__PURE__ */ jsx(Box, { marginTop: 1, children: /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2191\u2193 to select, Enter to confirm, or type /y /n /ya" }) })
+    ] }),
     isProcessing ? /* @__PURE__ */ jsxs(Box, { children: [
       /* @__PURE__ */ jsx(Text, { color: THEME.status.running, children: /* @__PURE__ */ jsx(Spinner, { type: "dots" }) }),
       /* @__PURE__ */ jsxs(Text, { color: THEME.text.muted, children: [
@@ -4654,20 +4736,38 @@ var App = ({ autoApprove = false, target }) => {
           "s)"
         ] })
       ] })
-    ] }) : /* @__PURE__ */ jsxs(Box, { children: [
-      /* @__PURE__ */ jsx(Text, { color: THEME.status.success, children: "\u276F " }),
-      /* @__PURE__ */ jsx(
-        TextInput,
-        {
-          value: input,
-          onChange: setInput,
-          onSubmit: handleSubmit,
-          placeholder: "Message or /help..."
-        }
-      )
+    ] }) : /* @__PURE__ */ jsxs(Box, { flexDirection: "column", children: [
+      showCommandHints && input.startsWith("/") && /* @__PURE__ */ jsx(Box, { flexDirection: "column", marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { dimColor: true, children: [
+        "/target <ip>",
+        "/start",
+        "/stop",
+        "/findings",
+        "/sessions",
+        "/yolo",
+        "/clear",
+        "/exit",
+        pendingApproval ? "/y /n /ya" : ""
+      ].filter((cmd) => cmd && cmd.toLowerCase().includes(input.toLowerCase().slice(1))).slice(0, 5).join("  \u2502  ") }) }),
+      /* @__PURE__ */ jsxs(Box, { children: [
+        /* @__PURE__ */ jsx(Text, { color: mode === "agent" ? THEME.status.success : "yellow", children: mode === "agent" ? "\u2728 " : "$ " }),
+        /* @__PURE__ */ jsx(
+          TextInput,
+          {
+            value: input,
+            onChange: (val) => {
+              setInput(val);
+              setShowCommandHints(val.startsWith("/") && val.length > 0);
+            },
+            onSubmit: handleSubmit,
+            placeholder: mode === "agent" ? "Message or /help..." : "Shell command..."
+          }
+        )
+      ] })
     ] }),
     /* @__PURE__ */ jsxs(Box, { marginTop: 1, justifyContent: "space-between", children: [
       /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
+        mode === "agent" ? "\u{1F916}" : "$",
+        " ",
         state.target.primary || "No target",
         " \u2502",
         state.findings.length,
@@ -4678,7 +4778,7 @@ var App = ({ autoApprove = false, target }) => {
         state.currentPhase !== AGENT_STATUS.IDLE && ` ${state.currentPhase} \u2502`
       ] }),
       /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
-        "/help \u2502 Ctrl+C ",
+        "Ctrl+X mode \u2502 /help \u2502 Ctrl+C ",
         isProcessing ? "stop" : "exit"
       ] })
     ] })

package/package.json

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