.bash_prompt

#!/usr/bin/env bash

# Shell prompt based on the Solarized Dark theme.
# Screenshot: http://i.imgur.com/EkEtphC.png
# Heavily inspired by @necolas’s prompt: https://github.com/necolas/dotfiles
# iTerm → Profiles → Text → use 13pt Monaco with 1.1 vertical spacing.

if [[ $COLORTERM = gnome-* && $TERM = xterm ]] && infocmp gnome-256color >/dev/null 2>&1; then
	export TERM='gnome-256color';
elif infocmp xterm-256color >/dev/null 2>&1; then
	export TERM='xterm-256color';
fi;

function prompter() {
	PS1="\[\033]0;\w\007\]";
	PS1+="\[${bold}\]\n"; # newline
	PS1+="\[${userStyle}\]\u"; # username
	PS1+="\[${white}\] at ";
	PS1+="\[${hostStyle}\]\h"; # host
	PS1+="\[${white}\] in ";
	PS1+="\[${green}\]\w"; # working directory
	PS1+="\$(prompt_git \"${white} on ${violet}\")"; # Git repository details
	#PS1+="\[${bold}\]\n"; # newline
	PS1+="\[${blue}\] k8s\[${white}\]:";
	PS1+="\${blue}$(k8s \")"; # Current Kubernet cluster
	PS1+="\[${orange}\] aws\[${white}\]:";
	PS1+="\${orange}$(awsf \")"; # Current aws context
	PS1+="\n";
	PS1+="\[${white}\]\$ \[${reset}\]"; # `$` (and reset color)
	export PS1;
}

k8s() {
	kube="$(kubectl config current-context)"
	echo -e "${kube}${blue}"
}

awsf() {

	echo -e "$AWS_DEFAULT_PROFILE${orange}"
}

prompt_git() {
	local s='';
	local branchName='';

	# Check if the current directory is in a Git repository.
	if [ $(git rev-parse --is-inside-work-tree &>/dev/null; echo "${?}") == '0' ]; then

		# check if the current directory is in .git before running git checks
		if [ "$(git rev-parse --is-inside-git-dir 2> /dev/null)" == 'false' ]; then

			# Ensure the index is up to date.
			git update-index --really-refresh -q &>/dev/null;

			# Check for uncommitted changes in the index.
			if ! $(git diff --quiet --ignore-submodules --cached); then
				s+='+';
			fi;

			# Check for unstaged changes.
			if ! $(git diff-files --quiet --ignore-submodules --); then
				s+='!';
			fi;

			# Check for untracked files.
			if [ -n "$(git ls-files --others --exclude-standard)" ]; then
				s+='?';
			fi;

			# Check for stashed files.
			if $(git rev-parse --verify refs/stash &>/dev/null); then
				s+='$';
			fi;

		fi;

		# Get the short symbolic ref.
		# If HEAD isn’t a symbolic ref, get the short SHA for the latest commit
		# Otherwise, just give up.
		branchName="$(git symbolic-ref --quiet --short HEAD 2> /dev/null || \
			git rev-parse --short HEAD 2> /dev/null || \
			echo '(unknown)')";

		[ -n "${s}" ] && s=" [${s}]";

		echo -e "${1}${branchName}${blue}${s}";
	else
		return;
	fi;
}

if tput setaf 1 &> /dev/null; then
	tput sgr0; # reset colors
	bold=$(tput bold);
	reset=$(tput sgr0);
	# Solarized colors, taken from http://git.io/solarized-colors.
	black=$(tput setaf 0);
	blue=$(tput setaf 117);
	cyan=$(tput setaf 37);
	green=$(tput setaf 114);
	orange=$(tput setaf 208);
	purple=$(tput setaf 125);
	red=$(tput setaf 197);
	violet=$(tput setaf 139);
	white=$(tput setaf 15);
	yellow=$(tput setaf 222);
else
	bold='';
	reset="\e[0m";
	black="\e[1;30m";
	blue="\e[1;34m";
	cyan="\e[1;36m";
	green="\e[1;32m";
	orange="\e[1;33m";
	purple="\e[1;35m";
	red="\e[1;31m";
	violet="\e[1;35m";
	white="\e[1;37m";
	yellow="\e[1;33m";
fi;

# Highlight the user name when logged in as root.
if [[ "${USER}" == "root" ]]; then
	userStyle="${red}";
else
	userStyle="${orange}";
fi;

# Highlight the hostname when connected via SSH.
if [[ "${SSH_TTY}" ]]; then
	hostStyle="${bold}${red}";
else
	hostStyle="${yellow}";
fi;

vpnOn() {
	if [ $(ifconfig | grep -Eo 'tun[0-9]*' | wc -l) > 0 ]; then
		vpnStyle="${yellow}";
		vpnOn="yes";
	else
		vpnStyle="${bold}${red}";
		vpnOn="no";
	fi;
	echo -e "\[${vpnStyle}\] vpn:${vpnOn}"
}

PS2="\[${yellow}\]→ \[${reset}\]";
export PS2;

#get k8s nodes external IPs

node-ip(){
	kubectl get node $1 -o json | jq -r '.status.addresses[] | select(.type == "ExternalIP") | .address'
}

nodes-ip(){
for each in $(kubectl get nodes --no-headers=true | cut -d " " -f1);
  do
  echo "Node" $each "has this External IP:"  ${yellow}$(kubectl get node $each -o json | jq -r '.status.addresses[] | select(.type == "ExternalIP") | .address')
	tput sgr0
done
}
#get k8s nodes aws zone
nodes-zone(){
for each in $(kubectl get nodes --no-headers=true | cut -d " " -f1);
  do
  echo "Node" $each "is located in:"  ${yellow}$(kubectl get node $each -o json | jq -r '.metadata.labels."failure-domain.beta.kubernetes.io/zone"')
	tput sgr0
done
}