-
-
Save Hagith/8e589a4ffd3091dea6cb219adcfb16d1 to your computer and use it in GitHub Desktop.
multi-distro puppet install script
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| # This script installs puppet on most linux distros. | |
| # To run, simply execute as root. | |
| # $ install_puppet.sh [ puppet_version ] | |
| # e.g. $ install_puppet.sh 3.4.1 | |
| # Successfully run on Debian 6/7, Ubuntu 12.04 LTS and 14.04 LTS, Fedora 20, and arch. | |
| # Not tested on OS X, Solaris, AIX, etc, simply because I lack test environments. | |
| # https://gist.github.com/danieldreier/8172bee4467127a992d8 | |
| set -e | |
| if [ -n "$1" ]; then | |
| puppet_version="$1" | |
| fi | |
| require_root_access() { | |
| if [[ $EUID -ne 0 ]]; then | |
| echo "ERROR: This script requires root access" | |
| exit 1 | |
| fi | |
| } | |
| # Check if a package is installed | |
| is_installed() { | |
| package="$1" | |
| detect_os | |
| case $DistroBasedOn in | |
| redhat) | |
| rpm -qi "$package" > /dev/null 2>&1 && return 0 | |
| ;; | |
| debian) | |
| dpkg-query --status "$package" >/dev/null 2>&1 && return 0 | |
| ;; | |
| arch) | |
| pacman -Qi "$package" >/dev/null 2>&1 && return 0 | |
| ;; | |
| *) | |
| echo "unknown package management system" | |
| ;; | |
| esac | |
| return 1 | |
| } | |
| # Check if the package is installed, and install it if not present | |
| ensure_package_present() { | |
| package="$1" | |
| is_installed "$package" && return 0 # return if it's installed already | |
| echo "installing $package and dependencies..." | |
| detect_os | |
| case $DistroBasedOn in | |
| redhat) | |
| yum -y --quiet install "$package" | |
| ;; | |
| debian) | |
| apt-get -qq update | |
| apt-get -qq install -y "$package" > /dev/null | |
| ;; | |
| arch) | |
| # Update the pacman repositories | |
| pacman -Sy | |
| # Install Ruby (arch uses gem install of puppet) | |
| pacman -S --noconfirm --needed "$package" | |
| ;; | |
| *) | |
| echo "Unable to install \"$package\": unknown package management system" | |
| ;; | |
| esac | |
| } | |
| lowercase(){ | |
| #echo "$1" | sed "y/ABCDEFGHIJKLMNOPQRSTUVWXYZ/abcdefghijklmnopqrstuvwxyz/" | |
| echo "$1" | tr "[:upper:]" "[:lower:]" | |
| } | |
| # Figure out what distribution or OS we're running on | |
| detect_os() { | |
| # Inspired by http://stackoverflow.com/questions/394230/detect-the-os-from-a-bash-script | |
| if [ -n "$DIST" ] ; then | |
| # Avoid running these checks repeatedly even if called again | |
| return | |
| fi | |
| # Identify major distro type | |
| if [ -f /etc/redhat-release ] ; then | |
| DistroBasedOn='RedHat' | |
| elif [ -f /etc/debian_version ] ; then | |
| DistroBasedOn='Debian' | |
| elif [ -f /etc/arch-release ] ; then | |
| DistroBasedOn='Arch' | |
| fi | |
| DistroBasedOn=$(lowercase $DistroBasedOn) | |
| # Determine further distro details | |
| # We take this approach to avoid forcing lsb-base package installs | |
| case $DistroBasedOn in | |
| redhat) | |
| DIST=$(cat /etc/redhat-release |sed s/\ release.*//) | |
| CODENAME=$(cat /etc/redhat-release | sed s/.*\(// | sed s/\)//) | |
| REV=$(cat /etc/redhat-release | sed s/.*release\ // | sed s/\ .*//) | |
| ;; | |
| debian) | |
| if hash lsb_release ; then | |
| DIST=$(lsb_release --id --short) | |
| CODENAME=$(lsb_release --codename --short) | |
| REV=$(lsb_release --release --short) | |
| elif [ -f /etc/lsb-release ] ; then | |
| DIST=$(grep '^DISTRIB_ID' /etc/lsb-release | awk -F= '{ print $2 }') | |
| CODENAME=$(grep '^DISTRIB_CODENAME' /etc/lsb-release | awk -F= '{ print $2 }') | |
| REV=$(grep '^DISTRIB_RELEASE' /etc/lsb-release | awk -F= '{ print $2 }') | |
| elif [ -f /etc/os-release ] ; then | |
| DIST=$(grep '^ID' /etc/os-release | awk -F= '{ print $2 }') | |
| CODENAME=$(grep '^VERSION=' /etc/os-release | cut -d '(' -f 2 | cut -d ')' -f 1) | |
| REV=$(grep '^VERSION_ID' /etc/os-release | awk -F= '{ print $2 }') | |
| else | |
| REV=$(cat /etc/debian_version) | |
| fi | |
| ;; | |
| arch) | |
| # Arch is rolling release based so revision numbers don't apply | |
| DIST=$DistroBasedOn | |
| ;; | |
| *) | |
| ;; | |
| esac | |
| MACH=$(uname -m) | |
| MAJOR_REV=$(echo $REV | cut -d '.' -f 1) | |
| } | |
| # Idempotent puppet and puppet repository installer | |
| ensure_puppet() { | |
| is_puppet_installed && echo "Puppet already installed" && return 0 | |
| ensure_package_present wget | |
| detect_os | |
| OS_DESCRIPTION="Detected ${DistroBasedOn}-based distro: $DIST $CODENAME $REV for $MACH" | |
| case $DistroBasedOn in | |
| redhat) | |
| echo "$OS_DESCRIPTION" | |
| REPO_BASE_URL="https://yum.puppetlabs.com/el/${MAJOR_REV}/products/${MACH}" | |
| if [ "$(lowercase $DIST)" == 'fedora' ]; then | |
| REPO_BASE_URL="https://yum.puppetlabs.com/fedora/f${MAJOR_REV}/products/${MACH}" | |
| ensure_package_present 'gnupg' | |
| fi | |
| # Figure out what the most recent puppetlabs-release rpm is | |
| TEMPLIST=$(mktemp) | |
| wget --convert-links --output-document="$TEMPLIST" "$REPO_BASE_URL" | |
| REPO_URL="$(wget --spider --force-html -i "$TEMPLIST" 2>&1 | grep --only-matching 'https://yum.puppetlabs.com/.*/puppetlabs-release-[0-9]*-[0-9]*.noarch.rpm' | sort | head -n 1)" | |
| #echo "wget --spider --force-html -i "$TEMPLIST" 2>&1 | grep --only-matching 'https://yum.puppetlabs.com/.*/puppetlabs-release-[0-9]*-[0-9]*.noarch.rpm' | sort | head -n 1" | |
| rm "$TEMPLIST" | |
| # Install GPG key | |
| if ! rpm -qi gpg-pubkey-4bd6ec30-4ff1e4fa > /dev/null 2>&1 ; then | |
| gpg_key=$(mktemp) | |
| gpg --recv-key --trust-model direct --keyserver pool.sks-keyservers.net 4BD6EC30 | |
| gpg --export --armor 47B320EB4C7C375AA9DAE1A01054B7A24BD6EC30 > $gpg_key | |
| echo "Installing puppetlabs GPG key" | |
| rpm --import $gpg_key | |
| rm $gpg_key | |
| rpm -qi gpg-pubkey-4bd6ec30-4ff1e4fa > /dev/null 2>&1 && echo "Puppet GPG key installed successfully" || echo "Error: Puppet GPG key install failed" | |
| fi | |
| if ! is_installed puppetlabs-release ; then | |
| # Install puppet yum repository | |
| echo "installing puppetlabs yum repository from $REPO_URL" | |
| repo_path="$(mktemp).rpm" | |
| wget --quiet --output-document=${repo_path} ${REPO_URL} | |
| yum -y --quiet install $repo_path | |
| rm $repo_path | |
| is_installed puppetlabs-release && echo "Puppetlabs yum repo installed sucessfully" || echo "Error: puppetlabs repo install failed" | |
| fi | |
| ensure_package_present 'ruby-devel' #needed for librarian-puppet and many others | |
| # Install puppet itself | |
| if [ -n "$puppet_version" ]; then | |
| puppet_package_name="puppet-${puppet_version}" | |
| else | |
| puppet_package_name="puppet" | |
| fi | |
| ensure_package_present "$puppet_package_name" | |
| ;; | |
| debian) | |
| echo "$OS_DESCRIPTION" | |
| # Ensure the puppet repository is available | |
| if ! is_installed "puppetlabs-release-pc1" ; then | |
| echo "Puppetlabs repository not detected. Installing..." | |
| REPO_URL="https://apt.puppetlabs.com/puppetlabs-release-pc1-${CODENAME}.deb" | |
| repo_path=$(mktemp) | |
| wget --quiet --output-document=${repo_path} ${REPO_URL} | |
| dpkg --install ${repo_path} | |
| apt-get -qq update | |
| rm ${repo_path} | |
| is_installed "puppetlabs-release-pc1" && echo "puppetlabs repository installed successfully" | |
| else | |
| echo "puppet repo already installed" | |
| fi | |
| # Install puppet | |
| if [ -n "$puppet_version" ]; then | |
| puppet_package_name="puppet=${puppet_version}-1puppetlabs1" | |
| else | |
| puppet_package_name="puppet" | |
| fi | |
| apt-get -qq update | |
| apt-get -qq install -y ${puppet_package_name} > /dev/null | |
| ensure_gem 'librarian-puppet' | |
| ;; | |
| arch) | |
| echo "detected arch distro" | |
| echo "always installs latest version of puppet - version parameter ignored" | |
| # Install Puppet and Facter as ruby gems | |
| ensure_gem puppet | |
| ensure_gem facter | |
| ensure_group puppet | |
| ;; | |
| *) | |
| echo "Unsupported OS - cannot install puppet" | |
| return 1 | |
| ;; | |
| esac | |
| } | |
| is_gem_installed() { | |
| gem list | grep -q "${gem} (" | |
| } | |
| ensure_gem() { | |
| gem="$1" | |
| ensure_package_present ruby | |
| if ! is_gem_installed "$gem" ; then | |
| gem install "$gem" --no-ri --no-rdoc --no-user-install | |
| is_gem_installed "$gem" || exit 1 | |
| fi | |
| } | |
| ensure_group() { | |
| [ -z "$1" ] && exit 1 | |
| group="$1" | |
| if ! getent group "$group" > /dev/null 2>&1 ; then | |
| groupadd "$group" | |
| fi | |
| } | |
| # Both facter and puppet need to be available | |
| is_puppet_installed() { | |
| ( hash puppet && hash facter && hash librarian-puppet) > /dev/null 2>&1 | |
| } | |
| require_root_access | |
| # Install puppet if it's not installed already | |
| if ! is_puppet_installed; then | |
| echo "No existing puppet install detected" | |
| echo "Preparing to install puppet and facter" | |
| ensure_puppet | |
| fi | |
| # Verify puppet install before telling user it's installed | |
| echo "Verifying that puppet and facter installed successfully..." | |
| if is_puppet_installed ; then | |
| echo "Detected puppet version $(puppet -V)" | |
| echo "Detected facter version $(facter --version)" | |
| echo "Detected librarian-puppet version $(librarian-puppet version)" | |
| echo "Puppet and facter installed successfully" | |
| else | |
| echo "Error: puppet and/or facter not detected following install attempt" | |
| exit 1 | |
| fi |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment