#!/bin/bash
##############################################################################
# Copyright (c) 2016 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
##############################################################################


##############################################
#  Usage: ./deploy.sh paras_conf outout_dir
##############################################
SCRIPT_DIR=$(cd "$(dirname "$0")"; pwd)
tool_dir=${0%/*}

function print_usage()
{
    echo "Usage: ./deploy.sh paras_conf output_dir"
}

## sanity check
if [ ! -f $tool_dir/mac_generator.sh ] || \
    [ ! -f $tool_dir/libvirt_template.xml ] || \
    [ ! -f $tool_dir/meta-data.template ] || \
    [ ! -f $tool_dir/user-data.template ] || \
    [ ! -f $tool_dir/p-master-user-data.template ] || \
    [ ! -f $tool_dir/p-agent-user-data.template ]; then
      echo "Lack some necessary files for this tool!"
      echo "deploy.sh"
      echo "mac_generator.sh"
      echo "1 xml, and 4 template"
      exit 1
fi

## Check input
if [ $# != 2 ]; then
    print_usage
    exit 1
fi
if [ ! -f $1 ]; then
    echo "Cannot find file: "$1
    exit 1
fi
if [ -d $2 ]; then
    echo "Ouput dir $2 exist!"
    exit 1
fi

## Assign parameters
host_names=""
puppet_enable="true"
master_host=""
vm_mem=
vm_cpu_cores=
image_url=
image_name=
ipaddr_start=
trusted_ssh_pub_keys_file=
while read line
do
    line=(${line//=/ })
    case ${line[0]} in
        "host_names" )
            host_names=${line[1]}
        ;;
        "puppet_enable" )
            puppet_enable=${line[1]}
        ;;
        "master_host" )
            master_host=${line[1]}
        ;;
        "vm_mem" )
            vm_mem=${line[1]}
        ;;
        "vm_cpu_cores" )
            vm_cpu_cores=${line[1]}
        ;;
        "image_url" )
            image_url=${line[1]}
        ;;
        "image_name" )
            image_name=${line[1]}
        ;;
        "ipaddr_start" )
            ipaddr_start=${line[1]}
        ;;
        "trusted_ssh_pub_keys_file" )
            trusted_ssh_pub_keys_file=${line[1]}
        ;;
    esac
done < $1

echo "puppet_enable="$puppet_enable

# Check parameters in conf file
if [ $puppet_enable == "true" ] ; then
    if [ ${#master_host} == 0 ];then
        echo "Should specify master_host!"
        exit 1
    else
        result=$(echo ${host_names} | grep  "${master_host}")
        if [ ${result} == "" ]; then
            echo "Specified master_host is invalid!"
            exit 1
        fi
    fi
fi

# Define and Prepare needed data
mac_arr=
hostname_arr=(${host_names//,/ })
virt_num=${#hostname_arr[@]}
ip_arr=()
replaced_hosts=""
replaced_ssh_keys=""
output_dir=$2
work_dir=
host_vm_dir=
cache_dir=
function init(){
    # Generate mac address
    local mac_generator=${tool_dir}/mac_generator.sh
    chmod +x $mac_generator
    mac_str=$($mac_generator $virt_num)
    mac_arr=($mac_str)

    # Generate hosts info
    local ip=""
    i=0
    for host in "${hostname_arr[@]}"; do
        ip=${ipaddr_prefix}$((i+$ipaddr_idx))
        ip_arr+=($ip)
        # Note the format, especially the space
        replaced_hosts=${replaced_hosts}"      "${ip}" "${host}"\n"
        let i=i+1
    done

    # Generate ssh public keys
    echo "## trusted_ssh_pub_keys_file --> "${trusted_ssh_pub_keys_file}
    if [ ${trusted_ssh_pub_keys_file} != "" ] && [ -f ${trusted_ssh_pub_keys_file} ]; then
        while read line
        do
            # Note the format, especially space
            replaced_ssh_keys=${replaced_ssh_keys}"    - "${line}"\n"
        done < ${trusted_ssh_pub_keys_file}
        # delete last "\n" in replaced_ssh_keysi
        replaced_ssh_keys=${replaced_ssh_keys%\\n}
    fi

    # Prepare needed folder and files
    if [ ${output_dir:0-1} == "/" ]; then
        output_dir=${output_dir%/*}
    fi
    output_file=$output_dir/hosts.info
    mkdir -p $output_dir
    touch $output_file
    echo "## Output host_info file --> "$output_file

    work_dir=$output_dir/work
    host_vm_dir=$work_dir/vm
    cache_dir=$work_dir/cache
    mkdir -p $work_dir
    mkdir -p $host_vm_dir
    mkdir -p $cache_dir

    # Cache img file
    echo "## Cache img file"
    curl --connect-timeout 10 -o ${cache_dir}/$image_name $image_url
}

# Bring up instances/vms
sub_ip_arr=(${ipaddr_start//./ })
ipaddr_prefix=${sub_ip_arr[0]}"."${sub_ip_arr[1]}"."${sub_ip_arr[2]}"."
ipaddr_idx=${sub_ip_arr[3]}
function bring_up() {
    i=0
    while (($i < $virt_num))
    do
        echo "Bring up a vm, hostname: ${hostname_arr[$i]}, ip: ${ip_arr[$i]}, mac: ${mac_arr[$i]}"
        vm_dir=$host_vm_dir/${hostname_arr[$i]}
        mkdir -p $vm_dir
        cp ${cache_dir}/$image_name $vm_dir

        sed -e "s/REPLACE_IPADDR/${ip_arr[$i]}/g" \
            -e "s/REPLACE_GATEWAY/${ipaddr_prefix}1/g" \
            -e "s/REPLACE_HOSTNAME/${hostname_arr[$i]}/g" \
            ${tool_dir}/meta-data.template > ${cache_dir}/meta-data

        if [ ${puppet_enable} == "true" ]; then
            # Use puppet user data
            echo "hostname: "${hostname_arr[$i]}
            if [ ${hostname_arr[$i]} == ${master_host} ]; then
                cp ${tool_dir}/p-master-user-data.template ${cache_dir}/user-data.template
            else
                cp ${tool_dir}/p-agent-user-data.template ${cache_dir}/user-data.templatate
            fi
            sed -e "s#REPLACED_TRUSTED_PUB_SSH_KEYS#${replaced_ssh_keys}#g" \
                -e "s#REPLACED_HOSTS_INFO#${replaced_hosts}#g" \
                -e "s/REPLACED_PUPPET_MASTER_SERVER/${master_host}/g" \
                ${cache_dir}/user-data.template > ${cache_dir}/user-data
        else
           # Use common user data
           echo "## Use common user-data.template"
           cp ${tool_dir}/user-data.template ${cache_dir}/user-data
        fi

        genisoimage -output seed.iso -volid cidata -joliet -rock ${cache_dir}/user-data ${cache_dir}/meta-data
        mv seed.iso ${vm_dir}/
        # Create vm xml
        sed -e "s/REPLACE_MEM/$vm_mem/g" \
            -e "s/REPLACE_CPU/$vm_cpu_cores/g" \
            -e "s/REPLACE_NAME/${hostname_arr[$i]}/g" \
            -e "s#REPLACE_IMAGE#$vm_dir/disk.img#g" \
            -e "s#REPLACE_SEED_IMAGE#$vm_dir/seed.iso#g" \
            -e "s/REPLACE_MAC_ADDR/${mac_arr[$i]}/g" \
            ${tool_dir}/libvirt_template.xml > ${vm_dir}/libvirt.xml

        echo "${ip_arr[$i]} ${hostname_arr[$i]}" >> $output_file

        echo "Will define xml from:"${vm_dir}"/libvirt.xml"
        echo "start: "${hostname_arr[$i]}
        sudo virsh define ${vm_dir}/libvirt.xml
        sudo virsh start ${hostname_arr[$i]}
        let i=i+1
        rm -rf ${cache_dir}/meta-data ${cache_dir}/user-data
    done
}

function clean(){
    rm -rf ${cache_dir}
}

init
bring_up
clean