hpm.linux

#!/bin/bash

platform='unknown'

PROTO="https"
HCORP_RELEASE="releases.hashicorp.com"
HCORP_DIR="${HOME}/.hpm/hashicorp"
HCORP_PRODUCTS=("terraform" "vault" "consul" "packer" "nomad" "serf" "sentinel" "vagrant")
PACKAGED_PRODUCTS=("vagrant")

DEBUG=0
SHOW_VERSION=1
TIMEOUT=10

bold=$(tput bold)
normal=$(tput sgr0)

function get_os() {
  case "${OSTYPE}" in
    solaris*) echo "solaris" ;;
    darwin*)  echo "osx" ;; 
    linux*)   echo "linux" ;;
    bsd*)     echo "freebsd" ;;
    msys*)    echo "windows" ;;
    *)        echo "unknown: ${OSTYPE}" ;;
  esac
}

function get_arch() {
  arch=$(uname -m)
  case "${arch}" in
    x86_64*) echo "amd64" ;;
    x86_32*) echo "x86" ;;
    *)       echo "unknown: ${arch}" ;;
  esac
}

function test_depends() {
  local current_os=${1}
  case "${current_os}" in
    solaris*) echo "solaris todo" ;;
    darwin*)  echo "install chtf formula with hombrew : brew instaff chtf"; exit 0 ;; 
    linux*)
      if [[ ! -x /usr/bin/unzip ]]; then
        echo "install unzip"
        exit 65
      fi
      if [[ ! -x /usr/bin/wget ]]; then
        echo "install wget"
        exit 65
      fi
    ;;
    bsd*)     echo "freebsd todo" ;;
    msys*)    echo "windows is not supported." && exit 666 ;;
    *)        echo "unknown: ${current_os}" ;;
  esac
}

function list_installed_releases() {
  local p=${1}
  local c=${2}
  if [[ -d ${HCORP_DIR}/${p} ]]
  then
    for rel in `find ${HCORP_DIR}/${p} -mindepth 1 -maxdepth 1 -type d|sort --version-sort`
    do
      release=$(echo ${rel} | awk -F\/ '{print $NF}')
      if [[ "${release}" == "${c}" ]]
      then
        echo " * ${release}"
      else
        echo "   ${release}"
      fi
    done
  fi
}

function get_activ_release() {
  local p=${1}
  local activ=""
  if [[ -L "${HOME}/bin/${p}" ]]
  then
    activ=$(ls -l ${HOME}/bin/${p} |awk '{print $NF}'|awk -F\/ '{print $5}')
  fi
  echo ${activ}
}

function check_release_is_installed() {
  local p=${1}
  local r=${2}

  if [[ " ${PACKAGED_PRODUCTS[@]} " =~ " ${p} " ]]
  then
    check_release_is_installed_from_package ${p} ${r}
  else
    check_release_is_installed_from_archive ${p} ${r}
  fi
}

function check_release_is_installed_from_archive() {
  local p=${1}
  local r=${2}
  if [[ ! -x ${HCORP_DIR}/${p}/${r}/${p} ]]
  then
    echo "0"
  else
    echo "1"
  fi
}

function check_release_is_installed_from_package() {
  local d=$(get_distribution)

  case "${d}" in
    debian|ubuntu)
        #dpkg -l vagrant|grep vagrant|awk '{print $3}'
        res=$(dpkg -l ${p}|grep -c ${p})
      ;;
    redhat|centos|fedora)
        res=$(rpm -qa|grep -c ${p})
      ;; 
  esac
  echo ${res}
}

function get_releases_url() {
  local p=${1}
  local releases_url=$(printf "%s://%s/%s/%s" ${PROTO} ${HCORP_RELEASE} ${p}) 
  echo ${releases_url}
}

function get_archive_url() {
  local p=${1}
  local r=${2}
  local z=${3}
  local archive_url=$(printf "%s://%s/%s/%s/%s" ${PROTO} ${HCORP_RELEASE} ${p} ${r} ${z}) 
  echo ${archive_url}
}

# https://gist.github.com/hrwgc/7455343
#if [[ `wget -S --spider ${url} 2>&1 | grep 'HTTP/1.1 200 OK'` ]]; then echo "true"; else echo "false"; fi
function validate_url() {
  local url=${1}  
  if [[ ${DEBUG} -eq "1" ]]; then
      echo "wget --no-hsts --timeout=${TIMEOUT} --server-response --spider ${url} 2>&1|grep -c 'HTTP/1.1 200 OK'"
  fi

  local valid=$(wget --no-hsts --timeout=${TIMEOUT} --server-response --spider ${url} 2>&1|grep -c 'HTTP/1.1 200 OK')

  if [[ ${valid} -eq "0" ]]; then
    echo "This release is not available."
    exit 0
  fi
}

function get_distribution() {
    distrib=$(lsb_release -i -s|tr '[:upper:]' '[:lower:]')
    echo ${distrib}
}

function get_archive_extension() {
  local p=${1}
  local d=$(get_distribution)

  if [[ ! " ${PACKAGED_PRODUCTS[@]} " =~ " ${p} " ]]
  then
    echo "zip"
  else
    case "${d}" in
      debian*) echo "deb" ;;
      ubuntu*) echo "deb" ;;
      redhat*) echo "rpm" ;; 
      centos*) echo "rpm" ;; 
      fedora*) echo "rpm" ;; 
      *)       echo "tar.xz" ;;
    esac
  fi
}

function get_archive_name() {
  local p=${1}
  local r=${2}
  local o=${3}
  local a=${4}
  local e=${5}

  if [[ " ${PACKAGED_PRODUCTS[@]} " =~ " ${p} " ]]
  then
    if [[ ${a} -eq "amd64" ]]
    then
      na="x86_64"
    else
      na="i686"
    fi
    local archive_name=$(printf "%s_%s_%s.%s" ${p} ${r} ${na} ${e}) 
  else
    local archive_name=$(printf "%s_%s_%s_%s.%s" ${p} ${r} ${o} ${a} ${e}) 
  fi

  echo ${archive_name}
}

function wget_archive() {
	local url=${1}
	local output=${2}
	if [[ ${DEBUG} -eq "1" ]]; then
		#echo "wget --no-hsts --quiet ${url} --output-document=/tmp/${output}"
		echo "wget --show-progress --progress=bar --no-hsts ${url} --output-document=/tmp/${output}"
	fi
	#wget --no-hsts --quiet ${url} --output-document=/tmp/${output}
	wget --show-progress --progress=bar --no-hsts ${url} --output-document=/tmp/${output}
}

function get_release_folder() {
  local p=${1}
  local r=${2}
  local folder=$(printf "%s/%s/%s" ${HCORP_DIR} ${p} ${r})
  echo ${folder}
}

function create_release_folder() {
  local f=${1}
  mkdir -p ${f}
}

function unzip_archive() {
  local z=${1}
  local d=${2}
  if [[ ${DEBUG} -eq "1" ]]; then
    echo "unzip -qq /tmp/${z} -d ${d}/"
    echo "rm -f /tmp/${z}"
  fi
  unzip -qq /tmp/${z} -d ${d}/
  rm -f /tmp/${z}
}


function install_package() {
  local d=${1}
  local p=${2}

  case "${d}" in
    debian|ubuntu)
        dpkg -i /tmp/${p}
        apt-get -f install
      ;;
    redhat|centos|fedora)
        rpm -ivh /tmp/${p}
      ;; 
  esac
}

function create_link() {
  local p=${1}
  local d=${2}
  mkdir -p /${HOME}/bin
  rm -f ${HOME}/bin/${p}
  ln -s ${d}/${p} ${HOME}/bin/${p}
}

function run_version() {
  local p=${1}
  ${p} --version
}

function show_help() {
  echo ""
  echo "${bold}Hashicorp Product Manager${normal} aka hpm"
  echo ""
  echo "Usage :"
  echo ""
  echo "hpm <product> [release]"
  echo ""
  echo "Where ${bold}product${normal} can be :"
  for p in "${HCORP_PRODUCTS[@]}"
  do
    echo " - ${p}"
  done
  echo ""
  echo "And ${bold}release${normal} the version tou want to be activated"
  echo ""
}

if [ $# -eq 0 ]
then
  show_help
  exit 0
fi


if [ $# -eq 1 ]
then
  product=${1}
  distribution=$(lsb_release -i -s|tr '[:upper:]' '[:lower:]')

  if [[ ! " ${HCORP_PRODUCTS[@]} " =~ " ${product} " ]]
  then
    echo "Product ${product} not found or not supporeted. Sorry."
    exit 100
  fi

  active_release=$(get_activ_release ${product})
  list_installed_releases ${product} ${active_release}
    
  url_releases=$(get_releases_url ${product} ${release})
  echo ""
  echo "To get avaliable releases check this link :"
  echo " - ${url_releases}"
  echo ""
  exit 0
fi

product=${1}
release=${2}
current_os=$(get_os)
current_arch=$(get_arch)

if [[ ${DEBUG} -eq "1" ]]; then
  echo ${current_os}
  echo ${current_arch}
fi

test_depends ${current_os}

distrib=$(get_distribution)
archive_ext=$(get_archive_extension ${product})

if [[ ${DEBUG} -eq "1" ]]; then
  echo ${distrib}
  echo ${archive_ext}
fi

if [[ ! " ${PACKAGED_PRODUCTS[@]} " =~ " ${product} " ]]
then
  release_folder=$(get_release_folder ${product} ${release})
else
  release_folder=""
fi
release_is_installed=$(check_release_is_installed ${product} ${release})

if [[ ${DEBUG} -eq "1" ]]; then
  echo ${release_folder}
  echo ${release_is_installed}
fi

if [[ "${release_is_installed}" -ne "1" ]]
then
  archive_name=$(get_archive_name ${product} ${release} ${current_os} ${current_arch} ${archive_ext})
  archive=$(get_archive_url ${product} ${release} ${archive_name})
  url_exists=$(validate_url ${archive})

  if [[ ${DEBUG} -eq "1" ]]; then
    echo ${archive_name}
    echo ${archive}
    echo ${url_exists}
  fi

  wget_archive ${archive} ${archive_name}

  if [[ " ${PACKAGED_PRODUCTS[@]} " =~ " ${product} " ]]
  then
    install_package ${distrib} ${archive_name}
  else
    create_release_folder ${release_folder}
    unzip_archive ${archive_name} ${release_folder}
  fi
fi

if [[ ! " ${PACKAGED_PRODUCTS[@]} " =~ " ${product} " ]]
then
  create_link ${product} ${release_folder}
fi

if [[ ${SHOW_VERSION} -eq "1" ]]; then
  run_version ${product}
fi

exit 0