Update kernel-clean.sh

Code overhaul with a fresh start.

misc/kernel-clean.sh

@@ -6,6 +6,7 @@
 # https://github.com/tteck/Proxmox/raw/main/LICENSE
 
 function header_info {
+  clear
   cat <<"EOF"
     __ __                     __   ________
    / //_/__  _________  ___  / /  / ____/ /__  ____ _____
@@ -15,32 +16,17 @@ function header_info {
 
 EOF
 }
-set -euo pipefail
-shopt -s inherit_errexit nullglob
 YW=$(echo "\033[33m")
-BL=$(echo "\033[36m")
 RD=$(echo "\033[01;31m")
-BGN=$(echo "\033[4;92m")
 GN=$(echo "\033[1;92m")
-DGN=$(echo "\033[32m")
 CL=$(echo "\033[m")
 BFR="\\r\\033[K"
 HOLD="-"
 CM="${GN}✓${CL}"
-CROSS="${RD}✗${CL}"
-PARTY="🎉"
 current_kernel=$(uname -r)
-clear
+available_kernels=$(dpkg --list | grep 'kernel-.*-pve' | awk '{print $2}' | grep -v "$current_kernel" | sort -V)
 header_info
-while true; do
-    read -p "This will Clean Unused Kernel Images, USE AT YOUR OWN RISK. Proceed(y/n)?" yn
-    case $yn in
-    [Yy]*) break ;;
-    [Nn]*) exit ;;
-    *) echo -e "${RD}Please answer y/n${CL}" ;;
-    esac
-done
-clear
+
 function msg_info() {
   local msg="$1"
   echo -ne " ${HOLD} ${YW}${msg}..."
@@ -51,90 +37,34 @@ function msg_ok() {
   echo -e "${BFR} ${CM} ${GN}${msg}${CL}"
 }
 
-function check_root() {
-    if [[ $EUID -ne 0 ]]; then
-        echo -e "${CROSS}${RD}Error: This script must be ran as the root user.\n${CL}"
-        exit 1
-    else
-        header_info
-        other_kernel
-        kernel_info
-        kernel_clean
-    fi
-}
-
-  VERSION="$(awk -F'=' '/^VERSION_CODENAME=/{ print $NF }' /etc/os-release)"
-function other_kernel() {
-  if [[ "${VERSION}" == "bullseye" ]]; then      
-    if [[ "$current_kernel" == *"edge"* ]]; then
-        echo -e "\n${CROSS} ${RD}ERROR:${CL} Proxmox ${BL}${current_kernel}${CL} Kernel Active"
-        echo -e "\nAn Active PVE Kernel is required to use Kernel Clean\n"
-        exit 1
-    fi
-    if [[ "$current_kernel" == *"6.1"* || "$current_kernel" == *"6.2"* ]]; then
-        echo -e "\n${CROSS} ${RD}ERROR:${CL} Proxmox ${BL}${current_kernel}${CL} Kernel Active"
-        echo -e "\nThe script cannot be used when running opt-in kernels. \nProxmox VE's package management relies directly on the current default kernel, which is 5.15. \nTherefore, it is not possible to utilize this script. In this case, you should use autoremove instead. \n`apt-get autoremove`\n"
-        exit 1
-    fi
-  fi
-}
-
-function kernel_info() {
-    if [[ "$MODE" != "PBS" ]]; then
-        echo -e "${YW}PVE Version: ${BL}$(pveversion)\n${CL}"
-    fi
-    if [[ "$current_kernel" == *"pve"* ]]; then
-        echo -e "${YW}Current Kernel: ${BL}$current_kernel\n${CL}"
-    else
-        echo -e "\n${CROSS} ${RD}ERROR: No PVE Kernel Found\n${CL}"
-        exit 1
-    fi
-}
+whiptail --backtitle "Proxmox VE Helper Scripts" --title "Proxmox VE Kernel Clean" --yesno "This will Clean Unused Kernel Images, USE AT YOUR OWN RISK. Proceed?" 10 68 || exit
 
-function kernel_clean() {
-    kernels=$(dpkg --list | grep 'kernel-.*-pve' | awk '{print $2}' | sort -V)
-    remove_kernels=""
-    for kernel in $kernels; do
-        if [ "$(echo $kernel | grep $current_kernel)" ]; then
-            break
-        else
-            echo -e "${BL}'$kernel' ${CL}${YW}has been added to the remove Kernel list\n${CL}"
-            remove_kernels+=" $kernel"
-        fi
-    done
-    msg_ok "Kernel Search Completed\n"
-    if [[ "$remove_kernels" != *"pve"* ]]; then
-        echo -e "${PARTY}  ${GN}It appears there are no old Kernels on your system. \n${CL}"
+if [ -z "$available_kernels" ]; then
+  whiptail --backtitle "Proxmox VE Helper Scripts" --title "No Other Kernels" --msgbox "It appears there are no old Kernels on your system. \nCurrent kernel ($current_kernel)." 10 68
   msg_info "Exiting"
   sleep 2
   msg_ok "Done"
-    else
-        read -p "Would you like to remove the $(echo $remove_kernels | awk '{print NF}') selected Kernels listed above? [y/n]: " -n 1 -r
-        echo
-        if [[ $REPLY =~ ^[Yy]$ ]]; then
-            msg_info "Removing ${CL}${RD}$(echo $remove_kernels | awk '{print NF}') ${CL}${YW}old Kernels${CL}"
-            /usr/bin/apt purge -y $remove_kernels >/dev/null 2>&1
-            msg_ok "Successfully Removed Kernels"
-            msg_info "Updating GRUB"
-            /usr/sbin/update-grub >/dev/null 2>&1
-            msg_ok "Successfully Updated GRUB"
-            msg_info "Exiting"
-            sleep 2
-            msg_ok "Done"
-        else
-            msg_info "Exiting"
-            sleep 2
-            msg_ok "Done"
-        fi
-    fi
-}
-
-if ! command -v pveversion >/dev/null 2>&1; then
-    echo -e " Switching to PBS mode"
-    MODE="PBS"
-    sleep 2
-else
-    MODE="PVE"
+  exit
 fi
 
-check_root
+while read -r TAG ITEM; do
+  OFFSET=2
+  ((${#ITEM} + OFFSET > MSG_MAX_LENGTH)) && MSG_MAX_LENGTH=${#ITEM}+OFFSET
+  CTID_MENU+=("$TAG" "$ITEM " "OFF")
+done < <(echo "$available_kernels")
+
+remove_kernels=$(whiptail --backtitle "Proxmox VE Helper Scripts" --title "Current Kernel $current_kernel" --checklist "\nSelect Kernels to remove:\n" \
+  16 $((MSG_MAX_LENGTH + 58)) 6 "${CTID_MENU[@]}" 3>&1 1>&2 2>&3 | tr -d '"') || exit
+
+whiptail --backtitle "Proxmox VE Helper Scripts" --title "Remove Kernels" --yesno "Would you like to remove the selected Kernels\n $remove_kernels\n Proceed?" 10 68 || exit
+
+msg_info "Removing ${CL}${RD}$(echo $remove_kernels | awk '{print NF}') ${CL}${YW}old Kernels${CL}"
+/usr/bin/apt purge -y $remove_kernels >/dev/null 2>&1
+msg_ok "Successfully Removed Kernels"
+
+msg_info "Updating GRUB"
+/usr/sbin/update-grub >/dev/null 2>&1
+msg_ok "Successfully Updated GRUB"
+msg_info "Exiting"
+sleep 2
+msg_ok "Done"