#!/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=("$@")

# 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 "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
    if [[ ! " ${EXISTING_HOSTNAMES[*]} " =~ " ${HOSTNAME} " ]]; then
        echo "address=/$HOSTNAME/$IP" >> "$DNSMASQ_CONF"
        echo "Ajout de : $HOSTNAME -> $IP"
        MODIFIED=1
    else
        echo "L'association $HOSTNAME -> $IP existe déjà."
    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