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 () {

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

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

}

npm_check () {

  if [ "$npmver" = "$corever" ]; 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/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/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/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 nodejs npm libpq-dev ntp gawk jq > /dev/null 2>&1
  sudo npm install -g n grunt-cli pm2 yarn lerna > /dev/null 2>&1
  sudo n 10 > /dev/null 2>&1
  pm2 install pm2-logrotate > /dev/null 2>&1

  local pm2startup="$(pm2 startup | tail -n1)"
  eval $pm2startup > /dev/null 2>&1
  pm2 save > /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 () {

  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

  yarn global add $repo/$package > /dev/null 2>&1

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

  setefile

  echo 'export PATH=$(yarn global bin):$PATH' >> $HOME/.bashrc

}

update () {

  yarn global add $repo/$package > /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}')

  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
  yarn global remove $repo/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/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/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/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/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

    npm_check

    echo -e -n "${cyan}${name}-core${nc} v${cyan}${corever}${nc} remote: v${cyan}${npmver}${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
      yarn global add $npmrepo/$2 > /dev/null 2>&1 &

      echo -ne "\n${cyan}Installing $2...  ${red}"

      while [ -d /proc/$! ]; do
        printf "\b${sp:i++%${#sp}:1}" && sleep .1
      done

      echo -e "\b${green}Done${nc}"

      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

      yarn global remove $npmrepo/$2  > /dev/null 2>&1 &

      echo -ne "\n${cyan}Removing $2...  ${red}"

      while [ -d /proc/$! ]; do
        printf "\b${sp:i++%${#sp}:1}" && sleep .1
      done

      echo -e "\b${green}Done${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

      rem=$(npm view $npmrepo/$2 version)
      loc=$(cat $HOME/.config/yarn/global/node_modules/$npmrepo/$2/package.json | jq -r '.version')

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

      yarn global add $npmrepo/$2 > /dev/null 2>&1 &

      echo -ne "\n${cyan}Installing $2...  ${red}"

      while [ -d /proc/$! ]; do
        printf "\b${sp:i++%${#sp}:1}" && sleep .1
      done

      echo -e "\b${green}Done${nc}"
      echo -e "\n${red}Restart Core for the changes to take effect.${nc}\n"

    else

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

    fi

}