install.sh

#!/bin/bash

# LocalTransfer - Script d'installation automatique pour Raspberry Pi et Linux
# Version: 1.0

set -e

echo "=========================================="
echo "  LocalTransfer - Installation automatique"
echo "=========================================="
echo ""

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

# Fonction pour afficher des messages colorés
print_step() {
    echo -e "${BLUE}[ÉTAPE]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[SUCCÈS]${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}[ATTENTION]${NC} $1"
}

print_error() {
    echo -e "${RED}[ERREUR]${NC} $1"
}

# Vérification si le script est exécuté en tant que root
if [[ $EUID -eq 0 ]]; then
   print_error "Ce script ne doit pas être exécuté en tant que root (sudo)"
   echo "Utilisez: ./install.sh"
   exit 1
fi

# Détection du système
print_step "Détection du système..."
if [ -f /etc/os-release ]; then
    . /etc/os-release
    OS=$NAME
    VER=$VERSION_ID
elif type lsb_release >/dev/null 2>&1; then
    OS=$(lsb_release -si)
    VER=$(lsb_release -sr)
else
    OS=$(uname -s)
    VER=$(uname -r)
fi

echo "Système détecté: $OS $VER"

# Vérifier si nous sommes sur Raspberry Pi
if grep -q "Raspberry Pi" /proc/cpuinfo 2>/dev/null; then
    print_success "Raspberry Pi détecté!"
    IS_RPI=true
else
    IS_RPI=false
fi

# Mise à jour du système
print_step "Mise à jour du système..."
sudo apt update && sudo apt upgrade -y

# Installation de Python et pip
print_step "Installation de Python et des dépendances..."
sudo apt install -y python3 python3-pip python3-venv git curl

# Vérification de la version Python
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
print_success "Python $PYTHON_VERSION installé"

# Création du répertoire d'installation
INSTALL_DIR="$HOME/LocalTransfer"
print_step "Création du répertoire d'installation: $INSTALL_DIR"

if [ -d "$INSTALL_DIR" ]; then
    print_warning "Le répertoire $INSTALL_DIR existe déjà"
    read -p "Voulez-vous le supprimer et réinstaller? (y/N): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        rm -rf "$INSTALL_DIR"
        print_success "Répertoire supprimé"
    else
        print_error "Installation annulée"
        exit 1
    fi
fi

# Copie des fichiers (si on exécute depuis le répertoire source)
if [ -f "main.py" ] && [ -f "config.py" ]; then
    print_step "Copie des fichiers vers $INSTALL_DIR..."
    cp -r . "$INSTALL_DIR"
    cd "$INSTALL_DIR"
else
    print_error "Fichiers source non trouvés. Assurez-vous d'exécuter ce script depuis le répertoire LocalTransfer."
    exit 1
fi

# Création de l'environnement virtuel
print_step "Création de l'environnement virtuel Python..."
python3 -m venv venv
source venv/bin/activate

# Installation des dépendances
print_step "Installation des dépendances Python..."
pip install --upgrade pip
pip install flask werkzeug

# Création du répertoire uploads
print_step "Création du répertoire uploads..."
mkdir -p uploads

# Configuration pour production
print_step "Configuration pour la production..."
if [ "$IS_RPI" = true ]; then
    # Configuration optimisée pour Raspberry Pi
    cat > config_prod.py << 'EOF'
# LocalTransfer Configuration - Production Raspberry Pi

# Server settings
HOST = "0.0.0.0"  # Allow connections from any IP on the local network
PORT = 5000
DEBUG = False  # Désactivé en production

# File upload settings
UPLOAD_FOLDER = "uploads"
MAX_FILE_SIZE = 2 * 1024 * 1024 * 1024  # 2GB per file (optimisé pour RPi)
MAX_TOTAL_SIZE = 8 * 1024 * 1024 * 1024  # 8GB total storage

# Allowed file extensions
ALLOWED_EXTENSIONS = {
    # Documents
    "txt", "pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "odt", "ods", "odp",
    # Images
    "png", "jpg", "jpeg", "gif", "bmp", "webp", "svg",
    # Audio
    "mp3", "wav", "ogg", "flac", "m4a",
    # Video
    "mp4", "avi", "mov", "mkv", "wmv", "flv", "webm",
    # Archives
    "zip", "rar", "7z", "tar", "gz",
    # Code
    "py", "js", "html", "css", "json", "xml", "csv",
}

# Security
SECRET_KEY = "raspberry-pi-localtransfer-secret-key-2025"
EOF
else
    # Configuration standard pour autres systèmes Linux
    cp config.py config_prod.py
    sed -i 's/DEBUG = True/DEBUG = False/' config_prod.py
fi

# Création du script de démarrage
print_step "Création du script de démarrage..."
cat > start.sh << 'EOF'
#!/bin/bash

# LocalTransfer - Script de démarrage
cd "$(dirname "$0")"

echo "=========================================="
echo "    LocalTransfer - Démarrage"
echo "=========================================="

# Activation de l'environnement virtuel
source venv/bin/activate

# Obtention de l'adresse IP locale
LOCAL_IP=$(hostname -I | awk '{print $1}')

echo "🌐 Application accessible à:"
echo "   • Localement: http://localhost:5000"
echo "   • Sur le réseau: http://$LOCAL_IP:5000"
echo ""
echo "💡 Partagez l'URL du réseau avec votre famille!"
echo "📁 Les fichiers sont stockés dans le dossier 'uploads'"
echo "⚠️  Appuyez sur Ctrl+C pour arrêter l'application"
echo "=========================================="
echo ""

# Utilisation de la configuration de production
export FLASK_ENV=production
python3 main.py
EOF

chmod +x start.sh

# Création du script d'arrêt
print_step "Création du script d'arrêt..."
cat > stop.sh << 'EOF'
#!/bin/bash

echo "Arrêt de LocalTransfer..."
pkill -f "python3 main.py"
echo "LocalTransfer arrêté."
EOF

chmod +x stop.sh

# Modification de main.py pour utiliser config_prod en production
print_step "Configuration pour la production..."
if ! grep -q "config_prod" main.py; then
    sed -i '/from config import \*/a\\n# Load production config if available\ntry:\n    from config_prod import *\n    print("Configuration de production chargée")\nexcept ImportError:\n    pass' main.py
fi

# Création du service systemd (optionnel)
print_step "Création du service systemd..."
cat > localtransfer.service << EOF
[Unit]
Description=LocalTransfer - Application de transfert de fichiers
After=network.target

[Service]
Type=simple
User=$USER
WorkingDirectory=$INSTALL_DIR
ExecStart=$INSTALL_DIR/venv/bin/python $INSTALL_DIR/main.py
Restart=always
RestartSec=10
Environment=FLASK_ENV=production

[Install]
WantedBy=multi-user.target
EOF

print_warning "Pour installer le service systemd (démarrage automatique):"
echo "sudo cp $INSTALL_DIR/localtransfer.service /etc/systemd/system/"
echo "sudo systemctl enable localtransfer"
echo "sudo systemctl start localtransfer"

# Test de l'installation
print_step "Test de l'installation..."
source venv/bin/activate
python3 -c "import flask; print('Flask installé correctement')"

# Obtention de l'adresse IP
LOCAL_IP=$(hostname -I | awk '{print $1}')

echo ""
echo "=========================================="
print_success "Installation terminée avec succès!"
echo "=========================================="
echo ""
echo "📍 LocalTransfer installé dans: $INSTALL_DIR"
echo "🚀 Pour démarrer l'application:"
echo "   cd $INSTALL_DIR"
echo "   ./start.sh"
echo ""
echo "🌐 L'application sera accessible à:"
echo "   • Localement: http://localhost:5000"
echo "   • Sur le réseau: http://$LOCAL_IP:5000"
echo ""
echo "🔧 Commandes utiles:"
echo "   • Démarrer: ./start.sh"
echo "   • Arrêter: ./stop.sh"
echo "   • Logs: journalctl -u localtransfer -f (si service activé)"
echo ""
echo "📚 Documentation complète dans README.md"
echo ""
if [ "$IS_RPI" = true ]; then
    echo "🍓 Configuration optimisée pour Raspberry Pi détectée"
    echo "   • Limite par fichier: 2GB"
    echo "   • Stockage total: 8GB"
fi
echo "=========================================="