#!/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.3 Quantum${NC}"
echo -e "${BLUE}==========================================${NC}\n"

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

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

# 1.1 Initialisation de rkhunter
echo -e "\n${BLUE}[*] Étape 1.1 : Initialisation de la base de données anti-rootkit...${NC}"
sudo rkhunter --update --quiet || true
sudo rkhunter --propupd --quiet || true
echo -e "${GREEN}[+] Empreintes système enregistrées.${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"

# 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 en cours...${NC}"
    cp 7lna.py "$APP_DIR/"
else
    echo -e "${RED}[!] Fichier 7lna.py introuvable localement.${NC}"
    echo -e "${BLUE}[*] Téléchargement automatique depuis le serveur officiel Git...${NC}"
    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 téléchargé avec succès.${NC}"
fi

# 3. Configuration de l'environnement virtuel (venv)
echo -e "\n${BLUE}[*] Étape 3 : Configuration de l'environnement isolé (venv)...${NC}"
python3 -m venv "$APP_DIR/.venv"
# Exécution directe via le pip du venv (Ajout de cryptography pour l'AES-256)
"$APP_DIR/.venv/bin/pip" install --upgrade pip --quiet
"$APP_DIR/.venv/bin/pip" install customtkinter watchdog cryptography --quiet
echo -e "${GREEN}[+] Modules Python installés.${NC}"

# 4. Création du script de lancement
echo -e "\n${BLUE}[*] Étape 4 : Création du lanceur...${NC}"
LAUNCHER_PATH="$APP_DIR/launcher.sh"
cat << EOF > "$LAUNCHER_PATH"
#!/bin/bash
# Exporter le display pour s'assurer que l'interface graphique s'ouvre bien
export DISPLAY=:0
"$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 du 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.3 Quantum
Comment=Suite de sécurité totale : ClamAV, Rootkits, OSINT, WiFi Guard, Vault AES-256
Exec=$LAUNCHER_PATH
Icon=security-high
Terminal=false
Type=Application
Categories=System;Security;Utility;
Keywords=antivirus;security;clamav;scan;virus;rootkit;cleaner;wifi;network;osint;vault;crypto;
EOF
chmod +x "$DESKTOP_FILE"

# Rafraîchir la base de données des applications du bureau
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 "🛡️  Vous pouvez maintenant lancer ${GREEN}7LnA Antivirus V10.3 Quantum${NC} depuis votre menu d'applications."
echo -e "💡  ${BLUE}Astuce : Utilisez le bouton 'MaJ Signatures' au premier lancement pour initialiser ClamAV.${NC}\n"