install.sh

#!/bin/bash

CONFIG_FILE='nextcoin.conf'
COIN_DAEMON='nextd'
COIN_CLI='next-cli'
COIN_REPO='https://github.com/ImTheDeveloper/next_mn/raw/master/next_linux_bin.zip'
COIN_NAME='NEXT'
COIN_PORT=7077
RPC_PORT=10001
SENTINEL_REPO='https://github.com/NextExchange/sentinel.git'
OS_VERSION='U18'
NODEIP=$(curl -s4 icanhazip.com)

RED='\033[0;31m'
GREEN='\033[0;32m'
NC='\033[0m'

progressfilt () {
  local flag=false c count cr=$'\r' nl=$'\n'
  while IFS='' read -d '' -rn 1 c
  do
    if $flag
    then
      printf '%c' "$c"
    else
      if [[ $c != $cr && $c != $nl ]]
      then
        count=0
      else
        ((count++))
        if ((count > 1))
        then
          flag=true
        fi
      fi
    fi
  done
}

function compile_node() {
  echo -e "Preparing to download and compile $COIN_NAME${NC} node wallet"
  TMP_FOLDER=$(mktemp -d)
  cd $TMP_FOLDER
  wget -q $COIN_REPO >/dev/null 2>&1
  compile_error
  COIN_ZIP=$(echo $COIN_REPO | awk -F'/' '{print $NF}')
  unzip $COIN_ZIP >/dev/null 2>&1
  compile_error
  rm -f $COIN_ZIP >/dev/null 2>&1
  chmod +x $COIN_DAEMON $COIN_CLI
  cp $COIN_DAEMON $COIN_CLI $NEXTCOINFOLDER
  cd - >/dev/null 2>&1
  rm -rf $TMP_FOLDER >/dev/null 2>&1
  echo -e "Node files downloaded and setup.${NC}"
  clear
}

function configure_systemd() {
  cat << EOF > /etc/systemd/system/$COIN_NAME.service
[Unit]
Description=$COIN_NAME service
After=network.target
[Service]
User=$NEXTCOINUSER
Group=$NEXTCOINUSER
Type=forking
#PIDFile=$NEXTCOINFOLDER/$COIN_NAME.pid
ExecStart=$NEXTCOINFOLDER/$COIN_DAEMON -daemon -conf=$NEXTCOINFOLDER/$CONFIG_FILE -datadir=$NEXTCOINFOLDER
ExecStop=$NEXTCOINFOLDER/$COIN_CLI -conf=$NEXTCOINFOLDER/$CONFIG_FILE -datadir=$NEXTCOINFOLDER stop
Restart=always
PrivateTmp=true
TimeoutStopSec=60s
TimeoutStartSec=10s
StartLimitInterval=120s
StartLimitBurst=5
[Install]
WantedBy=multi-user.target
EOF

  systemctl daemon-reload
  sleep 3
  systemctl start $COIN_NAME.service >/dev/null 2>&1
  systemctl enable $COIN_NAME.service >/dev/null 2>&1

  if [[ -z "$(ps axo cmd:100 | egrep $COIN_DAEMON)" ]]; then
    echo -e "${RED}$COIN_NAME is not running${NC}, please investigate. You should start by running the following commands as root:"
    echo -e "${GREEN}systemctl start $COIN_NAME.service"
    echo -e "systemctl status $COIN_NAME.service"
    echo -e "less /var/log/syslog${NC}"
    exit 1
  fi
}

function configure_startup() {
  cat << EOF > /etc/init.d/$COIN_NAME
#! /bin/bash
### BEGIN INIT INFO
# Provides: $COIN_NAME
# Required-Start: $remote_fs $syslog
# Required-Stop: $remote_fs $syslog
# Default-Start: 2 3 4 5
# Default-Stop: 0 1 6
# Short-Description: $COIN_NAME
# Description: This file starts and stops $COIN_NAME MN server
#
### END INIT INFO
case "\$1" in
 start)
   $NEXTCOINFOLDER/$COIN_DAEMON -daemon
   sleep 5
   ;;
 stop)
   $NEXTCOINFOLDER/$COIN_CLI stop
   ;;
 restart)
   $NEXTCOINFOLDER/$COIN_CLI stop
   sleep 10
   $NEXTCOINFOLDER/$COIN_DAEMON -daemon
   ;;
 *)
   echo "Usage: $COIN_NAME {start|stop|restart}" >&2
   exit 3
   ;;
esac
EOF
chmod +x /etc/init.d/$COIN_NAME >/dev/null 2>&1
update-rc.d $COIN_NAME defaults >/dev/null 2>&1
/etc/init.d/$COIN_NAME start 
if [ "$?" -gt "0" ]; then
 sleep 5
 /etc/init.d/$COIN_NAME start
fi
}


function create_config() {
  RPCUSER=$(tr -cd '[:alnum:]' < /dev/urandom | fold -w10 | head -n1)
  RPCPASSWORD=$(tr -cd '[:alnum:]' < /dev/urandom | fold -w22 | head -n1)
  cat << EOF > $NEXTCOINFOLDER/$CONFIG_FILE
rpcuser=$RPCUSER
rpcpassword=$RPCPASSWORD
rpcport=$RPC_PORT
#rpcallowip=127.0.0.1
listen=1
server=1
daemon=1
port=$COIN_PORT
EOF
}

function create_key() {
  echo -e "Please provide your ${RED}$COIN_NAME Masternode Private Key generated from your local wallet${NC}.\nLeave blank and hit [ENTER] if you wish for the script to generate a new ${RED}$COIN_NAME Masternode Private Key${NC} for you:"
  read -e COINKEY
  if [[ -z "$COINKEY" ]]; then
  echo -e "Please wait whilst we generate your masternode private key. It will be displayed once installation has completed."
  $NEXTCOINFOLDER/$COIN_DAEMON -daemon >/dev/null 2>&1
  sleep 30
  if [ -z "$(ps axo cmd:100 | grep $COIN_DAEMON)" ]; then
   echo -e "${RED}$COIN_NAME server couldn not start. Check /var/log/syslog for errors.${NC}"
   exit 1
  fi
  COINKEY=$($NEXTCOINFOLDER/$COIN_CLI masternode genkey)
  if [ "$?" -gt "0" ];
    then
    echo -e "${RED}Wallet not fully loaded. Let us wait and try again to generate the Private Key${NC}"
    sleep 30
    COINKEY=$($NEXTCOINFOLDER/$COIN_CLI masternode genkey)
  fi
  $NEXTCOINFOLDER/$COIN_CLI stop >/dev/null 2>&1
fi
clear
}

function update_config() {
  sed -i 's/daemon=1/daemon=0/' $NEXTCOINFOLDER/$CONFIG_FILE
  cat << EOF >> $NEXTCOINFOLDER/$CONFIG_FILE
logintimestamps=1
maxconnections=64
testnet=0
debug=0
masternode=1
externalip=$NODEIP
port=$COIN_PORT
masternodeprivkey=$COINKEY
EOF
}

function enable_firewall() {
  echo -e "Installing and setting up firewall to allow ingress on port ${GREEN}$COIN_PORT${NC}"
  ufw allow ssh >/dev/null 2>&1
  ufw allow $COIN_PORT >/dev/null 2>&1
  ufw default allow outgoing >/dev/null 2>&1
  echo "y" | ufw enable >/dev/null 2>&1
}

function get_ip() {
  declare -a NODE_IPS
  for ips in $(netstat -i | awk '!/Kernel|Iface|lo/ {print $1," "}')
  do
    NODE_IPS+=($(curl --interface $ips --connect-timeout 2 -s4 icanhazip.com))
  done

  if [ ${#NODE_IPS[@]} -gt 1 ]
    then
      echo -e "${GREEN}More than one IP. Please type 0 to use the first IP, 1 for the second and so on...${NC}"
      INDEX=0
      for ip in "${NODE_IPS[@]}"
      do
        echo ${INDEX} $ip
        let INDEX=${INDEX}+1
      done
      read -e choose_ip
      NODEIP=${NODE_IPS[$choose_ip]}
  else
    NODEIP=${NODE_IPS[0]}
  fi
}

function compile_error() {
if [ "$?" -gt "0" ];
 then
  echo -e "${RED}Failed to compile $COIN_NAME. Please investigate.${NC}"
  exit 1
fi
}

function detect_os() {
 if lsb_release -d | grep -q 'Ubuntu 18.04'; then
   OS_VERSION='U18'
 elif lsb_release -d | grep -q 'Ubuntu 18.10'; then
   OS_VERSION='U18'
 elif lsb_release -d | grep -q 'Ubuntu 16.04'; then
   OS_VERSION='U16'
 elif lsb_release -d | grep -q 'Ubuntu 14.04'; then
   OS_VERSION='U14'
 elif lsb_release -d | grep -q 'Debian'; then
   OS_VERSION='D'
else
   echo -e "${RED}Script currently only supports Ubuntu 14.04, 16.04, 18.04 or Debian. Installation will need to be carried out manually.${NC}"
   exit 1
fi
}

function checks() {
 detect_os 
if [[ $EUID -ne 0 ]]; then
   echo -e "${RED}$0 must be run as root.${NC}"
   exit 1
fi

if [ -n "$(pidof $COIN_DAEMON)" ] || [ -e "$COIN_DAEMOM" ] ; then
  echo -e "${RED}$COIN_NAME is already installed.${NC}"
  exit 1
fi
}

function prepare_system() {
echo -e "Prepare the system to install ${GREEN}$COIN_NAME${NC} master node."
apt-get update >/dev/null 2>&1
echo -e "Installing required packages.${NC}"
apt-get install -y wget curl ufw binutils net-tools unzip git >/dev/null 2>&1
echo -e "Checking if swap space is needed."
PHYMEM=$(free -g|awk '/^Mem:/{print $2}')
SWAP=$(swapon -s)
if [[ "$PHYMEM" -lt "2" && -z "$SWAP" ]];
  then
    echo -e "${GREEN}Server is running with less than 2G of RAM, creating 2G swap file.${NC}"
    dd if=/dev/zero of=/swapfile bs=1024 count=2M >/dev/null 2>&1
    chmod 600 /swapfile >/dev/null 2>&1
    mkswap /swapfile >/dev/null 2>&1
    swapon -a /swapfile >/dev/null 2>&1
else
  echo -e "${GREEN}The server running with at least 2G of RAM, or SWAP exists.${NC}"
fi
clear
}

function important_information() {
 echo
 echo -e "================================================================================"
 echo -e "${GREEN}$COIN_NAME${NC} Masternode is up and running listening on port ${RED}$COIN_PORT${NC}."
 echo -e "Configuration file is: ${RED}$NEXTCOINFOLDER/$CONFIG_FILE${NC}"
 echo -e "VPS_IP:PORT ${RED}$NODEIP:$COIN_PORT${NC}"
 echo -e "MASTERNODE PRIVATEKEY is: ${RED}$COINKEY${NC}"
 if [[ -n $SENTINEL_REPO  ]]; then
  echo -e "Sentinel is installed in ${RED}$NEXTCOINFOLDER/sentinel${NC}"
  echo -e "Sentinel logs is: ${RED}$NEXTCOINFOLDER/sentinel.log${NC}"
 fi
 echo -e "Useful commands for checking your node can be found in the ${GREEN}README${NC} here:"
 echo -e "${RED}https://github.com/ImTheDeveloper/next_mn/blob/master/README.md${NC}"
 echo -e "Produced by @munkee - any questions catch me on telegram!${NC}"
 echo -e "================================================================================"
}


function install_sentinel() {
   echo -e "Installing sentinel.${NC}"
   apt-get -y install python-virtualenv virtualenv >/dev/null 2>&1
   git clone $SENTINEL_REPO $NEXTCOINFOLDER/sentinel >/dev/null 2>&1
   cd $NEXTCOINFOLDER/sentinel
   virtualenv ./venv >/dev/null 2>&1
   ./venv/bin/pip install -r requirements.txt >/dev/null 2>&1
   cat << EOF > $NEXTCOINFOLDER/sentinel/sentinel.conf
next_conf=$NEXTCOINFOLDER/$CONFIG_FILE
EOF
   echo  "*/2 * * * * cd $NEXTCOINFOLDER/sentinel && ./venv/bin/python bin/sentinel.py >> $NEXTCOINFOLDER/sentinel.log 2>&1" > $NEXTCOINFOLDER/$COIN_NAME.cron
   crontab $NEXTCOINFOLDER/$COIN_NAME.cron
   rm $NEXTCOINFOLDER/$COIN_NAME.cron >/dev/null 2>&1
   clear
}


function ask_user() {
  DEFAULTNEXTCOINUSER=$COIN_NAME
  read -p "Which user do you wish to use/create to run this masternode as?: " -i $DEFAULTNEXTCOINUSER -e NEXTCOINUSER
  : ${NEXTCOINUSER:=$DEFAULTNEXTCOINUSER}

  if [ -z "$(getent passwd $NEXTCOINUSER)" ]; then
    useradd -m $NEXTCOINUSER
    USERPASS=$(tr -cd '[:alnum:]' < /dev/urandom | fold -w12 | head -n1)
    echo "$NEXTCOINUSER:$USERPASS" | chpasswd
  fi

  if (($OS_VERSION == 'D')); then
    NEXTCOINHOME=$(su - $NEXTCOINUSER bash -c 'echo $HOME')
  else
    NEXTCOINHOME=$(sudo -H -u $NEXTCOINUSER bash -c 'echo $HOME')
  fi
    DEFAULTNEXTCOINFOLDER="$NEXTCOINHOME/.next"
    read -p "Configuration folder: " -i $DEFAULTNEXTCOINFOLDER -e NEXTCOINFOLDER
    : ${NEXTCOINFOLDER:=$DEFAULTNEXTCOINFOLDER}
    mkdir -p $NEXTCOINFOLDER
    chown -R $NEXTCOINUSER: $NEXTCOINFOLDER >/dev/null
}



function setup_node() {
  ask_user
  get_ip
  compile_node
  create_config
  create_key
  update_config
  enable_firewall
  install_sentinel
  important_information
  if (( $OS_VERSION == 'U16' || $OS_VERSION == 'U18' || $OS_VERSION == 'D' )); then
    configure_systemd
  else
    configure_startup
  fi
}


##### Main #####
checks
prepare_system
setup_node