#! /bin/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.
#--------------------------------------------------------------------

# Utility script for running the new integration tests, since the Maven
# commands are unwieldy.

export DRUID_DEV=$(cd $(dirname $0) && pwd)

function usage
{
	cat <<EOF
Usage: $0 cmd [category]
  build
      build Druid and the distribution
  dist
      build the Druid distribution (only)
  image
      build the test image
  up <category>
      start the cluster for category
  down <category>
      stop the cluster for category
  test <category>
      start the cluster, run the test for category, and stop the cluster
  tail <category>
      show the last 20 lines of each container log
  travis <category>
      run one IT in Travis (build dist, image, run test, tail logs)
  prune
      prune Docker volumes
EOF
}

function tail_logs
{
	category=$1
	cd integration-tests-ex/cases/target/$category/logs
	ls *.log | while read log;
	do
		echo "----- $category/$log -----"
		tail -20 $log
	done
}

CMD=$1
shift
MAVEN_IGNORE="-P skip-static-checks,skip-tests -Dmaven.javadoc.skip=true"

case $CMD in
 	"help" )
 		usage
 		;;
	"build" )
		mvn clean package -P dist $MAVEN_IGNORE -T1.0C
		;;
	"dist" )
		mvn package -P dist $MAVEN_IGNORE -pl :distribution
		;;
	"image" )
		cd $DRUID_DEV/integration-tests-ex/image
		mvn install -P test-image $MAVEN_IGNORE
		;;
	"up" )
		if [ -z "$1" ]; then
			usage
			exit 1
		fi
		cd $DRUID_DEV/integration-tests-ex/cases
		./cluster.sh up $1
		;;
	"down" )
		if [ -z "$1" ]; then
			usage
			exit 1
		fi
		cd $DRUID_DEV/integration-tests-ex/cases
		./cluster.sh down $1
		;;
	"test" )
		if [ -z "$1" ]; then
			usage
			exit 1
		fi
		cd $DRUID_DEV/integration-tests-ex/cases
		mvn verify -P skip-static-checks,docker-tests,IT-$1 \
            -Dmaven.javadoc.skip=true -DskipUTs=true \
            -pl :druid-it-cases
		;;
	"tail" )
		if [ -z "$1" ]; then
			usage
			exit 1
		fi
		tail_logs $1
		;;
    "travis" )
		if [ -z "$1" ]; then
			usage
			exit 1
		fi
    	$0 dist
    	$0 image
    	$0 test $1
    	$0 tail $1
    	;;
	"prune" )
		# Caution: this removes all volumes, which is generally what you
		# want when testing.
		docker system prune --volumes
		;;
	* )
		usage
		exit -1
		;;
esac