#!/bin/sh

getabsolutepath() {
	[ -d "$1" ] && { cd "$1"; echo "$(pwd -P)"; } ||
	{ cd "$(dirname "$1")"; echo "$(pwd -P)/$(basename "$1")"; }
}

pfx=$(getabsolutepath "$1")

if [ -z "$1" ]; then
	echo "Usage: ./env.sh [destdir|prefix] [program]"
	exit 1
fi

if [ ! -d "$pfx" ]; then
	echo "Cannot find $pfx directory"
	exit 1
fi

# Support DESTDIR
if [ -d "$pfx/usr/bin" ]; then
	pfx="$pfx/usr"
fi

new_env='
R2_ENV_IS_SET=1
R2_LIBR_PLUGINS=${pfx}/lib/radare2
PATH=$pfx/bin${PATH:+:${PATH}}
LIBRARY_PATH=$pfx/lib${LIBRARY_PATH:+:${LIBRARY_PATH}}
LD_LIBRARY_PATH=$pfx/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
DYLD_LIBRARY_PATH=$pfx/lib${DYLD_LIBRARY_PATH:+:${DYLD_LIBRARY_PATH}}
PKG_CONFIG_PATH=$pfx/lib/pkgconfig${PKG_CONFIG_PATH:+:${PKG_CONFIG_PATH}}
'

shift
if [ "$R2_ENV_IS_SET" = 1 ]; then
	echo "Nested call"
	exec $*
	exit $?
fi

if [ -z "$*" ]; then
	echo
	echo "==> Entering radare2 environment shell..."
	echo
	echo $new_env $* \
	   | sed -e 's, ,\n,g' \
	   | sed -e 's,^,  ,g' \
	   | sed -e 's,$, \\,'
	echo
	export PS1="r2env.sh$ "
	eval $new_env $SHELL
	echo
	echo "==> Back to system shell..."
	echo
else
	if [ $# -gt 1 ]; then
		par=""
		p=0
		while : ; do
			p=$(($p+1))
			[ $p -gt $# ] && break
			a=`eval echo "\$\{$p\}"`
			par="$par\"$a\" "
		done
		eval $new_env $par
	else
		eval $new_env $*
	fi
fi