Update kernel-clean.sh

misc/kernel-clean.sh

@@ -1,28 +1,29 @@
 #!/bin/bash
-YW=`echo "\033[33m"`
-BL=`echo "\033[36m"`
-RD=`echo "\033[01;31m"`
-CM='\xE2\x9C\x94\033'
-GN=`echo "\033[1;92m"`
-CL=`echo "\033[m"`
+RED=$(tput setaf 1)
+GREEN=$(tput setaf 2)
+YELLOW=$(tput setaf 3)
+BLUE=$(tput setaf 4)
+WHITE=$(tput setaf 7)
+NORMAL=$(tput sgr0)
 while true; do
-    read -p "This will Clean unused Kernel images. Proceed(y/n)?" yn
+    read -p "${YELLOW}This will Clean unused Kernel images. Proceed(y/n)?${NORMAL}" yn
     case $yn in
         [Yy]* ) break;;
         [Nn]* ) exit;;
-        * ) echo "Please answer yes or no.";;
+        * ) echo -e "${RED}Please answer y/n${NORMAL}";;
     esac
 done
 clear
 current_kernel=$(uname -r)
+pve=$(pveversion)
 function check_root {
         if [[ $EUID -ne 0 ]]; then
-                printf "[!] Error: this script must be ran as the root user.\n" 
+                echo -e "${RED}Error: This script must be ran as the root user.\n${NORMAL}" 
                 exit 1
         fi
 }
 function header_info {
-echo -e "${RD}
+echo -e "${RED}
   _  __                    _    _____ _                  
  | |/ /                   | |  / ____| |                 
  |   / ___ _ __ _ __   ___| | | |    | | ___  __ _ _ __  
@@ -30,57 +31,59 @@ echo -e "${RD}
  |   \  __/ |  | | | |  __/ | | |____| |  __/ (_| | | | |
  |_|\_\___|_|  |_| |_|\___|_|  \_____|_|\___|\__,_|_| |_|
 
-${CL}"
+${NORMAL}"
 }
-function kernel_info {
-        latest_kernel=$(dpkg --list| grep 'pve-kernel-.*-pve' | awk '{print $2}' | tac | head -n 1)
-        printf "OS: $(cat /etc/os-release | grep "PRETTY_NAME" | sed 's/PRETTY_NAME=//g' | sed 's/["]//g' | awk '{print $0}')\n"
-        printf "Current Kernel: pve-kernel-$current_kernel\n"
+function kernel_info() {
+    latest_kernel=$(dpkg --list| grep 'kernel-.*-pve' | awk '{print $2}' | tac | head -n 1)
+    echo -e "${YELLOW}OS: ${GREEN}$(cat /etc/os-release | grep "PRETTY_NAME" | sed 's/PRETTY_NAME=//g' | sed 's/["]//g' | awk '{print $0}')\r${NORMAL}"
+    echo -e "${YELLOW}PVE Version: ${GREEN}$pve\n${NORMAL}"
     if [[ "$current_kernel" == *"pve"* ]]; then
       if [[ "$latest_kernel" != *"$current_kernel"* ]]; then
-                        printf "Latest Kernel: $latest_kernel\n"
+        echo -e "${GREEN}Latest Kernel: $latest_kernel\n${NORMAL}"
       fi
     else
-                printf "___________________________________________\n\n"
-                printf "[!] Warning, you're not running a PVE Kernel\n"
+        echo -e "\n${RED}ERROR: No PVE Kernel found\n${NORMAL}"
         exit 1
     fi
 }
-function kernel_clean {
-        kernels=$(dpkg --list| grep 'pve-kernel-.*-pve' | awk '{print $2}' | sort -V)
+
+function kernel_clean() {
+    kernels=$(dpkg --list| grep 'kernel-.*-pve' | awk '{print $2}' | sort -V)
     kernels_to_remove=""
     for kernel in $kernels
       do
         if [ "$(echo $kernel | grep $current_kernel)" ]; then
             break
         else
-                        printf "\"$kernel\" has been added to the Kernel remove list\n"
+            echo -e "${RED}'$kernel' ${NORMAL}${YELLOW}has been added to the Kernel remove list\n${NORMAL}"
                     kernels_to_remove+=" $kernel"
         fi
     done
-        printf "Kernel Search Complete!\n"
+echo -e "${YELLOW}Kernel Search Complete!\n${NORMAL}"
     if [[ "$kernels_to_remove" != *"pve"* ]]; then
-                printf "It appears there are no old Kernels on your system \n"
+        echo -e "${GREEN}It appears there are no old Kernels on your system \n${NORMAL}"
     else
-                        read -p "[!] Would you like to remove the $(echo $kernels_to_remove | awk '{print NF}') selected Kernels listed above? [y/n]: " -n 1 -r
-                        printf "\n"
+    read -p "${YELLOW}Would you like to remove the${RED} $(echo $kernels_to_remove | awk '{print NF}') ${NORMAL}${YELLOW}selected Kernels listed above? [y/n]: ${NORMAL}" -n 1 -r
     fi
       if [[ $REPLY =~ ^[Yy]$ ]]; then
-                        printf "Removing $(echo $kernels_to_remove | awk '{print NF}') old Kernels..."
+        echo -e "${YELLOW}\nRemoving ${NORMAL}${RED}$(echo $kernels_to_remove | awk '{print NF}') ${NORMAL}${YELLOW}old Kernels...${NORMAL}"
         /usr/bin/apt purge -y $kernels_to_remove > /dev/null 2>&1
-                        printf "Finished!\n"
-                        printf "Updating GRUB... \n"
+        echo -e "${YELLOW}Finished!\n${NORMAL}"
+        echo -e "${YELLOW}Updating GRUB... \n${NORMAL}"
         /usr/sbin/update-grub > /dev/null 2>&1
-                        printf "Finished!\n"
+        echo -e "${YELLOW}Finished!\n${NORMAL}"
       else
-                        printf "\nExiting...\n"
+        echo -e "${YELLOW}Exiting...\n${NORMAL}"
+        sleep 1
       fi
 }
-function main {
+
+function main() {
     check_root
     header_info
     kernel_info
 }
+
 while true; do
     case "$1" in
         * )