PROG=${0##*/}

build_dir="build-ci-debug"

red=$(tput setaf 1)

green=$(tput setaf 2)

yellow=$(tput setaf 3)

blue=$(tput setaf 4)

magenta=$(tput setaf 5)

cyan=$(tput setaf 6)

normal=$(tput sgr0)

print_full_help() {

}

abort() {

}

check_dependency() {

}

check_dependency ctest

check_dependency cmake

if [ x"$1" = x"-h" ] || [ x"$1" = x"--help" ]; then

print_full_help >&2

exit 0

gdb_mode=false

while getopts "g" opt; do

case $opt in

g)

gdb_mode=true

echo "gdb_mode Mode Enabled"

;;

esac

shift $((OPTIND - 1))

if [ -z "${1}" ]; then

abort "Test regex is required"

test_suite=${1:-}

test_number=${2:-}

repo_root=$(git rev-parse --show-toplevel)

if [ ! -d "$repo_root" ]; then

abort "Not in a Git repository."

pushd "$repo_root"

rm -rf "$build_dir" && mkdir "$build_dir" || abort "Failed to make $build_dir"

cmake -B "./$build_dir" -DCMAKE_BUILD_TYPE=Debug -DGGML_CUDA=1 -DLLAMA_CURL=1 || abort "Failed to build environment"

pushd "$build_dir"

make -j || abort "Failed to compile"

popd > /dev/null || exit 1

printf "\n\nGathering tests that fit REGEX: ${test_suite} ...\n"

pushd "$build_dir"

tests=($(ctest -R ${test_suite} -V -N | grep -E " +Test +#[0-9]+*" | cut -d':' -f2 | awk '{$1=$1};1'))

if [ ${#tests[@]} -eq 0 ]; then

abort "No tests available... check your compilation process..."

popd > /dev/null || exit 1

if [ -z $test_number ]; then

# List out available tests

printf "Which test would you like to debug?\n"

id=0

for s in "${tests[@]}"

do

echo "Test# ${id}"

echo " $s"

((id++))

done

# Prompt user which test they wanted to run

printf "\nRun test#? "

read test_number

printf "\nUser Already Requested #${test_number}\n"

pushd "$build_dir"

sIFS=$IFS # Save Initial IFS (Internal Field Separator)

IFS=$'\n' # Change IFS (Internal Field Separator) (So we split ctest output by newline rather than by spaces)

test_args=($(ctest -R ${test_suite} -V -N | grep "Test command" | cut -d':' -f3 | awk '{$1=$1};1' )) # Get test args

IFS=$sIFS # Reset IFS (Internal Field Separator)

popd > /dev/null || exit 1

single_test_name="${tests[test_number]}"

single_test_command="${test_args[test_number]}"

printf "${magenta}Running Test #${test_number}: ${single_test_name}${normal}\n"

printf "${cyan}single_test_command: ${single_test_command}${normal}\n"

if [ "$gdb_mode" = "true" ]; then

# Execute debugger

pushd "$repo_root" || exit 1

eval "gdb --args ${single_test_command}"

popd > /dev/null || exit 1

# Execute Test

pushd "$repo_root" || exit 1

eval "${single_test_command}"

exit_code=$?

popd > /dev/null || exit 1

# Print Result

printf "${blue}Ran Test #${test_number}: ${single_test_name}${normal}\n"

printf "${yellow}Command: ${single_test_command}${normal}\n"

if [ $exit_code -eq 0 ]; then

printf "${green}TEST PASS${normal}\n"

else

printf "${red}TEST FAIL${normal}\n"

fi

popd > /dev/null || exit 1