#!/usr/bin/env bash
set -e

version() {
  echo "Bats 0.4.0"
}

usage() {
  version
  echo "Usage: bats [-c] [-p | -t] <test> [<test> ...]"
}

help() {
  usage
  echo
  echo "  <test> is the path to a Bats test file, or the path to a directory"
  echo "  containing Bats test files."
  echo
  echo "  -c, --count    Count the number of test cases without running any tests"
  echo "  -h, --help     Display this help message"
  echo "  -p, --pretty   Show results in pretty format (default for terminals)"
  echo "  -t, --tap      Show results in TAP format"
  echo "  -v, --version  Display the version number"
  echo
  echo "  For more information, see https://github.com/bats-core/bats-core"
  echo
}

BATS_READLINK=

resolve_link() {
  if [[ -z "$BATS_READLINK" ]]; then
    if command -v 'greadlink' >/dev/null; then
      BATS_READLINK='greadlink'
    elif command -v 'readlink' >/dev/null; then
      BATS_READLINK='readlink'
    else
      BATS_READLINK='true'
    fi
  fi
  "$BATS_READLINK" "$1" || return 0
}

abs_dirname() {
  local cwd="$PWD"
  local path="$1"

  while [ -n "$path" ]; do
    cd "${path%/*}"
    local name="${path##*/}"
    path="$(resolve_link "$name")"
  done

  printf -v "$2" -- '%s' "$PWD"
  cd "$cwd"
}

expand_path() {
  local path="${1%/}"
  local dirname="${path%/*}"

  if [[ "$dirname" == "$path" ]]; then
    dirname="$PWD"
  elif cd "$dirname" 2>/dev/null; then
    dirname="$PWD"
    cd "$OLDPWD"
  else
    printf '%s' "$path"
    return
  fi
  printf -v "$2" '%s/%s' "$dirname" "${path##*/}"
}

abs_dirname "$0" 'BATS_LIBEXEC'
abs_dirname "$BATS_LIBEXEC" 'BATS_PREFIX'
abs_dirname '.' 'BATS_CWD'

export BATS_PREFIX
export BATS_CWD
export BATS_TEST_PATTERN="^[[:blank:]]*@test[[:blank:]]+(.*[^[:blank:]])[[:blank:]]+\{(.*)\$"
export PATH="$BATS_LIBEXEC:$PATH"

options=()
arguments=()
for arg in "$@"; do
  if [ "${arg:0:1}" = "-" ]; then
    if [ "${arg:1:1}" = "-" ]; then
      options[${#options[*]}]="${arg:2}"
    else
      index=1
      while option="${arg:$index:1}"; do
        [ -n "$option" ] || break
        options[${#options[*]}]="$option"
        let index+=1
      done
    fi
  else
    arguments[${#arguments[*]}]="$arg"
  fi
done

unset count_flag pretty
count_flag=''
pretty=''
[ -t 0 ] && [ -t 1 ] && pretty="1"
[ -n "${CI:-}" ] && pretty=""

if [[ "${#options[@]}" -ne '0' ]]; then
  for option in "${options[@]}"; do
    case "$option" in
    "h" | "help" )
      help
      exit 0
      ;;
    "v" | "version" )
      version
      exit 0
      ;;
    "c" | "count" )
      count_flag="-c"
      ;;
    "t" | "tap" )
      pretty=""
      ;;
    "p" | "pretty" )
      pretty="1"
      ;;
    * )
      usage >&2
      exit 1
      ;;
    esac
  done
fi

if [ "${#arguments[@]}" -eq 0 ]; then
  usage >&2
  exit 1
fi

filenames=()
for filename in "${arguments[@]}"; do
  expand_path "$filename" 'filename'

  if [ -d "$filename" ]; then
    shopt -s nullglob
    for suite_filename in "$filename"/*.bats; do
      filenames["${#filenames[@]}"]="$suite_filename"
    done
    shopt -u nullglob
  else
    filenames["${#filenames[@]}"]="$filename"
  fi
done

if [ "${#filenames[@]}" -eq 1 ]; then
  command="bats-exec-test"
else
  command="bats-exec-suite"
fi

set -o pipefail execfail
if [ -z "$pretty" ]; then
  exec "$command" $count_flag "${filenames[@]}"
else
  extended_syntax_flag="-x"
  formatter="bats-format-tap-stream"
  exec "$command" $count_flag $extended_syntax_flag "${filenames[@]}" | "$formatter"
fi
