mem_watch.sh

#!/bin/bash
# Version: 1.0.12
clear

show_logo() {
    echo -e "\e[92m"
    curl -s https://raw.githubusercontent.com/Dr0ff/Useful-scripts/refs/heads/main/tt.logo.txt
    echo -e "\e[0m"
}

show_logo

# Function to display error and exit
error_exit() {
  echo "Error: $1"
  exit 1
}



echo -e "\e[33m"
echo "/////////////////////////////////////////////////////////////////////////////////////////////"
echo "||                                                                                         //"
echo -e "||       \e[0mThis script will help you to configure and install the memory usage monitor     \e[33m  //"
echo -e "||       \e[0mfor your validator node, just follow few simple steps bellow \e[33m                     //"
echo -e "||                                                                                         //"
echo -e "||       \e[0mЭтот скрипт поможет вам установить и настроить мониторинг памяти \e[33m                 //"
echo -e "||       \e[0mдля вашей ноды, следуйте инструкциям и выполните несколько простых шагов\e[33m          //"
echo "||                                                                                         //"
echo "/////////////////////////////////////////////////////////////////////////////////////////////"
echo " "
echo " "
echo " "


echo -e "\e[32m Now we need to find the process to monitor the memory consumption\e[0m"
echo -e "\e[32m Сейчас необходимо найти процесс за которым мы будем вести наблюдение\e[0m"
echo  ""

# Function to prompt for manual or search input
prompt_process_name() {
  while true; do
    echo -e "\e[33mEnter the process name (node binary) or part of it (e.g xpl, ju, juno ...)\e[0m"
    echo -e "\e[33mВведите имя процесса (название бинарника) или его часть (например: xpl, ju, juno ...)\e[0m"

    read -p "" PROCESS_PART

    if [[ -z "$PROCESS_PART" ]] || ! [[ "$PROCESS_PART" =~ ^[a-zA-Z0-9._-]+$ ]]; then
    echo -e "\e[31mInvalid input. Please try again.\e[0m\n"
    echo -e "\e[31mНеправильный ввод. Попробуйте ещё раз.\e[0m\n"
    else
      MATCHING_PROCESSES=$(ps -eo comm | grep -i "$PROCESS_PART" | sort | uniq)

      if [[ -z "$MATCHING_PROCESSES" ]]; then
        echo -e "\e[31mNo matching processes found. Please try again.\e[0m\n"
        echo -e "\e[31mПодходящих процессов не найдено. Попробуйте ещё.\e[0m\n"
      else
        # Display the matching processes
        echo -e "\n\e[33mMATCHING PROCESSES:\e[0m"
        echo -e "\n\e[33mПОДХОДЯЩИЕ ПРОЦЕССЫ:\e[0m"
        PROCESS_LIST=()
        i=1

        while IFS= read -r process; do
          echo "$i. $process"
          PROCESS_LIST+=("$process")
          ((i++))
        done <<< "$MATCHING_PROCESSES"

        while true; do
          echo -e "\n\e[33mEnter process number\e[0m"
          echo -e "\n\e[33mВведит номер процесса\e[0m"
          read -p "(1,2,3...):" SELECTION


	if [[ "$SELECTION" =~ ^[0-9]+$ ]]; then
	  if [[ "$SELECTION" -ge 1 && "$SELECTION" -le "${#PROCESS_LIST[@]}" ]]; then
	    SELECTED_PROCESS="${PROCESS_LIST[$SELECTION - 1]}"
	    echo -e "\n\e[33mPROCESS NAME SELECTED: \e[32m$SELECTED_PROCESS\e[0m\n"
	    echo -e "\n\e[33mИМЯ ПРОЦЕССА: \e[32m$SELECTED_PROCESS\e[0m\n"
	    break 2
	  else
	    echo -e "\n\e[31mInvalid selection. Please try again.\e[0m"
	    echo -e "\n\e[31mПлохой выбор. Пробуйте ещё.\e[0m"
	  fi
	else
	  echo -e "\n\e[31mInvalid input. Please enter a number.\e[0m"
	  echo -e "\n\e[31mНеверный ввод. Пожалуйста, выберите цифру.\e[0m"
	fi
        done
      fi
    fi
  done
}

# Function to prompt for directory
prompt_directory() {
  while true; do
    echo -e "\n\e[32mWe need to find the directory to save this script (e.g. .xpla or .juno...)\e[0m"
    echo "Searching for matching directories..."
    echo -e "\n\e[32mТеперь нужно найти директорию где сохранить скрипт(e.g. .xpla or .juno...)\e[0m"
    echo "Поиск подходящей директории..."


    # Searching for directory, filtering .(dot)
    DIRECTORIES=$(find . -maxdepth 1 -type d -name ".${SELECTED_PROCESS:0:4}*" 2>/dev/null | sed 's/^\.\///g')

    if [[ -z "$DIRECTORIES" ]]; then
      echo -e "\n\e[31mNo matching directories found.\n\e[33mLooks like the directory doesn't exist.\e[0m"
      echo -e "\n\e[33mPlease check the installation directory and try again\nExiting...\e[0m"
      echo -e "\n\e[31mНужная директория не найдена.\n\e[33mПохоже её не существует.\e[0m"
      echo -e "\n\e[33mПожалуйста убедитесь в том,что нода установлена корректно\nЗавершаем работу...\e[0m"
      exit 1
    else
      echo -e "\n\e[33mMATCHING DIRECTORIES:\e[0m"
      echo -e "\n\e[33mПОДХОДЯЩИИЕ ДИРЕКТОРИИ:\e[0m"
      i=1
      DIRECTORY_LIST=()
      for DIR in $DIRECTORIES; do
        echo "$i. $DIR"
        DIRECTORY_LIST+=("$DIR")
        ((i++))
      done
#      echo -e "\n\e[33mWe have found some\e[0m"
      echo -e "\n\e[33mEnter the number corresponding to the right directory\e[0m"
      echo -e "\n\e[33mВведите номер соответсвующей директории\e[0m"
      read -p "(1,2,3...):" DIR_SELECTION
      if [[ "$DIR_SELECTION" -ge 1 && "$DIR_SELECTION" -le "${#DIRECTORY_LIST[@]}" ]]; then
        SELECTED_DIRECTORY="${DIRECTORY_LIST[$DIR_SELECTION - 1]}"
        break
      else
        echo -e "\n\e[31mError: Invalid selection. Please try again.\e[0m"
	echo -e "\n\e[31mОшибка: Неправильный выбор. Попробуйте снова.\e[0m"
      fi
    fi
  done

  echo -e "\n\e[33mDIRECTORY SELECTED: \e[32m$SELECTED_DIRECTORY\e[0m"
  echo -e "\n\e[33mВЫБРАННАЯ ДИРЕКТОРИЯ: \e[32m$SELECTED_DIRECTORY\e[0m"
}

# Main script logic
prompt_process_name
prompt_directory

# Prompt for memory limit
#while true; do
#  echo -e "\n\e[33mEnter the memory limit in kilobytes (e.g., 3000000 for 3 GB)\e[0m"
#  read -p "" MEMORY_LIMIT
#  if [[ "$MEMORY_LIMIT" =~ ^[0-9]+$ ]]; then
#    MEMORY_LIMIT_GB=$(awk -v limit="$MEMORY_LIMIT" 'BEGIN {printf "%.2f", limit / 1048576}')
#echo ""
#    printf "\033[33mMemory limit set to:\e[32m %.2f GB.\033[0m\n" "$MEMORY_LIMIT_GB"
#
#    break
#  else
#    echo -e "\n\e[31mInvalid input. Please enter a numeric value.\e[0m"
#  fi
#done

# Prompt for memory limit
while true; do
  echo -e "\n\e[33mEnter the memory limit in kilobytes (e.g., 3000000 for 2.8 GB) or press Enter to use the default (3 GB):\e[0m"
  echo -e "\n\e[33mВведите лимит памяти для процесса в килобайтах (e.g., 3000000 для 2.8 GB) или нажмите клавишу ВВОД, чтобы использовать значение по умолчанию (3 GB):\e[0m"
  read -p "" MEMORY_LIMIT

  # Check if input is empty, and use default value if so
  if [[ -z "$MEMORY_LIMIT" ]]; then
    MEMORY_LIMIT=3150000
    echo -e "\n\e[33mNo input provided. Using default memory limit: \e[32m${MEMORY_LIMIT}\e[0m"
    MEMORY_LIMIT_GB=$(awk -v limit="$MEMORY_LIMIT" 'BEGIN {printf "%.2f", limit / 1048576}')
    printf "\033[33mMemory limit set to:\e[32m %.2f GB.\033[0m\n" "$MEMORY_LIMIT_GB"
    break
  fi

  # Validate numeric input
  if [[ "$MEMORY_LIMIT" =~ ^[0-9]+$ ]]; then
    MEMORY_LIMIT_GB=$(awk -v limit="$MEMORY_LIMIT" 'BEGIN {printf "%.2f", limit / 1048576}')
    echo ""
    printf "\033[33mMemory limit set to:\e[32m %.2f GB.\033[0m\n" "$MEMORY_LIMIT_GB"
    break
  else
    echo -e "\n\e[31mInvalid input. Please enter a numeric value.\e[0m"
    echo -e "\n\e[31mНеправильный ввод. Пожалуйста введице цифровые значения.\e[0m"
  fi
done




# Print the selected directory
echo -e "\n\e[33mSELECTED DIRECTORY:\e[32m '$SELECTED_DIRECTORY'.\n\n\e[33mProceeding with the next steps....\e[0m"


# Form the path to save the script
SCRIPT_PATH="$SELECTED_DIRECTORY/memory_watch.sh"

# Create the main script (overwrites the existing file)
cat <<EOF > "$SCRIPT_PATH"
#!/bin/bash
# Version: 1.0.12

# Process name to monitor
PROCESS_NAME="$SELECTED_PROCESS"

# Memory limit in kilobytes
MEMORY_LIMIT_MB="$MEMORY_LIMIT"

# Check if the process is running
PROCESS_COUNT=\$(pgrep -c "\$PROCESS_NAME")

if [[ "\$PROCESS_COUNT" -eq 0 ]]; then
  echo -e "\$(date): Process \$PROCESS_NAME is not running. Skipping memory check.\n" >> /var/log/mem_watch.log
  exit 0
fi

# Get the memory usage of the process (in KB)
MEMORY_USAGE=\$(ps --no-headers -o rss -p \$(pgrep "\$PROCESS_NAME") | awk '{sum+=\$1} END {print sum}')

# Check if the limit is exceeded
if [[ "\$MEMORY_USAGE" -gt "\$MEMORY_LIMIT_MB" ]]; then
  echo -e "\$(date): Memory limit \$MEMORY_USAGE KB exceeded for \$PROCESS_NAME. Restarting service...\n" >> /var/log/mem_watch.log

  # Get the service name and restart it
  SERVICE_NAME=\$(cat /proc/\$(pidof "\$PROCESS_NAME")/cgroup | sed -n 's|.*/\([^.]*\)\.service|\1|p')

  if systemctl restart "\$SERVICE_NAME"; then
    echo -e "\$(date): Service \$SERVICE_NAME restarted successfully.\n" >> /var/log/mem_watch.log
  else
    echo -e "\$(date): Failed to restart \$SERVICE_NAME.\n" >> /var/log/mem_watch.log
  fi
fi
EOF

#Now we need to add this memory watch to cron (scheduler) 

if ! which cron > /dev/null; then
  echo "Cron is not installed yet. We need to install it..."
  sudo apt-get install cron
  echo "Opening crontab by crontab -e"
  sudo EDITOR=/bin/true crontab -e
  echo "Closing crontab -e"

fi
echo "Trying crontab -l"

if ! sudo crontab -l | grep -Fxq "*/30 * * * * bash /home/$USER/$SCRIPT_PATH"; then
  sudo crontab -l > /tmp/crontab
  echo "*/30 * * * * bash /home/$USER/$SCRIPT_PATH" >> /tmp/crontab
  sudo crontab /tmp/crontab
  rm /tmp/crontab
fi



# Make the new script executable
#chmod +x "$SCRIPT_PATH"

# Information about the created script
echo -e "\n\e[33mScript created and saved in:\e[32m $SCRIPT_PATH\e[0m"
echo -e "\n\e[33mLogs are stored in:\e[32m /var/log/mem_watch.log\e[0m"
echo ""
echo -e "\e[33mYou can check logs by using:\e[32m cat /var/log/mem_watch.log\e[0m\n"