#!/bin/sh

set -e

if [ -z "$SRCDIR" -o -z "$DATADIR" ]; then
	echo "Missing either SRCDIR or DATADIR env vars"
	exit 1
fi

export RESULTDIR="$DATADIR/uploads"
export ECHO_INPUT=1

SCRIPTPATH="$(dirname $(readlink -f "$0"))"
TESTDATA="$SCRIPTPATH/data"

shopt -s expand_aliases
alias pushd="pushd &>/dev/null"
alias popd="popd &>/dev/null"

pushd "$SRCDIR"

announce() {
	count=$(echo "$1" | wc -c)
	python3 -c "
import math
s = '$1'
c = 80
print()
print('#'*c)
print('#' + ' '*math.floor((c - len(s))/2-1) + s + ' '*math.ceil((c - len(s))/2-1) + '#')
print('#'*c)
print()
	"
}

checkleaks() {
	valgrind --leak-check=full --show-leak-kinds=all ./build/stldoctor 2>&1 | tee /tmp/testlog
	if [ -z "$(grep "no leaks are possible" /tmp/testlog)" ]; then
		echo "Valgrind exited with errors!"
		exit 1
	fi
}

connect() {
	if [ "$RUNTYPE" == "remote" ]; then
		nc localhost 9090
	elif [ "$RUNTYPE" == "debug" ]; then
		checkleaks
	else
		./build/stldoctor
	fi
}

cleanuploads() {
	[ ! -z "$RESULTDIR" ] && rm -rf "$RESULTDIR"
	mkdir -p "$RESULTDIR"
}

if [ "$1" == "stl-leaks" ]; then
	cleanuploads

	announce "Testing ASCII STL Parsing"
	(
		echo "echo"
		echo "upload"
		cat "$TESTDATA/sample-ascii.stl" | wc -c
		cat "$TESTDATA/sample-ascii.stl"
		echo "ASCII-testname"
	) | checkleaks

	announce "Testing BIN STL Parsing"
	(
		echo "echo"
		echo "upload"
		cat "$TESTDATA/sample-binary.stl" | wc -c
		cat "$TESTDATA/sample-binary.stl"
		echo "BIN-testname"
	) | checkleaks

elif [ "$1" == "stl-upload" ]; then
	cleanuploads

	popd
	file="$(realpath $2)"
	if [ ! -e "$file" ]; then
		echo "Supply a file to upload"
		exit 1
	fi
	pushd "$SRCDIR"

	name="${3:-samplefile}"
	(
		echo "echo"
		echo "upload"
		cat "$file" | wc -c
		cat "$file"
		echo "$name"
	) | checkleaks

elif [ "$1" == "auth-upload" ]; then
	cleanuploads

	(
		echo "echo"

		echo "auth test"
		echo "upload"
		cat "$TESTDATA/sample-ascii.stl" | wc -c
		cat "$TESTDATA/sample-ascii.stl"
		echo "testname"
	) | connect

	(
		echo "echo"

		echo "auth test"
		echo "list"
		echo "search testname"
	) | connect
else
	connect
fi

popd