functions.sh

#!/bin/bash

wrong_arguments () {

  echo -e "\nMissing: arg1 [arg2]\n"
  echo -e " ------------------------------------------------------------------------------"
  echo -e "| arg1     | arg2                         | Description                        |"
  echo -e " ------------------------------------------------------------------------------"
  echo -e "| install  | core                         | Install Core                       |"
  echo -e "| update   | core / self / check          | Update Core / Core-Control / Check |"
  echo -e "| remove   | core / self                  | Remove Core / Core-Control         |"
  echo -e "| secret   | set / clear                  | Delegate Secret Set / Clear        |"
  echo -e "| start    | relay / forger / all         | Start Core Services                |"
  echo -e "| restart  | relay / forger / all / safe  | Restart Core Services              |"
  echo -e "| stop     | relay / forger / all         | Stop Core Services                 |"
  echo -e "| status   | relay / forger / all         | Show Core Services Status          |"
  echo -e "| logs     | relay / forger / all         | Show Core Logs                     |"
  echo -e "| snapshot | create / restore             | Snapshot Create / Restore          |"
  echo -e "| system   | info / update                | System Info / Update               |"
  echo -e "| config   | reset                        | Reset Config Files to Defaults     |"
  echo -e "| database | clear                        | Clear the Database                 |"
  echo -e "| rollback |                              | Rollback to Specified Height       |"
  echo -e "| plugin   | list / add / remove / update | Manage Core Plugins                |"
  echo -e " ------------------------------------------------------------------------------\n"
  exit 1

}

pm2status () {

   echo $(pm2 describe $1 2>/dev/null | grep "status")

}

git_check_core () {

  git fetch > /dev/null 2>&1
  loc=$(git rev-parse --short @)
  rem=$(git rev-parse --short origin/$branch)

  if [ "$loc" = "$rem" ]; then
    up2date="yes"
  else
    up2date="no"
  fi

}

git_check () {

  git fetch > /dev/null 2>&1
  loc=$(git rev-parse --short @)
  rem=$(git rev-parse --short origin/$branch)

  if [ "$loc" = "$rem" ]; then
    up2date="yes"
  else
    up2date="no"
  fi

}

setefile () {

  local envFile="$config/.env"

  if [ -f $envFile ]; then
    rm $envFile
  fi

  touch "$envFile"

  echo "CORE_LOG_LEVEL=$log_level" >> "$envFile" 2>&1
  echo "CORE_LOG_LEVEL_FILE=$log_level" >> "$envFile" 2>&1
  echo "CORE_DB_HOST=localhost" >> "$envFile" 2>&1
  echo "CORE_DB_PORT=5432" >> "$envFile" 2>&1
  echo "CORE_DB_USERNAME=$USER" >> "$envFile" 2>&1
  echo "CORE_DB_PASSWORD=password" >> "$envFile" 2>&1
  echo "CORE_DB_DATABASE=${name}_$network" >> "$envFile" 2>&1
  echo "CORE_P2P_HOST=0.0.0.0" >> "$envFile" 2>&1
  echo "CORE_P2P_PORT=$p2p_port" >> "$envFile" 2>&1
  echo "CORE_API_HOST=0.0.0.0" >> "$envFile" 2>&1
  echo "CORE_API_PORT=$api_port" >> "$envFile" 2>&1
  echo "CORE_WEBHOOKS_HOST=0.0.0.0" >> "$envFile" 2>&1
  echo "CORE_WEBHOOKS_PORT=$wh_port" >> "$envFile" 2>&1
  echo "CORE_WALLET_API_HOST=0.0.0.0" >> "$envFile" 2>&1
  echo "CORE_WALLET_API_PORT=$wapi_port" >> "$envFile" 2>&1
  echo "CORE_EXCHANGE_JSON_RPC_HOST=0.0.0.0" >> "$envFile" 2>&1
  echo "CORE_EXCHANGE_JSON_RPC_PORT=$rpc_port" >> "$envFile" 2>&1

}

start () {

  local secrets=$(cat $config/delegates.json | jq -r '.secrets')

  if [ "$1" = "all" ]; then

    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [ "$rstatus" != "online" ]; then
      pm2 --name "${name}-relay" start $core/packages/core/bin/run -- relay:run --network $network --token $name > /dev/null 2>&1
    else
      echo -e "\n${red}Process relay already running. Skipping...${nc}"
    fi

    if [ "$secrets" = "[]" ]; then
      echo -e "\n${red}Delegate secret is missing. Forger start aborted!${nc}"
    elif [ "$fstatus" != "online" ]; then
      pm2 --name "${name}-forger" start $core/packages/core/bin/run -- forger:run --network $network --token $name > /dev/null 2>&1
    else
      echo -e "\n${red}Process forger already running. Skipping...${nc}"
    fi

    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [ "$rstatus" != "online" ]; then
      echo -e "\n${red}Process startup failed.${nc}"
    fi

  else

    local pstatus=$(pm2status "${name}-$1" | awk '{print $4}')

    if [[ "$secrets" = "[]" && "$1" = "forger" ]]; then
      echo -e "\n${red}Delegate secret is missing. Forger start aborted!${nc}"
    elif [ "$pstatus" != "online" ]; then
      pm2 --name "${name}-$1" start $core/packages/core/bin/run -- ${1}:run --network $network --token $name > /dev/null 2>&1
    else
      echo -e "\n${red}Process $1 already running. Skipping...${nc}"
    fi

    local pstatus=$(pm2status "${name}-$1" | awk '{print $4}')

    if [[ "$pstatus" != "online" && "$1" = "relay" ]]; then
      echo -e "\n${red}Process startup failed.${nc}"
    fi

  fi

  pm2 save > /dev/null 2>&1

}

restart () {

  if [ "$1" = "all" ]; then

    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [ "$rstatus" = "online" ]; then
      pm2 restart ${name}-relay > /dev/null 2>&1
    else
      echo -e "\n${red}Process relay not running. Skipping...${nc}"
    fi

    if [ "$fstatus" = "online" ]; then
      pm2 restart ${name}-forger > /dev/null 2>&1
    else
      echo -e "\n${red}Process forger not running. Skipping...${nc}"
    fi

  elif [ "$1" = "safe" ]; then

    local api=$(curl -Is http://127.0.0.1:5001)
    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [[ "$rstatus" != "online" || "$fstatus" != "online" ]]; then
      echo -e "\n${red}Core forger is offline. Use '${cyan}ccontrol restart${red}' instead.${nc}\n"
      exit 1
    elif [ -z "$api" ]; then
      echo -e "\n${red}Plugin round-monitor not active. Use '${cyan}ccontrol restart${red}' instead.${nc}\n"
      exit 1
    else
      curl -X POST http://127.0.0.1:5001/restart > /dev/null 2>&1
      echo -e "\n${green}Restart requested. Check logs to monitor progress.${nc}"
    fi

  else

    local pstatus=$(pm2status "${name}-$1" | awk '{print $4}')

    if [ "$pstatus" = "online" ]; then
      pm2 restart ${name}-$1 > /dev/null 2>&1
    else
      echo -e "\n${red}Process $1 not running. Skipping...${nc}"
    fi

  fi

}

stop () {

  if [ "$1" = "all" ]; then

    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [ "$rstatus" = "online" ]; then
      pm2 stop ${name}-relay > /dev/null 2>&1
    else
      echo -e "\n${red}Process relay not running. Skipping...${nc}"
    fi

    if [ "$fstatus" = "online" ]; then
      pm2 stop ${name}-forger > /dev/null 2>&1
    else
      echo -e "\n${red}Process forger not running. Skipping...${nc}"
    fi

  else

    local pstatus=$(pm2status "${name}-$1" | awk '{print $4}')

    if [ "$pstatus" = "online" ]; then
      pm2 stop ${name}-$1 > /dev/null 2>&1
    else
      echo -e "\n${red}Process $1 not running. Skipping...${nc}"
    fi

  fi

  pm2 save > /dev/null 2>&1

}

status () {

  echo -e -n "\n${cyan}${name}-core${nc} v${cyan}${corever}${nc} "

  if [ "$1" = "all" ]; then

    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    if [ "$rstatus" = "online" ]; then
      echo -ne "relay: ${green}online${nc} "
    else
      echo -ne "relay: ${red}offline${nc} "
    fi

    if [ "$fstatus" = "online" ]; then
      echo -e "forger: ${green}online${nc}\n"
    else
      echo -e "forger: ${red}offline${nc}\n"
    fi

  else

    local pstatus=$(pm2status "${name}-$1" | awk '{print $4}')

    if [ "$pstatus" = "online" ]; then
      echo -e "$1: ${green}online${nc}\n"
    else
      echo -e "$1: ${red}offline${nc}\n"
    fi

  fi

}

install_deps () {

  sudo timedatectl set-ntp no > /dev/null 2>&1
  sudo apt install -y htop curl build-essential python git libpq-dev ntp gawk jq mc libjemalloc-dev make g++ > /dev/null 2>&1

  sudo curl -sL https://deb.nodesource.com/setup_12.x | sudo -E bash - > /dev/null 2>&1
  sudo curl -sL https://dl.yarnpkg.com/debian/pubkey.gpg | sudo apt-key add - > /dev/null 2>&1
  echo "deb https://dl.yarnpkg.com/debian/ stable main" | sudo tee /etc/apt/sources.list.d/yarn.list > /dev/null 2>&1
  sudo apt-get update > /dev/null 2>&1
  sudo apt-get install -y nodejs yarn > /dev/null 2>&1

  sudo npm install -g grunt-cli pm2 lerna > /dev/null 2>&1
  pm2 install pm2-logrotate > /dev/null 2>&1

  pm2 startup > /dev/null 2>&1
  pm2 save --force > /dev/null 2>&1

}

secure () {

  sudo apt install -y ufw fail2ban > /dev/null 2>&1
  sudo ufw allow 22/tcp > /dev/null 2>&1
  sudo ufw allow ${p2p_port}/tcp > /dev/null 2>&1
  sudo ufw allow ${api_port}/tcp > /dev/null 2>&1
  sudo ufw allow ${wapi_port}/tcp > /dev/null 2>&1
  sudo ufw --force enable > /dev/null 2>&1
  sudo sed -i "/^PermitRootLogin/c PermitRootLogin prohibit-password" /etc/ssh/sshd_config > /dev/null 2>&1
  sudo systemctl restart sshd.service > /dev/null 2>&1

}

install_db () {

  sudo apt install -y postgresql postgresql-contrib > /dev/null 2>&1
  sudo -u postgres psql -c "CREATE USER $USER WITH PASSWORD 'password' CREATEDB;" > /dev/null 2>&1
  dropdb ${name}_$network > /dev/null 2>&1
  createdb ${name}_$network > /dev/null 2>&1

}

install_core () {

  git clone $repo $core -b $branch > /dev/null 2>&1

  if [ -d $HOME/.config ]; then
    sudo chown -R $USER:$USER $HOME/.config > /dev/null 2>&1
  else
    mkdir $HOME/.config > /dev/null 2>&1
  fi

  mkdir $data > /dev/null 2>&1
  cd $core > /dev/null 2>&1

  yarn setup > /dev/null 2>&1
  cp -rf "$core/packages/core/bin/config/$network" "$data" > /dev/null 2>&1

  setefile

}

update () {

  yarn setup > /dev/null 2>&1

  local api=$(curl -Is http://127.0.0.1:5001)
  local added="$(cat $config/plugins.js | grep round-monitor)"
  local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
  local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

  # INSTALL HYDRA IF IT IS NOT YET THERE
  if [ -z "$(cat $config/plugins.js | grep '@internet-of-people/hydra-plugin')" ]; then
    if [ -z "$(cat $config/plugins.js | grep '@arkecosystem/core-transaction-pool')" ]; then
      echo "ERROR: $config/plugins.js does not contain @arkecosystem/core-transaction-pool which is essential to have Hydra Core."
      exit 1;
    fi

    node $basedir/hydra-pluginsjs-installer.js $config/plugins.js
  fi

  if [ -z "$(cat $config/app.js | grep '@internet-of-people/hydra-plugin')" ]; then
    if [ -z "$(cat $config/app.js | grep '@arkecosystem/core-magistrate-transactions')" ]; then
      echo "ERROR: $config/app.js does not contain @arkecosystem/core-magistrate-transactions which is essential to have Hydra Core."
      exit 1;
    fi

    node $basedir/hydra-appjs-installer.js $config/app.js
  fi

  ## CHECKING IF API RATE LIMIT INCREASE IS THERE
  if [ -z "$(cat $config/.env | grep 'CORE_API_RATE_LIMIT')" ]; then
    echo CORE_API_RATE_LIMIT=true >> $config/.env
    echo CORE_API_RATE_LIMIT_USER_LIMIT=1000 >> $config/.env
  fi

  for plugin in $(ls $basedir/plugins); do

    if [ ! -z "$(cat $config/plugins.js | grep $plugin)" ]; then

      . "$basedir/plugins/$plugin"
      mkdir $core/node_modules/$npmrepo > /dev/null 2>&1
      git clone $gitrepo/$plugin $core/node_modules/$npmrepo/$plugin > /dev/null 2>&1
      cd $core/node_modules/$npmrepo/$plugin > /dev/null 2>&1
      yarn install > /dev/null 2>&1

    fi

  done

  yarn setup > /dev/null 2>&1

  if [[ "$rstatus" = "online" && "$fstatus" = "online" && ! -z "$api" && ! -z "$added" ]]; then

    curl -X POST http://127.0.0.1:5001/restart > /dev/null 2>&1

  else

    if [ "$rstatus" = "online" ]; then
      pm2 restart ${name}-relay > /dev/null 2>&1
    fi

    if [ "$fstatus" = "online" ]; then
      pm2 restart ${name}-forger > /dev/null 2>&1
    fi

  fi

}

remove () {

  pm2 delete ${name}-forger > /dev/null 2>&1
  pm2 delete ${name}-relay > /dev/null 2>&1
  pm2 save > /dev/null 2>&1
  rm -rf $core > /dev/null 2>&1
  rm -rf $data > /dev/null 2>&1
  rm -rf $HOME/.cache/${name}-core > /dev/null 2>&1
  rm -rf $HOME/.local/share/${name}-core > /dev/null 2>&1
  rm -rf $HOME/.local/state/${name}-core > /dev/null 2>&1
  rm -rf /tmp/$USER/${name}-core > /dev/null 2>&1
  dropdb ${name}_$network > /dev/null 2>&1
  sudo ufw delete allow $p2p_port/tcp > /dev/null 2>&1
  sudo ufw delete allow $api_port/tcp > /dev/null 2>&1
  sudo ufw delete allow $wapi_port/tcp > /dev/null 2>&1

}

config_reset () {

  stop all > /dev/null 2>&1
  rm -rf $config > /dev/null 2>&1
  cp -rf "$core/packages/core/bin/config/$network" "$data" > /dev/null 2>&1
  setefile

}

sysinfo () {

  local sockets="$(lscpu | grep "Socket(s):" | head -n1 | awk '{ printf $2 }')"
  local cps="$(lscpu | grep "Core(s) per socket:" | awk '{ printf $4 }')"
  local tpc="$(lscpu | grep "Thread(s) per core:" | awk '{ printf $4 }')"
  local os="$(lsb_release -d | awk '{ for (i=2;i<=NF;++i) printf $i " " }')"
  local cpu="$(lscpu | grep "Model name" | awk '{ for (i=3;i<=NF;++i) printf $i " " }')"
  local mhz="$(lscpu | grep "CPU MHz:" | awk '{ printf $3 }' | cut -f1 -d".")"
  local maxmhz="$(lscpu | grep "CPU max MHz:" | awk '{ printf $4 }' | cut -f1 -d".")"
  local hn="$(hostname --fqdn)"
  local ips="$(hostname --all-ip-address)"

  echo -e "\n${cyan}System: ${nc}$os"
  w | head -n1

  echo -e "\n${cyan}CPU(s): ${nc}${sockets}x ${cpu}with $cps Cores and $[cps*tpc] Threads"
  echo -ne " ${cyan}Total: ${nc}$[sockets*cps] Cores and $[sockets*cps*tpc] Threads"
  if [ -z "$maxmhz" ]; then
    echo -e " @ ${mhz}MHz"
  else
    echo -e " @ ${maxmhz}MHz"
  fi

  echo -e "\n${cyan}Hostname: ${nc}$hn"
  echo -e " ${cyan}IP(s): ${nc}$ips"

  echo -e "${yellow}"
  free -h

  echo -e "${magenta}"
  df -h /

  echo -e "${nc}"

}

sysupdate () {

  sudo apt-get -o Dpkg::Options::="--force-confdef" -o Dpkg::Options::="--force-confold" dist-upgrade -y
  sudo apt-get autoremove -y
  sudo apt-get autoclean -y

}

logs () {

  if [ "$1" = "all" ]; then
    pm2 logs
  else
    pm2 logs ${name}-$1
  fi

}

secret () {

  if [ "$1" = "set" ]; then
    local scrt="$2 $3 $4 $5 $6 $7 $8 $9 ${10} ${11} ${12} ${13}"
    jq --arg scrt "$scrt" '.secrets = [$scrt]' $config/delegates.json > delegates.tmp
  else
    jq '.secrets = []' $config/delegates.json > delegates.tmp
  fi

  mv delegates.tmp $config/delegates.json

}

snapshot () {

  if [ "$1" = "restore" ]; then

    local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
    local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

    stop all > /dev/null 2>&1

    dropdb ${name}_$network > /dev/null 2>&1
    createdb ${name}_$network > /dev/null 2>&1

    $core/packages/core/bin/run snapshot:restore --network $network --token $name

    if [ "$rstatus" = "online" ]; then
      start relay > /dev/null 2>&1
    fi

    if [ "$fstatus" = "online" ]; then
      start forger > /dev/null 2>&1
    fi

  else

    $core/packages/core/bin/run snapshot:dump --network $network --token $name

  fi

}

selfremove () {

  cd $HOME > /dev/null 2>&1
  rm -rf $basedir > /dev/null 2>&1
  sed -i '/ccontrol/d' $HOME/.bashrc > /dev/null 2>&1
  sed -i '/cccomp/d' $HOME/.bashrc > /dev/null 2>&1

}

rollback () {

  local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
  local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

  stop all > /dev/null 2>&1

  $core/packages/core/bin/run snapshot:rollback --height $1 --network $network --token $name

  if [ "$rstatus" = "online" ]; then
    start relay > /dev/null 2>&1
  fi

  if [ "$fstatus" = "online" ]; then
    start forger > /dev/null 2>&1
  fi

}

update_info () {

  cd $basedir > /dev/null 2>&1
  git_check

  echo -e -n "\n${cyan}core-control${nc} v${cyan}${version}${nc} hash: ${cyan}${loc}${nc} status: "

  if [ "$up2date" = "yes" ]; then
    echo -e "${green}current${nc}\n"
  else
    echo -e "${red}stale${nc}\n"
  fi

  if [ -d $core ]; then

    cd $core > /dev/null 2>&1
    git_check_core

    echo -e -n "${cyan}${name}-core${nc} v${cyan}${corever}${nc} hash: ${cyan}${loc}${nc} status: "

    if [ "$up2date" = "yes" ]; then
      echo -e "${green}current${nc}\n"
    else
      echo -e "${red}stale${nc}\n"
    fi

  fi

}

db_clear () {

  local fstatus=$(pm2status "${name}-forger" | awk '{print $4}')
  local rstatus=$(pm2status "${name}-relay" | awk '{print $4}')

  stop all > /dev/null 2>&1

  dropdb ${name}_$network > /dev/null 2>&1
  createdb ${name}_$network > /dev/null 2>&1

  if [ "$rstatus" = "online" ]; then
    start relay > /dev/null 2>&1
  fi

  if [ "$fstatus" = "online" ]; then
    start forger > /dev/null 2>&1
  fi

}

plugin_list () {

  echo -e "\nAvailable plugins:\n"

  for plugin in $(ls plugins); do

    . "plugins/$plugin"

    if [ -z "$(cat $config/plugins.js | grep $plugin)" ]; then
      echo -e "${cyan}$plugin${nc} - ${red}inactive${nc} [$desc]"
    else
      echo -e "${cyan}$plugin${nc} - ${green}active${nc} [$desc]"
    fi

  done

  echo

}

plugin_manage () {

    if [ ! -f plugins/$2 ]; then
      echo -e "\n${red}Plugin $2 not found.${nc}\n"
      exit 1
    else
      . "plugins/$2"
    fi

    added="$(cat $config/plugins.js | grep $2)"
    lastline='};'
    blockend='},'
    stab='    '


    if [[ "$1" = "add" && -z "$added" ]]; then

      alen=${#options[@]}
      insert="$stab\"$npmrepo\/$2\": {\n"

      for i in ${!options[@]}; do
        insert="$insert\t${options[$i]}"
        comp=$((i+1))
        if [ "$comp" -lt "$alen" ]; then
          insert="$insert,\n"
        else
          insert="$insert\n"
        fi
      done

      insert="$insert$stab$blockend\n"
      sed -i "s/$lastline/$insert$lastline/" $config/plugins.js

      mkdir $core/node_modules/$npmrepo > /dev/null 2>&1
      git clone $gitrepo/$2 $core/node_modules/$npmrepo/$2 > /dev/null 2>&1
      cd $core/node_modules/$npmrepo/$2
      yarn install > /dev/null 2>&1

      echo -e "\n${green}Plugin $2 installed with default settings.${nc}\n"
      echo -e "${red}Restart Core for the changes to take effect.${nc}\n"
      echo -e "${cyan}For more information and custom configuration${nc}"
      echo -e "${cyan}visit $gitrepo/$2${nc}\n"

    elif [[ "$1" = "add" && ! -z "$added" ]]; then

      echo -e "\n${red}Plugin $2 already installed.${nc}\n"


    elif [[ "$1" = "remove" && ! -z "$added" ]]; then

      sed -i "/$2/,/$blockend/d" $config/plugins.js
      rm -rf $core/node_modules/$npmrepo/$2 > /dev/null 2>&1

      echo -e "\n${green}Plugin $2 removed successfully.${nc}\n"
      echo -e "${red}Restart Core for the changes to take effect.${nc}\n"

    elif [[ "$1" = "remove" && -z "$added" ]]; then

      echo -e "\n${red}Plugin $2 not installed.${nc}\n"

    elif [[ "$1" = "update" && ! -z "$added" ]]; then

      cd $core/node_modules/$npmrepo/$2 > /dev/null 2>&1
      git_check

      if [ "$up2date" = "yes" ]; then
        echo -e "Already up-to-date."
        exit 1
      fi

      git pull > /dev/null 2>&1
      yarn install > /dev/null 2>&1

      echo -e "\n${green}Plugin $2 updated successfully.${nc}\n"
      echo -e "${red}Restart Core for the changes to take effect.${nc}\n"

    else

      echo -e "\n${red}Plugin $2 not installed.${nc}\n"

    fi

}