run.sh

#!/bin/bash

# Script de gestion pour l'environnement Docker OQS
set -e

PROJECT_NAME="oqs-tls-env"
IMAGE_NAME="oqs-tls:latest"
NETWORK_NAME="oqs-network"

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

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 créer le réseau Docker
create_network() {
    if ! docker network ls | grep -q "$NETWORK_NAME"; then
        log_info "Création du réseau Docker $NETWORK_NAME..."
        docker network create $NETWORK_NAME
    fi
}

# Fonction pour construire l'image Docker
build_image() {
    log_info "Construction de l'image Docker OQS..."
    
    # Build avec cache et parallélisme optimisé
    docker build \
        --tag $IMAGE_NAME \
        --build-arg BUILDKIT_INLINE_CACHE=1 \
        --progress=plain \
        . || {
        log_error "Échec de la construction de l'image"
        exit 1
    }
    
    log_success "Image Docker construite avec succès!"
}

# Fonction pour démarrer le conteneur serveur en arrière-plan
start_server() {
    log_info "Démarrage du serveur TLS OQS..."
    
    # Créer le réseau si nécessaire
    create_network
    
    # Arrêter le serveur existant s'il existe
    docker stop "${PROJECT_NAME}_server" 2>/dev/null || true
    docker rm "${PROJECT_NAME}_server" 2>/dev/null || true
    
    # Démarrer le nouveau serveur
    docker run -d \
        --name "${PROJECT_NAME}_server" \
        --network $NETWORK_NAME \
        -p 8443:8443 \
        -p 8080:8080 \
        -v "$(pwd)":/app \
        $IMAGE_NAME python3 src/tls/server.py
    
    log_success "Serveur TLS OQS démarré en arrière-plan!"
    log_info "🌐 Interface web: https://localhost:8443"
    log_info "📊 Pour voir les logs: ./run.sh logs"
    log_info "🛑 Pour arrêter: ./run.sh stop"
    
    # Attendre que le serveur soit prêt
    log_info "⏳ Vérification du démarrage du serveur..."
    sleep 5
    
    # Tester la connectivité
    if curl -k -s https://localhost:8443 > /dev/null 2>&1; then
        log_success "✅ Serveur accessible et opérationnel!"
    else
        log_warning "⚠️  Le serveur démarre encore... Vérifiez les logs avec: ./run.sh logs"
    fi
}

# Fonction pour démarrer le client
start_client() {
    log_info "Démarrage du client de test..."
    
    # Vérifier si le serveur est en cours d'exécution
    if ! docker ps --format '{{.Names}}' | grep -q "${PROJECT_NAME}_server"; then
        log_warning "Aucun serveur détecté. Démarrage automatique du serveur..."
        start_server
        sleep 8  # Attendre que le serveur démarre complètement
    fi
    
    # Créer le réseau si nécessaire
    create_network
    
    # Lancer le client avec connexion via le réseau Docker
    docker run --rm \
        --name "${PROJECT_NAME}_client" \
        --network $NETWORK_NAME \
        --add-host=host.docker.internal:host-gateway \
        -v "$(pwd)":/app \
        $IMAGE_NAME python3 src/tls/client.py
}

# Fonction pour démarrer le conteneur interactif
start_interactive() {
    log_info "Démarrage du conteneur interactif..."
    
    create_network
    
    docker run -it --rm \
        --name "${PROJECT_NAME}_interactive" \
        --network $NETWORK_NAME \
        -p 8443:8443 \
        -p 8080:8080 \
        -v "$(pwd)":/app \
        $IMAGE_NAME /bin/bash
}

# Fonction pour arrêter tous les conteneurs
stop_containers() {
    log_info "Arrêt des conteneurs OQS..."
    
    # Arrêter et supprimer tous les conteneurs du projet
    for container in "${PROJECT_NAME}_server" "${PROJECT_NAME}_client" "${PROJECT_NAME}_interactive" "${PROJECT_NAME}_test" "${PROJECT_NAME}_demo"; do
        docker stop "$container" 2>/dev/null || true
        docker rm "$container" 2>/dev/null || true
    done
    
    # Supprimer le réseau
    docker network rm $NETWORK_NAME 2>/dev/null || true
    
    log_success "Tous les conteneurs ont été arrêtés!"
}

# Fonction pour voir les logs du serveur
show_logs() {
    if docker ps --format '{{.Names}}' | grep -q "${PROJECT_NAME}_server"; then
        log_info "Affichage des logs du serveur (Ctrl+C pour quitter)..."
        docker logs -f "${PROJECT_NAME}_server"
    else
        log_warning "Aucun serveur en cours d'exécution"
    fi
}

# Fonction pour voir le statut
show_status() {
    log_info "Statut des conteneurs OQS:"
    echo ""
    
    # Vérifier chaque conteneur
    containers=("${PROJECT_NAME}_server" "${PROJECT_NAME}_client" "${PROJECT_NAME}_interactive" "${PROJECT_NAME}_test")
    
    for container in "${containers[@]}"; do
        if docker ps --format '{{.Names}}' | grep -q "$container"; then
            echo -e "  ✅ $container: ${GREEN}En cours d'exécution${NC}"
        elif docker ps -a --format '{{.Names}}' | grep -q "$container"; then
            echo -e "  🔶 $container: ${YELLOW}Arrêté${NC}"
        else
            echo -e "  ⚪ $container: ${NC}Non créé${NC}"
        fi
    done
    
    echo ""
    if docker ps --format '{{.Names}}' | grep -q "${PROJECT_NAME}_server"; then
        echo -e "🌐 Interface web: ${BLUE}https://localhost:8443${NC}"
        echo -e "📊 Logs du serveur: ${BLUE}./run.sh logs${NC}"
    fi
    
    # Test de connectivité
    if curl -k -s https://localhost:8443 > /dev/null 2>&1; then
        echo -e "🔗 Connectivité: ${GREEN}✅ Serveur accessible${NC}"
    else
        echo -e "🔗 Connectivité: ${RED}❌ Serveur inaccessible${NC}"
    fi
}

# Fonction pour exécuter des tests
run_tests() {
    log_info "Exécution des tests OQS..."
    
    create_network
    
    docker run --rm \
        --name "${PROJECT_NAME}_test" \
        --network $NETWORK_NAME \
        -v "$(pwd)":/app \
        $IMAGE_NAME python3 -m pytest tests/ -v
}

# Fonction pour lancer une démonstration
run_demo() {
    log_info "Lancement de la démonstration OQS..."
    
    create_network
    
    docker run --rm \
        --name "${PROJECT_NAME}_demo" \
        --network $NETWORK_NAME \
        -v "$(pwd)":/app \
        $IMAGE_NAME python3 src/oqs_demo.py
}

# Fonction pour nettoyer
clean() {
    log_info "Nettoyage des ressources Docker..."
    
    # Arrêter tous les conteneurs
    stop_containers
    
    # Supprimer l'image
    docker rmi $IMAGE_NAME 2>/dev/null || true
    
    # Nettoyage du cache Docker
    docker builder prune -f
    
    log_success "Nettoyage terminé!"
}

# Fonction pour afficher l'aide
show_help() {
    echo "Usage: $0 [COMMAND]"
    echo ""
    echo "Commands:"
    echo "  build     Construire l'image Docker"
    echo "  server    Démarrer le serveur TLS OQS (en arrière-plan)"
    echo "  client    Démarrer le client de test"
    echo "  run       Démarrer un conteneur interactif"
    echo "  stop      Arrêter tous les conteneurs"
    echo "  logs      Afficher les logs du serveur"
    echo "  status    Afficher le statut des conteneurs"
    echo "  test      Exécuter les tests"
    echo "  demo      Lancer la démonstration OQS"
    echo "  clean     Nettoyer les ressources Docker"
    echo "  help      Afficher cette aide"
    echo ""
    echo "Examples:"
    echo "  $0 build          # Construire l'environnement"
    echo "  $0 server         # Démarrer le serveur"
    echo "  $0 client         # Tester avec le client"
    echo "  $0 logs           # Voir les logs du serveur"
    echo "  $0 status         # Vérifier l'état"
    echo "  $0 stop           # Arrêter tous les services"
    echo ""
    echo "Workflow typique:"
    echo "  1. $0 build       # Une seule fois"
    echo "  2. $0 server      # Démarrer le serveur"
    echo "  3. $0 client      # Tester la connexion"
    echo "  4. $0 status      # Vérifier l'état"
    echo "  5. $0 stop        # Quand vous avez terminé"
}

# Parsing des arguments
case "${1:-help}" in
    "build")
        build_image
        ;;
    "server")
        start_server
        ;;
    "client")
        start_client
        ;;
    "run")
        start_interactive
        ;;
    "stop")
        stop_containers
        ;;
    "logs")
        show_logs
        ;;
    "status")
        show_status
        ;;
    "test")
        run_tests
        ;;
    "demo")
        run_demo
        ;;
    "clean")
        clean
        ;;
    "help"|"-h"|"--help")
        show_help
        ;;
    *)
        log_error "Commande inconnue: $1"
        show_help
        exit 1
        ;;
esac