#!/usr/bin/env bash
# bash -c "$(wget -qLO - https://github.com/tteck/Proxmox/raw/main/misc/edge-kernel.sh)"
set -e
KERNEL_ON=$(uname -r)
PVE_KERNEL=$(dpkg --list | grep 'kernel-.*-pve' | awk '{print substr($2, 12, length($2)-1) }' | tac | head -n 1)
EDGE_KERNEL=$(dpkg --list | grep 'kernel-.*-edge' | awk '{print substr($2, 12, length($2)-1) }' | tac | head -n 1)
clear
while true; do
    read -p "This is a Proxmox Edge Kernel Tool, USE AT YOUR OWN RISK. Proceed(y/n)?" yn
    case $yn in
    [Yy]*) break ;;
    [Nn]*) exit ;;
    *) echo "Please answer yes or no." ;;
    esac
done
clear

show_menu() {
    normal=$(echo "\033[m")
    safe=$(echo "\033[32m")
    menu=$(echo "\033[36m")
    number=$(echo "\033[33m")
    bgred=$(echo "\033[41m")
    fgred=$(echo "\033[31m")
    proxmox-boot-tool kernel list
    echo -e "\nCurrent Kernel: ${menu}${KERNEL_ON}${normal}"
    printf "\n${menu}*********************************************${normal}\n"
    printf "${menu}**${number} 1)${normal} Install Proxmox Edge Kernel & Reboot\n"
    printf "${menu}**${number} 2)${normal} Switch to Proxmox VE 7 ${menu}${PVE_KERNEL}${normal} Kernel & Reboot\n"
    printf "${menu}**${number} 3)${normal} Switch to Proxmox Edge ${menu}${EDGE_KERNEL}${normal} Kernel & Reboot\n"
    printf "${menu}**${number} 4)${normal} Unpin Current Kernel\n"
    printf "${menu}**${number} 5)${normal} Remove Proxmox Edge Kernel & Reboot\n"
    printf "${menu}*********************************************${normal}\n"
    printf "Please choose an option from the menu and enter or ${fgred}x${normal} to exit."
    read opt
}
option_picked() {
    msgcolor=$(echo "\033[01;31m")
    normal=$(echo "\033[00;00m")
    message=${@:-"${normal}Error: No message passed"}
    printf "${msgcolor}${message}${normal}\n"
}
clear
show_menu
while [ $opt != '' ]; do
    if [ $opt = '' ]; then
        exit
    else
        case $opt in
        1)
            while true; do
                read -p "Are you sure you want to Install Proxmox Edge Kernel & Reboot? Proceed(y/n)?" yn
                case $yn in
                [Yy]*) break ;;
                [Nn]*) exit ;;
                *) echo "Please answer yes or no." ;;
                esac
            done
            clear
            option_picked "Installing Proxmox Edge Kernel & Rebooting"
            apt-get install -y gnupg
            curl -1sLf 'https://dl.cloudsmith.io/public/pve-edge/kernel/gpg.8EC01CCF309B98E7.key' | apt-key add -
            echo "deb https://dl.cloudsmith.io/public/pve-edge/kernel/deb/debian bullseye main" >/etc/apt/sources.list.d/pve-edge-kernel.list
            apt-get -y update
            apt-get -y install pve-kernel-6.0-edge
            reboot
            break
            ;;
        2)
            while true; do
                read -p "Are you sure you want to Switch to Proxmox VE 7 ${PVE_KERNEL} Kernel & Reboot? Proceed(y/n)?" yn
                case $yn in
                [Yy]*) break ;;
                [Nn]*) exit ;;
                *) echo "Please answer yes or no." ;;
                esac
            done
            clear
            option_picked "Switching to Proxmox VE 7 Kernel & Rebooting"
            proxmox-boot-tool kernel pin ${PVE_KERNEL}
            reboot
            break
            ;;
        3)
            while true; do
                read -p "Are you sure you want to Switch to Proxmox ${EDGE_KERNEL} Edge Kernel & Reboot? Proceed(y/n)?" yn
                case $yn in
                [Yy]*) break ;;
                [Nn]*) exit ;;
                *) echo "Please answer yes or no." ;;
                esac
            done
            clear
            option_picked "Switching to Proxmox Edge Kernel & Rebooting"
            proxmox-boot-tool kernel pin ${EDGE_KERNEL}
            reboot
            break
            ;;
        4)
            while true; do
                read -p "Are you sure you want to Unpin the Current Kernel? Proceed(y/n)?" yn
                case $yn in
                [Yy]*) break ;;
                [Nn]*) exit ;;
                *) echo "Please answer yes or no." ;;
                esac
            done
            clear
            option_picked "Unpinning Current Kernel"
            proxmox-boot-tool kernel unpin
            clear
            break
            ;;
        5)
            while true; do
                read -p "Are you sure you want to Remove Proxmox Edge Kernel & Reboot? Proceed(y/n)?" yn
                case $yn in
                [Yy]*) break ;;
                [Nn]*) exit ;;
                *) echo "Please answer yes or no." ;;
                esac
            done
            clear
            option_picked "Removing Proxmox Edge Kernel & Rebooting"
            apt-get purge -y ${EDGE_KERNEL}
            rm -rf /etc/apt/sources.list.d/pve-edge-kernel.list
            proxmox-boot-tool kernel unpin
            reboot
            break
            ;;
        x)
            exit
            ;;
        \n)
            exit
            ;;
        *)
            clear
            option_picked "Please choose an option from the menu"
            show_menu
            ;;
        esac
    fi
done

show_menu