pentesting 0.2.6 → 0.2.7

package/dist/index.js

@@ -1376,7 +1376,7 @@ const { chromium } = require('playwright');
 }
 
 // src/config/constants.ts
-var APP_VERSION = "0.2.6";
+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,8 +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 });
@@ -4604,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...");
@@ -4618,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();
@@ -4631,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 = {
@@ -4659,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: [
@@ -4690,7 +4749,7 @@ var App = ({ autoApprove = false, target }) => {
         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: THEME.status.success, children: "\u276F " }),
+        /* @__PURE__ */ jsx(Text, { color: mode === "agent" ? THEME.status.success : "yellow", children: mode === "agent" ? "\u2728 " : "$ " }),
         /* @__PURE__ */ jsx(
           TextInput,
           {
@@ -4700,13 +4759,15 @@ var App = ({ autoApprove = false, target }) => {
               setShowCommandHints(val.startsWith("/") && val.length > 0);
             },
             onSubmit: handleSubmit,
-            placeholder: "Message or /help..."
+            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,
@@ -4717,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.6",
+  "version": "0.2.7",
   "description": "Autonomous Penetration Testing AI Agent",
   "type": "module",
   "main": "dist/index.js",