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

count_only_flag=""
if [[ "$1" = "-c" ]]; then
  count_only_flag=1
  shift
fi

extended_syntax_flag=""
if [[ "$1" = "-x" ]]; then
  extended_syntax_flag="-x"
  shift
fi

trap "kill 0; exit 1" int

count=0
for filename in "$@"; do
  while IFS= read -r line; do
    if [[ "$line" =~ $BATS_TEST_PATTERN ]]; then
      let count+=1
    fi
  done <"$filename"
done

if [[ -n "$count_only_flag" ]]; then
  printf '%d\n' "$count"
  exit
fi

printf '1..%d\n' "$count"
status=0
offset=0
for filename in "$@"; do
  index=0
  {
    IFS= read -r # 1..n
    while IFS= read -r line; do
      case "$line" in
      "begin "* )
        let index+=1
        printf '%s\n' "${line/ $index / $(($offset + $index)) }"
        ;;
      "ok "* | "not ok "* )
        if [[ -z "$extended_syntax_flag" ]]; then
          let index+=1
        fi
        printf '%s\n' "${line/ $index / $(($offset + $index)) }"
        if [[ "${line:0:6}" == "not ok" ]]; then
          status=1
        fi
        ;;
      * )
        printf '%s\n' "$line"
        ;;
      esac
    done
  } < <( bats-exec-test $extended_syntax_flag "$filename" )
  offset=$(($offset + $index))
done

exit "$status"
