From 1150c69735c7f33a34d43fc474f6ff87f454224d Mon Sep 17 00:00:00 2001
From: Pedro Algarvio <pedro@algarvio.me>
Date: Wed, 17 Oct 2012 14:02:09 +0100
Subject: [PATCH] Bootstrap script reworked.

* Dropped the log function, instead we pipe stderr and stdout to both the console and the logfile.
* We now make use of 2 scripts from shtool which are involved in system detection, downloaded at runtime.
* In order to support a distribution there are some functions which need to be added to this script, all explained on the source.
---
 bootstrap-salt-minion.sh | 471 ++++++++++++++++++++++++++++++++-------
 1 file changed, 388 insertions(+), 83 deletions(-)

diff --git a/bootstrap-salt-minion.sh b/bootstrap-salt-minion.sh
index 3b4fb60..3b38181 100644
--- a/bootstrap-salt-minion.sh
+++ b/bootstrap-salt-minion.sh
@@ -1,98 +1,403 @@
-#!/bin/sh
-LOGFILE=/var/log/bootstrap-salt-minion.log 
+#!/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
 
-log() {
-    message="$@"
-    echo $message
-    echo $message >>$LOGFILE
-}
+ScriptVersion="1.0"
+NOCOLORS=0
 
-UNAME=`uname`
-if [ "$UNAME" != "Linux" ] ; then
-    log "Sorry, this OS is not supported."
+
+# 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
 
-set -e
-trap "echo Installation failed." EXIT
+# 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
 
-if [ "$UNAME" = "Linux" ] ; then
-    do_with_root() {
-        if [ `whoami` = 'root' ] ; then
-            RESULT=$($*)
-            log $RESULT
-        else
-            log "Salt requires root privileges to install. Please re-run this script as root."
-            exit 1
-        fi
-    }
- 
-    if [ -f /etc/lsb-release ] ; then
-        OS=$(lsb_release -si)
-        CODENAME=$(lsb_release -sc)
+# 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
 
-        if [ $OS = 'Ubuntu' ]; then
-            if [ $CODENAME = 'oneiric' ]; then
-                log "Installing for Ubuntu Oneiric."
-                do_with_root apt-get update
-                do_with_root apt-get -y install python-software-properties
-                do_with_root add-apt-repository -y 'deb http://us.archive.ubuntu.com/ubuntu/ oneiric universe'
-                do_with_root add-apt-repository -y ppa:saltstack/salt
-                do_with_root apt-get update
-                do_with_root apt-get -y install msgpack-python salt-minion
-                do_with_root add-apt-repository -y --remove 'deb http://us.archive.ubuntu.com/ubuntu/ oneiric universe'
-            elif [ $CODENAME = 'lucid' -o $CODENAME = 'precise' ]; then
-                log "Installing for Ubuntu Lucid/Precise."
-                do_with_root apt-get update
-                do_with_root apt-get -y install python-software-properties
-                if [ $CODENAME = 'lucid' ]; then
-                    do_with_root add-apt-repository ppa:saltstack/salt
-                else
-                    do_with_root add-apt-repository -y ppa:saltstack/salt
-                fi
-                do_with_root apt-get update
-                do_with_root apt-get -y install salt-minion
-            else
-                log "Ubuntu $CODENAME is not supported."
-                exit 1
-            fi
-        elif [ $OS = 'Debian' ]; then
-            if [ $CODENAME = 'wheezy' -o $CODENAME = 'jessie' ]; then
-                log "Installing for Debian Weezy/Jessie."
-                do_with_root apt-get -y install salt-minion
-            else
-                log "Debian $CODENAME is not supported."
-                exit 1
-            fi
-        else
-            log "Debian variant $OS not supported."
+# 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
-    elif [ -f /etc/debian_version ] ; then
-        DVER=$(cat /etc/debian_version)
-        if [ $DVER = '6.0'  ]; then
-            log "Installing for Debian Squeeze."
-            do_with_root echo "deb http://backports.debian.org/debian-backports squeeze-backports main" >> /etc/apt/sources.list.d/backports.list
-            do_with_root apt-get update
-            do_with_root apt-get -t squeeze-backports -y install salt-minion
-        else
-            log "Debian version $VER not supported."
+    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
-    elif [ -f /etc/redhat-release ] ; then
-        OS=$(cat /etc/redhat-release)
-        CODENAME=$(cat /etc/redhat-release)
-        log "Installing for Redhat/CentOS"
-        do_with_root rpm -Uvh --force http://mirrors.kernel.org/fedora-epel/6/x86_64/epel-release-6-7.noarch.rpm
-        do_with_root yum update
-        do_with_root yum -y install salt-minion --enablerepo=epel-testing
-        do_with_root /sbin/chkconfig salt-minion on
-        do_with_root salt-minion start
-    else
-        log "Unable to install. Bootstrapping only supported on *Some OS.*"
+    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
 
-log "Salt has been installed!"
-trap - EXIT
+# 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}"