run.sh

#!/bin/bash

function check_pip_installed() {
	PIP=$(which pip)
	if [ -z "$PIP" ] ; then
		echo ERROR: Python not installed
		exit 1
	fi
	for module in $@ ; do
		INSTALLED=$(pip show $module 2>/dev/null)
		if [ -z "$INSTALLED" ] ; then
			echo ERROR: module $module not installed. 	
			echo ERROR: please run pip install $@
			exit 1
		fi
	done
}

check_pip_installed boto netaddr

if [ ! -f ~/.aws/credentials ] ; then
	echo ERROR: ~/.aws/credentials are missing. 
	echo '	[default]'
	echo '	aws_access_key_id = <ACCESS_KEY>'
	echo '	aws_secret_access_key = <SECRET_KEY>'
	exit 1
fi

read -e -p 'AWS Profile [default]:' profilename
profilename=${profilename:-default}
read -e -p 'AWS Region [eu-central-1]:' profileregion
profileregion=${profileregion:-eu-central-1}

rm -rf target/$profilename
mkdir -p target/$profilename/default
mkdir -p target/$profilename/securitygroups
mkdir -p target/$profilename/subnets

echo INFO: graphing default dependencies
python graph_region.py -p $profilename --directory target/$profilename -r $profileregion $@
echo INFO: graphing with subnets
python graph_region.py -p $profilename --directory target/$profilename/subnets -r $profileregion --use-subnets $@
echo INFO: graphing with security groups 
python graph_region.py -p $profilename --directory target/$profilename/securitygroups -r $profileregion --use-security-group-subgraphs $@


DOT=$(which dot)
DOTFILE=$(find target/$profilename -type f -name '*.dot')

if [ -n "$DOT" ] ; then
	for file in $DOTFILE; do
		echo INFO: generating png for $file
		dot -Tpng -o $(dirname $file)/$(basename $file .dot).png  $file
	done
	if [ "$(uname)" == "Darwin" ] ; then
		open target/*/*.png
	else
		echo INFO: Done. checkout target/ subdirectories
	fi
else
	echo WARN: dot not installed.
fi