mirror of
https://github.com/GH05TCREW/pentestagent.git
synced 2026-03-07 22:33:38 +00:00
94 lines
2.5 KiB
PowerShell
94 lines
2.5 KiB
PowerShell
# PentestAgent PowerShell Setup Script
|
|
|
|
Write-Host "=================================================================="
|
|
Write-Host " PENTESTAGENT"
|
|
Write-Host " AI Penetration Testing"
|
|
Write-Host "=================================================================="
|
|
Write-Host ""
|
|
Write-Host "Setup"
|
|
Write-Host ""
|
|
|
|
# Check Python version
|
|
Write-Host "Checking Python version..."
|
|
try {
|
|
$pythonVersion = python --version 2>&1
|
|
if ($pythonVersion -match "Python (\d+)\.(\d+)") {
|
|
$major = [int]$Matches[1]
|
|
$minor = [int]$Matches[2]
|
|
if ($major -lt 3 -or ($major -eq 3 -and $minor -lt 10)) {
|
|
Write-Host "Error: Python 3.10 or higher is required"
|
|
exit 1
|
|
}
|
|
Write-Host "[OK] $pythonVersion"
|
|
}
|
|
} catch {
|
|
Write-Host "Error: Python not found. Please install Python 3.10+"
|
|
exit 1
|
|
}
|
|
|
|
# Create virtual environment
|
|
Write-Host "Creating virtual environment..."
|
|
if (-not (Test-Path "venv")) {
|
|
python -m venv venv
|
|
Write-Host "[OK] Virtual environment created"
|
|
} else {
|
|
Write-Host "[OK] Virtual environment exists"
|
|
}
|
|
|
|
# Activate virtual environment
|
|
Write-Host "Activating virtual environment..."
|
|
& .\venv\Scripts\Activate.ps1
|
|
|
|
# Upgrade pip
|
|
Write-Host "Upgrading pip..."
|
|
pip install --upgrade pip
|
|
|
|
# Install dependencies
|
|
Write-Host "Installing dependencies..."
|
|
pip install -e ".[all]"
|
|
Write-Host "[OK] Dependencies installed"
|
|
|
|
# Install playwright browsers
|
|
Write-Host "Installing Playwright browsers..."
|
|
playwright install chromium
|
|
Write-Host "[OK] Playwright browsers installed"
|
|
|
|
# Create .env file if not exists
|
|
if (-not (Test-Path ".env")) {
|
|
Write-Host "Creating .env file..."
|
|
@"
|
|
# PentestAgent Configuration
|
|
# Add your API keys here
|
|
|
|
# OpenAI API Key (required for GPT models)
|
|
OPENAI_API_KEY=
|
|
|
|
# Anthropic API Key (required for Claude models)
|
|
ANTHROPIC_API_KEY=
|
|
|
|
# Model Configuration
|
|
PENTESTAGENT_MODEL=gpt-5
|
|
|
|
# Debug Mode
|
|
PENTESTAGENT_DEBUG=false
|
|
|
|
# Max Iterations
|
|
PENTESTAGENT_MAX_ITERATIONS=50
|
|
"@ | Set-Content -Path ".env" -Encoding UTF8
|
|
Write-Host "[OK] .env file created"
|
|
Write-Host "[!] Please edit .env and add your API keys"
|
|
}
|
|
|
|
# Create loot directory for reports
|
|
New-Item -ItemType Directory -Force -Path "loot" | Out-Null
|
|
Write-Host "[OK] Loot directory created"
|
|
|
|
Write-Host ""
|
|
Write-Host "Setup complete!"
|
|
Write-Host ""
|
|
Write-Host "To get started:"
|
|
Write-Host " 1. Edit .env and add your API keys"
|
|
Write-Host " 2. Activate: .\venv\Scripts\Activate.ps1"
|
|
Write-Host " 3. Run: pentestagent or python -m pentestagent"
|
|
Write-Host ""
|