pentesting 0.5.4 → 0.5.5

package/dist/index.js

@@ -3232,9 +3232,53 @@ ${prompt}`
   // ===== Target Setting =====
   setTarget(target) {
     this.state.target.primary = target;
+    if (!this.state.target.discovered.includes(target)) {
+      this.state.target.discovered.push(target);
+    }
     this.think(THOUGHT_TYPE.OBSERVATION, `Target Setting: ${target}`);
     this.emit(AGENT_EVENT.TARGET_SET, target);
   }
+  /**
+   * Add a target to the discovered list (multi-target support)
+   */
+  addTarget(target) {
+    if (this.state.target.discovered.includes(target)) {
+      return false;
+    }
+    this.state.target.discovered.push(target);
+    if (!this.state.target.primary) {
+      this.state.target.primary = target;
+    }
+    this.think(THOUGHT_TYPE.OBSERVATION, `Target added: ${target}`);
+    this.emit(AGENT_EVENT.TARGET_SET, { target, action: "added" });
+    return true;
+  }
+  /**
+   * Remove a target from the discovered list
+   */
+  removeTarget(target) {
+    const index = this.state.target.discovered.indexOf(target);
+    if (index === -1) {
+      return false;
+    }
+    this.state.target.discovered.splice(index, 1);
+    if (this.state.target.primary === target) {
+      this.state.target.primary = this.state.target.discovered[0] || "";
+    }
+    this.think(THOUGHT_TYPE.OBSERVATION, `Target removed: ${target}`);
+    this.emit(AGENT_EVENT.TARGET_SET, { target, action: "removed" });
+    return true;
+  }
+  /**
+   * Get all targets (primary + discovered)
+   */
+  getAllTargets() {
+    const targets = [...this.state.target.discovered];
+    if (this.state.target.primary && !targets.includes(this.state.target.primary)) {
+      targets.unshift(this.state.target.primary);
+    }
+    return targets;
+  }
   // ===== Phase Management =====
   getCurrentPhase() {
     return this.state.phases.find((p) => p.id === this.state.currentPhase);
@@ -5461,7 +5505,8 @@ var App = ({ autoApprove = false, target }) => {
           addMessage(
             MESSAGE_TYPE.SYSTEM,
             `\u2500\u2500 Core \u2500\u2500
-/target <ip>     Set target
+/target [domain|ip]  Set/show targets
+  add <t>  list  rm <t>  set <t>
 /start [goal]    Start autonomous pentest  
 /stop            Stop operation
 /status          Show status report
@@ -5497,11 +5542,76 @@ pentesting v${APP_VERSION}`
           return;
         case CLI_COMMAND.TARGET:
         case "t":
-          if (args[0]) {
-            agent.setTarget(args[0]);
-            addMessage(MESSAGE_TYPE.SYSTEM, `Target \u2192 ${args[0]}`);
-          } else {
-            addMessage(MESSAGE_TYPE.ERROR, "Usage: /target <ip>");
+          const subCmd = args[0]?.toLowerCase();
+          const targetArg = args.slice(1).join(" ") || args[0];
+          if (!subCmd) {
+            const allTargets = agent.getAllTargets();
+            if (allTargets.length === 0) {
+              addMessage(MESSAGE_TYPE.SYSTEM, `
+\u{1F3AF} No targets set
+
+Usage:
+  /target <domain|ip>       Set primary target
+  /target add <target>      Add target to list
+  /target list              Show all targets
+  /target rm <target>       Remove target
+  /target set <target>      Set as primary`);
+            } else {
+              const primary = agent.getState().target.primary;
+              const targetList = allTargets.map(
+                (t) => t === primary ? `  \u2605 ${t} (primary)` : `    ${t}`
+              ).join("\n");
+              addMessage(MESSAGE_TYPE.SYSTEM, `\u{1F3AF} Targets:
+${targetList}`);
+            }
+            return;
+          }
+          switch (subCmd) {
+            case "add":
+            case "+":
+              if (args[1]) {
+                const added = agent.addTarget(args[1]);
+                addMessage(MESSAGE_TYPE.SYSTEM, added ? `\u2713 Target added: ${args[1]}` : `Already exists: ${args[1]}`);
+              } else {
+                addMessage(MESSAGE_TYPE.ERROR, "Usage: /target add <domain|ip>");
+              }
+              break;
+            case "list":
+            case "ls":
+              const targets = agent.getAllTargets();
+              if (targets.length === 0) {
+                addMessage(MESSAGE_TYPE.SYSTEM, "No targets");
+              } else {
+                const primary = agent.getState().target.primary;
+                const list = targets.map(
+                  (t, i) => t === primary ? `  ${i + 1}. \u2605 ${t}` : `  ${i + 1}.   ${t}`
+                ).join("\n");
+                addMessage(MESSAGE_TYPE.SYSTEM, `\u{1F3AF} Targets (${targets.length}):
+${list}`);
+              }
+              break;
+            case "rm":
+            case "remove":
+            case "-":
+              if (args[1]) {
+                const removed = agent.removeTarget(args[1]);
+                addMessage(MESSAGE_TYPE.SYSTEM, removed ? `\u2713 Target removed: ${args[1]}` : `Not found: ${args[1]}`);
+              } else {
+                addMessage(MESSAGE_TYPE.ERROR, "Usage: /target rm <domain|ip>");
+              }
+              break;
+            case "set":
+            case "primary":
+              if (args[1]) {
+                agent.setTarget(args[1]);
+                addMessage(MESSAGE_TYPE.SYSTEM, `\u2605 Primary target \u2192 ${args[1]}`);
+              } else {
+                addMessage(MESSAGE_TYPE.ERROR, "Usage: /target set <domain|ip>");
+              }
+              break;
+            default:
+              agent.setTarget(subCmd);
+              addMessage(MESSAGE_TYPE.SYSTEM, `\u{1F3AF} Target \u2192 ${subCmd}`);
           }
           return;
         case CLI_COMMAND.START:
@@ -5987,12 +6097,14 @@ pentesting v${APP_VERSION}`
         mode === "agent" ? "\u{1F916}" : "$",
         " ",
         state.target.primary || "No target",
+        state.target.discovered.length > 1 && ` (+${state.target.discovered.length - 1})`,
         " \u2502",
         state.findings.length,
         " findings \u2502",
         state.credentials.length,
         " creds \u2502",
-        tokenUsage.total > 0 && `${(tokenUsage.total / 1e3).toFixed(1)}k tokens \u2502`,
+        `${(tokenUsage.total / 1e3).toFixed(1)}k tokens`,
+        " \u2502",
         state.currentPhase !== AGENT_STATUS.IDLE && ` ${state.currentPhase} \u2502`
       ] }),
       /* @__PURE__ */ jsxs2(Text2, { dimColor: true, children: [

package/package.json

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