#!/bin/bash
##############################################################################
# Copyright (c) 2015 Ericsson AB, Huawei Technologies Co.,Ltd and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

# Perepare the environment to run yardstick ci

: ${DEPLOY_TYPE:='bm'} # Can be any of 'bm' (Bare Metal) or 'virt' (Virtual)
: ${INSTALLER_TYPE:='unknown'}
: ${NODE_NAME:='unknown'}
: ${EXTERNAL_NETWORK:='admin_floating_net'}
: ${USER_NAME:='ubuntu'}
: ${SSH_KEY:='/root/.ssh/id_rsa'}
: ${DEPLOY_SCENARIO:='unknown'}

# Extract network name from EXTERNAL_NETWORK
#  e.g. EXTERNAL_NETWORK='ext-net;flat;192.168.0.2;192.168.0.253;192.168.0.1;192.168.0.0/24'
EXTERNAL_NETWORK=$(echo ${EXTERNAL_NETWORK} | cut -f1 -d \;)
export EXTERNAL_NETWORK

# Create openstack credentials
echo "INFO: Creating openstack credentials .."
mkdir -p /etc/yardstick
OPENRC=/etc/yardstick/openstack.creds
INSTALLERS=(apex compass fuel joid)

RC_VAR_EXIST=false
if [[ "${OS_AUTH_URL}" && "${OS_USERNAME}" && "${OS_PASSWORD}" && "${EXTERNAL_NETWORK}" ]];then
    RC_VAR_EXIST=true
fi

if [[ "${RC_VAR_EXIST}" = false && -f ${OPENRC} ]]; then
    . ${OPENRC}
fi

export EXTERNAL_NETWORK INSTALLER_TYPE DEPLOY_TYPE NODE_NAME

# Prepare a admin-rc file for StorPerf integration
"${YARDSTICK_REPO_DIR}"/tests/ci/prepare_storperf_admin-rc.sh

# copy Storperf related files to the deployment location
if [ "$INSTALLER_TYPE" == "compass" ]; then
    . "${YARDSTICK_REPO_DIR}"/tests/ci/scp_storperf_files.sh
fi

# Fetching id_rsa file from jump_server..."
verify_connectivity() {
    local ip=$1
    echo "Verifying connectivity to $ip..."
    for i in $(seq 0 10); do
        if ping -c 1 -W 1 "${ip}" > /dev/null; then
            echo "$ip is reachable!"
            return 0
        fi
        sleep 1
    done
    error "Can not talk to $ip."
}

ssh_options="-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no"
if [ "$INSTALLER_TYPE" == "apex" ]; then

    # check the connection
    verify_connectivity "${INSTALLER_IP}"

    pod_yaml="$YARDSTICK_REPO_DIR/etc/yardstick/nodes/apex_baremetal/pod.yaml"

    # update "ip" according to the CI env
    ssh -l root "${INSTALLER_IP}" -i ${SSH_KEY} ${ssh_options} \
         "source /home/stack/stackrc && openstack server list -f yaml" > node_info

    controller_ips=($(awk '/control/{getline; {print $2}}' < node_info | grep -o '[0-9]\{1,3\}\.[0-9]\{1,3\}\.[0-9]\{1,3\}\.[0-9]\{1,3\}'))
    compute_ips=($(awk '/compute/{getline; {print $2}}' < node_info | grep -o '[0-9]\{1,3\}\.[0-9]\{1,3\}\.[0-9]\{1,3\}\.[0-9]\{1,3\}'))
    odl_ip=""
    # Get ODL's cluster default module-shard Leader IP in HA scenario
    if [[ ${DEPLOY_SCENARIO} == os-odl-*-ha ]]; then
        for ip in "${controller_ips[@]}";
        do
           if [[ "$odl_ip" ]]; then
               break
           fi
           for ((i=0; i<${#controller_ips[@]}; i++));
           do
               ODL_STATE=$(curl -s -u admin:admin -H "Accept: application/json" -H "Content-Type: application/json" \
                           "http://"${ip}":8081/jolokia/read/org.opendaylight.controller:Category=Shards,name=member-"${i}"-shard-default-operational,type=DistributedOperationalDatastore" \
                           | grep -o \"RaftState\"\:\"Leader\" | tr ":" "\n" | sed -n '2p' | sed 's/\"//g');

               if [[ ${ODL_STATE} == "Leader" ]]; then
                    odl_ip=${ip}
                    break
               fi
           done;
        done

        if [[ -z "$odl_ip" ]]; then
          echo "ERROR: Opendaylight Leader IP is emtpy"
          exit 1
        fi

    elif [[ ${DEPLOY_SCENARIO} == *"odl"* ]]; then
        odl_ip=${controller_ips[0]}
    fi

    if [[ ${controller_ips[0]} ]]; then
        sed -i "s|ip1|${controller_ips[0]}|" "${pod_yaml}"
    fi
    if [[ ${controller_ips[1]} ]]; then
        sed -i "s|ip2|${controller_ips[1]}|" "${pod_yaml}"
    fi
    if [[ ${controller_ips[2]} ]]; then
        sed -i "s|ip3|${controller_ips[2]}|" "${pod_yaml}"
    fi
    if [[ ${compute_ips[0]} ]]; then
        sed -i "s|ip4|${compute_ips[0]}|" "${pod_yaml}"
    fi
    if [[ ${compute_ips[1]} ]]; then
        sed -i "s|ip5|${compute_ips[1]}|" "${pod_yaml}"
    fi
    if [[ ${odl_ip} ]]; then
        sed -i "s|ip6|${odl_ip}|" "${pod_yaml}"
    fi


    # update 'key_filename' according to the CI env
    sed -i "s|node_keyfile|${SSH_KEY}|" "${pod_yaml}"

fi
if [ "$INSTALLER_TYPE" == "fuel" ]; then

    # check the connection
    verify_connectivity "${INSTALLER_IP}"

    pod_yaml="$YARDSTICK_REPO_DIR/etc/yardstick/nodes/fuel_baremetal/pod.yaml"

    # update "ip" according to the CI env
    ssh -l ubuntu "${INSTALLER_IP}" -i ${SSH_KEY} ${ssh_options} \
         "sudo salt -C 'ctl* or cmp* or odl01* or gtw*' grains.get fqdn_ip4  --out yaml">node_info

    controller_ips=($(awk '/ctl/{getline; print $2}' < node_info))
    compute_ips=($(awk '/cmp/{getline; print $2}' < node_info))
    odl_ip=($(awk '/odl01/{getline; print $2}' < node_info))
    gateway_ip=($(awk '/gtw/{getline; print $2}' < node_info))

    if [[ ${controller_ips[0]} ]]; then
        sed -i "s|ip1|${controller_ips[0]}|" "${pod_yaml}"
    fi
    if [[ ${controller_ips[1]} ]]; then
        sed -i "s|ip2|${controller_ips[1]}|" "${pod_yaml}"
    fi
    if [[ ${controller_ips[2]} ]]; then
        sed -i "s|ip3|${controller_ips[2]}|" "${pod_yaml}"
    fi
    if [[ ${compute_ips[0]} ]]; then
        sed -i "s|ip4|${compute_ips[0]}|" "${pod_yaml}"
    fi
    if [[ ${compute_ips[1]} ]]; then
        sed -i "s|ip5|${compute_ips[1]}|" "${pod_yaml}"
    fi
    if [[ ${odl_ip[0]} ]]; then
        sed -i "s|ip6|${odl_ip[0]}|" "${pod_yaml}"
    fi
    if [[ ${gateway_ip[0]} ]]; then
        sed -i "s|ip7|${gateway_ip[0]}|" "${pod_yaml}"
    fi

    # update 'user' and 'key_filename' according to the CI env
    sed -i "s|node_username|${USER_NAME}|;s|node_keyfile|${SSH_KEY}|" "${pod_yaml}"

fi