functions

#!/bin/sh
CONF="vm.conf"
SCCDATA="SCCDATA.conf"

## color
# NORMAL
N=`echo -en "\\033[0;32m"`
# SUCCESS
S=`echo -en "\\033[1;32m"`
#INFO
I=`echo -en "\\033[1;34m"`
# WARNING / DELETE
W=`echo -en "\\033[1;33m"`
# OTHER
#O=`echo -en "\\033[0;39m"`
O=`echo -en "\\033[0m"`
#O=`echo -en ""
# FAILURE
F=`echo -en "\\033[1;31m"`

check_load_config_file() {
if [ "$1" == "other" ]; then CDIR=`dirname $PWD`; else CDIR=$PWD; fi
if [ -f $CDIR/${CONF} ]; then
    source ${CDIR}/${CONF}
else
    echo "!! ${CONF} not found in ${CDIR} path !!"
    exit 1
fi
if [ -f $CDIR/${SCCDATA} ]; then
    source ${CDIR}/${SCCDATA}
else
    echo "!! ${SCCDATA} not found in ${CDIR} path !!"
    echo "!! Please create one but never push this file in the git repo !!"
    echo ""
    echo "Typical content of this file: ${SCCDATA}"
    echo "SCCREGCODE=YOUR_SCC_REG_CODE"
    echo "VERSION=15.4"
    exit 1
fi
}

install_package() {
    echo $I "############ START install_package" $O
    if [ $# -lt 1 ]
    then
        echo $F "! Package(s) name needed ! ; Exiting" $O
        exit 1
    fi
    PKG=$1
#    for i in `seq 1 $NBNODE`
#    do
#        exec_on_node ${NODENAME}${i} "zypper in -y screen"
#    done
    for i in `seq 1 $NBNODE`
    do
        echo "- Installing $PKG on node ${NODENAME}${i}"
        ssh ${NODENAME}${i} "screen -d -m zypper in -y $PKG"
    done
}

# execute a command on a NODE from host
exec_on_node() {
    # first arg is NODE name, second arg is command
    # third arg is optionnal, will ignore returning an error in case output command result is not 0
    if [ $# -lt 2 ]
    then
	echo $F "! Two args needed (NODE / CMD / OPTIONNAL_IGNORE); Exiting" $O
        exit 1
    fi
    NODE="$1"
    CMD="$2"
    IGNORE="$3"
    # avoid: No pseudo-tty detected! Use -t option to ssh if calling remotely
    echo "${NODE}: ssh -T -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no ${NODE} \"${CMD}\"";
    ssh -q -T -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no ${NODE} "${CMD}"
    if [ "$?" -eq 0 ]
    then
	echo $S "- ssh command sucessful" $O
    else
	if [ "${IGNORE}" == "" ]
	then
	    echo $F "- ! Return of the ssh command is not 0" $O
	    echo $F "-  # PRESS [ENTER] TO CONTINUE # " $O
	    read
	fi
    fi
}

exec_on_node_screen() {
    # first arg is NODE name, second arg is command
    if [ $# -lt 2 ]
    then
	echo $F "! Two args needed (NODE / CMD); Exiting" $O
        exit 1
    fi
    NODE="$1"
    CMD="$2"
    HASH=`echo $NODE$CMD |md5sum | cut -d " " -f 1`
    # avoid: No pseudo-tty detected! Use -t option to ssh if calling remotely
#    screen -A -d -m -S sle15sp332390dc28adf81c4f7d1b3ecf9f8c76cd ssh sle15sp33 zypper ref
    echo "${NODE}: screen -A -d -m -S ${NODE}-${HASH} ssh -T -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no ${NODE} ${CMD}";
    screen -A -d -m -S ${NODE}${HASH} ssh -T -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no ${NODE} ${CMD}
}

exec_pssh() {
    if [ $# -lt 1 ]
    then
        echo "! 1 arg needed (CMD);Exiting"
        exit 1
    fi
    CMD="$1"
    for nb in `seq 2 $NBNODE` 
    do
        echo "ssh root@${NODENAME}${nb} -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no $CMD"
        ssh root@${NODENAME}${nb} -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no "$CMD"
    done
    # echo "pssh -h ${PSSHCONF} -x \"-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no\" \"$CMD\""
    # pssh -h ${PSSHCONF} -x "-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no" "$CMD"
}

# scp a file to a NODE from host
scp_on_node() {
    # first arg is SOURCE file, last arg is NODE:/destination_dir
    if [ $# -lt 2 ];then
	echo "! 2 args needed (FILE NODE:DEST); Exiting"
        exit 1
    fi
    FILE="$1"
    DEST="$2"
    # avoid: No pseudo-tty detected! Use -t option to ssh if calling remotely
    echo "${NODE}: scp -o StrictHostKeyChecking=no ${FILE} ${DEST}"
    scp -q -o StrictHostKeyChecking=no ${FILE} ${DEST}
    if [ "$?" -eq 0 ]
	then
	echo $S "- scp command sucessful" $O
    else
	echo $F "- ! Return of scp command is not 0" $O
    fi
}

# create a pool (on host)
create_pool() {
    if [ $# -lt 1 ];then echo "! 1 arg needed: POOL_NAME; Exiting" ; exit 1; fi
    POOL=$1
    echo "############ START create a pool ${POOL} #############"
    mkdir -p ${STORAGEP}/${POOL}
    virsh pool-list --all | grep ${POOL} > /dev/null
    if [ $? == "0" ]; then
        echo "- ${POOL} already present, deleting it"
        virsh pool-destroy ${POOL}
        virsh pool-undefine ${POOL}
        rm -rvf ${STORAGEP}/${POOL}
    fi
    echo "- Create ${POOL}"
    mkdir -p ${STORAGEP}/${POOL}
    virsh pool-define-as --name ${POOL} --type dir --target ${STORAGEP}/${POOL}
    virsh pool-start ${POOL}
    virsh pool-autostart ${POOL}
}

start_vm() {
    echo $I "############ START start_vm" $O
    for i in `seq 1 $NBNODE`
    do
        echo "- Starting domain ${NODENAME}${i}"
        virsh start ${NODENAME}${i}
        sleep 2
    done
}

stop_vm() {
    echo $I "############ START stop_vm" $O
    for i in `seq 1 $NBNODE`
    do
        echo "- Stoping domain ${NODENAME}${i}"
        virsh destroy ${NODENAME}${i}
        sleep 1
    done
}

enable_DVD_media() {
echo $I "############ START enable_DVD_media"$O
    for i in `seq 1 $NBNODE`
    do
        exec_on_node ${NODENAME}${i} "zypper mr -e 1 2 3 4 5 6 7"
    done
}

enable_SCC_repo() {
echo $I "############ START enable_SCC_repo"$O
    for i in `seq 1 $NBNODE`
    do
        exec_on_node ${NODENAME}${i} "SUSEConnect -r ${SCCREGCODE}"
        exec_on_node ${NODENAME}${i} "SUSEConnect -p PackageHub/$VERSION/$HOSTTYPE"
        exec_on_node ${NODENAME}${i} "SUSEConnect -p sle-module-server-applications/$VERSION/$HOSTTYPE"
        exec_on_node ${NODENAME}${i} "SUSEConnect -p sle-module-web-scripting/$VERSION/$HOSTTYPE"
        exec_on_node ${NODENAME}${i} "SUSEConnect -p sle-module-development-tools/$VERSION/$HOSTTYPE"
        exec_on_node ${NODENAME}${i} "SUSEConnect -p sle-module-hpc/$VERSION/$HOSTTYPE"
    done
}

cleanup_zypper_repo() {
echo $I "############ START cleanup_zypper_repo"$O
    for i in `seq 1 $NBNODE`
    do
        exec_on_node ${NODENAME}${i} "SUSEConnect --cleanup"
        exec_on_node ${NODENAME}${i} "zypper removerepo 1 2 3 4 5 6 7 8 9"
    done
}

update_nodes() {
echo $I "############ START update_nodes"$O
    for i in `seq 1 $NBNODE`
    do
#       exec_on_node ${NODENAME}${i} "zypper ref"
        echo "AA"
    done
    for i in `seq 1 $NBNODE`
    do
        echo ${NODENAME}${i} "zypper up -y; zypper up -y --auto-agree-with-licenses"
        exec_on_node_screen ${NODENAME}${i} "zypper ref -y ; zypper up -y --auto-agree-with-licenses"
    done
    echo " - Update running in background, check in screen:"
    screen -list
    echo " screen -r SCREENID"
}