#!/usr/bin/env sh

# ARGUMENTS:
# $1 == Domain to scan
# $2 == Nameserver to do the zone transfer from
# $3 == Default username for connections

scan_domain=$1
scan_ns=$2

zone_xfer=`dig $scan_domain @$scan_ns AXFR | egrep -v 'SOA|NS|MX|DiG' | grep $scan_domain | awk '{print $1,$4,$5}'`
for host in `echo $zone_xfer | awk '{print $1}'`; do
	user_name=$3
	connect_cmd="ssh -t"
	host_short=`echo ${host} | awk -F. '{print $1}'`
	if echo $zone_xfer | grep $host | grep TXT; then
		if [ "`echo $zone_xfer | grep $host | grep TXT | awk '{print $3}'`" = '"lxc"' ]; then
			user_name="root"
			host_ip=`echo $zone_xfer | grep ${host} | grep A | awk '{print $3}'`
			alias ${host_short}="${connect_cmd} ${user_name}@${host_ip}"
		elif [ "`echo $zone_xfer | grep $host | grep TXT | awk '{print $3}'`" = '"windows"' ]; then
			connect_cmd="rdesktop -g 1600x900"
			host_ip=`echo $zone_xfer | grep ${host} | grep A | awk '{print $3}'`
			alias ${host_short}="${connect_cmd} -u ${user_name} ${host_ip}"
		elif [ "`echo $zone_xfer | grep $host | grep TXT | awk '{print $3}'`" = '"cisco"' ]; then
			connect_cmd="telnet"
			host_ip=`echo $zone_xfer | grep ${host} | grep A | awk '{print $3}'`
			alias ${host_short}="${connect_cmd} ${host_ip}"
		fi
	else
		host_ip=`echo $zone_xfer | grep $host | grep A | awk '{print $3}'`
		alias ${host_short}="${connect_cmd} ${user_name}@${host_ip}"
	fi >/dev/null
done

return 0