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 "| install  | core advanced                       | Install Core (without UFW)         |"
  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 / core / all         | Start Core Services                |"
  echo -e "| restart  | relay / forger / core / all / safe  | Restart Core Services              |"
  echo -e "| stop     | relay / forger / core / all         | Stop Core Services                 |"
  echo -e "| status   | relay / forger / core / all         | Show Core Services Status          |"
  echo -e "| logs     | relay / forger / core / 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

}

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
  echo "CORE_IPFS_PORT=$ipfs_port" >>"$envFile" 2>&1
  echo "CORE_IPFS_WS_PORT=$ipfs_ws_port" >>"$envFile" 2>&1
  echo "CORE_IPFS_GATEWAY=$ipfs_gateway" >>"$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}')
    local cstatus=$(pm2status "${name}-core" | 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

    if [ "$cstatus" = "online" ]; then
      pm2 stop ${name}-core >/dev/null 2>&1
    else
      echo -e "\n${red}Process core 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}')
    local cstatus=$(pm2status "${name}-core" | awk '{print $4}')

    if [ "$cstatus" = "online" ]; then
      echo -e "core: ${green}online${nc}"
    else
      echo -e "core: ${red}offline${nc}"
    fi

    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() {
  curl -sS https://dl.yarnpkg.com/debian/pubkey.gpg | sudo apt-key add -
  echo "deb https://dl.yarnpkg.com/debian/ stable main" | sudo tee /etc/apt/sources.list.d/yarn.list
  sudo apt-get update -y
  sudo DEBIAN_FRONTEND=noninteractive apt-get -o Dpkg::Options::="--force-confdef" -o Dpkg::Options::="--force-confold" dist-upgrade -yq
  sudo apt-get autoremove -y
  sudo apt-get autoclean -y
  sudo timedatectl set-ntp no
  sudo DEBIAN_FRONTEND=noninteractive apt install -yq htop curl build-essential python git nodejs npm libpq-dev libjemalloc-dev ntp gawk jq tor
  sudo npm install -g n
  sudo n 12
  sudo npm install -g yarn
  sudo npm install -g grunt-cli
  sudo npm install -g pm2@3
  sudo npm install -g lerna
  pm2 install pm2-logrotate

  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 443/tcp >/dev/null 2>&1
  sudo ufw allow 8000/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 allow ${ipfs_port}/tcp >/dev/null 2>&1
  sudo ufw allow ${ipfs_ws_port}/tcp >/dev/null 2>&1
  sudo ufw allow ${storage_api_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() {
  echo -e "${nc}"
  git clone -b $branch $repo $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

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

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

  setefile

}

update() {
  cd $core
  echo -e "${nc}"
  yarn setup

  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}')

  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

  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

    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

  rm $core/plugins/storage/databases/$network.sqlite >/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
    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

}