#!/bin/bash

# --- Couleurs pour le terminal ---
GREEN='\033[0;32m'
BLUE='\033[0;34m'
RED='\033[0;31m'
NC='\033[0m' # No Color

echo -e "${BLUE}==========================================${NC}"
echo -e "${GREEN}    🛡️ Installation de 7LnA Security V10.1 Quantum${NC}"
echo -e "${BLUE}==========================================${NC}\n"

# Arrêter le script immédiatement si une erreur survient
set -e

# 1. Installation des dépendances système
echo -e "${BLUE}[*] Étape 1 : Installation des dépendances système (Sudo requis)...${NC}"
sudo apt update
sudo apt install -y clamav clamav-daemon rkhunter arp-scan cron python3-venv python3-full python3-tk libnotify-bin x11-xserver-utils wget curl

# 1.1 Correction et Initialisation de rkhunter
echo -e "\n${BLUE}[*] Étape 1.1 : Configuration de la base anti-rootkit...${NC}"
# Correction du bug WEB_CMD pour permettre les mises à jour
sudo sed -i 's|WEB_CMD="/bin/false"|WEB_CMD=""|g' /etc/rkhunter.conf || true
sudo rkhunter --update --quiet || true
sudo rkhunter --propupd --quiet || true
echo -e "${GREEN}[+] Protection système initialisée.${NC}"

# 2. Création du répertoire de l'application
echo -e "\n${BLUE}[*] Étape 2 : Création des dossiers de l'application...${NC}"
APP_DIR="$HOME/.7lna_app"
mkdir -p "$APP_DIR"
mkdir -p "$HOME/.7lna_quarantine"

# Auto-détection du script Python : Copie locale OU téléchargement
if [ -f "7lna.py" ]; then
    echo -e "${GREEN}[+] Fichier local 7lna.py détecté. Copie...${NC}"
    cp 7lna.py "$APP_DIR/"
else
    echo -e "${BLUE}[*] Téléchargement du code source depuis GitHub...${NC}"
    # METS BIEN TON LIEN GITHUB CI-DESSOUS :
    wget -qO "$APP_DIR/7lna.py" "https://git.7ka1.com/7ka1/7LnA_Antivirus_Linux_Free_ClamAV_Based/raw/branch/main/7lna.py"
    echo -e "${GREEN}[+] Code source récupéré.${NC}"
fi

# 3. Configuration de l'environnement virtuel (venv)
echo -e "\n${BLUE}[*] Étape 3 : Configuration de l'environnement Python (venv)...${NC}"
python3 -m venv "$APP_DIR/.venv"
"$APP_DIR/.venv/bin/pip" install --upgrade pip --quiet
"$APP_DIR/.venv/bin/pip" install customtkinter watchdog --quiet
echo -e "${GREEN}[+] Environnement Python prêt.${NC}"

# 4. Création du lanceur AVEC CORRECTIF D'AFFICHAGE (DISPLAY)
echo -e "\n${BLUE}[*] Étape 4 : Création du lanceur sécurisé...${NC}"
LAUNCHER_PATH="$APP_DIR/launcher.sh"
cat << EOF > "$LAUNCHER_PATH"
#!/bin/bash
# Autorise l'accès à l'affichage pour la session actuelle (évite TclError)
xhost +local:\$(whoami) > /dev/null 2>&1
export DISPLAY=:0
# Lancement de l'application
"$APP_DIR/.venv/bin/python3" "$APP_DIR/7lna.py"
EOF
chmod +x "$LAUNCHER_PATH"

# 5. Création de l'icône dans le menu des applications
echo -e "\n${BLUE}[*] Étape 5 : Intégration au menu système...${NC}"
DESKTOP_DIR="$HOME/.local/share/applications"
mkdir -p "$DESKTOP_DIR"
DESKTOP_FILE="$DESKTOP_DIR/7lna.desktop"

cat << EOF > "$DESKTOP_FILE"
[Desktop Entry]
Name=7LnA Antivirus V10.1 Quantum
Comment=Antivirus ClamAV avec Mode Zero USB
Exec=$LAUNCHER_PATH
Icon=security-high
Terminal=false
Type=Application
Categories=System;Security;Utility;
StartupNotify=true
EOF
chmod +x "$DESKTOP_FILE"

# Rafraîchir la base de données des applications
update-desktop-database "$DESKTOP_DIR" 2>/dev/null || true

echo -e "\n${BLUE}==========================================${NC}"
echo -e "${GREEN} ✅ INSTALLATION TERMINÉE AVEC SUCCÈS !${NC}"
echo -e "${BLUE}==========================================${NC}"
echo -e "🛡️  Lancez ${GREEN}7LnA Antivirus V10.1 Quantum${NC} depuis votre menu."
echo -e "💡  ${BLUE}Rappel : Activez le 'Mode Zero USB' dans l'app pour le scan auto.${NC}\n"