Install Docker Machine with dependencies

gistfile1.sh

#!/bin/bash
#
# A script for setting up a productive development environment with Docker
# on OS X. See https://github.com/brikis98/docker-osx-dev for more info.

set -e

# Environment variable file constants
readonly BASH_PROFILE="$HOME/.bash_profile"
readonly BASH_RC="$HOME/.bashrc"
readonly ZSH_RC="$HOME/.zshrc"

# Docker environment variable constants
readonly DOCKER_HOST_EXPORT="export DOCKER_HOST"
readonly DOCKER_HOST="tcp://localhost:2375"

# Url constants
readonly HOSTS_FILE="/etc/hosts"
readonly VAGRANT_HOST="192.168.99.100"
readonly DOCKER_HOST_NAME="docker.me"

# Console colors
readonly COLOR_INFO='\033[0;32m[INFO]'
readonly COLOR_WARN='\033[1;33m[WARN]'
readonly COLOR_ERROR='\033[0;31m[ERROR]'
readonly COLOR_INSTRUCTIONS='\033[0;37m[INSTRUCTIONS]'
readonly COLOR_END='\033[0m'

# Script constants
readonly BIN_DIR="/usr/local/bin"
readonly DOCKER_OSX_DEV_SCRIPT_NAME="docker-machine"
readonly DOCKER_OSX_DEV_URL="https://github.com/docker/machine/releases/download/v0.2.0/docker-machine_darwin-amd64"

function log_info {
  log "$1" $COLOR_INFO
}

function log_warn {
  log "$1" $COLOR_WARN
}

function log_error {
  log "$1" $COLOR_ERROR
}

function log_instructions {
  log "$1" $COLOR_INSTRUCTIONS
}

function log {
  local readonly message=$1
  local readonly color=$2 || $COLOR_INFO
  echo -e "${color} ${message}${COLOR_END}"
}

function brew_install {
  local readonly package_name="$1"
  local readonly readable_name="$2"
  local readonly command_name="$3"
  local readonly use_cask="$4"

  local brew_command="brew"
  if [[ "$use_cask" = true ]]; then
    brew_command="brew cask"
  fi

  if eval "$brew_command list $package_name" > /dev/null 2>&1 ; then
    log_warn "$readable_name is already installed by HomeBrew, skipping"
  elif type "$command_name" > /dev/null 2>&1 ; then
    log_warn "Found command $command_name, assuming $readable_name is already installed and skipping"
  else
    log_info "Installing $readable_name"
    eval "$brew_command install $package_name"
  fi
}

function vagrant_plugin_install {
  local readonly plugin_name=$1
  local readonly readable_name=$2

  if vagrant plugin list | grep -q $plugin_name ; then
    log_warn "$readable_name already installed, skipping"
  else
    log_info "Installing $readable_name"
    vagrant plugin install $plugin_name
  fi
}

function get_env_file {
  if [[ -f "$BASH_RC" ]]; then
    echo "$BASH_RC"
  elif [[ -f "$ZSH_RC" ]]; then
    echo "$ZSH_RC"
  else
    echo "$BASH_PROFILE"
  fi
}

function check_prerequisites {
  local readonly os=$(uname)

  if [[ ! "$os" = "Darwin" ]]; then
    log_error "This script should only be run on OS X"
    exit 1
  fi

  if ! type brew > /dev/null 2>&1 ; then
    log_error "This script requires HomeBrew, but it's not installed. Aborting."
    exit 1
  fi
}

function install_dependencies {
  log_info "Updating HomeBrew"
  brew update

  brew_install "caskroom/cask/brew-cask" "Cask" "" false
  brew_install "virtualbox" "VirtualBox" "vboxwebsrv" true
  brew_install "vagrant" "Vagrant" "vagrant" true
  brew_install "docker" "Docker" "docker" false
  brew_install "docker-compose" "Docker Compose" "docker-compose" false
}

function install_vagrant_plugins {
  vagrant_plugin_install vagrant-gatling-rsync "Vagrant Gatling Rsync"
}

function add_environment_variables {
  local readonly env_file=$(get_env_file)
  local readonly docker_host_export="\n# docker-osx-dev\n$DOCKER_HOST_EXPORT=$DOCKER_HOST"

  if grep -q "$DOCKER_HOST_EXPORT" "$env_file" ; then
    log_warn "$env_file already contains \"$DOCKER_HOST_EXPORT\", will not overwrite"
  else
    log_info "Adding \"$DOCKER_HOST_EXPORT\" to $env_file"
    echo -e "$docker_host_export" >> "$env_file"
    log_instructions "Please run the following command to pick up new environment variables: source $env_file"
  fi
}

function install_local_scripts {
  local readonly script_path="$BIN_DIR/$DOCKER_OSX_DEV_SCRIPT_NAME"
  if [[ -f "$script_path" ]]; then
    log_warn "$script_path already exists, will not overwrite"
  else
    log_info "Adding $script_path"
    curl -L "$DOCKER_OSX_DEV_URL" > "$script_path"
    chmod +x "$script_path"
  fi
}

function add_docker_host {
  local readonly host_entry="$VAGRANT_HOST $DOCKER_HOST_NAME"

  if grep -q "$DOCKER_HOST_NAME" "$HOSTS_FILE" ; then
    log_warn "$HOSTS_FILE already contains $DOCKER_HOST_NAME, will not overwrite"
  else
    log_info "Adding $DOCKER_HOST_NAME entry to $HOSTS_FILE so you can use http://$DOCKER_HOST_NAME URLs for testing"
    log_instructions "Modifying $HOSTS_FILE requires sudo privileges, please enter your password."
    sudo -k sh -c "echo $host_entry >> $HOSTS_FILE"
  fi
}

check_prerequisites
install_dependencies
install_vagrant_plugins
install_local_scripts
add_docker_host
add_environment_variables