# GhostCrew PowerShell Setup Script

Write-Host "GhostCrew Setup" -ForegroundColor Blue
Write-Host "AI Penetration Testing" -ForegroundColor Green
Write-Host ""

# Check Python version
Write-Host "Checking Python version..." -ForegroundColor Yellow
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" -ForegroundColor Red
            exit 1
        }
        Write-Host "[OK] $pythonVersion" -ForegroundColor Green
    }
} catch {
    Write-Host "Error: Python not found. Please install Python 3.10+" -ForegroundColor Red
    exit 1
}

# Create virtual environment
Write-Host "Creating virtual environment..." -ForegroundColor Yellow
if (-not (Test-Path "venv")) {
    python -m venv venv
    Write-Host "[OK] Virtual environment created" -ForegroundColor Green
} else {
    Write-Host "[OK] Virtual environment exists" -ForegroundColor Green
}

# Activate virtual environment
Write-Host "Activating virtual environment..." -ForegroundColor Yellow
& .\venv\Scripts\Activate.ps1

# Upgrade pip
Write-Host "Upgrading pip..." -ForegroundColor Yellow
pip install --upgrade pip

# Install dependencies
Write-Host "Installing dependencies..." -ForegroundColor Yellow
pip install -e ".[all]"
Write-Host "[OK] Dependencies installed" -ForegroundColor Green

# Install playwright browsers
Write-Host "Installing Playwright browsers..." -ForegroundColor Yellow
playwright install chromium
Write-Host "[OK] Playwright browsers installed" -ForegroundColor Green

# Create .env file if not exists
if (-not (Test-Path ".env")) {
    Write-Host "Creating .env file..." -ForegroundColor Yellow
    @"
# GhostCrew 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
GHOSTCREW_MODEL=gpt-5

# Debug Mode
GHOSTCREW_DEBUG=false

# Max Iterations
GHOSTCREW_MAX_ITERATIONS=50
"@ | Set-Content -Path ".env" -Encoding UTF8
    Write-Host "[OK] .env file created" -ForegroundColor Green
    Write-Host "[!] Please edit .env and add your API keys" -ForegroundColor Yellow
}

# Create loot directory for reports
New-Item -ItemType Directory -Force -Path "loot" | Out-Null
Write-Host "[OK] Loot directory created" -ForegroundColor Green

Write-Host ""
Write-Host "Setup complete!" -ForegroundColor Green
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: ghostcrew or python -m ghostcrew"
Write-Host ""