#!/usr/bin/env bash
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.

export MAVEN_OPTS='-XX:+TieredCompilation -XX:TieredStopAtLevel=1 -XX:+CMSClassUnloadingEnabled -XX:+UseConcMarkSweepGC -XX:-UseGCOverheadLimit -Xmx3g'

base_dir=$(pwd)
build=0
cases=()

DIST_PACKAGE=${DIST_PACKAGE:-apache-skywalking-apm-bin.tar.gz}
ES_VERSION=${ES_VERSION:-6.3.2}

# Parse the arguments
# --build-dist: build the distribution package ignoring the existance of `dist` folder, useful when running e2e locally

while [[ $# -gt 0 ]]; do
  case "$1" in
    --build)
      build=1
      ;;
    --profiles=*)
      profiles=${1#*=}
      ;;
    --storage=*)
      storage=${1#*=}
      ;;
    *)
      cases+=($1)
  esac
  shift
done

[[ ! -f "${base_dir}/mvnw" ]] \
  && echo 'Please run run.sh in the root directory of SkyWalking' \
  && exit 1

[[ ${#cases[@]} -le 0 ]] \
  && echo 'Usage: sh test/e2e/run.sh [--build-dist] [--fast-fail] <case1 maven module>[<case2 maven module>...<caseN maven module>]' \
  && exit 1

[[ ${build} -eq 1 ]] \
  && echo 'Building distribution package...' \
  && ./mvnw --batch-mode --activate-profiles "${profiles}" -q -Dcheckstyle.skip -Drat.skip -T2 -Dmaven.compile.fork -DskipTests -am clean install

echo "Running cases: $(IFS=$' '; echo "${cases[*]}")"

for test_case in "${cases[@]}"
do
  echo "Running case: $test_case"

  [[ -d "$base_dir/$test_case" ]] && rm -rf "$base_dir/$test_case"

  # Some of the tests will modify files in the distribution folder, e.g. cluster test will modify the application.yml
  # so we give each test a separate distribution folder here
  mkdir -p "$test_case" && tar -zxf dist/${DIST_PACKAGE} -C "$test_case"

  SW_STORAGE=${storage}
  if [[ ${storage} = "elasticsearch" && ${ES_VERSION} =~ ^7 ]]; then
    SW_STORAGE=elasticsearch7
  fi

  ./mvnw --batch-mode -Dbuild.id="${BUILD_ID:-local}" \
         -De2e.container.version="${E2E_VERSION}" \
         -Delasticsearch.version="${ES_VERSION}" \
         -Dsw.home="${base_dir}/$test_case/${DIST_PACKAGE//.tar.gz/}" \
         -DSW_STORAGE=${SW_STORAGE} \
         `if [ ! -z "${storage}" ] ; then echo -P"${storage}"; fi` \
         -f test/e2e/pom.xml -pl "$test_case" -am verify

  status_code=$?

  if [[ ${status_code} -ne 0 ]]; then
    echo "Fast failing due to previous failure: ${test_case}, exit status code: ${status_code}" && exit ${status_code}
  fi
done

exit 0