#!/usr/bin/env bash

#
# Copyright 2017-18 IBM Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

# errors otherwise: nvm is not compatible with the "npm_config_prefix" environment var
npm config delete prefix
rm -f ~/.bashrc ~/.bash_profile
rm -f .nvmrc

set -e
set -o pipefail

# composer/kui doesn't support travis osx and travis osx doesn't have docker
if [ "$TRAVIS_OS_NAME" == "osx" ] && ([ "$TRAVIS_REPO_SLUG" == "composer/kui" ] || [ "$NEEDS_KUBERNETES" == true ]); then
  exit 0
fi

SCRIPTDIR=$(cd $(dirname "$0") && pwd)

if [ "$TRAVIS_OS_NAME" == "linux" ] && [ -n "$NEEDS_TOP" ]; then
   "$SCRIPTDIR"/top.sh | tee /tmp/top.out &
   TOP_PID=$!
fi

children=()
childrenNames=()
childrenStartTimes=()

if [ -n "$SCRIPTS" ]; then
    #
    # then we were asked to run one or more test.d/ scripts
    #
    for script in $SCRIPTS; do
        "$SCRIPTDIR"/test.d/$script &
        PID=$!
        echo "spawning test script $script with PID $PID"
        children+=("$PID")
        childrenNames+=("$script")
        childrenStartTimes+=("$(date +%s)")
    done
fi

function wait_and_get_exit_codes() {
    children=("$@")
    EXIT_CODE=0
    # the ! gives us indices
    for jobIdx in "${!children[@]}"; do
       job="${children[$jobIdx]}"
       jobName="${childrenNames[$jobIdx]}"
       echo "$(tput setaf 3)waiting on ${jobName} with PID ${job}$(tput sgr0)" # yellow text
       CODE=0;
       wait ${job} || CODE=$?
       if [[ "${CODE}" != "0" ]]; then
           echo "$(tput setaf 1)failing: job ${jobName} exited with a non-zero code ${CODE}$(tput sgr0)" # red text
           EXIT_CODE=1;
       else
           then="${childrenStartTimes[$jobIdx]}"
           now=$(date +%s)
           delta=$((now-then))
           echo "$(tput setaf 2)ok: travis job ${jobName} exited with success in <=${delta} seconds$(tput sgr0)" # red text
       fi
   done
}

function count {
    cnt=0
    for i in $1; do
        cnt=$((cnt+1))
    done
    echo $cnt
}

if [ -n "$LAYERS" ]; then
    #
    # we were asked to run one or more mocha test suites (which suites
    # as indicated by $LAYERS)
    #

    export KEY=${TRAVIS_JOB_NUMBER}
    echo "Using KEY=$KEY"

    # remove HEADLESS from the list, as we are handling in specially
    # in the else clause below
    NON_HEADLESS_LAYERS=${LAYERS#HEADLESS}

    # first, run any "wait layers", i.e. layers that need to be
    # run sequentially, e.g. because they mess with some global
    # state such as current context
    if [ -n "$WAIT_LAYERS" ]; then
        if [ -n "$NON_HEADLESS_LAYERS" ] && [ -n "$MOCHA_TARGETS" ]; then
            PORT_OFFSET_BASE=1

            for MOCHA_RUN_TARGET in $MOCHA_TARGETS; do
                if [ "$MOCHA_RUN_TARGET" == "webpack" ] && [ "$TRAVIS_OS_NAME" == "osx" ]; then
                    echo "skip travis osx Webpack test since travis doesn't support docker on osx"
                else
                    export MOCHA_RUN_TARGET
                    export PORT_OFFSET_BASE

                    ls node_modules/.bin/mocha

                    echo "running these non-headless layers with $MOCHA_RUN_TARGET and wait: $WAIT_LAYERS"
                    MONOREPO_MODE=true npm run test $WAIT_LAYERS
                fi
            done
            echo "done with wait layers"
        fi
    fi

    # is "HEADLESS" on the LAYERS list?
    if [ "$NON_HEADLESS_LAYERS" != "$LAYERS" ]; then
        #
        # for now, the headless test suite (which is also a mocha suite)
        # is a bit of a special case
        #
        echo "running HEADLESS layer"

        # When testing against build headless, we set TEST_SPACE manually
        # since we can't get the env var TEST_SPACE from the previous
        # runMochaLayers.sh => runTest.sh process. Namespace Current tests will
        # fail if we don't have TEST_SPACE.

        export TEST_SPACE="${TEST_SPACE_PREFIX-ns}${KEY}_1"
        (cd /tmp/kui && MOCHA_RUN_TARGET=headless npm run test) & # see ./install.sh for the /tmp/kui target
        children+=("$!")
        childrenNames+=("headless layer")
        childrenStartTimes+=("$(date +%s)")
    fi

    # here is where we start scheduling the mocha test layers that can be run concurrently
    if [ -n "$NON_HEADLESS_LAYERS" ] && [ -n "$MOCHA_TARGETS" ]; then
        PORT_OFFSET_BASE=1

        for MOCHA_RUN_TARGET in $MOCHA_TARGETS; do
          echo "mocha target: $MOCHA_RUN_TARGET"
          if [ "$MOCHA_RUN_TARGET" == "webpack" ] && [ "$TRAVIS_OS_NAME" == "osx" ]; then
            echo "skip travis osx Webpack test since travis doesn't support docker on osx"
          else
            echo "PORT_OFFSET_BASE=$PORT_OFFSET_BASE"
            export MOCHA_RUN_TARGET
            export PORT_OFFSET_BASE

            echo "running these non-headless layers with $MOCHA_RUN_TARGET: $NON_HEADLESS_LAYERS"

            if [ "$KUI_USE_CLIENT" == test ]; then
              (cd clients/"$MOCHA_RUN_TARGET"; npm run test $NON_HEADLESS_LAYERS) &
            else
              (MONOREPO_MODE=true npm run test $NON_HEADLESS_LAYERS) &
            fi

            children+=("$!")
            childrenNames+=("mocha layers")
            childrenStartTimes+=("$(date +%s)")
          fi

          NLAYERS=$(count "$LAYERS")
          PORT_OFFSET_BASE=$((PORT_OFFSET_BASE+NLAYERS))
        done
    fi
fi

# finally, wait for the parallel subtasks to finish
wait_and_get_exit_codes "${children[@]}"
if [ $EXIT_CODE != 0 ]; then exit $EXIT_CODE; fi

if [ "$TRAVIS_OS_NAME" == "linux" ] && [ -n "$NEEDS_TOP" ]; then
    kill $TOP_PID
fi

# code coverage report generation
if [ -n "$NYC" ]; then
    set +e # we don't want report generation failures to induce a test failure
    "$SCRIPTDIR"/../../../tools/codecov/report.sh
fi

exit $EXIT_CODE