#!/usr/bin/env bash

set -e

function usage {
  cat <<EOF
USAGE

  $0 [-adDg] [-s SEED] [-l LANGUAGE] [-e EXAMPLE]

OPTIONS

  -h  Print this message

  -a  Compile C code with the Clang address sanitizer

  -e  Run only the corpus tests whose name contain the given string

  -i  Run the given number of iterations of randomized tests (default 10)

  -s  Set the seed used to control random behavior

  -d  Print parsing log to stderr

  -D  Generate an SVG graph of parsing logs

  -g  Run the tests with a debugger

EOF
}

export RUST_BACKTRACE=full

mode=normal
test_flags=()

while getopts "adDghl:e:s:i:" option; do
  case ${option} in
    h)
      usage
      exit
      ;;
    a)
      export CFLAGS=-fsanitize=undefined,address

      # When the Tree-sitter C library is compiled with the address sanitizer, the address sanitizer
      # runtime library needs to be linked into the final test executable. When using Xcode clang,
      # the Rust linker doesn't know where to find that library, so we need to specify linker flags directly.
      runtime_dir=$(cc -print-runtime-dir)
      if [[ $runtime_dir == */Xcode.app/* ]]; then
        export RUSTFLAGS="-C link-arg=-L${runtime_dir} -C link-arg=-lclang_rt.asan_osx_dynamic -C link-arg=-Wl,-rpath,${runtime_dir}"
      fi

      # Specify a `--target` explicitly. This is required for address sanitizer support.
      toolchain=$(rustup show active-toolchain)
      toolchain_regex='(stable|beta|nightly)-([_a-z0-9-]+).*'
      if [[ $toolchain =~ $toolchain_regex ]]; then
        release=${BASH_REMATCH[1]}
        current_target=${BASH_REMATCH[2]}
      else
        printf "Failed to parse toolchain '%s'\n" "$toolchain" >&2
      fi

      test_flags+=("--target=$current_target")
      ;;
    e)
      export TREE_SITTER_EXAMPLE=${OPTARG}
      ;;
    s)
      export TREE_SITTER_SEED=${OPTARG}
      ;;
    i)
      export TREE_SITTER_ITERATIONS=${OPTARG}
      ;;
    d)
      export TREE_SITTER_LOG=1
      ;;
    D)
      export TREE_SITTER_LOG_GRAPHS=1
      ;;
    g)
      mode=debug
      ;;
    *)
      usage
      exit 1
      ;;
  esac
done

shift $((OPTIND - 1))

if [[ ${mode} == debug ]]; then
  test_binary=$(
    cargo test "${test_flags[@]}" --no-run --message-format=json 2> /dev/null |
    jq -rs 'map(select(.target.name == "tree-sitter-cli" and .executable))[0].executable'
  )
  lldb "${test_binary}" -- "$1"
else
  cargo test "${test_flags[@]}" "$1" -- --nocapture
fi