pentesting 0.2.6 → 0.3.0

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.3.0";
 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;
@@ -4254,45 +4254,51 @@ var THEME = {
     // Purple-gray
     muted: "#6b6b7d",
     // Muted purple-gray
-    accent: "#b794f6"
+    accent: "#b794f6",
     // Soft purple (pentesting identity)
+    highlight: "#f0abfc"
+    // Pink highlight
   },
-  // Status colors (cyber-security themed)
+  // Status colors (cyber-security themed) - Enhanced!
   status: {
-    success: "#6ee7b7",
-    // Mint green (shell access)
-    warning: "#fcd34d",
-    // Amber (vulnerabilities)
-    error: "#fca5a5",
-    // Soft red (failed)
-    info: "#93c5fd",
-    // Light blue (scanning)
-    running: "#c4b5fd"
+    success: "#4ade80",
+    // Bright green (shell access)
+    warning: "#fbbf24",
+    // Golden amber (vulnerabilities)
+    error: "#f87171",
+    // Coral red (failed)
+    info: "#60a5fa",
+    // Sky blue (scanning)
+    running: "#a78bfa",
     // Violet (active operations)
+    pending: "#facc15"
+    // Yellow (waiting)
   },
-  // Severity colors (CVE-style)
+  // Severity colors (CVE-style) - More vibrant!
   semantic: {
-    critical: "#dc2626",
-    // Critical (red)
-    high: "#ea580c",
-    // High (orange)
-    medium: "#d97706",
-    // Medium (amber)
-    low: "#16a34a",
-    // Low (green)
-    info: "#7c3aed"
-    // Informational (purple)
+    critical: "#ef4444",
+    // Vivid red
+    high: "#f97316",
+    // Bright orange
+    medium: "#eab308",
+    // Vivid yellow
+    low: "#22c55e",
+    // Bright green
+    info: "#8b5cf6"
+    // Violet
   },
   // Border colors (purple-tinted)
   border: {
     default: "#2e2e42",
     // Subtle purple-gray
-    focus: "#8b5cf6",
+    focus: "#a78bfa",
     // Violet focus
-    error: "#f87171"
+    error: "#f87171",
     // Red error
+    success: "#4ade80"
+    // Green success
   },
-  // Phase colors (attack lifecycle)
+  // Phase colors (attack lifecycle) - Vibrant gradient-inspired
   phase: {
     recon: "#818cf8",
     // Indigo (reconnaissance)
@@ -4300,14 +4306,40 @@ var THEME = {
     // Emerald (enumeration)
     vuln: "#fbbf24",
     // Amber (vulnerability)
-    exploit: "#f97316",
+    exploit: "#fb923c",
     // Orange (exploitation)
-    privesc: "#ef4444",
+    privesc: "#f87171",
     // Red (privilege escalation)
-    persist: "#a78bfa",
+    persist: "#c084fc",
     // Purple (persistence)
     report: "#22d3ee"
     // Cyan (reporting)
+  },
+  // Rich accent colors for UI elements
+  accent: {
+    purple: "#a855f7",
+    violet: "#8b5cf6",
+    indigo: "#6366f1",
+    blue: "#3b82f6",
+    cyan: "#06b6d4",
+    teal: "#14b8a6",
+    emerald: "#10b981",
+    green: "#22c55e",
+    lime: "#84cc16",
+    yellow: "#eab308",
+    amber: "#f59e0b",
+    orange: "#f97316",
+    red: "#ef4444",
+    pink: "#ec4899",
+    rose: "#f43f5e"
+  },
+  // Gradients (for special UI elements)
+  gradient: {
+    purple: ["#7c3aed", "#a855f7"],
+    cyber: ["#06b6d4", "#8b5cf6"],
+    danger: ["#ef4444", "#f97316"],
+    success: ["#10b981", "#22c55e"],
+    gold: ["#f59e0b", "#fbbf24"]
   }
 };
 var ASCII_BANNER = `
@@ -4326,8 +4358,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 +4638,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 +4673,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 +4709,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 +4742,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 +4781,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 +4791,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 +4810,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.3.0",
   "description": "Autonomous Penetration Testing AI Agent",
   "type": "module",
   "main": "dist/index.js",