#!/usr/bin/env bash
trap ctrl_c INT
ctrl_c() {
	pkill -P $$
	echo -e "Interrupted and cleaned up."
	return 255
}

_defaults() {
	match=""
	match_begin=""
	match_end=""
	match_sub=""

	tst() {
		echo "dummy test! please set me up properly" > /dev/stderr
		exit 1
	}

	prepare() {
		:
	}

	cleanup () {
		:
	}
}

_defaults

on_error() {
	on_error_default
}

on_success() {
	on_success_default
}

on_success_default() {
	echo "OK: $test_name"
	(( ok_count++ ))
	return 0
}

on_error_default() {
	echo "FAIL: $test_name"
	echo "(res: $res)"
	(( fail_count++ ))
	return 0
}

on_fatal() {
	echo "FATAL: $test_name"
	_final_cleanup
	exit 1
}

IFS=$'\n'
for i in "$@"; do
	if [[ ! -f "$i" ]]; then
		echo "ENOENT: $i"
		echo -e "$0 - basic test framework\n\nusage: $0 <test> [test] [...]"
		exit 1
	fi
done
unset IFS

ok_count=0
fail_count=0

_a() {
	[[ "$res_code" == 255 ]] && on_fatal

	# Q: why not `[[ ... ]] && a || b`?
	# A: simple; if `a` returns 1, `b` will get called erroneously.
	#	 normally one wouldn't care, but those functions are meant to
	#    be overriden. I don't want to fund anyone a lot of frustration,
	#	 so splitting the ifs is a saner option here :)
	
	if [[ "$match" ]]; then
		if [[ "$res" == "$match" ]]; then
			on_success
		else
			on_error
		fi
	elif [[ "$match_sub" ]]; then
		if [[ "$res" == *"$match_sub"* ]]; then
			on_success
		else
			on_error
		fi
	elif [[ "$match_begin" ]]; then
		if [[ "$res" == "$match_begin"* ]]; then
			on_success
		else
			on_error
		fi
	elif [[ "$match_end" ]]; then
		if [[ "$res" == *"$match_end" ]]; then
			on_success
		else
			on_error
		fi
	elif [[ "$match_not" ]]; then
		if [[ "$res" == *"$match_not"* ]]; then
			on_error
		else
			on_success
		fi
	else
		if [[ "$res_code" == 0 ]]; then
			on_success
		else
			on_error
		fi
	fi
	unset match match_sub match_begin match_end match_not
	prepare() { :; }
}

_final_cleanup() {
	# handle spawned processes
	for i in $(jobs -p); do
		pkill -P $i
	done
	if [[ "$(jobs -p)" != '' ]]; then
		sleep 1
		for i in $(jobs -p); do
			pkill -9 -P $i
		done
	fi
	pkill -P $$
}

for j in "$@"; do
	source "$j"
	if [[ "${#subtest_list[@]}" == 0 ]]; then
		test_name="$j"
		prepare
		res="$(tst)"
		res_code=$?
		cleanup
		_a
	else
		echo "--- $j ---"
		for i in "${subtest_list[@]}"; do
			test_name="$i"
			"$i"
			prepare
			res="$(tst)"
			res_code=$?
			cleanup
			_a
		done
	fi
	_defaults
done

_final_cleanup

echo -e "\n\nTesting done!
OK:   $ok_count
FAIL: $fail_count"