From b15bd4d2e5d4097632096f731d0106edb3fa2960 Mon Sep 17 00:00:00 2001 From: tteckster Date: Fri, 15 Apr 2022 07:43:04 -0400 Subject: [PATCH] Update kernel-clean.sh --- misc/kernel-clean.sh | 129 ++++++++++++++++++++++--------------------- 1 file changed, 66 insertions(+), 63 deletions(-) diff --git a/misc/kernel-clean.sh b/misc/kernel-clean.sh index 0561c105..054144fd 100644 --- a/misc/kernel-clean.sh +++ b/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,64 +31,66 @@ 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" - if [[ "$current_kernel" == *"pve"* ]]; then - if [[ "$latest_kernel" != *"$current_kernel"* ]]; then - printf "Latest Kernel: $latest_kernel\n" - fi +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 + echo -e "${GREEN}Latest Kernel: $latest_kernel\n${NORMAL}" + fi + else + echo -e "\n${RED}ERROR: No PVE Kernel found\n${NORMAL}" + exit 1 + fi +} + +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 "___________________________________________\n\n" - printf "[!] Warning, you're not running a PVE Kernel\n" - exit 1 + echo -e "${RED}'$kernel' ${NORMAL}${YELLOW}has been added to the Kernel remove list\n${NORMAL}" + kernels_to_remove+=" $kernel" fi + done +echo -e "${YELLOW}Kernel Search Complete!\n${NORMAL}" + if [[ "$kernels_to_remove" != *"pve"* ]]; then + echo -e "${GREEN}It appears there are no old Kernels on your system \n${NORMAL}" + else + 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 + 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 + echo -e "${YELLOW}Finished!\n${NORMAL}" + echo -e "${YELLOW}Updating GRUB... \n${NORMAL}" + /usr/sbin/update-grub > /dev/null 2>&1 + echo -e "${YELLOW}Finished!\n${NORMAL}" + else + echo -e "${YELLOW}Exiting...\n${NORMAL}" + sleep 1 + fi } -function kernel_clean { - kernels=$(dpkg --list| grep 'pve-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" - kernels_to_remove+=" $kernel" - fi - done - printf "Kernel Search Complete!\n" - if [[ "$kernels_to_remove" != *"pve"* ]]; then - printf "It appears there are no old Kernels on your system \n" - 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" - fi - if [[ $REPLY =~ ^[Yy]$ ]]; then - printf "Removing $(echo $kernels_to_remove | awk '{print NF}') old Kernels..." - /usr/bin/apt purge -y $kernels_to_remove > /dev/null 2>&1 - printf "Finished!\n" - printf "Updating GRUB... \n" - /usr/sbin/update-grub > /dev/null 2>&1 - printf "Finished!\n" - else - printf "\nExiting...\n" - fi -} -function main { - check_root - header_info - kernel_info + +function main() { + check_root + header_info + kernel_info } + while true; do - case "$1" in - * ) - main - kernel_clean - exit 1 - ;; + case "$1" in + * ) + main + kernel_clean + exit 1 + ;; esac - shift + shift done