set -e

# cd into script's directory
cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null
root=$(pwd)

export UTILS="$root"/utils

REGEXS=()

while [[ $# -gt 0 ]]; do
  case $1 in
    -h|--help)
        cat <<EOF
USAGE: [flags] [test regexs...]
FLAGS
    --keep-tmp
    --verbose (-v)
    --help    (-h)
EOF
      exit 1
      ;;
    -v|--verbose)
      VERBOSE=1
      shift
      ;;
    --keep-tmp)
      KEEP_TMP=1
      shift
      ;;
    *)
      REGEXS+=("$1")
      shift
      ;;
  esac
done

[ -n "$VERBOSE" ] && set -x

ROOT_TMPDIR="$(mktemp --tmpdir -d isle-tests.XXXXXX)"
if [ -z "$KEEP_TMP" ]; then trap 'rm -rf $ROOT_TMPDIR' EXIT; fi

TMPDIR="$ROOT_TMPDIR"

export ROOT_TMPDIR TMPDIR
echo "tmp dir is $ROOT_TMPDIR"

# Blackhole these directories so that tests don't accidentally use the host's
# real ones.
export XDG_RUNTIME_DIR=/dev/null
export XDG_DATA_HOME=/dev/null

test_files=$(
    find ./cases -type f -name '*.sh' \
    | sed "s|^\./cases/||" \
    | grep -v entrypoint.sh \
    | sort
)

for r in "${REGEXS[@]}"; do
    test_files="$(echo "$test_files" | grep "$r")"
done

echo -e "number of tests: $(echo "$test_files" | wc -l)\n"
for file in $test_files; do
    echo "Running test case: $file"

    if [ -z "$VERBOSE" ]; then
        output="$TMPDIR/$file.log"
        mkdir -p "$(dirname "$output")"
    else
        output=/dev/stdout
    fi

    (
        export TEST_CASE_FILE="$file"

        if ! $SHELL -e -x "$root/cases/$file" >"$output" 2>&1; then
            echo "$file FAILED"
            if [ -z "$VERBOSE" ]; then
                echo "output of test is as follows"
                echo "------------------------------"
                cat "$output"
                echo "------------------------------"
            fi
            exit 1
        fi
    ) || TESTS_FAILED=1

    if [ -n "$TESTS_FAILED" ]; then break; fi
done


# Clean up any shared running clusters. Each cleanup script is responsible for
# figuring out if its shared cluster was actually instantiated during any tests.

if [ -e "$ROOT_TMPDIR/cleanup-pids" ]; then
    echo "Cleaning up running pids"
    tac "$ROOT_TMPDIR/cleanup-pids" | while read -r line; do
        pid="$(echo "$line" | cut -d' ' -f1)"
        descr="$(echo "$line" | cut -d' ' -f2-)"
        echo "Killing $descr ($pid)"
        kill "$pid"
    done

    # This is easier than checking if the pids are still running, and for some
    # reason it doesn't occur until after the pids have died anyway
    echo "Waiting for appimage mounts to unmount"
    while [ "$(find "$ROOT_TMPDIR" -type d -name '*.mount_isle*' | wc -l)" -ge "1" ]; do
        sleep 1
    done
fi

if [ -z "$TESTS_FAILED" ]; then echo -e '\nall tests succeeded!'; fi