#!/bin/bash

# Importer les fonctions communes
source "$(dirname "$0")/../common/common_utils.sh"

# Vérifier si le script est exécuté en root
check_root

# Fichier de configuration de dnsmasq
DNSMASQ_CONF="/etc/dnsmasq.d/custom_hosts"

# Vérifier que dnsmasq est installé
if ! command -v dnsmasq &> /dev/null; then
    echo "dnsmasq n'est pas installé. Installez-le avec : sudo ./setup_dhcp.sh"
    exit 1
fi

# Vérifier les arguments
if [[ $# -lt 2 ]]; then
    echo "Usage: $0 <IP> <HOSTNAME1> [HOSTNAME2] [HOSTNAME3] ..."
    exit 1
fi

IP="$1"
shift  # Supprime le premier argument pour ne garder que les hostnames
HOSTNAMES=("$@")

# Fonction pour obtenir le nom de domaine de la machine
get_domain_name() {
    local domain=$(dnsdomainname)
    echo "$domain"
}

# Fonction pour obtenir le FQDN d'un hostname
get_fqdn() {
    local hostname="$1"
    local domain_name=$(get_domain_name)
    if [[ -n "$domain_name" ]]; then
        echo "$hostname.$domain_name"
    else
        echo "$hostname"
    fi
}

# Fonction pour vérifier si un hostname est déjà enregistré
is_hostname_registered() {
    local hostname="$1"
    grep -q "address=/$hostname/$IP" "$DNSMASQ_CONF"
    return $?
}

# Variable pour suivre si un changement a été fait
MODIFIED=0

# Vérifier si l'IP existe déjà dans le fichier
EXISTING_HOSTNAMES=($(grep -E "address=/[^ ]+/$IP$" "$DNSMASQ_CONF" | sed -E "s/address=\/([^\/]+)\/$IP/\1/"))

if [[ ${#EXISTING_HOSTNAMES[@]} -gt 0 ]]; then
    echo "L'adresse IP $IP existe déjà avec les noms suivants : ${EXISTING_HOSTNAMES[*]}"
else
    echo "Nouvelle adresse IP détectée : $IP"
fi

# Vérifier et ajouter chaque hostname
for HOSTNAME in "${HOSTNAMES[@]}"; do
    FQDN=$(get_fqdn "$HOSTNAME")
    echo "Le FQDN détecté pour $HOSTNAME est : $FQDN"
    
    if is_hostname_registered "$FQDN"; then
        echo "L'association $FQDN -> $IP existe déjà."
    else
        read -p "Le FQDN $FQDN n'est pas dans la liste. Voulez-vous l'ajouter ? (o/n) " REPLY
        if [[ "$REPLY" =~ ^[Oo]$ ]]; then
            echo "address=/$FQDN/$IP" >> "$DNSMASQ_CONF"
            echo "Ajout de : $FQDN -> $IP"
            MODIFIED=1
        else
            echo "address=/$HOSTNAME/$IP" >> "$DNSMASQ_CONF"
            echo "Ajout de : $HOSTNAME -> $IP"
            MODIFIED=1
        fi
    fi

    if is_hostname_registered "$HOSTNAME"; then
        echo "L'association $HOSTNAME -> $IP existe déjà."
    else
        echo "address=/$HOSTNAME/$IP" >> "$DNSMASQ_CONF"
        echo "Ajout de : $HOSTNAME -> $IP"
        MODIFIED=1
    fi

done

# Redémarrer dnsmasq seulement si des changements ont été faits
if [[ $MODIFIED -eq 1 ]]; then
    systemctl restart dnsmasq
    echo "Redémarrage de dnsmasq effectué."
else
    echo "Aucune modification nécessaire."
fi