#!/bin/bash

# Define all the arguments
declare -A argExpected
argExpected['fqdn|FQDN']="FQDN - Fully Qualified Domain Name"
argExpected['record|R']="recordType=A - The type of DNS record to check (A, AAAA, CNAME, etc)"
argExpected['target|t']="target - The target value for the DNS record"
argExpected['h|help']="help - This help message"

# Get the source directory
SOURCE_ROOT="${BASH_SOURCE%/*}"

# Set the library root path
LIBRARY_PATH_ROOT="$SOURCE_ROOT/libs"

# Include all libraries in the libs directory
for f in "$LIBRARY_PATH_ROOT"/*.sh; do
	# Include the directory
	source "$f"
done

# Show the help text
if argPassed 'help'; then
	argList
	exit 0
fi

FQDN="$(argValue 'FQDN')"
recordType="$(argValue 'recordType')"

declare -A NAMESERVERS

NAMESERVERS['Google']="8.8.8.8 8.8.4.4"
NAMESERVERS['OpenDNS']="208.67.222.222 208.67.220.220 208.67.222.220 208.67.220.222"
NAMESERVERS['OpenDNSFamilyShield']="208.67.222.123 208.67.220.123"
NAMESERVERS['DNSAdvantage']="156.154.70.1 156.154.71.1"
NAMESERVERS['Verisign']="64.6.64.6 64.6.65.6"
NAMESERVERS['DNSWatch']="84.200.69.80 84.200.70.40"
NAMESERVERS['DYN']="216.146.35.35 216.146.36.36"
NAMESERVERS['Norton']="199.85.126.10 199.85.127.10"

table=""

table+="DNS Provider	\e[29mServer 1\e[0m	\e[29mServer 2\e[0m	\e[29mServer 3\e[0m	\e[29mServer 4\e[0m\n"
table+="\n"

for NSGroupName in "${!NAMESERVERS[@]}"; do

	NSGroupIPs=(${NAMESERVERS[$NSGroupName]})

	table+="$NSGroupName	"

	for NSGroupIP in "${NSGroupIPs[@]}"; do
		DIGOUTPUT=$(dig +nocmd +noall +answer "$FQDN" "$recordType" @$NSGroupIP)

		regexDNSRecord="^([[:alnum:]\-\.]+)[[:space:]]+([0-9]+)[[:space:]]+(IN)[[:space:]]+([[:alnum:]]+)[[:space:]]+(.+)$"
		[[ $DIGOUTPUT =~ $regexDNSRecord ]]

		reportedValue="${BASH_REMATCH[5]}"

		if [ "$reportedValue" == "$(argValue 'target')" ]; then
			table+="\e[32m"
		elif [ "$reportedValue" == "" ]; then
			reportedValue="[NO RECORD]"
			table+="\e[33m"
		else
			table+="\e[31m"
		fi

		table+="$reportedValue\e[0m	"
	done

	table+="\n"

done

echo -ne "$table" | column -tes $'\t'