#!/bin/bash

# This script ensures RPM's build properly and can
# be installed as expected

set -o nounset
set -o pipefail


# Values that can be overridden
RPM_TEST_PRODUCT=${RPM_TEST_PRODUCT:-"origin"}           # origin or atomic-enterprise
RPM_TEST_OUTPUT_DIR=${RPM_TEST_OUTPUT_DIR:-"/tmp/tito/"} # Output for all build artifacts
RPM_TEST_SKIP_LINT=${RPM_TEST_SKIP_LINT:-""}             # Set to anything to disable rpmlint test

# Values that should be left alone
REQUIRED_PACKAGES="rpmlint rpm-build tito"               # Required packages to build and test
RPM_DIR=$RPM_TEST_OUTPUT_DIR/`arch`                      # Convenience. Path to the RPM output directory
SERVICE_PREFIX="origin"                                  # Used as both RPM name and service script prefix
if [ $RPM_TEST_PRODUCT == "atomic-enterprise" ]; then
    SERVICE_PREFIX="atomic-openshift"
fi

# ===
# Testing helper functions

# Show info line
function info()
{
    printf "\033[1;37mINFO\033[0m: $1\n"
}

# Show a test pass line
function pass()
{
    printf "\033[0;32mPASS\033[0m: $1\n"
}

# Show an error line
function error()
{
    printf "\033[0;33mError\033[0m: $1\n"
}

# Show a test fail line
function fail()
{
    printf "\033[0;31mFAIL\033[0m: $1\n"
}

# Show a failure and exit if the expected return code isn't returned
function fail_out()
{
    if [ $1 -ne 0 ]; then
        fail "$2"
        exit 1
    fi
}

# Show an error and exit if the expected return code isn't returned
function error_out()
{
    if [ $1 -ne 0 ]; then
        error "$2"
        exit 1
    fi
}
# ====


# Root check
if [ `id -u` -eq 0 ]; then
    error_out 1 'Do not run tests as root.'
fi


# Verifies the environment can produce rpms
function check_environment()
{
    info "Checking environment for suitability"
    # Check for required packages
    for required_rpm in $REQUIRED_PACKAGES; do
        rpm -q $required_rpm > /dev/null
        error_out $? "$required_rpm is missing. $REQUIRED_PACKAGES must all be installed."
    done
    pass "Environment looks good. Tests can run."
}

# Cleans out the generated RPM directory
function clean_output_dir_of_rpms()
{
    rm -rf $RPM_TEST_OUTPUT_DIR/`arch`
    info "Cleaned output dir of rpms."
}

# Builds the RPMs for a product
function build_rpm()
{
    mkdir -p $RPM_TEST_OUTPUT_DIR
    clean_output_dir_of_rpms
    info "Starting tito build."
    dist=""
    if [ $RPM_TEST_PRODUCT == "atomic-enterprise" ]; then
        dist="--dist=.el7aos"
    fi
    tito build --rpm --test --offline $dist -o "$RPM_TEST_OUTPUT_DIR"
    if [ $? -ne 0 ]; then
        fail "tito failed to build rpms"
        exit 1
    fi

    pass "Build RPMS"
}

# Uses rpmlint to check for errors
function lint_rpms()
{
    rpmlint -V
    rpmlint -i $RPM_DIR/*rpm
    if [ $? -eq 64 ]; then
        fail "rpmlint reported errors. (Warnings ignored...)"
        exit 1
    fi

    pass "Lint RPMS"
}

# Ensures the proper services are in the generated RPMs
function verify_expected_services()
{
    rpm -qpl $RPM_DIR/$SERVICE_PREFIX-master*rpm | grep $SERVICE_PREFIX-master.service > /dev/null
    fail_out $? "$SERVICE_PREFIX-master.service not in the $SERVICE_PREFIX-master rpm"

    rpm -qpl $RPM_DIR/$SERVICE_PREFIX-node*rpm | grep $SERVICE_PREFIX-node.service > /dev/null
    fail_out $? "$SERVICE_PREFIX-node.service not in the $SERVICE_PREFIX-node rpm"

    pass "Verify Expected Services"
}

# Verifies that installation can happen
function test_install()
{
    info "Verifying install cases..."
    info "Testing install of all rpms"
    rpm -ivh --test $RPM_DIR/*rpm
    fail_out $? "Unable to install all packages together"

    rpm_version=`rpm -qp --qf "%{VERSION}" $RPM_DIR/$SERVICE_PREFIX-master*.rpm`

    info "Testing install of main and master"
    rpm -ivh --test $RPM_DIR/$SERVICE_PREFIX-$rpm_version*.rpm $RPM_DIR/$SERVICE_PREFIX-master*.rpm
    fail_out $? "Unable to install main and master"

    info "Testing install of main, node and tuned"
    rpm -ivh --test $RPM_DIR/$SERVICE_PREFIX-$rpm_version*.rpm $RPM_DIR/$SERVICE_PREFIX-node*.rpm $RPM_DIR/tuned-profiles-$SERVICE_PREFIX-node*rpm
    fail_out $? "Unable to install main, node and tuned"
    pass "Test Install"
}



# Run the build tests
check_environment
build_rpm
if [ -e $RPM_TEST_SKIP_LINT ]; then
    lint_rpms
fi
test_install
verify_expected_services
exit 0