#!/usr/bin/env bash

italic="\033[3m"
underline="\033[4m"
ita_under="\033[3;4m"
bgd="\033[1;4;31m"
red="\033[1;31m"
bold="\033[1m"
bold_ita="\033[1;3m"
box="\033[1;41m"
redbold="\033[1;31m"
redbox="\033[1;41m"
green="\033[0;32m"
reset="\033[0m"


absent=()
delete=()
domain=".photos-nas.ovh"
not_required=("drive" "files" "gitea" "home-assistant" "homebridge" "portainer" "wg" "yatch")
not_used=("ds916" "musiiic" "tunes" "notif" "pastefy")

pihole1=192.168.2.116
pihole2=192.168.2.216

cat < /dev/null > /dev/tcp/1.1.1.1/53
if [[ $? -ne 0 ]]; then
	echo -e "\n${red}No Internet connection !${reset}"
	echo -e "Exit !"
	exit 1
fi

# Sauvegarde du lan.list

echo -e "${bold}Backup ${italic}lan.list...${reset}"

ping -q -c1 "$pihole1" &>/dev/null && ssh -p51322 root@"$pihole1" "cat /etc/pihole/lan.list" > lan.list.1 || echo -e "\n${red}Pihole1 is unreachable !'${reset}"
ping -q -c1 "$pihole2" &>/dev/null && ssh -p51522 root@"$pihole2" "cat /etc/pihole/lan.list" > lan.list.2 || echo -e "\n${red}Pihole2 is unreachable !'${reset}"


if [ -f lan.list.bak ]; then
	cp lan.list.bak /tmp/lan.list
	bzip2 /tmp/lan.list		# lan.list.bz2
	mv /tmp/lan.list.bz2 .
fi
if [ -f lan.list ]; then
	cp lan.list lan.list.bak
elif [ -f lan.list.1 ]; then
	cp lan.list.1 lan.list
elif [ -f lan.list.2 ]; then
	cp lan.list.2 lan.list
else
	echo "Error ! No lan.list file !"
	exit
fi

nb_dynhost=$(awk 'END { print NR }' lan.list)
echo "$nb_dynhost dynhost found in lan.list !"


# Suppression des 'not_required' et 'not_used' dans le lan.list

echo -e "${bold}\nFind extra dynhost in ${italic}lan.list...${reset}"

while IFS= read -r line; do
	
	ndd=$(echo "${line}" | awk '{print $2}' | awk -F"." '{print $1}')
	if [[ " ${not_required[*]} " =~ " $ndd " ]] || [[ " ${not_used[*]} " =~ " $ndd " ]]; then
		delete+=(${ndd})
	fi
		
done < lan.list

if [ ${#delete[@]} -ge 1 ]; then

	echo -e "${red}To delete: ${delete[@]}${reset}"

	b=$(echo -e "Do you want to delete ${#delete[@]} extra dynhost in ${italic}lan.list${reset} ? (y/n)")
	read -p "$b" choice
	if [ "$choice" == "y" ] || [ "$choice" == "Y" ]; then

		for val in ${!delete[@]}
		do
   			value="${delete[$val]}$domain"
   			value=$(echo "$value" | sed 's/\./\\\./g')
   			# -i '' for sed BSD ; -i for sed GNU
			sed -i '' "/$value/d" lan.list
		done
		nb_dynhost=$(awk 'END { print NR }' lan.list)
		echo "$nb_dynhost dynhost found in lan.list !"
	fi
	echo ""
fi


# Ajout des DynHOST de OVH (Zones DNS) dans les 2 lan.list (pihole1 & pihole2)
# Zones DNS -> photos-nas.ovh -> Historique des zones

echo -e "${bold}\nFind missing dynhost in ${italic}lan.list...${reset}"

dynhost_list=$(cat photos-nas.ovh_dns_data.txt | grep "60 IN A" | sed '1d')

while IFS= read -r line; do
				
	dynhost=$(echo "${line}" | awk '{print $1}')
	grep -q $dynhost lan.list
	if [ $? != 0 ]; then
		if [[ ! " ${not_required[*]} " =~ " $dynhost " ]] && [[ ! " ${not_used[*]} " =~ " $dynhost " ]]; then
			absent+=(${dynhost})
		fi
	fi
					
done <<< "$dynhost_list"

echo
echo -e "${bold}Update ${italic}lan.list...${reset}"
echo -e "${green}Not required: ${not_required[@]}${reset}"
echo -e "${italic}Not used: ${not_used[@]}${reset}"
echo -e "${red}Missing: ${absent[@]}${reset}"


if [ ${#absent[@]} -ge 1 ]; then

	[ -f /tmp/temp_file.list ] && rm /tmp/temp_file.list

	for val in ${absent[@]}
	do
		echo -e "192.168.2.57    ${val}.photos-nas.ovh    ${val}" >> /tmp/temp_file.list	
	done

	cat /tmp/temp_file.list >> lan.list
else
	echo -e "\n${green}No new dynhost !${reset}"
fi

cat lan.list | sort -k2 > lan.list.sorted
mv lan.list.sorted lan.list

echo
echo -e "${bold}Display ${italic}lan.list...${reset}"
cat lan.list

nb_dynhost=$(awk 'END { print NR }' lan.list)
echo -e "$nb_dynhost dynhost !\n"


a=$(echo -e "Do you want to export ${italic}lan.list${reset} to ${bold}pihole1${reset} and ${bold}pihole2${reset} ? (y/n)")
read -p "$a" choice
if [ "$choice" == "y" ] || [ "$choice" == "Y" ]; then
	
	if ping -q -c1 "$pihole1" &>/dev/null; then
		echo "Export to pihole1..."
		cat lan.list | ssh -p51322 root@"$pihole1" 'cat > /etc/pihole/lan.list'
		[ $? != 0 ] && echo -e "${red}Error during transfer to pihole1 ($pihole1)${reset}" || echo -e "${green}Ok${reset}"
		
		echo "Restarting dns on pihole1..."
		ssh -p51322 root@"$pihole1" 'pihole restartdns'
		[ $? != 0 ] && echo -e "${red}Error during restarting dns on pihole1 ($pihole1)${reset}" || echo -e "${green}Ok${reset}"
	fi
	

	if ping -q -c1 "$pihole2" &>/dev/null; then	
		echo "Export to pihole2..."
		cat lan.list | ssh -p51522 root@"$pihole2" 'cat > /etc/pihole/lan.list'
		[ $? != 0 ] && echo -e "${red}Error during transfer to pihole2 ($pihole2)${reset}" || echo -e "${green}Ok${reset}"

		echo "Restarting dns on pihole2..."
		ssh -p51522 root@"$pihole2" 'pihole restartdns'
		[ $? != 0 ] && echo -e "${red}Error during restarting dns on pihole2 ($pihole2)${reset}" || echo -e "${green}Ok${reset}"
	fi
else
	echo -e "OK, let's continue..."
fi

[ -f /tmp/temp_file.list ] && rm /tmp/temp_file.list
[ -f lan.list.sorted ] && rm lan.list.sorted