#!/bin/sh

# ----------------------------------------------------------------------
# Copyright (c) 2013, 2014, 2015, 2016 Thomas Levine
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as 
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# ----------------------------------------------------------------------


# ----------------------------------------------------------------------
# Copyright (c) 2014, Michael Klement
# Copyright (c) 2012, ScraperWiki Limited
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# Redistributions of source code must retain the above copyright notice, this
# list of conditions and the following disclaimer.
#
# Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
# ----------------------------------------------------------------------

set -e

# Delimiters
LF="$(printf '\n')"
HT="$(printf '\t')"
DELIMITERS="[${LF}${HT}]"

# Urchin version number
VERSION=0.0.0-master

# Kill subprocesses on interrupt.
trap "kill -$$; exit" HUP INT TERM

DEFAULT_SHELLS='
sh
bash
dash
ksh
posh
pdksh
mksh
yash
zsh
'
if [ -n "${ZSH_VERSION}" ]; then
  # avoid "no matches found: *" error when directories are empty
  setopt NULL_GLOB
  emulate sh
fi

# -------------------- Usage -------------------- 
USAGE="usage: $0 [options]... [test file or directory]..."

urchin_help() {
  cat <<EOF

${USAGE}

By default, Urchin checks for the following shells and runs every
particular test file once per shell.

  $(echo "${DEFAULT_SHELLS}" | sed 's/ /\n  /g')

On each run,

1. The TEST_SHELL environment variable is set to the particular shell.
2. If the test file lacks a shebang line or has a shebang line of
   "#!/bin/sh", the test script is also executed in that shell.

The following flags affect how this multiple-shell testing is handled.

-s, --shell <shell>      Tell Urchin to use a different list of shells.
                         (You can pass this flag multiple times.)

-n, --disable-cycling    Disable the cycling of shells; Urchin will
                         execute test files ordinarily, implicitly using
                         sh for files that lack shebang lines. It will
                         set the TEST_SHELL variable to "/bin/sh" if and
                         only if TEST_SHELL is empty or undefined.

The following flags affect how Urchin processes tests.

-b, --run-in-series      Run tests in series. The default is to run tests
                         in parallel where possible.
-e, --exit-on-fail       Stop running if any single test fails.
                         This can be useful if you are running something
                         other than test files with Urchin.
-T, --timeout <seconds>  Kill a test if it runs for longer than the
                         specified duration. The default is no timeout.
-f, --force              Force running even if the test directory's name
                         does not contain the word "test".

These options affect how results are formatted. Options -q, and -v
have no effect when combined with formats other than "urchin".
-vv, -vvv, and -vvvv do have effect when combined with formats "urchin"
or "tap".

-p, --pretty             Print results in color and with fancy symbols.
-F, --format <name>      XXX

And these options affect how much is printed.

-q, --quiet              Print nothing to stdout;
                         the only output is the exit code.
(default verbosity)      Print names of failed tests and counts
                         of passed, failed, and skipped tests.
-v                       Print stdout from failing tests.
-vv                      Print names of passed tests.
-vvv, --verbose          Print stdout from all tests.
-vvvv, --debug           Run with set -x.

The remaining flags provide information about urchin.

-h, --help               Display this help.
--version                Display the version number.

Urchin recognizes certain environment variables.

TEST_SHELL           This is sometimes over-ridden; see -s.
RUN_IN_SERIES        Set this to have the same effect as
                     -b/--run-in-series. This is helpful if you are
                     calling urchin inside an urchin test suite.

Exit codes have the following meanings

0                    All tests were ok
1                    At least one test was not ok.
2                    No tests were found.
10                   Dependencies are missing (locally, not on remotes).
11                   Flags were not valid.
12                   File names contain unsupported delimiters (HT or LF).

Go to https://thomaslevine.com/!/urchin/ for documentation on writing tests.

EOF
}

# -------------------- Dependency checks -------------------- 
if command -v md5 1> /dev/null 2> /dev/null; then
  urchin_md5=md5
elif command -v md5sum 1> /dev/null 2> /dev/null; then
  urchin_md5=md5sum
else
  echo Could not find MD5 hash command >&2
  exit 10
fi

if epoch_date 2>&1 > /dev/null; then
  epoch=epoch_date
elif epoch_pax 2>&1 > /dev/null; then
  epoch=epoch_pax
else
  echo I could not find a seconds counter. >&2
  exit 10
fi


# -------------------- Portable wrappers -------------------- 
mktemp_dir() {
  # Support HP-UX mktemp that has wrong exit codes and
  # can't make directories.
  tmp=$(mktemp /tmp/urchin.XXXXXXXX)
  if test -f "${tmp}"; then
    rm "${tmp}"
  fi
  mkdir "${tmp}"
  echo "${tmp}"
}
mktemp_file() {
  tmp=$(mktemp /tmp/urchin.XXXXXXXX)
  if ! test -f "${tmp}"; then
    > "${tmp}"
  fi
  echo "${tmp}"
}

md5 () {
  case "${urchin_md5}" in
    md5sum) echo "${1}" | md5sum | sed 's/ .*//' ;;
    md5) echo "${1}" | md5 | sed 's/.* //' ;;
  esac
}

# -------------------- Temporary directory -------------------- 
urchin_tmp=$(mktemp_dir)
> "${urchin_tmp}/log"

urchin_exit() {
  rm -Rf "${urchin_tmp}"
  exit "$@"
}

# -------------------- Utilities -------------------- 
epoch_date() {
  date +%s
}
epoch_pax() {
  # Based on http://stackoverflow.com/a/7262588/407226
  tmp="$(mktemp_file)"
  echo "ibase=8;$({ pax -wx cpio "${tmp}"; echo; } | cut -c 48-59)" | bc
  rm "${tmp}"
}

plural () {
  # Make $1 a plural according to the number $2.
  # If $3 is supplied, use that instead of "${1}s".
  # Result is written to stdout.
  if [ "${2}" = 1 ]; then
    echo "${1}"
  else
    echo "${3-${1}s}"
  fi
}

has_shebang_line() {
  head -n 1 "${1}" | grep -v '^#!/bin/sh$' | grep -q '^#!'
}

indent() {
  level="${1}"
  if test "${level}" -gt 0; then
    printf "%$((2 * ${level}))s"
  fi
}

stdout_file() {
  the_test="${1}"
  the_shell="${2}"

  x="${urchin_tmp}/stdout$(fullpath "$the_test")"
  mkdir -p "${x}"
  echo "${x}/$(md5 "${the_shell}")"
}

# Expand relative paths
fullpath() {
  readlink -f -- "${1}"
}

contains() {
  case "$#" in
    1) grep "${1}" > /dev/null ;;
    2) echo "${1}" | contains "${2}" ;;
    3) contains "${1}" "${2}" && contains "${1}" "${3}" ;;
    *) container="${1}" && shift
       contains "${container}" "${1}" && shift &&
       contains "${container}" "${@}" ;;
   esac
}

is_set() {
  set | grep "^${1}=" > /dev/null
}

remove_trailing_slash() {
  echo "$1" | sed s/\\/$//
}

# Find the root directory of the present test suite
urchin_root() {
  # Call recursively but remember the original argument.
  orig="${2:-$1}"
  current="${1%/}"

  abscurrent="$(fullpath "${current}")"
  if test "${abscurrent}" = / ||
    basename "${abscurrent}" | contains '^\.' ; then
    # Stop traversing upwards at / and at hidden directories.
    if test -d "${orig}"; then
      echo "${orig}"
    else
      dirname -- "${orig}"
    fi
  elif ! test -e "${current}"; then
    echo "${current}: No such file or directory">&2
    return 1
  elif test -f "${current}"; then
    urchin_root "$(dirname -- "${current}")" "${orig}"
  elif test -f "${current}"/.urchin_root; then
    echo "${current}"
  else
    urchin_root "${current}"/.. "${orig}"
  fi
}

# -------------------- Metafunctions -------------------- 
meta_verbosity() {
  echo "if test \${${1}} -gt ${2}; then ${3}=true; fi"
}

# -------------------- Printing output -------------------- 
# Format functions may read a log file from stdin.

format_tap() {
  v="${1}"
  tmp_dir="${2}"
  elapsed="${3}"

  $(verbosity v 2 print_not_ok_stdout)
  $(verbosity v 3 print_ok_stdout)

  print_stdout() {
    echo '# ------------ Begin output ------------'
    sed 's/^/# /' "$(stdout_file "${path}" "${the_shell}")"
    echo '# ------------ End output ------------'
  }

  while IFS="${HT}" read remote the_shell path result file_elapsed; do
    # Number of files that have run, including this one
    n=$(( ${n:-0} + 1))

    if test -z "${the_shell}"; then
      the_shell='File is not executable.'
    fi
    case "${result}" in
      ok) echo "ok $n - ${path} (${the_shell}${remote})"
          if "${print_ok_stdout}"; then print_stdout; fi ;;
      not_ok) echo "not_ok $n - ${path} (${the_shell}${remote})"
              if "${print_not_ok_stdout}"; then print_stdout; fi ;;
      skip) "ok $n - ${path} (${the_shell}${remote}) # SKIP" ;;
    esac
    echo "# Previous test took ${file_elapsed} seconds."

  done

  echo "# Full test suite took ${elapsed} $(plural second ${elapsed})."
  echo 1.."${n}"
}

format_urchin() {
  v="${1}"
  tmp_dir="${2}"
  verbosity="${3}"
  print_in_color="${4}"

  $(verbosity v 1 print_margins)
  $(verbosity v 1 print_not_ok)
  $(verbosity v 2 print_not_ok_stdout)
  $(verbosity v 2 print_ok)
  $(verbosity v 3 print_ok_stdout)

  if $print_in_color; then
    success_mark=$(printf "\033[32m✓ \033[0m")
    fail_mark=$(printf "\033[31m✗ \033[0m")
  else
    success_mark=.\ 
    fail_mark=F\ 
  fi

  header() {
    if test "${prevdir}" != "${currentdir}"; then
      echo
    fi
    if test "${prevpath}" != "${path}"; then
      printf "$(dirname -- "${path}")/\n> $(basename -- "${path}")\n"
    fi
  }

  print_stdout() {
    sed 's/^/  | /' "$(stdout_file "${path}" "${the_shell}")"
  }

  while IFS="${HT}" read remote the_shell path result file_elapsed; do
    abspath=${urchin_tmp}/${path}

    currentdir="$(dirname -- "${path}")"
    prevdir="${currentdir}"

    if test -z "${the_shell}"; then
      message="${the_shell} on ${remote} is not executable."
    else
      unit="$(plural second "${file_elapsed}")"
      message="${the_shell} on ${remote} (${file_elapsed} ${unit})"
    fi

    # Keep track of how many files have been ok, not ok, and skipped.
    eval "${result}s=$((${result}s+1))"

    # Print the result.
    case "${result}" in
      ok) if "${print_ok}"; then
            header && echo "${success_mark} ${message}"
          fi ;;
      not_ok) if "${print_not_ok}"; then
                header && echo "${fail_mark} ${message}"
                if "${print_not_ok_stdout}"; then print_stdout; fi
              fi ;;
      skip) if "${print_ok}"; then
              header && echo "${skip_mark} ${message}"
              if "${print_ok_stdout}"; then print_stdout; fi
            fi ;;
    esac
    prevpath="${path}"
  done

  if "${print_margins}"; then
    echo
    echo "Done, took ${elapsed} $(plural second ${elapsed})."
    echo "${oks} $(plural test "${oks}") passed."
    echo "${skips} $(plural test "${skips}") skipped."
    echo "${not_oks} $(plural test "${not_oks}") failed."
  fi
}

# -------------------- Main stuff -------------------- 
recurse() {
  requested_path="${1}"
  potential_test="$(fullpath "${2}")"
  cycle_shell="${3}"
  TEST_SHELL="${4}"

  root="$(urchin_root "${1}")"

  for ignore in setup_dir teardown_dir setup teardown; do
    if test "$(basename "${potential_test}")" = "${ignore}"; then
      return
    fi
  done

  # Return if we should not run this file
  if contains "${potential_test}" "^${requested_path}" ||
     contains "${requested_path}" "^${potential_test}" ; then
    if test "$(dirname "${potential_test}")" = \
            "$(dirname "${requested_path}")" &&
       test "${potential_test}" != "${requested_path}"; then
      return 0
    fi
  else
    return 0
  fi

  if contains "${potential_test}" "${HT}"; then
    echo 'Test file names may not contain tabs (HT).' >&2
    urchin_exit 11
  fi

  if [ -x "${potential_test}" ]; then
    if [ -d "${potential_test}" ]; then
    (
      cd -- "${potential_test}"
      if test -f .urchin_dir && grep series .urchin_dir > /dev/null; then
        run_in_series_dir=true
      else
        run_in_series_dir=false
      fi

      if test -f setup_dir; then
        . ./setup_dir
      fi

      for test in *; do
        if test "${test}" = '*' && ! test -e "${test}"; then
          # The directory is empty.
          break
        fi

        recurse "${requested_path}" "${test}" "${cycle_shell}" \
          "${TEST_SHELL}" &

        if "${run_in_series}" || "${run_in_series_dir}"; then
          if wait "${!}"; then exit_code=0; else exit_code="${?}"; fi
          if "${exit_on_not_ok}" && test "${exit_code}" -ne 0; then
            if test -f teardown_dir; then
              . ./teardown_dir
            fi
            return 1
          fi
        fi
      done
      wait
      if test -f teardown_dir; then
        . ./teardown_dir
      fi
    )
    elif [ -f "${potential_test}" ]; then
      cd -- "$(dirname -- "${potential_test}")"

      # Determine the environment variable to define for test scripts
      # that reflects the specified or implied shell to use for shell-code tests.
      echo "${shell_list}" | while read the_test_shell; do
        (
          if test -f setup; then
            . ./setup
          fi

          # Run the test
          start=$("${epoch}")
          set +e
          { 
            if "${cycle_shell}"; then
              if has_shebang_line "${potential_test}"; then
                TEST_SHELL="${the_test_shell}" $TIMEOUT "${potential_test}"
              else
                TEST_SHELL="${the_test_shell}" $TIMEOUT \
                  "${the_test_shell}" "${potential_test}"
              fi
            else
              # Shell cycling is disabled with -n; use the present value of
              # TEST_SHELL or default to /bin/sh
              if [ -n "${TEST_SHELL}" ]; then
                $TIMEOUT "${potential_test}"
              else
                TEST_SHELL=/bin/sh $TIMEOUT "${potential_test}"
              fi
            fi
          } > "$(stdout_file "${potential_test}" "${the_test_shell}")" 2>&1
          exit_code="${?}"
          set -e
          finish=$("${epoch}")

          if test -f teardown; then
            . ./teardown
          fi

          case "${exit_code}" in
            0) result=ok ;;
            3) result=skip ;;
            *) result=not_ok ;;
          esac

          elapsed=$(($finish - $start))
          rel="${potential_test##"${root}/"}"
          printf "\t${the_test_shell}\t${rel}\t${result}\t${elapsed}\n" \
            >> "${urchin_tmp}"/log
          exit "${exit_code}"
        ) &

        if "${run_in_series}"; then
          if wait "${!}"; then exit_code=0; else exit_code="${?}"; fi
          if "${exit_on_not_ok}" && test "${exit_code}" -ne 0; then
            if test -f teardown_dir; then
              . ./teardown_dir
            fi
            return 1
          fi
        fi
      done
      wait
    fi
  else
    # Shell is '' because "File is not executable".
    # Why does that happen?
    printf "\t\t${potential_test}\tskip\t0\n" >> "${urchin_tmp}"/log
  fi
}

report_outcome() {
  format="${2}"
  log_file="${3}"
  start="${4}"
  finish="${5}"

  elapsed=$(($finish - $start))

  # Use a temporary file rather than a pipe because a pipe starts a sub-shell
  # and thus makes the above variables local.
  sorted_log_file=$(mktemp_file)
  cat "${log_file}" | LC_COLLATE=C sort > "${sorted_log_file}"

  rm "${sorted_log_file}"

  test "${not_oks}" -eq '0'
}


main() {
  cycle_shell=true
  test_arg_list="${urchin_tmp}"/test_list
  > "${test_arg_list}"
  format=urchin

  while [ "${#}" -gt 0 ]
  do
      case "${1}" in
          -b|--run-in-series) run_in_series=true;;
          -e|--exit-on-fail) exit_on_not_ok=true;;
          -f|--force) force=true;;
          -s|--shell)
            shift
            shell_for_sh_tests="${1}"
           
            command -v "${shell_for_sh_tests}" > /dev/null || {
              echo "Cannot find specified shell: '${shell_for_sh_tests}'" >&2
              urchin_help >&2
              urchin_exit 13
            }

            if contains "${potential_test}" "${HT}"; then
              echo 'Shell paths may not contain tab characters (HT).' >&2
              urchin_exit 11
            fi

            if contains "${shell_for_sh_tests}" "[${IFS}]"; then
              echo "Warning: It is best if field-separator characters
(usually spaces) are absent from shell paths so that
you don't need to quote the TEST_SHELL variable." >&2
            fi
            shell_list="${shell_for_sh_tests}${LF}${shell_list}"
            ;;
          -n|--disable-cycling) cycle_shell=false;; 
          -F|--format) shift ; format="${1}";;
          -T|--timeout)
            shift
            urchin_timeout="${1}" 
            if ! contains "${urchin_timeout}" '[0-9][0-9.]*\(s\|m\|h\|d\|\)' ; then
              echo Bad timeout argument: "${urchin_timeout}" >&2
              urchin_exit 11
            fi ;;
          -p|--pretty) print_in_color=true;;

          -q|--quiet)     verbosity=0 ;;
          -v)             verbosity=2 ;;
          -vv)            verbosity=3 ;;
          -vvv|--verbose) verbosity=4 ;;
          -vvvv|--debug)  verbosity=5 ;;

          -h|--help) urchin_help
                     urchin_exit 0;;
          --version) echo "${VERSION}"
              urchin_exit;;
          -*) urchin_help >&2
              urchin_exit 11;;
          *)  if contains "${1}" "${HT}"; then
                echo 'Test file names may not contain tab characters (HT).' >&2
                urchin_exit 11
              elif [ ! -e "${1}" ]; then
                echo "No such file or directory: '${1}'" >&2
                echo "${USAGE}" >&2
                urchin_exit 11
              elif ! {
                # Molly guard
                root="$(urchin_root "${1}")"
                basename "$(fullpath "${root}")" |
                grep -i 'test' > /dev/null || "${force}"
              }; then
                echo 'The root directory of the tests that you are running urchin on
does not contain the word "test", so I am not running,
in case that was an accident. Use the -f flag if you really
want to run urchin on that directory.' >&2
                urchin_exit 12
              fi
              echo "${1}" >> "${test_arg_list}" ;;
      esac
      shift
  done
  if "${RUN_IN_SERIES}" 2> /dev/null; then
    run_in_series=true
  fi

  # -------------------- VALIDATE INPUT -------------------- #
  if ! "${cycle_shell}" && ! is_set shell_list; then
    echo "The -n/--disable-cycling and -s/--shell options clash with each other." >&2
    urchin_exit 11
  fi

  # If -s was not passed, use the available default shells.
  if ! is_set "${shell_list}"; then
    if $cycle_shell; then
      for shell in $DEFAULT_SHELLS; do
        if command -v "${shell}" 1> /dev/null 2> /dev/null; then
          shell_list="${shell}${HT}${shell_list}"
        fi
      done
    fi
  fi

  if $print_margins; then
    if test "${format}" = tap; then
      printf \#\ >> "${urchin_tmp}"/head
    fi
    if test "${format}" = urchin || test "${format}" = tap; then
      echo Running tests at $(date +%Y-%m-%dT%H:%M:%S) >> "${urchin_tmp}"/head
    fi

    if test "${format}" = tap; then
      printf '# ' >> "${urchin_tmp}"/head
    fi
    if test "${format}" = urchin || test "${format}" = tap; then
      printf 'Cycling with the following shells: ' >> "${urchin_tmp}"/head
    fi
    echo "${shell_list}" | tr "${HT}" \  >> "${urchin_tmp}"/head
    echo >> "${urchin_tmp}"/head
  fi

  if is_set urchin_timeout; then
    # Choose the timeout command
    if timeout -t 0 true 2> /dev/null; then
      TIMEOUT="timeout -t ${urchin_timeout}"
    elif timeout 0 true 2> /dev/null; then
      TIMEOUT="timeout ${urchin_timeout}"
    else
      echo I couldn\'t figure out how to use your version of timeout >&2
      urchin_exit 13
    fi
  fi

  if "${exit_on_not_ok}" && ! "${run_in_series}"; then
    echo 'You must also pass -b/--series in order to use -e/--exit-on-fail.' >&2
    urchin_exit 11
  fi

  # --------------------   REALLY RUN   -------------------- #
  start=$("${epoch}")

  # 1 test file or folder to run
  # 2 urchin root
  # 3 Should we cycle shells?
  # 4 TEST_SHELL
  while read seed; do
    recurse "$(fullpath "${seed}")" "${root}" "${cycle_shell}" \
      "${TEST_SHELL}" || break
  done < "${test_arg_list}"
  finish=$("${epoch}")

  if test $(cat "${urchin_tmp}"/log | wc -l) -eq 0; then
    echo 'No tests found' >&2
    urchin_exit 12
  fi
    
  cat "${urchin_tmp}"/head
  report_outcome "${root}" "${format}" "${urchin_tmp}"/log "${start}" \
    "${finish}"
# cat "${urchin_tmp}"/foot
  urchin_exit "${?}"
}

is_set TESTING_URCHIN_INTERNALS || main "$@"