#!/bin/bash
# ──────────────────────────────────────────────
# d0a1 quickstart — IA agéntica local, sin coste
# ──────────────────────────────────────────────
# Ejecuta:  curl -fsSL https://d0a1.es/quickstart | bash
#           o:    ./deploy.sh
# Abre:     http://localhost:3000
# ──────────────────────────────────────────────
set -e

# Colores
GREEN='\033[0;32m'
CYAN='\033[0;36m'
YELLOW='\033[0;33m'
RED='\033[0;31m'
BOLD='\033[1m'
NC='\033[0m'

echo ""
echo -e "${BOLD}${CYAN}╔══════════════════════════════════════════╗${NC}"
echo -e "${BOLD}${CYAN}║   d0a1 quickstart — IA local, sin coste ║${NC}"
echo -e "${BOLD}${CYAN}╚══════════════════════════════════════════╝${NC}"
echo ""

# ── 1. Comprobar Docker ──────────────────────
echo -e "${YELLOW}[1/4]${NC} Comprobando Docker..."

if ! command -v docker &>/dev/null; then
    echo -e "${RED}✗ Docker no está instalado.${NC}"
    echo ""
    echo "Instálalo primero:"
    echo "  Linux:  curl -fsSL https://get.docker.com | sh"
    echo "  macOS: https://docs.docker.com/desktop/install/mac-install/"
    echo "  Windows: https://docs.docker.com/desktop/install/windows-install/"
    echo ""
    echo "Y vuelve a ejecutar este script."
    exit 1
fi

if ! docker compose version &>/dev/null; then
    echo -e "${RED}✗ Docker Compose v2 no está disponible.${NC}"
    echo "Actualiza Docker: https://docs.docker.com/compose/install/"
    exit 1
fi

echo -e "${GREEN}✓${NC} Docker encontrado: $(docker compose version --short 2>/dev/null || echo 'OK')"

# ── 2. Comprobar RAM ────────────────────────
echo -e "${YELLOW}[2/4]${NC} Comprobando memoria..."

TOTAL_RAM_KB=$(grep MemTotal /proc/meminfo 2>/dev/null | awk '{print $2}' || echo "0")
TOTAL_RAM_GB=$((TOTAL_RAM_KB / 1024 / 1024))

if [ "$TOTAL_RAM_GB" -lt 6 ]; then
    echo -e "${RED}✗ Se necesitan al menos 8 GB de RAM (detectado: ${TOTAL_RAM_GB} GB).${NC}"
    echo "El modelo de IA necesita memoria para funcionar."
    exit 1
fi

echo -e "${GREEN}✓${NC} RAM: ${TOTAL_RAM_GB} GB"

# ── 3. Descargar y arrancar ─────────────────
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
COMPOSE_FILE="$SCRIPT_DIR/docker-compose.yml"

if [ ! -f "$COMPOSE_FILE" ]; then
    echo -e "${RED}✗ No encuentro docker-compose.yml junto a este script.${NC}"
    echo "Asegúrate de ejecutarlo desde la carpeta quickstart/."
    exit 1
fi

echo -e "${YELLOW}[3/4]${NC} Descargando contenedores (primera vez tarda 2-3 minutos)..."
docker compose -f "$COMPOSE_FILE" up -d

# ── 4. Descargar modelo ─────────────────────
echo -e "${YELLOW}[4/4]${NC} Descargando modelo de IA (llama3.2:3b, ~2 GB)..."
echo "        Esto también solo ocurre la primera vez."

# Esperar a que Ollama responda
echo "        Esperando a que Ollama arranque..."
MAX_WAIT=60
WAITED=0
until docker exec d0a1-ollama ollama ps &>/dev/null; do
    sleep 2
    WAITED=$((WAITED + 2))
    if [ "$WAITED" -ge "$MAX_WAIT" ]; then
        echo -e "${RED}✗ Ollama no arrancó en ${MAX_WAIT}s. Revisa los logs:${NC}"
        echo "  docker logs d0a1-ollama"
        exit 1
    fi
done

docker exec d0a1-ollama ollama pull llama3.2:3b

# ── Listo ────────────────────────────────────
echo ""
echo -e "${BOLD}${GREEN}¡Listo! Tu IA local está corriendo.${NC}"
echo ""
echo -e "  ${BOLD}Abre tu navegador:${NC} ${CYAN}http://localhost:3000${NC}"
echo ""
echo "  No necesitas cuenta ni API key. Empieza a chatear."
echo ""
echo -e "  ${BOLD}Modelos adicionales (opcionales):${NC}"
echo "  docker exec d0a1-ollama ollama pull qwen2.5:7b    # Mejor en español (5 GB RAM)"
echo "  docker exec d0a1-ollama ollama pull llama3.1:8b    # Más calidad (6 GB RAM)"
echo "  docker exec d0a1-ollama ollama pull codellama:7b  # Para código (5 GB RAM)"
echo ""
echo -e "  ${BOLD}Comandos útiles:${NC}"
echo "  docker compose -f $COMPOSE_FILE logs -f   # Ver logs"
echo "  docker compose -f $COMPOSE_FILE down       # Parar todo"
echo "  docker compose -f $COMPOSE_FILE down -v    # Borrar todo (incluidos datos)"
echo ""
echo -e "  ${BOLD}¿Quieres más?${NC} Despliega el stack completo en un VPS:"
echo "  ${CYAN}https://d0a1.es/vps${NC}"
echo ""
echo -e "${BOLD}${CYAN}═══════════════════════════════════════════${NC}"