#!/bin/bash
# Setup del Pentest Toolkit
# Ejecuta esto en tu portátil para construir el contenedor

echo "=========================================="
echo " Pentest Toolkit - Setup"
echo "=========================================="

# 1. Editar targets
echo ""
echo "[!] ANTES DE CONTINUAR:"
echo "  1. Edita targets.txt con la IP local de tu servidor"
echo "  2. Verifica los subdominios listados"
echo ""
read -p "  ¿Ya editaste targets.txt? (s/n): " confirm
if [ "$confirm" != "s" ]; then
    echo "  Edita targets.txt y vuelve a ejecutar este script."
    exit 1
fi

# 2. Construir imagen Docker
echo ""
echo "[*] Construyendo imagen Docker..."
docker compose build 2>&1 | tail -5

# 3. Verificar imagen
echo ""
echo "[*] Verificando imagen..."
docker images pentest-toolkit:latest --format "  Imagen: {{.Repository}}:{{.Tag}} | {{.Size}}"

# 4. Verificar herramientas
echo ""
echo "[*] Verificando herramientas instaladas..."
docker run --rm pentest-toolkit bash -c '
for tool in nmap whatweb nikto gobuster sqlmap nuclei ffuf testssl curl wget dig jq; do
    which "$tool" >/dev/null 2>&1 \
        && echo "  [OK] $tool" \
        || echo "  [MISSING] $tool"
done
'

echo ""
echo "=========================================="
echo " Setup completado."
echo " Para iniciar: docker compose run --rm pentest"
echo "=========================================="