#!/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 [[ $# -ne 2 ]]; then
    echo "Usage: $0 <IP> <HOSTNAME>"
    exit 1
fi

IP="$1"
HOSTNAME="$2"

# Vérifier si l'entrée existe déjà
if grep -qE "^address=/$HOSTNAME/$IP$" "$DNSMASQ_CONF"; then
    echo "L'association $HOSTNAME -> $IP existe déjà dans $DNSMASQ_CONF."
    exit 0
fi

# Ajouter l'entrée dans le fichier de configuration
echo "address=/$HOSTNAME/$IP" >> "$DNSMASQ_CONF"

# Redémarrer dnsmasq pour appliquer les changements
systemctl restart dnsmasq

echo "Ajout réussi : $HOSTNAME -> $IP"