#!/usr/bin/env bash

readlink -f / &> /dev/null || readlink() { greadlink "$@" ; } # for MacOS
MYDIR=$(dirname -- "$(readlink -f -- "$0")")
MYPROG=$(basename -- "$0")
source "$MYDIR/lib_options.sh"
ARTIFACTS=$MYDIR/gerrit/artifacts

die() { echo -e "\nERROR: $@" ; kill $$ ; exit 1 ; } # error_message

progress() { # message cmd [args]...
    local message=$1 ; shift
    echo -n "$message"
    "$@" &
    local pid=$!
    while kill -0 $pid 2> /dev/null ; do
        echo -n "."
        sleep 2
    done
    echo
    wait "$pid"
}

usage() { # [error_message]
    cat <<-EOF
Usage:
    $MYPROG [--task-plugin-jar|-t <FILE_PATH>]
            [--names-factory-provider-plugin-jar|-f <FILE_PATH>]
            [--gerrit-war|-g <FILE_PATH>]

    This tool runs the plugin functional tests in a Docker environment built
    from the gerritcodereview/gerrit base Docker image.

    The task plugin JAR, names-factory-provider plugin JAR and optionally a
    Gerrit WAR are expected to be in the $ARTIFACTS dir;
    however, the --task-plugin-jar, --names-factory-provider-plugin-jar and
    --gerrit-war switches may be used as helpers to specify which files to
    copy there.

    Options:
    --help|-h
    --gerrit-war|-g                           path to Gerrit WAR file
    --task-plugin-jar|-t                      path to task plugin JAR file
    --names-factory-provider-plugin-jar|-f    path to names-factory-provider plugin JAR file.
                                              It can be generated using 'bazel build names-factory-provider'
    --preserve                                To preserve the docker setup for debugging

EOF

    [ -n "$1" ] && echo -e "\nERROR: $1" && exit 1
    exit 0
}

check_prerequisite() {
    docker --version > /dev/null || die "docker is not installed"
    docker-compose --version > /dev/null || die "docker-compose is not installed"
}

build_images() {
    docker-compose "${COMPOSE_ARGS[@]}" build --quiet
}

run_task_plugin_tests() {
    echo "Running test suite with default root_project and root_branch"
    docker-compose "${COMPOSE_ARGS[@]}" up --detach
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        '/task/test/docker/run_tests/start.sh'
    # TODO: Once the 'retest' functionality is fixed, the re-run with custom root prj and
    #       branch can be done without needing to bring down and start the containers again.
    docker-compose "${COMPOSE_ARGS[@]}" down -v --rmi local 2>/dev/null

    ROOT_CFG_PRJ=task-config
    ROOT_CFG_BRANCH=refs/heads/master
    echo "Re-running test suite with root_project=$ROOT_CFG_PRJ and root_branch=$ROOT_CFG_BRANCH"
    docker-compose "${COMPOSE_ARGS[@]}" up --detach
    configure_root_prj_and_branch
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        sh -c "/task/test/docker/run_tests/start.sh \
            --root-config-project $ROOT_CFG_PRJ \
            --root-config-branch $ROOT_CFG_BRANCH"
}

configure_root_prj_and_branch() {
    docker-compose "${COMPOSE_ARGS[@]}" exec -T gerrit-01 \
        sh -c "git config -f \$GERRIT_SITE/etc/task.config rootConfig.project $ROOT_CFG_PRJ"
    docker-compose "${COMPOSE_ARGS[@]}" exec -T gerrit-01 \
        sh -c "git config -f \$GERRIT_SITE/etc/task.config rootConfig.branch $ROOT_CFG_BRANCH"
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        sh -c "./task/test/docker/run_tests/wait-for-it.sh \$GERRIT_HOST:29418 -t -60" || \
            die "Failed to start gerrit"
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        sh -c "cat \$USER_HOME/.ssh/id_rsa.pub | ssh -p 29418 -i /server-ssh-key/ssh_host_rsa_key \
            \"Gerrit Code Review@\$GERRIT_HOST\" suexec --as admin@example.com -- gerrit set-account \
            admin --add-ssh-key -" || die "Failed to add ssh key to admin account"
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        sh -c "ssh -p 29418 \$GERRIT_HOST gerrit plugin reload task" || die "Failed to reload plugin"
}

retest() {
    docker-compose "${COMPOSE_ARGS[@]}" exec -T --user=admin run_tests \
        sh -c "/task/test/docker/run_tests/start.sh --retest"
    RESULT=$?
    cleanup
}

get_run_test_container() {
    docker-compose "${COMPOSE_ARGS[@]}" ps | grep run_tests | awk '{ print $1 }'
}

cleanup() {
    if [ "$PRESERVE" = "true" ] ; then
        echo "Preserving the following docker setup"
        docker-compose "${COMPOSE_ARGS[@]}" ps
        echo ""
        echo "To exec into runtests container, use following command:"
        echo "docker exec -it $(get_run_test_container) /bin/bash"
        echo ""
        echo "Run the following command to bring down the setup:"
        echo "docker-compose ${COMPOSE_ARGS[@]} down -v --rmi local"
        echo ""
        options_prefix "COMPOSE_ARGS" "--compose-arg"
        echo "Use command below to re run tests after making changes to test scripts"
        echo " $MYDIR/$MYPROG --retest --preserve ${COMPOSE_ARGS[@]}"
    else
        docker-compose "${COMPOSE_ARGS[@]}" down -v --rmi local 2>/dev/null
    fi
}

PRESERVE="false"
RETEST="false"
COMPOSE_ARGS=()
while (( "$#" )) ; do
    case "$1" in
        --help|-h)                                usage ;;
        --gerrit-war|-g)                          shift ; GERRIT_WAR=$1 ;;
        --task-plugin-jar|-t)                     shift ; TASK_PLUGIN_JAR=$1 ;;
        --names-factory-provider-plugin-jar|-f)   shift ; NAMES_FACTORY_PROVIDER_PLUGIN_JAR=$1 ;;
        --preserve)                               PRESERVE="true" ;;
        --retest)                                 RETEST="true" ;;
        --compose-arg)                            shift ; COMPOSE_ARGS+=("$1") ;;
        *)                                        usage "invalid argument $1" ;;
    esac
    shift
done

[ "$RETEST" = "true" ] && { retest ; exit "$RESULT" ; }

PROJECT_NAME="task_$$"
COMPOSE_YAML="$MYDIR/docker-compose.yaml"
COMPOSE_ARGS=(--project-name "$PROJECT_NAME" -f "$COMPOSE_YAML")
check_prerequisite
mkdir -p -- "$ARTIFACTS"
[ -n "$TASK_PLUGIN_JAR" ] && cp -f -- "$TASK_PLUGIN_JAR" "$ARTIFACTS/task.jar"
if [ ! -e "$ARTIFACTS/task.jar" ] ; then
    MISSING="Missing $ARTIFACTS/task.jar"
    [ -n "$TASK_PLUGIN_JAR" ] && die "$MISSING, check for copy failure?"
    usage "$MISSING, did you forget --task-plugin-jar?"
fi
[ -n "$NAMES_FACTORY_PROVIDER_PLUGIN_JAR" ] && cp -f -- "$NAMES_FACTORY_PROVIDER_PLUGIN_JAR" \
    "$ARTIFACTS/names-factory-provider.jar"
if [ ! -e "$ARTIFACTS/names-factory-provider.jar" ] ; then
    MISSING="Missing $ARTIFACTS/names-factory-provider.jar"
    [ -n "$NAMES_FACTORY_PROVIDER_PLUGIN_JAR" ] && die "$MISSING, check for copy failure?"
    usage "$MISSING, did you forget --names-factory-provider-plugin-jar?"
fi
[ -n "$GERRIT_WAR" ] && cp -f -- "$GERRIT_WAR" "$ARTIFACTS/gerrit.war"
( trap cleanup EXIT SIGTERM
    progress "Building docker images" build_images
    run_task_plugin_tests
)