#!/bin/bash -xe

# source CI credentials
. /home/jenkins/ci_openrc
# source functions
. $FUNCTION_PATH/functions-common.sh
. $FUNCTION_PATH/functions-dib.sh

CLUSTER_HASH=${CLUSTER_HASH:-$RANDOM}
cluster_name="$HOST-$ZUUL_CHANGE-$CLUSTER_HASH"

SAHARA_PATH="/tmp/sahara"
SAHARA_TESTS_PATH="/tmp/sahara-tests"
sahara_conf_file="$SAHARA_PATH/etc/sahara/sahara.conf"
sahara_templates_path=$SAHARA_TESTS_PATH/sahara_tests/scenario/defaults
tests_etc=$sahara_templates_path

plugin="$1"
os="$2"
image_name=${HOST}_${plugin}_${os}_${ZUUL_CHANGE}
eval ${plugin//./_}_image=$image_name
mode="aio"
sahara_plugin=$(echo $plugin | awk -F '_' '{ print $1 } ')

# Clone Sahara
get_dependency "$SAHARA_PATH" "openstack/sahara" "$ZUUL_BRANCH"

# Clone Sahara Scenario tests
get_dependency "$SAHARA_TESTS_PATH" "openstack/sahara-tests" "master"

# make verbose the scripts execution of disk-image-create
export DIB_DEBUG_TRACE=1

case "${os}" in
    c6.6)
        username="cloud-user"
        os_type="centos"
        ;;
    c7)
        username="centos"
        os_type="centos7"
        ;;
    u12 | u14)
        username="ubuntu"
        os_type="ubuntu"
        ;;
    *)
        echo "Unrecognized OS: ${os}" >&2
        exit 1
        ;;
esac

case $plugin in
    vanilla_2.7.1)
      env ${os_type}_vanilla_hadoop_2_7_1_image_name=${vanilla_2_7_1_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p vanilla -i $os_type -v 2.7.1
      check_error_code $? ${vanilla_2_7_1_image}.qcow2
      upload_image "${plugin}" "${username}" ${vanilla_2_7_1_image}
      mode=distribute
      scenario_conf_file="$sahara_templates_path/vanilla-2.7.1.yaml.mako"
      template_image_prefix="vanilla_two_seven_one"
    ;;

    spark_1.3.1)
       env ubuntu_spark_image_name=${spark_1_3_1_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p spark -s 1.3.1
       check_error_code $? ${spark_1_3_1_image}.qcow2
       upload_image "${plugin}" "${username}" ${spark_1_3_1_image}
       scenario_conf_file="$sahara_templates_path/spark-1.3.1.yaml.mako"
       template_image_prefix="spark_1_3"
    ;;

    spark_1.6.0)
       env ubuntu_spark_image_name=${spark_1_6_0_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p spark -s 1.6.0
       check_error_code $? ${spark_1_6_0_image}.qcow2
       upload_image "${plugin}" "${username}" ${spark_1_6_0_image}
       scenario_conf_file="$sahara_templates_path/spark-1.6.0.yaml.mako"
       template_image_prefix="spark_1_6"
    ;;

    cdh_5.4.0)
       env cloudera_5_4_${os_type}_image_name=${cdh_5_4_0_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p cloudera -i $os_type -v 5.4
       check_error_code $? ${cdh_5_4_0_image}.qcow2
       upload_image "${plugin}" "${username}" ${cdh_5_4_0_image}
       scenario_conf_file="$sahara_templates_path/cdh-5.4.0.yaml.mako"
       template_image_prefix="cdh_5_4_0"
    ;;

    cdh_5.5.0)
       env cloudera_5_5_${os_type}_image_name=${cdh_5_5_0_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p cloudera -i $os_type -v 5.5
       check_error_code $? ${cdh_5_5_0_image}.qcow2
       upload_image "${plugin}" "${username}" ${cdh_5_5_0_image}
       scenario_conf_file="$sahara_templates_path/cdh-5.5.0.yaml.mako"
       template_image_prefix="cdh_5_5_0"
    ;;

    cdh_5.7.0)
       env cloudera_5_7_${os_type}_image_name=${cdh_5_7_0_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p cloudera -i $os_type -v 5.7
       check_error_code $? ${cdh_5_7_0_image}.qcow2
       upload_image "${plugin}" "${username}" ${cdh_5_7_0_image}
       scenario_conf_file="$sahara_templates_path/cdh-5.7.0.yaml.mako"
       template_image_prefix="cdh_5_7_0"
    ;;

    ambari_2.2)
       env ambari_${os_type}_image_name=${ambari_2_2_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p ambari -i $os_type
       check_error_code $? ${ambari_2_2_image}.qcow2
       upload_image "${plugin}" "${username}" ${ambari_2_2_image}
       # we use Ambari 2.1 management console for creating HDP 2.3 stack
       scenario_conf_file="$sahara_templates_path/ambari-2.3.yaml.mako"
       template_image_prefix="ambari_2_2"
    ;;

    mapr_5.1.0.mrv2)
       env mapr_ubuntu_image_name=${mapr_5_1_0_mrv2_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p mapr -i ubuntu
       check_error_code $? ${mapr_5_1_0_mrv2_image}.qcow2
       upload_image "${plugin}" "${username}" ${mapr_5_1_0_mrv2_image}
       mode=distribute
       scenario_conf_file="$sahara_templates_path/mapr-5.1.0.mrv2.yaml.mako"
       template_image_prefix="mapr_510mrv2"
    ;;
    storm_1.0.1)
       env ubuntu_storm_image_name=${storm_1_0_1_image} SIM_REPO_PATH=$WORKSPACE tox -e venv -- sahara-image-create -p storm -t 1.0.1
       check_error_code $? ${storm_1_0_1_image}.qcow2
       upload_image "${plugin}" "${username}" ${storm_1_0_1_image}
       scenario_conf_file="$sahara_templates_path/storm-1.0.1.yaml.mako"
       template_image_prefix="storm_1_0_1"
    ;;
esac

cd $SAHARA_PATH
sudo pip install . --no-cache-dir
enable_pypi
write_sahara_main_conf "$sahara_conf_file" "$sahara_plugin"
write_tests_conf "$cluster_name" "$template_image_prefix" "$image_name" "$scenario_conf_file"
start_sahara "$sahara_conf_file" "$mode" && run_tests "$scenario_conf_file"
print_python_env
cleanup_image "$plugin" "$os"