Envsettings

#! /bin/sh

test -n "$XCLUSTER" && return 0

if ! ./xcadmin.sh env_check; then
	cat <<EOF

# Install dependecies;
sudo apt install -y net-tools qemu-kvm xterm screen
# Add user to the "kvm" group;
sudo usermod -a -G kvm $USER
# (logout/login after usermod)

EOF
	return 0
fi

if test -z "$XCLUSTER_WORKSPACE"; then
	export XCLUSTER_WORKSPACE="$(readlink -f .)/workspace"
	if ! test -d "$XCLUSTER_WORKSPACE"; then
		mkdir -p "$XCLUSTER_WORKSPACE"
		cat <<EOF

The \$XCLUSTER_WORKSPACE is set to;

  XCLUSTER_WORKSPACE=$XCLUSTER_WORKSPACE

EOF
	fi
fi

test -n "$ARCHIVE" || export ARCHIVE=$HOME/Downloads
mkdir -p $ARCHIVE

cdo() {
	cd $($XCLUSTER ovld $1)
}

export XCLUSTER="$(readlink -f .)/xcluster.sh"
alias xc=$XCLUSTER
alias xcadmin="$(readlink -f .)/xcadmin.sh"

eval $($XCLUSTER env | grep -i DISKIM)
export DISKIM
alias diskim=$DISKIM

which mke2fs > /dev/null || export PATH=$PATH:/sbin:/usr/sbin

if ip netns id | grep -q -E "xcluster[0-9]+\$"; then
	# We are in a netns. Reset the $__net_setup is it is set for main
	# netns.
	echo $__net_setup | grep -q '/config/net-setup-userspace.sh' && \
		unset __net_setup

	vm() {
		local bg='#040'
		test "$1" -gt 200 && bg='#400'
		test "$1" -gt 220 && bg='#004'
		XXTERM=XCLUSTER exec xterm -T "vm-$1" -fg wheat -bg "$bg" $xtermopt \
			-e "telnet 192.168.0.$1" &
	}

else
	# We are not in a xcluster netns. Use user-space networking.
	export __net_setup="$(readlink -f .)/config/net-setup-userspace.sh"

	vm() {
		local bg='#040'
		test "$1" -gt 200 && bg='#400'
		test "$1" -gt 220 && bg='#004'
		local nodeid=$1
		local p=$((12000+nodeid))
		test -n "$XCLUSTER_TELNET_BASE" && p=$((XCLUSTER_TELNET_BASE+nodeid))
		XXTERM=XCLUSTER exec xterm -T "vm-$1" -fg wheat -bg "$bg" $xtermopt \
			-e "telnet 127.0.0.1 $p" &
	}
fi

mynetns=$(ip netns id)
if ! netstat -lutan | grep -q :::10053; then
	cdns=$GOPATH/bin/coredns
	test -x $cdns || cdns=$XCLUSTER_WORKSPACE/bin/coredns
	if test -x $cdns; then
		cdnslog=/tmp/$USER/coredns.log
		test -n "$mynetns" && cdnslog=/tmp/$USER/coredns-$mynetns.log
		mkdir -p /tmp/$USER
		echo "Starting a local coredns on port 10053. Log in $cdnslog"
		$cdns -conf "$(readlink -f .)/config/Corefile" > $cdnslog 2>&1 &
	else
		echo "WARNING: CoreDNS not found. DNS from within the cluster will not work!"
	fi
	unset cdns cdnslog
fi

alias images="$(readlink -f .)/ovl/images/images.sh"
test -n "$__dns_spoof" || export __dns_spoof=$(readlink -f .)/config/dns-spoof.txt

# completion is a bash thing
ps -p $$ | grep -q bash || return 0
test -n "$PS1" || return 0

_cdo_completion() {
	local XCLUSTER_OVLPATH d o
	eval $($XCLUSTER env | grep XCLUSTER_OVLPATH=)
	for d in $(echo $XCLUSTER_OVLPATH | tr : ' '); do
		test -d $d || continue
		o+=$(find $d -mindepth 1 -maxdepth 1 -type d -name "$2*" -printf "%f ")
	done
	COMPREPLY=($o)
}
complete -F _cdo_completion cdo
_xc_completion() {
	COMPREPLY=($($XCLUSTER completion $2))
}
complete -F _xc_completion xc

lso() {
	local XCLUSTER_OVLPATH d arg
	eval $($XCLUSTER env | grep XCLUSTER_OVLPATH=)
	for d in $(echo $XCLUSTER_OVLPATH | tr : '\n' | sort -u); do
		arg="$arg $d/*"
	done
	if echo $@ | grep -qF -- '-t'; then
		\ls -dt $arg | sed -E 's,.*/([^/]+),\1,'
	else
		if test -t 1 -o "$1" = '-c'; then
			\ls -d $arg | sed -E 's,.*/([^/]+),\1,' | sort -u | column
		else
			\ls -d $arg | sed -E 's,.*/([^/]+),\1,' | sort -u 
		fi
	fi
}