#!/bin/sh

set -e

info() {
	printf 'I: %s\n' "$*"
}

error() {
	printf 'E: %s\n' "$*"
}

gotestsum_runner() {
	gotestsum \
		--jsonfile "${junitOutput}.json" \
		--format "${gotestrunner_format}" \
		--raw-command \
		-- \
		"${GO}" test -json "$@"
}

gotestsum_postprocess() {
	touch "${junitOutput}.json"
	gotestsum \
		--junitfile "${junitOutput}" \
		--raw-command \
		-- \
		cat "${junitOutput}.json"
}

gotest_runner() {
	"${GO}" test "$@"
}

gotest_postprocess() {
	true
}

export GOFLAGS='-mod=vendor -trimpath'
export GO111MODULE='on'

GO='/usr/local/go/bin/go'
GO_TAGS='sylog singularity_engine fakeroot_engine apparmor selinux seccomp'

verbose=false
use_gotestsum=false
junitOutput=
gotestrunner_format=standard-quiet

test_runner=gotest_runner
test_postprocess=gotest_postprocess

skip=false

for arg in "$@" ; do
	shift

	if ${skip} ; then
		skip=false
		continue
	fi

	case "${arg}" in
		-sudo)
			# prepare sudo execution
			sudo_exec='-exec /home/gpsr/webserver/cgidocs/anshu/RAPID-engine/singularity-ce-4.2.2/scripts/test-sudo'
			if [ `id -u` = 0 ]; then
				error "Run $0 as user when specifying -sudo. Abort."
				exit 1
			fi
			if ! command -v sudo > /dev/null 2>&1; then
				error "sudo command not found in PATH. Abort."
				exit 1
			fi
			# ask for password or reset the session timeout
			if ! sudo -v; then
				exit 1
			fi
			;;

		-tags)
			GO_TAGS="${GO_TAGS} ${1}"
			skip=true
			;;

		-junit)
			if ! command -v gotestsum > /dev/null 2>&1 ; then
				error 'JUnit output requested but gotestsum not found in PATH. Abort.'
				info ''
				info 'Looked in the following directories, in order:'
				info ''
				IFS=:
				for dir in ${PATH} ; do
					info "    ${dir}"
				done
				exit 1
			fi

			use_gotestsum=true
			test_runner=gotestsum_runner
			test_postprocess=gotestsum_postprocess
			junitOutput="${1}"

			skip=true
			;;

		-v|-verbose)
			verbose=true
			set -- "$@" -v
			;;

		-e2e_groups)
		    e2e_groups="${1}"
			skip=true
			;;

		-e2e_tests)
			e2e_tests="${1}"
			skip=true
			;;

		*)
			set -- "$@" "${arg}"
			;;
	esac
done

if ${use_gotestsum} ; then
	if ${verbose} ; then
		gotestrunner_format=standard-verbose
	fi
fi

if ${use_gotestsum} ; then
	if ${verbose} ; then
		gotestrunner_format=standard-verbose
	fi
fi

if  [ -n "${e2e_groups}" ] ; then
	set -- "$@" "-e2e_groups"
	set -- "$@" "${e2e_groups}"
fi

if  [ -n "${e2e_tests}" ] ; then
	set -- "$@" "-e2e_tests"
	set -- "$@" "${e2e_tests}"
fi

# capture exit code
rc=0

"${test_runner}" \
	-count=1 \
	-timeout=30m \
	-tags "${GO_TAGS}" \
	-cover \
	${sudo_exec} \
	"$@" ||
rc=$?

"${test_postprocess}"

# return original exit code
exit ${rc}