murano-repository/muranorepository/Services/scripts/Linux/common.sh

#!/bin/bash
# 
DEBUGLVL=3
LOGFILE=/tmp/muranodeployment.log
PIPAPPS="pip python-pip pip-python"
PIPCMD=""
if [ "$DEBUGLVL" -eq 4 ]; then
    set -x
fi
function log {
    if [ "$DEBUGLVL" -gt 0 ]; then
        chars=$(echo "@$" | wc -c)
        case $DEBUGLVL in
            1 )
                echo -e "LOG:>$@"
                ;;
            2)
                echo -e "$(date +"%m-%d-%Y %H:%M") LOG:>$@" | tee --append $LOGFILE
                ;;
            3)
                echo -e "$(date +"%m-%d-%Y %H:%M") LOG:>$@" >> $LOGFILE
                ;;
            4)
                echo -e "$(date +"%m-%d-%Y %H:%M") LOG:>$@" | tee --append $LOGFILE
                ;;
        esac
    fi
}
function lowercase(){
    echo "$1" | sed "y/ABCDEFGHIJKLMNOPQRSTUVWXYZ/abcdefghijklmnopqrstuvwxyz/"
}
function find_pip()
{
    for cmd in $PIPAPPS
    do
        _cmd=$(which $cmd 2>/dev/null)
        if [ $? -eq 0 ];then
                break
        fi
    done
    if [ -z $_cmd ];then
        echo "Can't find \"pip\" in system, please install it first, exiting!"
        exit 1
    else
        PIPCMD=$_cmd
    fi
}
OPTIND=1 # Reset if getopts used previously
function collect_args(){
    _n=$1
    shift
    ARGS=''
    while true
    do
        if [[ "$_n" == -* ]] || [ -z "$_n" ]; then
            OPTIND=$((OPTIND - 1))
            break
        fi
        #echo "_n=$_n ; $OPTIND"
        if [ -z "$ARGS" ]; then
            ARGS=$OPTARG
        else
            ARGS="$ARGS $_n"
        fi
        eval _n=\$$OPTIND
        OPTIND=$((OPTIND + 1))
        #sleep 1
    done
    echo $ARGS
    unset _n
    unset ARGS
}
function get_os(){
    KERNEL=$(uname -r)
    MACH=$(uname -m)
    OS=$(uname)
    if [ "${OS}" = "Linux" ] ; then
        if [ -f /etc/redhat-release ] ; then
            DistroBasedOn='RedHat'
            Packager='yum'
            DIST=$(cat /etc/redhat-release |sed s/\ release.*//)
            PSUEDONAME=$(cat /etc/redhat-release | sed s/.*\(// | sed s/\)//)
            REV=$(cat /etc/redhat-release | sed s/.*release\ // | sed s/\ .*//)
        elif [ -f /etc/SuSE-release ] ; then
            DistroBasedOn='SuSe'
            Packager='zypper'
            PSUEDONAME=$(cat /etc/SuSE-release | tr "\n" ' '| sed s/VERSION.*//)
            REV=$(cat /etc/SuSE-release | tr "\n" ' ' | sed s/.*=\ //)
        elif [ -f /etc/mandrake-release ] ; then
            DistroBasedOn='Mandrake'
            Packager='urpmi urpme'
            PSUEDONAME=$(cat /etc/mandrake-release | sed s/.*\(// | sed s/\)//)
            REV=$(cat /etc/mandrake-release | sed s/.*release\ // | sed s/\ .*//)
        elif [ -f /etc/debian_version ] ; then
            DistroBasedOn='Debian'
            Packager='apt-get'
            DIST=$(cat /etc/lsb-release | grep '^DISTRIB_ID' | awk -F=  '{ print $2 }')
            PSUEDONAME=$(cat /etc/lsb-release | grep '^DISTRIB_CODENAME' | awk -F=  '{ print $2 }')
            REV=$(cat /etc/lsb-release | grep '^DISTRIB_RELEASE' | awk -F=  '{ print $2 }')
        fi
        if [ -f /etc/UnitedLinux-release ] ; then
            DIST="${DIST}[$(cat /etc/UnitedLinux-release | tr "\n" ' ' | sed s/VERSION.*//)]"
        fi
        OS=$(lowercase $OS)
        DistroBasedOn=$(lowercase $DistroBasedOn)
        readonly OS
        readonly DIST
        readonly DistroBasedOn
        readonly PSUEDONAME
        readonly REV
        readonly KERNEL
        readonly MACH
        #readonly Packager
    else
        OS=unknown
        readonly OS
        log "OS:$OS"
        exit 1
    fi
}
function add_fw_rule(){
    _rule_string=$@
    _tmp_fw_port=$(echo $_rule_string | grep -o -e "dport [0-9]*\s")
    _tmp_fw_proto=$(echo $_rule_string | grep -o -e "-p \w*\s")
    _fw_port=$(echo $_tmp_fw_port | awk '{print $2}')
    _fw_proto=$(echo $_tmp_fw_proto |awk '{print $2}')
    _fw_reload=""
    #find iptables and add rule
    case $DIST in
        "Fedora")
            _fw_cmd=$(which firewall-cmd)
            _fw_port=$(echo $_rule_string | grep -o -e "dport [0-9]*\s" | awk '{print $2}')
            _fw_proto=$(echo $_rule_string | grep -o -e "-p \w*\s" | awk '{print $2}')
            _fw_rule="--permanent --add-port=$_fw_port/$_fw_proto"
            _fw_enable_rules="$_fw_cmd --reload"
            ;;
        *)
            _fw_cmd=$(which iptables)
            _fw_rule=$_rule_string
            _fw_enable_rules="service $(basename $_fw_cmd) save"
            ;;
    esac
    iptcmdsave=$(which iptables-save)
    if [[ "$_fw_cmd" != '' ]] && [[ "$iptcmdsave" != '' ]]; then
        eval "$iptcmdsave | grep -e \"$_tmp_fw_port\" | grep -e \"$_tmp_fw_proto\"" > /dev/null 2>&1
        if [ $? -ne 0 ]; then
            eval $_fw_cmd $_fw_rule
            if [ $? -ne 0 ]; then
                log "Can't set firewall rules, exiting..."
                exit 1
            else
                if [ -n "$_fw_enable_rules" ]; then
                    log "Running \"$_fw_enable_rules\""
                    $_fw_enable_rules > /dev/null
                fi
                log "$_fw_cmd rule with $_fw_rule set."
            fi
        else
            log "$_fw_cmd rule exists."
        fi
    else
        log "There are no fw found..."
    fi
}
function enable_init(){
    _initctrl=""
    _init_suffix=""
    _service=$1
    case $DistroBasedOn in
        "debian")
            _initctrl="update-rc.d"
            _init_suffix="defaults"
            ;;
        *)
            _initctrl="chkconfig"
            _init_suffix="on"
            ;;
    esac
    $_initctrl $_service $_init_suffix
    if [ $? -ne 0 ]; then
        log "$_initctrl $_service $_init_suffix - fails!"
        exit 1
    fi
}
function restart_service(){
    _service=$1
    service $_service restart > /dev/null 2>&1
    if [ $? -ne 0 ]; then
        log "Can't start $_service service!"
        exit 1
    fi
}
function package_renamer(){
    _pkg=$1
    case $DistroBasedOn in
        "debian")
            _pkg=$(echo $_pkg | sed 's/-devel$/-dev/')
            ;;
        *)
            _pkg=$(echo $_pkg | sed 's/-dev$/-devel/')
            ;;
    esac
    echo $_pkg
}