#!/bin/bash

# Couleurs pour les messages
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Fonction pour afficher les messages
log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# Fonction pour nettoyer les processus à l'arrêt
cleanup() {
    log_warning "Arrêt de la stack OMOP Pipeline..."
    
    if [ ! -z "$API_PID" ]; then
        log_info "Arrêt de l'API (PID: $API_PID)"
        kill $API_PID 2>/dev/null
    fi
    
    if [ ! -z "$FRONTEND_PID" ]; then
        log_info "Arrêt du frontend (PID: $FRONTEND_PID)"
        kill $FRONTEND_PID 2>/dev/null
    fi
    
    log_success "Stack arrêtée proprement"
    exit 0
}

# Capturer Ctrl+C
trap cleanup INT TERM

# Banner
echo ""
echo "╔═══════════════════════════════════════════════════════════╗"
echo "║                                                           ║"
echo "║           🚀 OMOP PIPELINE - STACK COMPLÈTE 🚀            ║"
echo "║                                                           ║"
echo "╚═══════════════════════════════════════════════════════════╝"
echo ""

# Vérifier si on est dans le bon répertoire
if [ ! -f "run_api.py" ]; then
    log_error "Ce script doit être exécuté depuis le répertoire omop/"
    exit 1
fi

# 1. Vérifier Python
log_info "Vérification de Python..."
if ! command -v python3 &> /dev/null; then
    log_error "Python 3 n'est pas installé"
    exit 1
fi
PYTHON_VERSION=$(python3 --version)
log_success "Python trouvé: $PYTHON_VERSION"

# 2. Vérifier Node.js
log_info "Vérification de Node.js..."
if ! command -v node &> /dev/null; then
    log_error "Node.js n'est pas installé"
    exit 1
fi
NODE_VERSION=$(node --version)
log_success "Node.js trouvé: $NODE_VERSION"

# 3. Vérifier npm
log_info "Vérification de npm..."
if ! command -v npm &> /dev/null; then
    log_error "npm n'est pas installé"
    exit 1
fi
NPM_VERSION=$(npm --version)
log_success "npm trouvé: v$NPM_VERSION"

# 4. Vérifier PostgreSQL
log_info "Vérification de PostgreSQL..."
if ! command -v psql &> /dev/null; then
    log_warning "psql n'est pas trouvé dans le PATH"
else
    PSQL_VERSION=$(psql --version)
    log_success "PostgreSQL trouvé: $PSQL_VERSION"
fi

# 5. Installer les dépendances Python si nécessaire
log_info "Vérification des dépendances Python..."
if ! python3 -c "import fastapi" 2>/dev/null; then
    log_warning "Dépendances Python manquantes, installation..."
    pip install -r requirements.txt -q
    pip install -r requirements-api.txt -q
    log_success "Dépendances Python installées"
else
    log_success "Dépendances Python OK"
fi

# 6. Installer les dépendances npm si nécessaire
log_info "Vérification des dépendances frontend..."
if [ ! -d "frontend/node_modules" ]; then
    log_warning "node_modules manquant, installation..."
    cd frontend
    npm install --silent
    cd ..
    log_success "Dépendances frontend installées"
else
    log_success "Dépendances frontend OK"
fi

# 7. Vérifier la connexion à la base de données
log_info "Vérification de la connexion PostgreSQL..."
if psql -U dom -d omop_cdm -c "SELECT 1;" &> /dev/null; then
    log_success "Connexion à la base de données OK"
else
    log_warning "Impossible de se connecter à la base de données"
    log_warning "Assurez-vous que PostgreSQL est démarré et que la base 'omop_cdm' existe"
fi

echo ""
log_info "═══════════════════════════════════════════════════════════"
log_info "              DÉMARRAGE DE LA STACK"
log_info "═══════════════════════════════════════════════════════════"
echo ""

# 8. Démarrer l'API en arrière-plan
log_info "Démarrage de l'API FastAPI..."
python3 run_api.py > logs/api.log 2>&1 &
API_PID=$!

# Attendre que l'API démarre
sleep 3

# Vérifier si l'API est démarrée
if ps -p $API_PID > /dev/null; then
    log_success "API démarrée (PID: $API_PID)"
    log_success "API disponible sur: http://localhost:8001"
    log_success "Documentation API: http://localhost:8001/docs"
else
    log_error "Échec du démarrage de l'API"
    log_error "Consultez logs/api.log pour plus de détails"
    exit 1
fi

# 9. Démarrer le frontend en arrière-plan
log_info "Démarrage du frontend React..."
cd frontend
npm run dev > ../logs/frontend.log 2>&1 &
FRONTEND_PID=$!
cd ..

# Attendre que le frontend démarre
sleep 5

# Vérifier si le frontend est démarré
if ps -p $FRONTEND_PID > /dev/null; then
    log_success "Frontend démarré (PID: $FRONTEND_PID)"
    log_success "Frontend disponible sur: http://localhost:4400"
else
    log_error "Échec du démarrage du frontend"
    log_error "Consultez logs/frontend.log pour plus de détails"
    kill $API_PID 2>/dev/null
    exit 1
fi

echo ""
log_success "═══════════════════════════════════════════════════════════"
log_success "           ✅ STACK OMOP PIPELINE DÉMARRÉE ✅"
log_success "═══════════════════════════════════════════════════════════"
echo ""
echo "  📊 Frontend:        http://localhost:4400"
echo "  🔌 API:             http://localhost:8001"
echo "  📚 Documentation:   http://localhost:8001/docs"
echo ""
echo "  📝 Logs API:        logs/api.log"
echo "  📝 Logs Frontend:   logs/frontend.log"
echo ""
log_info "Appuyez sur Ctrl+C pour arrêter la stack"
echo ""

# Attendre indéfiniment (les processus tournent en arrière-plan)
wait