Initial commit

Co-Authored-By: Nellie Mullane <nellie@keygraph.io>
2026-06-07 16:03:58 +02:00 · 2025-10-03 19:35:08 -07:00
commit 8f52722d56
69 changed files with 16062 additions and 0 deletions
@@ -0,0 +1,136 @@
+import chalk from 'chalk';
+import {
+  selectSession, deleteSession, deleteAllSessions,
+  validateAgent, validatePhase
+} from '../session-manager.js';
+import {
+  runPhase, runAll, rollbackTo, rerunAgent, displayStatus, listAgents
+} from '../checkpoint-manager.js';
+import { logError, PentestError } from '../error-handling.js';
+import { cleanupMCP } from '../setup/environment.js';
+
+// Developer command handlers
+export async function handleDeveloperCommand(command, args, pipelineTestingMode, runClaudePromptWithRetry, loadPrompt) {
+  try {
+    let session;
+
+    // Commands that don't require session selection
+    if (command === '--list-agents') {
+      listAgents();
+      return;
+    }
+
+    if (command === '--cleanup') {
+      // Handle cleanup without needing session selection first
+      if (args[0]) {
+        // Cleanup specific session by ID
+        const sessionId = args[0];
+        const deletedSession = await deleteSession(sessionId);
+        console.log(chalk.green(`✅ Deleted session ${sessionId} (${new URL(deletedSession.webUrl).hostname})`));
+        // Clean up MCP agents when deleting specific session
+        await cleanupMCP();
+      } else {
+        // Cleanup all sessions - require confirmation
+        console.log(chalk.yellow('⚠️  This will delete all pentest sessions. Are you sure? (y/N):'));
+        const { createInterface } = await import('readline');
+        const readline = createInterface({
+          input: process.stdin,
+          output: process.stdout
+        });
+
+        await new Promise((resolve) => {
+          readline.question('', (answer) => {
+            readline.close();
+            if (answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes') {
+              deleteAllSessions().then(deleted => {
+                if (deleted) {
+                  console.log(chalk.green('✅ All sessions deleted'));
+                } else {
+                  console.log(chalk.yellow('⚠️  No sessions found to delete'));
+                }
+                // Clean up MCP agents after deleting sessions
+                return cleanupMCP();
+              }).then(() => {
+                resolve();
+              }).catch(error => {
+                console.log(chalk.red(`❌ Failed to delete sessions: ${error.message}`));
+                resolve();
+              });
+            } else {
+              console.log(chalk.gray('Cleanup cancelled'));
+              resolve();
+            }
+          });
+        });
+      }
+      return;
+    }
+
+    // Early validation for commands with agent names (before session selection)
+
+    if (command === '--run-phase') {
+      if (!args[0]) {
+        console.log(chalk.red('❌ --run-phase requires a phase name'));
+        console.log(chalk.gray('Usage: ./shannon.mjs --run-phase <phase-name>'));
+        process.exit(1);
+      }
+      validatePhase(args[0]); // This will throw PentestError if invalid
+    }
+
+    if (command === '--rollback-to' || command === '--rerun') {
+      if (!args[0]) {
+        console.log(chalk.red(`❌ ${command} requires an agent name`));
+        console.log(chalk.gray(`Usage: ./shannon.mjs ${command} <agent-name>`));
+        process.exit(1);
+      }
+      validateAgent(args[0]); // This will throw PentestError if invalid
+    }
+
+    // Get session for other commands
+    try {
+      session = await selectSession();
+    } catch (error) {
+      console.log(chalk.red(`❌ ${error.message}`));
+      process.exit(1);
+    }
+
+    switch (command) {
+
+      case '--run-phase':
+        await runPhase(args[0], session, pipelineTestingMode, runClaudePromptWithRetry, loadPrompt);
+        break;
+
+      case '--run-all':
+        await runAll(session, pipelineTestingMode, runClaudePromptWithRetry, loadPrompt);
+        break;
+
+      case '--rollback-to':
+        await rollbackTo(args[0], session);
+        break;
+
+      case '--rerun':
+        await rerunAgent(args[0], session, pipelineTestingMode, runClaudePromptWithRetry, loadPrompt);
+        break;
+
+      case '--status':
+        await displayStatus(session);
+        break;
+
+      default:
+        console.log(chalk.red(`❌ Unknown developer command: ${command}`));
+        console.log(chalk.gray('Use --help to see available commands'));
+        process.exit(1);
+    }
+  } catch (error) {
+    if (error instanceof PentestError) {
+      await logError(error, `Developer command ${command}`);
+      console.log(chalk.red.bold(`\n🚨 Command failed: ${error.message}`));
+    } else {
+      console.log(chalk.red.bold(`\n🚨 Unexpected error: ${error.message}`));
+      if (process.env.DEBUG) {
+        console.log(chalk.gray(error.stack));
+      }
+    }
+    process.exit(1);
+  }
+}