#!/bin/bash

# Couleurs pour le texte
GREEN="\e[32m"
RED="\e[31m"
BLUE="\e[34m"
RESET="\e[0m"

# Fonction pour afficher un message de succès
success() {
    echo -e "${GREEN}[SUCCESS]${RESET} $1"
}

# Fonction pour afficher un message d'erreur
error() {
    echo -e "${RED}[ERROR]${RESET} $1"
}

# Fonctions de vos opérations
update_and_clean_system() {
    echo -e "${BLUE}Mise à jour et nettoyage du système...${RESET}"
    sudo apt update && sudo apt full-upgrade -y && sudo apt autoremove -y && sudo apt autoclean -y
    success "Mise à jour et nettoyage terminés."
}

install_nfs_server() {
    echo -e "${BLUE}Installation et configuration du serveur NFS...${RESET}"
    sudo apt install -y nfs-kernel-server
    sudo systemctl enable nfs-server
    sudo systemctl start nfs-server
    success "Serveur NFS installé et configuré."
}

install_nfs_client() {
    echo -e "${BLUE}Installation du client NFS...${RESET}"
    sudo apt install -y nfs-common
    success "Client NFS installé."
}

install_samba_server() {
    echo -e "${BLUE}Installation et configuration du serveur Samba...${RESET}"
    sudo apt install -y samba
    sudo systemctl enable smbd
    sudo systemctl start smbd
    success "Serveur Samba installé et configuré."
}

add_user_to_sudo() {
    echo -e "${BLUE}Ajout de l'utilisateur actuel au groupe sudo...${RESET}"
    sudo apt install -y sudo
    sudo usermod -aG sudo "$USER"
    success "Utilisateur ajouté au groupe sudo."
}

install_docker() {
    echo -e "${BLUE}Installation de Docker...${RESET}"
    sudo apt install -y docker.io docker-compose
    sudo systemctl enable docker
    sudo systemctl start docker
    sudo usermod -aG docker "$USER"
    success "Docker installé et configuré."
}

configure_network() {
    echo -e "${BLUE}Configuration du réseau...${RESET}"
    echo "À configurer selon vos besoins spécifiques."
}

install_usual_tools() {
    echo -e "${BLUE}Installation des outils usuels...${RESET}"
    sudo apt install -y curl git
    success "Outils usuels installés."
}

# Menu principal avec `select`
while true; do
    echo -e "\n${BLUE}Post-installation Debian : Menu${RESET}"
    select option in \
        "Mise à jour et nettoyage du système" \
        "Installation et configuration du serveur NFS" \
        "Installation du client NFS" \
        "Installation et configuration du serveur Samba" \
        "Ajouter l'utilisateur actuel à sudo" \
        "Installation de Docker" \
        "Configuration du réseau" \
        "Installation des outils usuels" \
        "Quitter"; do

        case $REPLY in
            1) update_and_clean_system ; break ;;
            2) install_nfs_server ; break ;;
            3) install_nfs_clie