#!/bin/sh

usage () {

    cat <

    $0  [-c] [-a <ARTIFACTS DIR>]

    Run gating tests on test runners provisioned by linchpin and deployed with ansible,

    syncing artifacts to localhost.

   -c  Run configuration check.

   -a  Local host directory for fetching artifacts from test runner.

HELP_USAGE

}

CHECK_ONLY="no"

ARTIFACTS_VAR=""

while getopts "ca:" opt; do

    case $opt in

        c)

            # Run only configuration check

            CHECK_ONLY="yes"

            ;;

        a)

            # Set up directory for fetching artifacts

            ARTIFACTS_VAR="local_artifacts=${OPTARG}"

            ;;

        *)

            echo "Usage:"

            usage

            exit 1

            ;;

    esac

done

DEFAULT_CRED_FILENAME="clouds.yml"

CRED_DIR="${HOME}/.config/linchpin"

CRED_FILE_PATH=${CRED_DIR}/${DEFAULT_CRED_FILENAME}

TOPOLOGY_FILE_PATH="linchpin/topologies/gating-test.yml"

ANSIBLE_CFG_PATH="remote_config/ansible.cfg"

CHECK_RESULT=0

############################## Check the configuration

echo

echo "========= Dependencies are installed"

echo "linchpin and ansible are required to be installed."

echo "For linchpin installation instructions see:"

echo "https://linchpin.readthedocs.io/en/latest/installation.html"

echo

if ! type ansible &> /dev/null; then

    echo "=> FAILED: ansible package is not installed"

    CHECK_RESULT=1

else

    echo "=> OK: ansible is installed"

fi

if ! type linchpin &> /dev/null; then

    echo "=> FAILED: linchpin is not installed"

    CHECK_RESULT=1

else

    echo "=> OK: linchpin is installed"

fi

echo

echo "========= Linchpin cloud credentials configuration"

echo "The credentials file for linchpin provisioner should be in ${CRED_DIR}"

echo "The name of the file and the profile to be used is defined by"

echo "   resource_groups.credentials variables in the topology file"

echo "   (${TOPOLOGY_FILE_PATH})"

echo

config_changed=0

if [[ -f ${TOPOLOGY_FILE_PATH} ]]; then

    grep -q 'filename:.*'${DEFAULT_CRED_FILENAME} ${TOPOLOGY_FILE_PATH}

    config_changed=$?

fi

if [[ ${config_changed} -eq 0 ]]; then

    if [[ -f ${CRED_FILE_PATH} ]]; then

        echo "=> OK: ${CRED_FILE_PATH} exists"

    else

        echo "=> FAILED: ${CRED_FILE_PATH} does not exist"

        CHECK_RESULT=1

    fi

else

    echo "=> NOT CHECKING: seems like this has been configured in a different way"

fi

echo

echo "========== Deployment ssh key configuration"

echo "The ssh key used for deployment with ansible has to be defined by"

echo "private_key_file variable in ${ANSIBLE_CFG_PATH}"

echo "and match the key used for provisioning of the machines with linchpin"

echo "which is defined by resource_groups.resource_definitions.keypair variable"

echo "in topology file (${TOPOLOGY_FILE_PATH})."

echo

deployment_key_defined_line=$(grep 'private_key_file.*=.*[^\S]' ${ANSIBLE_CFG_PATH})

if [[ -n "${deployment_key_defined_line}" ]]; then

    echo "=> OK: ${ANSIBLE_CFG_PATH}: ${deployment_key_defined_line}"

else

    echo "=> FAILED: deployment ssh key not defined in ${ANSIBLE_CFG_PATH}"

    CHECK_RESULT=1

fi

linchpin_keypair=$(grep "keypair:" ${TOPOLOGY_FILE_PATH} | uniq)

echo "=> INFO: should be the same key as ${TOPOLOGY_FILE_PATH}: ${linchpin_keypair}"

if [[ ${CHECK_RESULT} -ne 0 ]]; then

echo

echo "=> Configuration check FAILED, see FAILED messages above."

echo

fi

if [[ ${CHECK_ONLY} == "yes" || ${CHECK_RESULT} -ne 0 ]]; then

    exit ${CHECK_RESULT}

fi

############################## Run the tests

set -x

### Clean the linchpin generated inventory

rm -rf linchpin/inventories/*.inventory

### Provision test runner

linchpin -v --workspace linchpin -p linchpin/PinFile -c linchpin/linchpin.conf up

### Pass inventory generated by linchpin to ansible

cp linchpin/inventories/*.inventory remote_config/inventory/linchpin.inventory

### Use ansible configuration for running tests on remote host

export ANSIBLE_CONFIG=${ANSIBLE_CFG_PATH}

### Configure remote user for playbooks

# By default root is used but it can be fedora or cloud-user for cloud images

for USER in root fedora cloud-user; do

  ansible-playbook --extra-vars="remote_user=$USER" check_and_set_remote_user.yml

done

### Prepare test runner

ansible-playbook prepare-test-runner.yml

### Run test on test runner (supply artifacts variable which is testing system's job)

ansible-playbook --extra-vars="artifacts=./artifacts" tests.yml

### Gather artifacts (into /tmp/artifacts by default)

ansible-playbook --extra-vars="${ARTIFACTS_VAR}" clean-test-runner.yml

### Destroy the test runner

linchpin -v --workspace linchpin -p linchpin/PinFile -c linchpin/linchpin.conf destroy