#!/bin/bash
#
# Installation des dépendances pour la capture d'écran
# Installe mss et pygetwindow si nécessaires
#

set -e

echo "============================================================"
echo "INSTALLATION DES DÉPENDANCES DE CAPTURE D'ÉCRAN"
echo "============================================================"
echo ""

# Couleurs
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

print_status() {
    echo -e "${GREEN}[✓]${NC} $1"
}

print_info() {
    echo -e "${YELLOW}[i]${NC} $1"
}

# Vérifier l'environnement virtuel
if [ ! -d "venv" ] && [ ! -d "venv_v3" ]; then
    echo "Erreur: Environnement virtuel non trouvé"
    echo "Exécute d'abord: python3 -m venv venv_v3"
    exit 1
fi

# Activer le venv
if [ -d "venv_v3" ]; then
    source venv_v3/bin/activate
    print_status "Environnement venv_v3 activé"
elif [ -d "venv" ]; then
    source venv/bin/activate
    print_status "Environnement venv activé"
fi

# Installer mss
print_info "Installation de mss (capture d'écran rapide)..."
pip install mss>=9.0.0 --quiet

# Installer pygetwindow
print_info "Installation de pygetwindow (détection de fenêtres)..."
pip install pygetwindow>=0.0.9 --quiet

print_status "Dépendances installées avec succès"

# Vérifier l'installation
echo ""
print_info "Vérification de l'installation..."

if python -c "import mss" 2>/dev/null; then
    print_status "mss est installé"
else
    echo "Erreur: Impossible d'importer mss"
    exit 1
fi

if python -c "import pygetwindow" 2>/dev/null; then
    print_status "pygetwindow est installé"
else
    echo "Erreur: Impossible d'importer pygetwindow"
    exit 1
fi

echo ""
echo "============================================================"
print_status "Installation terminée avec succès!"
echo "============================================================"
echo ""
print_info "Pour tester la capture d'écran:"
echo "  python examples/test_screen_capturer.py"
echo ""