install.sh

#!/bin/bash

# Copyright 2014 Intel Corporation, All Rights Reserved.

# Licensed under the Apache License, Version 2.0 (the"License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

#  http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.

#-------------------------------------------------------------------------------
#            Usage
#-------------------------------------------------------------------------------

function usage() {
    cat << EOF
Usage: install.sh

Auto deploy vsm:
    The tool can help you to deploy the vsm envirement automatically.
    Please run such as bash +x install.sh

Options:
  --help | -h
    Print usage information.
  --manifest [manifest directory] | -m [manifest directory]
    The directory to store the server.manifest and cluster.manifest.
  --version [master] | -v [master]
    The version of vsm dependences to download(Default=master).
  --keyfile [key file] | -k [key file]
    The key file required for ssh/scp connection at the environment where certificate based authentication is enabled.
  --user | -u
    The user will be used to connect remote nodes to deploy vsm
EOF
    exit 0
}

MANIFEST_PATH=""
DEPENDENCE_BRANCH="master"
USER=`whoami`
SSH='ssh'
SCP='scp'
while [ $# -gt 0 ]; do
  case "$1" in
    -h) usage ;;
    --help) usage ;;
    -m| --manifest) shift; MANIFEST_PATH=$1 ;;
    -v| --version) shift; DEPENDENCE_BRANCH=$1 ;;
	-k| --key) shift; keyfile=$1; export SSH='ssh -i $keyfile'; export SCP='scp -i $keyfile' ;;
    -u| --user) shift; USER=$1 ;;
    *) shift ;;
  esac
  shift
done


set -e
set -o xtrace

echo "Before auto deploy the vsm, please be sure that you have set the manifest 
such as manifest/192.168.100.100/server.manifest. And you have changed the file, too."
sleep 5

TOPDIR=$(cd $(dirname "$0") && pwd)
TEMP=`mktemp`; rm -rfv $TEMP >/dev/null; mkdir -p $TEMP;

HOSTNAME=`hostname`
#HOSTIP=`hostname -I|sed s/[[:space:]]//g`
HOSTIP=`hostname -I`

source $TOPDIR/installrc

is_controller=0
for ip in $HOSTIP; do
    if [ $ip == $CONTROLLER_ADDRESS ]; then
        is_controller=1
    fi
done

if [[ $HOSTNAME == $CONTROLLER_ADDRESS ]]; then
    is_controller=1
fi

if [ $is_controller -eq 0 ]; then
    echo "[Info]: You run the tool in a third server."
else
    echo "[Info]: You run the tool in the controller server."
fi


#-------------------------------------------------------------------------------
#            checking packages
#-------------------------------------------------------------------------------

echo "+++++++++++++++start checking packages+++++++++++++++"

if [ ! -d vsmrepo ]; then
    echo "You should have the vsmrepo folder, please check and try again"
    exit 1
fi

cd vsmrepo
is_python_vsmclient=`ls|grep python-vsmclient*.rpm|wc -l`
is_vsm=`ls|grep -v python-vsmclient|grep -v vsm-dashboard|grep -v vsm-deploy|grep vsm|wc -l`
is_vsm_dashboard=`ls|grep vsm-dashboard*.rpm|wc -l`
is_vsm_deploy=`ls|grep vsm-deploy*.rpm|wc -l`
if [ $is_python_vsmclient -gt 0 ] && [ $is_vsm -gt 0 ] && [ $is_vsm_dashboard -gt 0 ] && [ $is_vsm_deploy -gt 0 ]; then
    echo "The vsm pachages have been already prepared"
else
    echo "please check the vsm packages, then try again"
    exit 1
fi

cd $TOPDIR
echo "+++++++++++++++finish checking packages+++++++++++++++"


#-------------------------------------------------------------------------------
#            setting the iptables and selinux
#-------------------------------------------------------------------------------

echo "+++++++++++++++start setting the iptables and selinux+++++++++++++++"

function set_iptables_selinux() {
    $SSH $USER@$1 "service iptables stop"
    $SSH $USER@$1 "chkconfig iptables off"
    $SSH $USER@$1 "sed -i \"s/SELINUX=enforcing/SELINUX=disabled/g\" /etc/selinux/config"
#    $SSH $USER@$1 "setenforce 0"
}

if [ $is_controller -eq 0 ]; then
    set_iptables_selinux $CONTROLLER_ADDRESS
else
    service iptables stop
    chkconfig iptables off
    sed -i "s/SELINUX=enforcing/SELINUX=disabled/g" /etc/selinux/config
#    setenforce 0
fi

for ip in $AGENT_ADDRESS_LIST; do
    set_iptables_selinux $ip
done

echo "+++++++++++++++finish setting the iptables and selinux+++++++++++++++"


#-------------------------------------------------------------------------------
#            downloading the dependences
#-------------------------------------------------------------------------------

if [ ! -d /opt/vsm-dep-repo ] && [ ! -d vsm-dep-repo ]; then
    wget https://github.com/01org/vsm-dependencies/archive/"$DEPENDENCE_BRANCH".zip
    unzip $DEPENDENCE_BRANCH
    mv vsm-dependencies-$DEPENDENCE_BRANCH/repo vsm-dep-repo
    is_createrepo=`rpm -qa|grep createrepo|wc -l`
    if [[ $is_createrepo -gt 0 ]]; then
        createrepo vsm-dep-repo
    fi
    rm -rf vsm-dependencies-$DEPENDENCE_BRANCH
    rm -rf $DEPENDENCE_BRANCH
fi

if [ $is_controller -eq 0 ]; then
    $SSH $USER@$CONTROLLER_ADDRESS "rm -rf /opt/vsm-dep-repo"
    $SCP -r vsm-dep-repo $USER@$CONTROLLER_ADDRESS:/opt
else
    if [ -d vsm-dep-repo ]; then
        rm -rf /opt/vsm-dep-repo
        cp -rf vsm-dep-repo /opt
    fi
fi


#-------------------------------------------------------------------------------
#            setting the repo
#-------------------------------------------------------------------------------

echo "+++++++++++++++start setting the repo+++++++++++++++"

is_httpd=`rpm -qa|grep httpd|grep -v httpd-tools|wc -l`
if [[ $is_httpd -gt 0 ]]; then
    sed -i "s,#*Listen 80,Listen 80,g" /etc/httpd/conf/httpd.conf
    service httpd restart
fi

rm -rf vsm.repo
cat <<"EOF" >vsm.repo
[vsm-dep-repo]
name=vsm-dep-repo
baseurl=file:///opt/vsm-dep-repo
gpgcheck=0
enabled=1
proxy=_none_
EOF

oldurl="file:///opt/vsm-dep-repo"
newurl="http://$CONTROLLER_ADDRESS/vsm-dep-repo"
if [ $is_controller -eq 0 ]; then
    $SCP vsm.repo $USER@$CONTROLLER_ADDRESS:/etc/yum.repos.d
    $SSH $USER@$CONTROLLER_ADDRESS "yum makecache; yum -y install httpd; service httpd restart; rm -rf /var/www/html/vsm-dep-repo; cp -rf /opt/vsm-dep-repo /var/www/html"
    $SSH $USER@$CONTROLLER_ADDRESS "sed -i \"s,$oldurl,$newurl,g\" /etc/yum.repos.d/vsm.repo; yum makecache"
else
    cp vsm.repo /etc/yum.repos.d
    yum makecache; yum -y install httpd; service httpd restart; rm -rf /var/www/html/vsm-dep-repo; cp -rf /opt/vsm-dep-repo /var/www/html
    sed -i "s,$oldurl,$newurl,g" /etc/yum.repos.d/vsm.repo
    yum makecache
fi

sed -i "s,$oldurl,$newurl,g" vsm.repo

function set_repo() {
    $SSH $USER@$1 "rm -rf /etc/yum.repos.d/vsm.repo"
    $SCP vsm.repo $USER@$1:/etc/yum.repos.d
    $SSH $USER@$1 "yum makecache"
}

for ip in $AGENT_ADDRESS_LIST; do
    set_repo $ip
done

echo "+++++++++++++++finish setting the repo+++++++++++++++"


#-------------------------------------------------------------------------------
#            install vsm rpm and dependences
#-------------------------------------------------------------------------------

echo "+++++++++++++++install vsm rpm and dependences+++++++++++++++"

function install_vsm_controller() {
    $SSH $USER@$1 "mkdir -p /opt/vsm_install"
    $SCP vsmrepo/python-vsmclient*.rpm vsmrepo/vsm*.rpm $USER@$1:/opt/vsm_install
    $SSH $USER@$1 "cd /opt/vsm_install; yum -y localinstall python-vsmclient*.rpm vsm*.rpm"
    $SSH $USER@$1 "preinstall"
    $SSH $USER@$1 "cd /opt; rm -rf /opt/vsm_install"
}

function install_vsm_storage() {
    $SSH $USER@$1 "mkdir -p /opt/vsm_install"
    $SCP vsmrepo/vsm*.rpm $USER@$1:/opt/vsm_install
    $SSH $USER@$1 "cd /opt/vsm_install; rm -rf vsm-dashboard*; yum -y localinstall vsm*.rpm"
    $SSH $USER@$1 "preinstall"
    $SSH $USER@$1 "cd /opt; rm -rf /opt/vsm_install"
}

if [ $is_controller -eq 0 ]; then
    install_vsm_controller $CONTROLLER_ADDRESS
else
    yum -y localinstall vsmrepo/python-vsmclient*.rpm vsmrepo/vsm*.rpm
    preinstall
fi

for ip in $AGENT_ADDRESS_LIST; do
    install_vsm_storage $ip
done

echo "+++++++++++++++finish install vsm rpm and dependences+++++++++++++++"


#-------------------------------------------------------------------------------
#            setup vsm controller node
#-------------------------------------------------------------------------------

if [ -z $MANIFEST_PATH ]; then
    MANIFEST_PATH="manifest"
fi

function setup_controller() {
    $SSH $USER@$CONTROLLER_ADDRESS "rm -rf /etc/manifest/cluster_manifest"
    $SCP $MANIFEST_PATH/$CONTROLLER_ADDRESS/cluster.manifest $USER@$CONTROLLER_ADDRESS:/etc/manifest
    $SSH $USER@$CONTROLLER_ADDRESS "chown root:vsm /etc/manifest/cluster.manifest; chmod 755 /etc/manifest/cluster.manifest"
    is_cluster_manifest_error=`$SSH $USER@$CONTROLLER_ADDRESS "cluster_manifest|grep error|wc -l"`
    if [ $is_cluster_manifest_error -gt 0 ]; then
        echo "please check the cluster.manifest, then try again"
        exit 1
    else
        $SSH $USER@$CONTROLLER_ADDRESS "vsm-controller"
    fi
}

if [ $is_controller -eq 0 ]; then
    setup_controller
else
    rm -rf /etc/manifest/cluster.manifest
    cp $MANIFEST_PATH/$CONTROLLER_ADDRESS/cluster.manifest /etc/manifest
    chown root:vsm /etc/manifest/cluster.manifest
    chmod 755 /etc/manifest/cluster.manifest
    if [ `cluster_manifest|grep error|wc -l` -gt 0 ]; then
        echo "please check the cluster.manifest, then try again"
        exit 1
    else
        vsm-controller
    fi
fi


#-------------------------------------------------------------------------------
#            setup vsm storage node
#-------------------------------------------------------------------------------

count_ip=0
for ip in $AGENT_ADDRESS_LIST; do
    let count_ip=$count_ip+1
done
let count_ip=$count_ip+2+1
if [ `ls $MANIFEST_PATH|wc -l` != $count_ip ]; then
    echo "please check the manifest folder"
    exit 1
fi

success=""
failure=""
if [ $is_controller -eq 0 ]; then
    token=`$SSH $USER@$CONTROLLER_ADDRESS "agent-token"`
else
    token=`agent-token`
fi

function setup_storage() {
    $SSH $USER@$1 "rm -rf /etc/manifest/server.manifest"
    #sed -i "s/token-tenant/$token/g" $MANIFEST_PATH/$1/server.manifest
    #old_str=`cat $MANIFEST_PATH/$1/server.manifest| grep ".*-.*" | grep -v by | grep -v "\["`
    #sed -i "s/$old_str/$token/g" $MANIFEST_PATH/$1/server.manifest
    sed -i "/^\[auth_key\]$/,/^\[.*\]/ s/^.*-.*$/$TOKEN/" $MANIFEST_PATH/$1/server.manifest
    $SCP $MANIFEST_PATH/$1/server.manifest $USER@$1:/etc/manifest
    $SSH $USER@$1 "chown root:vsm /etc/manifest/server.manifest; chmod 755 /etc/manifest/server.manifest"
    is_server_manifest_error=`$SSH $USER@$1 "server_manifest" |grep ERROR|wc -l`
    if [ $is_server_manifest_error -gt 0 ]; then
        echo "[warning]: The server.manifest in $1 is wrong, so fail to setup in $1 storage node"
        failure=$failure"$1 "
    else
        $SSH $USER@$1 "vsm-node"
        success=$success"$1 "
    fi
}

for ip in $AGENT_ADDRESS_LIST; do
    setup_storage $ip
done


#-------------------------------------------------------------------------------
#            finish auto deploy
#-------------------------------------------------------------------------------

echo "Successful storage node ip: $success"
echo "Failure storage node ip: $failure"

set +o xtrace