Lisk - Linux script for installation

installLisk.sh

#!/bin/bash
#############################################################
# Lisk Installation Script                                  #
# by: Isabella Dell                                         #
# Date: 15/05/2016                                          #
# contribs: Walker de Alencar [4827888033055033464L]        #
#############################################################

#Variable Declaration
UNAME=$(uname)-$(uname -m)
defaultLiskLocation=~
defaultRelease=main

export LC_ALL=en_US.UTF-8
export LANG=en_US.UTF-8
export LANGUAGE=en_US.UTF-8

#Verification Checks
if [ "$USER" == "root" ]; then
  echo "Error: Lisk should not be installed be as root. Exiting."
  exit 1
fi

#Adding LC_ALL LANG and LANGUAGE to user profile
if [[ -f ~/.profile && ! "$(grep "en_US.UTF-8" ~/.profile)" ]]; then
  echo "LC_ALL=en_US.UTF-8" >> ~/.profile
  echo "LANG=en_US.UTF-8"  >> ~/.profile
  echo "LANGUAGE=en_US.UTF-8"  >> ~/.profile
elif [[ -f ~/.bash_profile && ! "$(grep "en_US.UTF-8" ~/.bash_profile)" ]]; then
  echo "LC_ALL=en_US.UTF-8" >> ~/.bash_profile
  echo "LANG=en_US.UTF-8"  >> ~/.bash_profile
  echo "LANGUAGE=en_US.UTF-8"  >> ~/.bash_profile
fi

user_prompts() {
  read -r -p "Where do you want to install Lisk to? (Default $defaultLiskLocation): " liskLocation
  liskLocation=${liskLocation:-$defaultLiskLocation}
  if [[ ! -r "$liskLocation" ]]; then
    echo "$liskLocation is not valid, please check and re-excute"
    exit 2;
  fi
  
  read -r -p "Would you like to install the Main or Test Client? (Default $defaultRelease): " release
  release=${release:-$defaultRelease}
  if [[ ! "$release" == "main" && ! "$release" == "test" ]]; then
    echo "$release is not valid, please check and re-excute"
    exit 2;
  fi
}

ntp_checks() {
  #Install NTP or Chrony for Time Management - Physical Machines only
  if [[ "$(uname)" == "Linux" ]]; then
    if [[ -f "/etc/debian_version" &&  ! -f "/proc/user_beancounters" ]]; then
      if sudo pgrep -x "ntpd" > /dev/null; then
        echo "√ NTP is running"
      else
        echo "X NTP is not running"
        read -r -n 1 -p "Would like to install NTP? (y/n): " $REPLY
        if [[  $REPLY =~ ^[Yy]$ ]]; then
          echo -e "\nInstalling NTP, please provide sudo password.\n"
          sudo apt-get install ntp -yyq
          sudo service ntp stop
          sudo ntpdate pool.ntp.org
          sudo service ntp start
          if sudo pgrep -x "ntpd" > /dev/null; then
            echo "√ NTP is running"
          else
            echo -e "\nLisk requires NTP running on Debian based systems. Please check /etc/ntp.conf and correct any issues."
            exit 0
          fi
        else
          echo -e "\nLisk requires NTP on Debian based systems, exiting."
          exit 0
        fi
      fi #End Debian Checks
    elif [[ -f "/etc/redhat-release" &&  ! -f "/proc/user_beancounters" ]]; then
      if sudo pgrep -x "ntpd" > /dev/null; then
        echo "√ NTP is running"
      else
        if sudo pgrep -x "chronyd" > /dev/null; then
          echo "√ Chrony is running"
        else
          echo "X NTP and Chrony are not running"
          read -r -n 1 -p "Would like to install NTP? (y/n): " $REPLY
          if [[  $REPLY =~ ^[Yy]$ ]]; then
            echo -e "\nInstalling NTP, please provide sudo password.\n"
            sudo yum -yq install ntp ntpdate ntp-doc
            sudo chkconfig ntpd on
            sudo service ntpd stop
            sudo ntpdate pool.ntp.org
            sudo service ntpd start
            if pgrep -x "ntpd" > /dev/null; then
              echo "√ NTP is running"
              else
              echo -e "\nLisk requires NTP running on Debian based systems. Please check /etc/ntp.conf and correct any issues."
              exit 0
            fi
          else
            echo -e "\nLisk requires NTP or Chrony on RHEL based systems, exiting."
            exit 0
          fi
        fi
      fi #End Redhat Checks
    elif [[ -f "/proc/user_beancounters" ]]; then
      echo "_ Running OpenVZ VM, NTP and Chrony are not required"
    fi
  elif [[ "$(uname)" == "FreeBSD" ]]; then
    if sudo pgrep -x "ntpd" > /dev/null; then
      echo "√ NTP is running"
    else
      echo "X NTP is not running"
      read -r -n 1 -p "Would like to install NTP? (y/n): " $REPLY
      if [[  $REPLY =~ ^[Yy]$ ]]; then
        echo -e "\nInstalling NTP, please provide sudo password.\n"
        sudo pkg install ntp
        sudo sh -c "echo 'ntpd_enable=\"YES\"' >> /etc/rc.conf"
        sudo ntpdate -u pool.ntp.org
        sudo service ntpd start
        if pgrep -x "ntpd" > /dev/null; then
          echo "√ NTP is running"
        else
          echo -e "\nLisk requires NTP running on FreeBSD based systems. Please check /etc/ntp.conf and correct any issues."
          exit 0
        fi
      else
        echo -e "\nLisk requires NTP FreeBSD based systems, exiting."
        exit 0
      fi
    fi #End FreeBSD Checks
  elif [[ "$(uname)" == "Darwin" ]]; then
    if pgrep -x "ntpd" > /dev/null; then
      echo "√ NTP is running"
    else
      sudo launchctl load /System/Library/LaunchDaemons/org.ntp.ntpd.plist
      sleep 1
      if pgrep -x "ntpd" > /dev/null; then
        echo "√ NTP is running"
      else
        echo -e "\nNTP did not start, Please verify its configured on your system"
        exit 0
      fi
    fi  #End Darwin Checks
  fi #End NTP Checks
}

install_lisk() {
  liskVersion=lisk-$UNAME.tar.gz
  liskDir=`echo $liskVersion | cut -d'.' -f1`

  echo -e "\nDownloading current Lisk binaries: "$liskVersion

  curl -Ls "https://downloads.lisk.io/lisk/$release/$liskVersion" -o $liskVersion

  curl -Ls "https://downloads.lisk.io/lisk/$release/$liskVersion.md5" -o $liskVersion.md5

  md5=`md5sum $liskVersion | awk '{print $1}'`
  md5_compare=`grep "$liskVersion" $liskVersion.md5 | awk '{print $1}'`

  if [[ "$md5" == "$md5_compare" ]]; then
    echo "Checksum Passed!"
  else
    echo "Checksum Failed, aborting installation"
    rm -f $liskVersion $liskVersion.md5
    exit 0
  fi

  echo -e "Extracting Lisk binaries to "$liskLocation/lisk-$release

  tar -xzf $liskVersion -C $liskLocation

  mv $liskLocation/$liskDir $liskLocation/lisk-$release

  echo -e "\nCleaning up downloaded files"
  rm -f $liskVersion $liskVersion.md5
 
}

configure_lisk() {

  cd $liskLocation/lisk-$release

  echo -e "\nColdstarting Lisk for the first time"
  bash lisk.sh coldstart
  
  sleep 5

  echo -e "\nStopping Lisk to perform database tuning"
  bash lisk.sh stop

  echo -e "\nExecuting database tuning operation"
  bash tune.sh

  echo -e "\nStarting Lisk with all parameters in place"
  bash lisk.sh start

  sleep 5
  blockHeight=`curl -s http://localhost:8000/api/loader/status/sync | cut -d: -f5 | cut -d} -f1`

  echo -e "\nCurrent Block Height: " $blockHeight

}

backup_lisk() {
  echo -e "\nStopping Lisk to perform a backup"
  cd $liskLocation/lisk-$release
  bash lisk.sh stop

  echo -e "\nBacking up existing Lisk Folder"
  mkdir -p $liskLocation/backup/  
  mv -f $liskLocation/lisk-$release $liskLocation/backup/    
  
}

upgrade_lisk() {
  
  echo -e "\nRestoring Database to new Lisk Install"
  mkdir -p -m700 $liskLocation/lisk-$release/pgsql/data
  cp -rf $liskLocation/backup/lisk-$release/pgsql/data/* $liskLocation/lisk-$release/pgsql/data/
  
  echo -e "\nStarting Lisk"
  cd $liskLocation/lisk-$release
  bash lisk.sh start
  
  echo -e "\nWaiting to check Block Height"
  sleep 5
  blockHeight=`curl -s http://localhost:8000/api/loader/status/sync | cut -d: -f5 | cut -d} -f1`

  echo -e "\nCurrent Block Height: " $blockHeight
}


case $1 in
"install")
  user_prompts
  ntp_checks
  install_lisk
  configure_lisk
  ;;
"upgrade")
  user_prompts
  backup_lisk
  install_lisk
  upgrade_lisk
  ;;
*)
  echo "Error: Unrecognized command."
  echo ""
  echo "Available commands are: install upgrade"
  ;;
esac