brewstrap_Linux.sh

#!/usr/bin/env bash

BREWSTRAPRC="${HOME}/.brewstraprc"
WORK_DIR="/tmp/${USER}-brewstrap"
RVM_URL="https://raw.github.com/wayneeseguin/rvm/master/binscripts/rvm-installer"
RVM_MIN_VERSION="185"
RBENV_RUBY_VERSION="1.9.3-p125"
RVM_RUBY_VERSION="ruby-1.9.3-p125"
CHEF_MIN_VERSION="10.12.0"
CHEF_LIBRARIAN_MIN_VERSION="0.0.24"
ORIGINAL_PWD=`pwd`
GIT_PASSWORD_SCRIPT="${WORK_DIR}/retrieve_git_password.sh"
RUBY_RUNNER=""
RUBY_GEM_OPTIONS="--no-ri --no-rdoc"
USING_RVM=0
USING_RBENV=0
TOTAL=11
STEP=1
clear

GIT_DEBUG=""
if [ ! -z ${DEBUG} ]; then
  GIT_DEBUG="--verbose --progress"
fi

function print_step() {
  echo -e "\033[1m($(( STEP++ ))/${TOTAL}) ${1}\033[0m\n"
}

function print_warning() {
  echo -e "\033[1;33m${1}\033[0m\n"
}

function print_error() {
  echo -e "\033[1;31m${1}\033[0m\n"
  exit 1
}

echo -e "\033[1m\nStarting brewstrap...\033[0m\n"
echo -e "\n"
echo -e "Brewstrap will make sure your machine is bootstrapped and ready to run chef"
echo -e "by making sure GCC, git and rbenv/RVM and chef are installed. From there it will"
echo -e "kick off a chef-solo run using whatever chef repository of cookbooks you point it at."
echo -e "\n"
echo -e "It expects the chef repo to exist as a public or private repository on github.com"
echo -e "You will need your github credentials so now might be a good time to login to your account."

[[ -s "$BREWSTRAPRC" ]] && source "$BREWSTRAPRC"

if [ -e .rvmrc ]; then
  print_error "Do not run brewstrap from within a directory with an existing .rvmrc!\nIt causes the wrong environment to load."
fi

if [ ! -d $WORK_DIR ]; then
  mkdir -p $WORK_DIR
fi
if [ ! -x $WORK_DIR ]; then
  print_error "Unable to access ${WORK_DIR}! Permissions problem?"
fi

if [ -d /tmp/chef ]; then
  print_step "Found old brewstrap directory, removing and symlinking to ${WORK_DIR}/chef"
  rm -rf /tmp/chef && ln -s ${WORK_DIR}/chef /tmp/chef
fi

print_step "Collecting information.."
if [ -z $GITHUB_LOGIN ]; then
  echo -n "Github Username: "
  stty echo
  read GITHUB_LOGIN
  echo ""
fi

if [ -z $GITHUB_PASSWORD ]; then
  echo -n "Github Password: "
  stty -echo
  read GITHUB_PASSWORD
  echo ""
fi

if [ -z $CHEF_REPO ]; then
  echo -n "Chef Repo (Take the github HTTP URL): "
  stty echo
  read CHEF_REPO
  echo ""
fi
stty echo

rm -f $BREWSTRAPRC
echo "GITHUB_LOGIN=${GITHUB_LOGIN}" >> $BREWSTRAPRC
echo "GITHUB_PASSWORD=${GITHUB_PASSWORD}" >> $BREWSTRAPRC
echo "CHEF_REPO=${CHEF_REPO}" >> $BREWSTRAPRC
chmod 0600 $BREWSTRAPRC

GCC_BIN=`which gcc`
if [ $? != 0 ]; then
  print_error "There is no GCC available. Please install before continuing."
  exit 1
else
  print_step "GCC is already installed"
fi

MAKE_BIN=`which make`
if [ $? != 0 ]; then
  print_error "There is no make available. Please install before continuing."
  exit 1
else
  print_step "make is already installed"
fi

GIT_PATH=`which git`
if [ $? != 0 ]; then
  print_error "There is no git available. Please install before continuing."
else
  print_step "git is already installed"
fi

if [ $RVM ]; then
  RUBY_RUNNER="rvm ${RVM_RUBY_VERSION} exec"
  if [ ! -e ~/.rvm/bin/rvm ]; then
    print_step "Installing RVM (Forced)"
    bash -s stable < <( curl -fsSL ${RVM_URL} )
    if [ ! $? -eq 0 ]; then
      print_error "Unable to install RVM!"
    fi
  else
    RVM_VERSION=`~/.rvm/bin/rvm --version | cut -f 2 -d ' ' | head -n2 | tail -n1 | sed -e 's/\.//g'`
    if [ "${RVM_VERSION}0" -lt "${RVM_MIN_VERSION}0" ]; then
      print_step "RVM needs to be upgraded..."
      ~/.rvm/bin/rvm get 1.8.5
    else
      print_step "RVM already installed"
    fi
  fi

  [[ -s "$HOME/.rvm/scripts/rvm" ]] && source "$HOME/.rvm/scripts/rvm"

  if [ ! -e ~/.bash_profile ]; then
      echo "[[ -s \"\$HOME/.rvm/scripts/rvm\" ]] && source \"\$HOME/.rvm/scripts/rvm\"" > ~/.bash_profile
  fi

  rvm list | grep ${RVM_RUBY_VERSION}
  if [ $? -gt 0 ]; then
    print_step "Installing RVM Ruby ${RVM_RUBY_VERSION}"
    gcc --version | head -n1 | grep llvm >/dev/null
    if [ $? -eq 0 ]; then
      export CC="gcc-4.2"
    fi
    rvm install ${RVM_RUBY_VERSION}
    unset CC
    if [ ! $? -eq 0 ]; then
      print_error "Unable to install RVM ${RVM_RUBY_VERSION}"
    fi
  else
    print_step "RVM Ruby ${RVM_RUBY_VERSION} already installed"
  fi
  USING_RVM=1
else
  if [ ! -d ~/.rvm ]; then
    if [ ! -d /usr/local/rvm ]; then
      print_step "Found no RVM on the system, installing rbenv by default. If you wish to use RVM instead, please re-run with RVM=true"
      if [ ! -d ~/.rbenv ]; then
        cd ~/ && git clone ${GIT_DEBUG} git://github.com/sstephenson/rbenv.git .rbenv
      fi
      unset GEM_PATH
      unset GEM_HOME
      unset MY_RUBY_HOME
      (echo $PATH | grep "rbenv") || (test -e ~/.bash_profile  && cat ~/.bash_profile | grep PATH | grep rbenv) || false
      if [ $? -eq 1 ]; then
        echo 'export PATH="$HOME/.rbenv/bin:$PATH"' >> ~/.bash_profile
      fi
      grep "rbenv init" ~/.bash_profile
      if [ $? -eq 1 ]; then
        echo "eval \"\$(rbenv init -)\"" >> ~/.bash_profile
      fi
      export PATH=$HOME/.rbenv/bin:$PATH
      if [ ! -d ~/.rbenv/plugins/ruby-build ]; then
        mkdir -p ~/.rbenv/plugins && cd ~/.rbenv/plugins && git clone $GIT_DEBUG git://github.com/sstephenson/ruby-build.git
      fi
      gcc --version | head -n1 | grep llvm >/dev/null
      if [ $? -eq 0 ]; then
        export CC="gcc-4.2"
      fi
      which rbenv
      if [ $? -eq 1 ]; then
        print_error "Unable to find rbenv in ${PATH} !"
        exit 1
      fi
      rbenv versions | grep ${RBENV_RUBY_VERSION}
      if [ ! $? -eq 0 ]; then
        rbenv install ${RBENV_RUBY_VERSION}
        rbenv rehash
        if [ ! $? -eq 0 ]; then
          print_error "Unable to install rbenv or ruby ${RBENV_RUBY_VERSION}!"
        fi
      fi
      USING_RBENV=1
      RUBY_RUNNER=""
      eval "$(rbenv init -)"
      rbenv shell ${RBENV_RUBY_VERSION}
    else
      print_step "Found multi-user RVM installation. Skipping installation..."
      RUBY_RUNNER="rvm ${RVM_RUBY_VERSION} exec"
      USING_RVM=1
    fi
  else
    print_step "Found local user RVM installation. Skipping installation..."
    RUBY_RUNNER="rvm ${RVM_RUBY_VERSION} exec"
    USING_RVM=1
  fi
fi
${RUBY_RUNNER} gem specification --version ">=${CHEF_MIN_VERSION}" chef 2>&1 | awk 'BEGIN { s = 0 } /^name:/ { s = 1; exit }; END { if(s == 0) exit 1 }'
if [ $? -gt 0 ]; then
  print_step "Installing chef gem"
  ${RUBY_RUNNER} gem install chef --version "${CHEF_MIN_VERSION}" ${RUBY_GEM_OPTIONS}
  if [ $USING_RBENV -eq 1 ]; then
    print_step "Rehashing RBEnv for chef"
    rbenv rehash
  fi
  if [ ! $? -eq 0 ]; then
    print_error "Unable to install chef!"
  fi
else
  print_step "Chef already installed"
fi

if [ ! -f ${GIT_PASSWORD_SCRIPT} ]; then
  echo "grep PASSWORD ~/.brewstraprc | sed s/^.*=//g" > ${GIT_PASSWORD_SCRIPT}
  chmod 700 ${GIT_PASSWORD_SCRIPT}
fi

export GIT_ASKPASS=${GIT_PASSWORD_SCRIPT}

if [ -d ${WORK_DIR}/chef ]; then
  if [ ! -d ${WORK_DIR}/chef/.git ]; then
    print_step "Existing git repo bad? Attempting to remove..."
    rm -rf ${WORK_DIR}/chef
  fi
fi

if [ ! -d ${WORK_DIR}/chef ]; then
  if [ ! -z ${GITHUB_LOGIN} ]; then
    CHEF_REPO=`echo ${CHEF_REPO} | sed -e "s|https://github.com|https://${GITHUB_LOGIN}@github.com|"`
  fi
  print_step "Cloning chef repo (${CHEF_REPO})"
  git clone ${GIT_DEBUG} ${CHEF_REPO} ${WORK_DIR}/chef

  if [ ! $? -eq 0 ]; then
    print_error "Unable to clone repo!"
  fi
  print_step "Updating submodules..."
  if [ -e ${WORK_DIR}/chef/.gitmodules ]; then
    if [ ! -z ${GITHUB_LOGIN} ]; then
      sed -i -e "s|https://github.com|https://${GITHUB_LOGIN}@github.com|" ${WORK_DIR}/chef/.gitmodules
    fi
  fi
  cd ${WORK_DIR}/chef && git submodule update --init
  if [ ! $? -eq 0 ]; then
    print_error "Unable to update submodules!"
  fi
else
  if [ -z ${LOCAL} ]; then
    print_step "Updating chef repo"
    if [ -e ${WORK_DIR}/chef/.rvmrc ]; then
      rvm rvmrc trust ${WORK_DIR}/chef/
    fi
    cd ${WORK_DIR}/chef && git pull && git submodule update --init
    if [ ! $? -eq 0 ]; then
      print_error "Unable to update repo!"
    fi
  else
    print_step "Using local chef repo"
  fi
fi

unset GIT_ASKPASS

if [ ! -e ${WORK_DIR}/chef/node.json ]; then
  print_error "The chef repo provided has no node.json at the toplevel. This is required to know what to run."
fi

if [ ! -e ${WORK_DIR}/chef/solo.rb ]; then
  print_warning "No solo.rb found, writing one..."
  echo "file_cache_path '${WORK_DIR}/chef-solo-brewstrap'" > ${WORK_DIR}/chef/solo.rb
  echo "cookbook_path '${WORK_DIR}/chef/cookbooks'" >> ${WORK_DIR}/chef/solo.rb
fi

if [ -e ${WORK_DIR}/chef/Cheffile ]; then
  print_step "Cheffile detected, checking for librarian"
  ${RUBY_RUNNER} gem specification --version ">=${CHEF_LIBRARIAN_MIN_VERSION}" librarian 2>&1 | awk 'BEGIN { s = 0 } /^name:/ { s = 1; exit }; END { if(s == 0) exit 1 }'
  if [ $? -gt 0 ]; then
    print_step "Installing librarian chef gem"
    ${RUBY_RUNNER} gem install librarian ${RUBY_GEM_OPTIONS}
    if [ $USING_RBENV -eq 1 ]; then
      print_step "Rehasing RBEnv for librarian chef"
      rbenv rehash
    fi
    if [ ! $? -eq 0 ]; then
      print_error "Unable to install librarian chef!"
    fi
  else
    print_step "Librarian Chef already installed"
  fi
  if [ -e ${WORK_DIR}/chef/Gemfile ]; then
    print_step "Installing bundler gem"
    ${RUBY_RUNNER} gem install bundler ${RUBY_GEM_OPTIONS}
    if [ $USING_RBENV -eq 1 ]; then
      print_step "Rehasing RBEnv for bundler"
      rbenv rehash
    fi
    if [ ! $? -eq 0 ]; then
      print_error "Unable to install bundler!"
    fi
    print_step "Bundler already installed"
  fi
  print_step "Kicking off libarian chef"
  BUNDLER_COMMAND="${RUBY_RUNNER} bundle install --without development"
  env ${BUNDLER_COMMAND}
  LIBRARIAN_COMMAND="${RUBY_RUNNER} librarian-chef install --clean"
  env ${LIBRARIAN_COMMAND}
fi

print_step "Kicking off chef-solo (password will be your local user password)"
echo $RUBY_RUNNER | grep "&&"
if [ $? -eq 0 ]; then
  RUBY_RUNNER=""
fi

CHEF_DEBUG=""
if [ ! -z ${DEBUG} ]; then
  CHEF_DEBUG="-l debug"
fi

CHEF_COMMAND="GITHUB_PASSWORD=$GITHUB_PASSWORD GITHUB_LOGIN=$GITHUB_LOGIN ${RUBY_RUNNER} chef-solo -j ${WORK_DIR}/chef/node.json -c ${WORK_DIR}/chef/solo.rb ${CHEF_DEBUG}"
if [ ! -z ${DEBUG} ]; then
  echo $CHEF_COMMAND
fi
sudo -E env ${CHEF_COMMAND}
if [ ! $? -eq 0 ]; then
  print_error "BREWSTRAP FAILED!"
else
  print_step "BREWSTRAP FINISHED"
fi
cd $ORIGINAL_PWD

if [ -n "$PS1" ]; then
  exec bash --login
fi