salt-bootstrap/bootstrap-salt-minion.sh

#!/bin/bash -
#===============================================================================
# vim: softtabstop=4 shiftwidth=4 expandtab fenc=utf-8 spell spelllang=en
#===============================================================================
#
#          FILE: bootstrap-salt-minion.sh
#
#         USAGE: bootstrap-salt-minion.sh
#
#   DESCRIPTION: Bootstrap salt installation for various systems/distros
#
#       OPTIONS: ---
#  REQUIREMENTS: ---
#          BUGS: ---
#         NOTES: ---
#        AUTHOR: Pedro Algarvio (s0undt3ch), pedro@algarvio.me
#  ORGANIZATION: UfSoft.org
#       CREATED: 10/15/2012 09:49:37 PM WEST
#      REVISION:  ---
#===============================================================================
set -o nounset                              # Treat unset variables as an error

ScriptVersion="1.0"
NOCOLORS=0


# Define our colors
if [ $NOCOLORS -eq 0 ]; then
    BLACK="\033[0;30m"
    BLUE="\033[0;34m"
    BROWN="\033[0;33m"
    CYAN="\033[0;36m"
    DARK_GRAY="\033[1;30m"
    DEFAULT_COLOR="\033[00m"
    ENDC="\033[0m"
    GREEN="\033[0;32m"
    LIGHT_BLUE="\033[1;34m"
    LIGHT_CYAN="\033[1;36m"
    LIGHT_GRAY="\033[0;37m"
    LIGHT_GREEN="\033[1;32m"
    LIGHT_PURPLE="\033[1;35m"
    LIGHT_RED="\033[1;31m"
    PURPLE="\033[0;35m"
    RED="\033[0;31m"
    RED_BOLD="\033[01;31m"
    WHITE="\033[1;37m"
    YELLOW="\033[1;33m"
else
    BLACK=''
    BLUE=''
    BROWN=''
    CYAN=''
    DARK_GRAY=''
    DEFAULT_COLOR=''
    ENDC=''
    GREEN=''
    LIGHT_BLUE=''
    LIGHT_CYAN=''
    LIGHT_GRAY=''
    LIGHT_GREEN=''
    LIGHT_PURPLE=''
    LIGHT_RED=''
    PURPLE=''
    RED=''
    RED_BOLD=''
    WHITE=''
    YELLOW=''
fi

#===  FUNCTION  ================================================================
#         NAME:  usage
#  DESCRIPTION:  Display usage information.
#===============================================================================
usage() {
    cat << EOT

  Usage :  ${0##/*/} [options] <install-type>

  Installation types:
    - stable (default)
    - daily
    - git

  Options:
  -h|help       Display this message
  -v|version    Display script version
  -N|nocolor    Do not use colors

EOT
}   # ----------  end of function usage  ----------

#-----------------------------------------------------------------------
#  Handle command line arguments
#-----------------------------------------------------------------------

while getopts ":hvN" opt
do
  case $opt in

    h|help     )  usage; exit 0   ;;

    v|version  )  echo "$0 -- Version $ScriptVersion"; exit 0   ;;

    N|nocolor  )  NOCOLORS=1    ;;

    \? )  echo -e "\n  Option does not exist : $OPTARG\n"
          usage; exit 1   ;;

  esac    # --- end of case ---
done
shift $(($OPTIND-1))

# Define installation type
if [ "$#" -eq 0 ];then
    ITYPE="stable"
else
    ITYPE=$1
fi

if [ "$ITYPE" != "stable" -a "$ITYPE" != "daily" -a "$ITYPE" != "git" ]; then
    echo "${LIGHT_RED} ERROR: Installation type \"$ITYPE\" is not known...${ENDC}"
    exit 1
fi

# Root permissions are required to run this script
if [ $(whoami) != "root" ] ; then
    echo "${LIGHT_RED} * ERROR: Salt requires root privileges to install. Please re-run this script as root.${ENDC}"
    exit 1
fi

# Create a temporary directory used for any temp files created
TMPDIR="/tmp/salty-temp"
if [ ! -d $TMPDIR ]; then
    echo "${LIGHT_BLUE} * Creating temporary directory ${TMPDIR} ${ENDC}"
    mkdir $TMPDIR
fi
# Store current directory
STORED_PWD=$(pwd)
# Change to temp directory
cd $TMPDIR
# When the script exits, change to the initial directory.
trap "cd $STORED_PWD" EXIT

# Define our logging file and pipe
LOGFILE="/tmp/$(basename $0).log"
LOGPIPE="/tmp/$(basename $0).logpipe"

# Remove the logging pipe when the script exits
trap "rm -f $LOGPIPE" EXIT

# Create our logging pipe
mknod $LOGPIPE p

# What ever is written to the logpipe gets written to the logfile
tee < $LOGPIPE $LOGFILE &

# Close STDOUT, reopen it directing it to the logpipe
exec 1>&-
exec 1>$LOGPIPE
# Close STDERR, reopen it directing it to the logpipe
exec 2>&-
exec 2>$LOGPIPE

# Define some SHTOOL variables
SHTOOL_COMMON="sh.common"
SHTOOL_COMMON_MD5='2fdd8ccf7122df039cdf79a9f7a083e4'
SHTOOL_COMMON_LINK='http://cvs.ossp.org/getfile?f=ossp-pkg/shtool/sh.common&v=1.24'
SHTOOL_PLATFORM="shtool.platform"
SHTOOL_PLATFORM_MD5='bf4c782746e1c92923fb66de513f9295'
SHTOOL_PLATFORM_LINK='http://cvs.ossp.org/getfile?f=ossp-pkg/shtool/sh.platform&v=1.31'


#===  FUNCTION  ================================================================
#         NAME:  shtool
#  DESCRIPTION:  Run shtool commands.
#===============================================================================
shtool() {
    OPWD=$(pwd)
    cd $TMPDIR
    echo $(sh ./shtool.platform "$@")
    cd $OPWD
}


#===  FUNCTION  ================================================================
#         NAME:  download_shtool
#  DESCRIPTION:  Download shtool required scripts
#===============================================================================
download_shtool() {
    if [ ! -f $SHTOOL_COMMON ]; then
        echo "Download SHTOOL sh.common from $SHTOOL_COMMON_LINK"
        wget $SHTOOL_COMMON_LINK -O $SHTOOL_COMMON
        MD5SUM=$(md5sum $SHTOOL_COMMON | awk '{ print $1 }')
        if [ "$MD5SUM" != "$SHTOOL_COMMON_MD5" ]; then
            echo "MD5 signature of sh.common does not match!"
            exit 1
        fi
    fi

    if [ ! -f $SHTOOL_PLATFORM ]; then
        echo "Download sh.platform from $SHTOOL_PLATFORM_LINK"
        wget $SHTOOL_PLATFORM_LINK -O $SHTOOL_PLATFORM
        MD5SUM=$(md5sum $SHTOOL_PLATFORM | awk '{ print $1 }')
        if [ "$MD5SUM" != "$SHTOOL_PLATFORM_MD5" ]; then
            echo "MD5 signature of shtool.platform does not match!"
            exit 1
        fi
    fi
}

echo "${LIGHT_BLUE} * Downloading shtool required scripts for system detection${ENDC}"
download_shtool
echo "${LIGHT_BLUE} * Downloaded required shtool scripts${ENDC}"

echo "${LIGHT_BLUE} * Detecting system:${ENDC}"

ARCH=$(shtool -F "%at")

FULL_SYSTEM=$(shtool -F '%<st>' -L -S '|' -C '+')
SYSTEM_NAME=$(echo $FULL_SYSTEM | cut -d \| -f1 )
SYSTEM_VERSION=$(echo $FULL_SYSTEM | cut -d \| -f2 )

FULL_DISTRO=$(shtool -F '%<sp>' -L -S '|' -C '+')
DISTRO_NAME=$(echo $FULL_DISTRO | cut -d \| -f1 )
DISTRO_VERSION=$(echo $FULL_DISTRO | cut -d \| -f2 )

echo "${YELLOW}    System Information:${ENDC}"
echo "${YELLOW}      System:\t\t${FULL_SYSTEM}${ENDC}"
echo "${YELLOW}      Architecture:\t${ARCH}${ENDC}"
echo "${YELLOW}      Distribution:\t${DISTRO_NAME} ${DISTRO_VERSION}${ENDC}"

if [ $SYSTEM_NAME != "linux" ]; then
    echo "${LIGHT_RED} * ERROR: Only Linux is currently supported${ENDC}"
    exit 1
fi

# Black Magic Below!!!
INSTALL_FUNC=""

###############################################################################################
#
#   Distribution install functions
#
#   In order to install salt for a distribution you need to define:
#
#   To Install Dependencies, which is required, one of:
#       1. install_<distro>_<distro_version>_<install_type>_deps
#       2. install_<distro>_<distro_version>_deps
#       3. install_<distro>_deps
#
#
#   To install salt, which, of course, is required, one of:
#       1. install_<distro>_<distro_version>_<install_type>
#       1. install_<distro>_<install_type>
#
#   And optionally, define a post install function, one of:
#       1. install_<distro>_<distro_versions>_<install_type>_post
#       2. install_<distro>_<distro_versions>_post
#       3. install_<distro>_post
#
###############################################################################################

###############################################################################################
#
#   Ubuntu Install Functions
#
###############################################################################################
install_ubuntu_deps() {
    apt-get update
    apt-get -y install python-software-properties
    add-apt-repository -y ppa:saltstack/salt
    apt-get update
}

install_ubuntu_1004_deps() {
    apt-get update
    apt-get -y install python-software-properties
    add-apt-repository ppa:saltstack/salt
    apt-get update
    apt-get -y install salt-minion
}

install_ubuntu_1110_deps() {
    apt-get update
    apt-get -y install python-software-properties
    add-apt-repository -y 'deb http://us.archive.ubuntu.com/ubuntu/ oneiric universe'
    add-apt-repository -y ppa:saltstack/salt
}

install_ubuntu_1110_post() {
    add-apt-repository -y --remove 'deb http://us.archive.ubuntu.com/ubuntu/ oneiric universe'
}

install_ubuntu_1004_stable() {
    apt-get -y install salt-minion
}

install_ubuntu_1104_stable() {
    apt-get -y install salt-minion
}

install_ubuntu_stable() {
    apt-get -y install salt-minion
}
###############################################################################################
#
#   End of Ubuntu Install Functions
#
###############################################################################################

###############################################################################################
#
#   Debian Install Functions
#
###############################################################################################

install_debian_60_stable_deps() {
    echo "deb http://backports.debian.org/debian-backports squeeze-backports main" >> /etc/apt/sources.list.d/backports.list
    apt-get update
}

install_debian_60_stable() {
    apt-get -t squeeze-backports -y install salt-minion
}
###############################################################################################
#
#   Ended Debian Install Functions
#
###############################################################################################


###############################################################################################
###############################################################################################
###############################################################################################
###
###   DO NOT CHANGE ANYTHING BELLOW
###
###############################################################################################
###############################################################################################
###############################################################################################

# Let's get the dependencies install function
DEPS_INSTALL_FUNC="install_${DISTRO_NAME}_$(echo $DISTRO_VERSION | tr -d '.')_${ITYPE}_deps"
if $( ! type ${DEPS_INSTALL_FUNC} | grep -q 'shell function' ); then
    echo "${BROWN} * INFO: ${DEPS_INSTALL_FUNC} not found..."
    # let's try and see if have a deps function which ignores the installation type
    DEPS_INSTALL_FUNC="install_${DISTRO_NAME}_$(echo $DISTRO_VERSION | tr -d '.')_deps"
    if $( ! type ${DEPS_INSTALL_FUNC} | grep -q 'shell function' ); then
        echo "${BROWN} * INFO: ${DEPS_INSTALL_FUNC} not found..."
        # Let's try to see if we have a deps function which also ignores the distro version
        DEPS_INSTALL_FUNC="install_${DISTRO_NAME}_deps"
        if $( ! type ${DEPS_INSTALL_FUNC} | grep -q 'shell function' ); then
            echo "${LIGHT_RED} * ERROR: Installation not supported not supported. Can't find ${DEPS_INSTALL_FUNC}()${ENDC}"
            exit 1
        fi
    fi
fi

# Let's get the install function
INSTALL_FUNC="install_${DISTRO_NAME}_$(echo $DISTRO_VERSION | tr -d '.')_${ITYPE}"
if $( ! type ${INSTALL_FUNC} | grep -q 'shell function' ); then
    echo "${BROWN} * INFO: ${INSTALL_FUNC} not found..."
    # Let see if we have an install function which ignores the distribution version
    INSTALL_FUNC="install_${DISTRO_NAME}_${ITYPE}"
    if $( ! type ${INSTALL_FUNC} | grep -q 'shell function' ); then
        echo "${LIGHT_RED}ERROR: Installation not supported not supported. Can't find ${INSTALL_FUNC}()${ENDC}"
        exit 1
    fi
fi

# Let's get the post function, if any, it's optional
POST_INSTALL_FUNC="install_${DISTRO_NAME}_$(echo $DISTRO_VERSION | tr -d '.')_${ITYPE}_post"
if $( ! type ${POST_INSTALL_FUNC} | grep -q 'shell function' ); then
    echo "${BROWN} * INFO: ${POST_INSTALL_FUNC} not found..."
    # let's try and see if have a post function which ignores the installation type
    POST_INSTALL_FUNC="install_${DISTRO_NAME}_$(echo $DISTRO_VERSION | tr -d '.')_post"
    if $( ! type ${POST_INSTALL_FUNC} | grep -q 'shell function' ); then
        echo "${BROWN} * INFO: ${POST_INSTALL_FUNC} not found..."
        # Let's try to see if we have a deps function which also ignores the distro version
        POST_INSTALL_FUNC="install_${DISTRO_NAME}_post"
        if $( ! type ${POST_INSTALL_FUNC} | grep -q 'shell function' ); then
            echo "${BROWN} * INFO: ${POST_INSTALL_FUNC} not found..."
            POST_INSTALL_FUNC=""
        fi
    fi
fi

# Install dependencies
echo "${LIGHT_BLUE} * Running ${DEPS_INSTALL_FUNC}()${ENDC}"
$DEPS_INSTALL_FUNC

# Install Salt
echo "${LIGHT_BLUE} * Running ${INSTALL_FUNC}()${ENDC}"
$INSTALL_FUNC

# Run any post install function
if [ "$POST_INSTALL_FUNC" != "" ]; then
    echo "${LIGHT_BLUE} * Running ${POST_INSTALL_FUNC}()${ENDC}"
    $POST_INSTALL_FUNC
fi

# Done!
echo "${LIGHT_BLUE} * Salt installed!${ENDC}"