#!/usr/bin/env zsh
setopt err_exit
typeset -i benchmark=0 test_extended=0 test_security=0

while [[ $1 == --* ]] {
	case $1 in
		--all) test_extended=1; test_security=1 ;;
		--benchmark) benchmark=1 ;;
		--extended) test_extended=1 ;;
		--security) test_security=1 ;;
	esac
	shift
}

typeset envstore=${1-envstore}
typeset store_file=${2-/tmp/.envstore-$UID}
trap "print -P '\n%N:%i: %B%F{red}Test faild!%F{default}%b'" ZERR
trap "$envstore clear" INT

cat <<- ente
	Usage: $0 [options] [path to envstore] [path to envstore store file]
	valid options are:
	     --all         enable all test options (except --benchmark)
	     --benchmark   do a little benchmark after the test (might cause CPU and HDD stress)
	     --extended    Test for more than just the new envstore API (invalid invocations etc)
	     --security    Test for security (world-writable store file etc)

	Note: this script will remove/overwrite your envstore store file.
	The envstore script needs to be compatible with the original envstore API,
	that is, it should implement the commends "clear", "rm", "eval",
	"save var" and "save var value"

ente

echo "# $envstore clear"
$envstore clear

echo "# $envstore save var"
export hello=world
$envstore save hello
unset hello

echo "# $envstore eval"
eval $($envstore eval)
[[ $hello == world ]]
unset hello

echo "# $envstore rm"
$envstore rm hello
eval $($envstore eval)
[[ -z $hello ]]

echo "# $envstore save var value"
$envstore save hello world

echo "# $envstore eval"
eval $($envstore eval)
[[ $hello == world ]]
unset hello

echo "# $envstore clear"
$envstore clear
eval $($envstore eval)
[[ -z $hello ]]

echo "# $envstore save + eval (spaces in value), save var"
export hello='your mom'
$envstore save hello
unset hello
eval $($envstore eval)
[[ $hello == 'your mom' ]]
unset hello
$envstore clear

echo "# $envstore save + eval (spaces in value), save var value"
$envstore save hello 'your mom'
eval $($envstore eval)
[[ $hello == 'your mom' ]]
unset hello

echo "# $envstore save (overwrite)"
$envstore save hello world
eval $($envstore eval)
[[ $hello == world ]]
$envstore clear

echo "# $envstore save (' in value)"
$envstore save hello "the ' dude"
eval $($envstore eval)
[[ $hello == "the ' dude" ]]
unset hello

echo "# $envstore save (UTF-8)"
export hello='mÿde Rentner… und so'
$envstore save hello
unset hello
eval $($envstore eval)
[[ $hello == 'mÿde Rentner… und so' ]]
unset hello
$envstore clear

if ((test_extended)) {
	echo "# invalid invocations"
	! $envstore save
	unset nonexistent
	! $envstore save nonexistent
	! $envstore rm

	echo "# other invocations"
	$envstore show
	$envstore eval
	$envstore rm nonexistent
	$envstore clear
}

if ((test_security)) {
	echo "# world-writable store file"
	$envstore save fucked yes
	chmod 777 $store_file
	[[ $($envstore eval) != *fucked=yes* ]]
	rm $store_file
	$envstore save fucked yes
	cat <<- ente
		Now, enter the following as root:
		chown root $store_file
	ente
	echo -n "[press return when done] "
	read
	[[ $($envstore eval) != *fucked=yes* ]]
	cat <<- ente
		Looking good, you may remove the file now
		rm $store_file
	ente
	echo -n "[press return when done] "
	read
}

print -P '\n%F{green}Test passed%F{default}\n'

if ((benchmark)) {
	trap '' ZERR
	echo "## some benchmarks now..."
	TIMEFMT='%*E real, %*U user, %*S system - %P%% CPU - %J'

	echo "# adding 5000 vars (this could take a _long_ time)"
	for i in {0..5000}; {
		(( i % 500 )) || echo "# $((5000-i)) to go"
		$envstore save $i $i$i$i
	}
	echo

	repeat 2 {time $envstore eval > /dev/null}
	time $envstore rm 999 > /dev/null
	repeat 2 {time $envstore clear}
}