aa8108f74b
Fixed the following issues : - The password prompt went all wonky if password was wrong - The password prompt did not use system password cache and kept asking for password again on reruns of script - Installation failed if the script was initially ran by "sudo" due to inability to preserve ownership of files during cp commands - Security issue where the password prompt could be interrupted (Ctrl+C) and rerun without sleeping for few seconds first just like normal sudo failure in terminal - The sudo command would no longer accept password from ANY commands system-wide if wrong password was given to the install script enough times!! (the only fix would be restarting the system)
509 lines
15 KiB
Bash
Executable file
509 lines
15 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# Grub2 Themes
|
|
set -o errexit
|
|
|
|
[ GLOBAL::CONF ]
|
|
{
|
|
readonly ROOT_UID=0
|
|
readonly Project_Name="GRUB2::THEMES"
|
|
readonly MAX_DELAY=20 # max delay for user to enter root password
|
|
tui_root_login=
|
|
|
|
THEME_DIR="/usr/share/grub/themes"
|
|
REO_DIR="$(cd $(dirname $0) && pwd)"
|
|
}
|
|
|
|
#COLORS
|
|
CDEF=" \033[0m" # default color
|
|
CCIN=" \033[0;36m" # info color
|
|
CGSC=" \033[0;32m" # success color
|
|
CRER=" \033[0;31m" # error color
|
|
CWAR=" \033[0;33m" # waring color
|
|
b_CDEF=" \033[1;37m" # bold default color
|
|
b_CCIN=" \033[1;36m" # bold info color
|
|
b_CGSC=" \033[1;32m" # bold success color
|
|
b_CRER=" \033[1;31m" # bold error color
|
|
b_CWAR=" \033[1;33m" # bold warning color
|
|
|
|
# echo like ... with flag type and display message colors
|
|
prompt () {
|
|
case ${1} in
|
|
"-s"|"--success")
|
|
echo -e "${b_CGSC}${@/-s/}${CDEF}";; # print success message
|
|
"-e"|"--error")
|
|
echo -e "${b_CRER}${@/-e/}${CDEF}";; # print error message
|
|
"-w"|"--warning")
|
|
echo -e "${b_CWAR}${@/-w/}${CDEF}";; # print warning message
|
|
"-i"|"--info")
|
|
echo -e "${b_CCIN}${@/-i/}${CDEF}";; # print info message
|
|
*)
|
|
echo -e "$@"
|
|
;;
|
|
esac
|
|
}
|
|
|
|
# Check command availability
|
|
function has_command() {
|
|
command -v $1 > /dev/null
|
|
}
|
|
|
|
usage() {
|
|
printf "%s\n" "Usage: ${0##*/} [OPTIONS...]"
|
|
printf "\n%s\n" "OPTIONS:"
|
|
printf " %-25s%s\n" "-b, --boot" "install grub theme into /boot/grub/themes"
|
|
printf " %-25s%s\n" "-l, --slaze" "slaze grub theme"
|
|
printf " %-25s%s\n" "-s, --stylish" "stylish grub theme"
|
|
printf " %-25s%s\n" "-t, --tela" "tela grub theme"
|
|
printf " %-25s%s\n" "-v, --vimix" "vimix grub theme"
|
|
printf " %-25s%s\n" "-w, --white" "Install white icon version"
|
|
printf " %-25s%s\n" "-u, --ultrawide" "Install 2560x1080 background image - not available for slaze grub theme"
|
|
printf " %-25s%s\n" "-C, --custom-background" "Use either background.jpg or custom-background.jpg as theme background instead"
|
|
printf " %-25s%s\n" "-2, --2k" "Install 2k(2560x1440) background image"
|
|
printf " %-25s%s\n" "-4, --4k" "Install 4k(3840x2160) background image"
|
|
printf " %-25s%s\n" "-r, --remove" "Remove theme (must add theme name option)"
|
|
printf " %-25s%s\n" "-h, --help" "Show this help"
|
|
}
|
|
|
|
install() {
|
|
if [[ ${theme} == 'slaze' ]]; then
|
|
local name="Slaze"
|
|
elif [[ ${theme} == 'stylish' ]]; then
|
|
local name="Stylish"
|
|
elif [[ ${theme} == 'tela' ]]; then
|
|
local name="Tela"
|
|
elif [[ ${theme} == 'vimix' ]]; then
|
|
local name="Vimix"
|
|
else
|
|
prompt -i "\n Run ./install.sh -h for help or install dialog"
|
|
install_dialog
|
|
prompt -i "\n Run ./install.sh again!"
|
|
exit 0
|
|
fi
|
|
|
|
if [[ ${screen} == '2k' ]]; then
|
|
local screen="2k"
|
|
elif [[ ${screen} == '4k' ]]; then
|
|
local screen="4k"
|
|
elif [[ ${screen} == '1080p_21:9' ]]; then
|
|
local screen="1080p_21:9"
|
|
else
|
|
local screen="1080p"
|
|
fi
|
|
|
|
if [[ ${screen} == '1080p_21:9' && ${name} == 'Slaze' ]]; then
|
|
prompt -e "ultrawide 1080p does not support Slaze theme"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ ${custom_background} == 'custom-background' ]]; then
|
|
local custom_background="custom-background"
|
|
else
|
|
local custom_background="default-background"
|
|
fi
|
|
|
|
if [[ ${icon} == 'white' ]]; then
|
|
local icon="white"
|
|
else
|
|
local icon="color"
|
|
fi
|
|
|
|
# Check for root access and proceed if it is present
|
|
if [ "$UID" -eq "$ROOT_UID" ]; then
|
|
clear
|
|
|
|
if [[ "${custom_background}" == "custom-background" ]]; then
|
|
if [[ -f "background.jpg" ]]; then
|
|
custom_background="background.jpg"
|
|
elif [[ -f "custom-background.jpg" ]]; then
|
|
custom_background="custom-background.jpg"
|
|
else
|
|
prompt -e "Neither background.jpg, or custom-background.jpg could be found, exiting"
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
# Create themes directory if it didn't exist
|
|
echo -e "\n Checking for the existence of themes directory..."
|
|
|
|
[[ -d "${THEME_DIR}/${name}" ]] && rm -rf "${THEME_DIR}/${name}"
|
|
mkdir -p "${THEME_DIR}/${name}"
|
|
|
|
# Copy theme
|
|
prompt -i "\n Installing ${name} ${icon} ${screen} theme..."
|
|
|
|
# Don't preserve ownership because the owner will be root, and that causes the script to crash if it is ran from terminal by sudo
|
|
cp -a --no-preserve=ownership "${REO_DIR}/common/"{*.png,*.pf2} "${THEME_DIR}/${name}"
|
|
cp -a --no-preserve=ownership "${REO_DIR}/config/theme-${screen}.txt" "${THEME_DIR}/${name}/theme.txt"
|
|
|
|
if [[ ${custom_background} == "background.jpg" ]] || [[ ${custom_background} == "custom-background.jpg" ]]; then
|
|
if [[ -f "$custom_background" ]]; then
|
|
prompt -i "\n Using ${custom_background} as background image..."
|
|
cp -a --no-preserve=ownership "${REO_DIR}/${custom_background}" "${THEME_DIR}/${name}/background.jpg"
|
|
convert -auto-orient "${THEME_DIR}/${name}/background.jpg" "${THEME_DIR}/${name}/background.jpg"
|
|
else
|
|
prompt -e "$custom_background couldn't be found, exiting"
|
|
exit 0
|
|
fi
|
|
else
|
|
cp -a --no-preserve=ownership "${REO_DIR}/backgrounds/${screen}/background-${theme}.jpg" "${THEME_DIR}/${name}/background.jpg"
|
|
fi
|
|
|
|
if [[ ${screen} == '1080p_21:9' ]]; then
|
|
cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/icons-1080p" "${THEME_DIR}/${name}/icons"
|
|
cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/select-1080p/"*.png "${THEME_DIR}/${name}"
|
|
else
|
|
cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/icons-${screen}" "${THEME_DIR}/${name}/icons"
|
|
cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/select-${screen}/"*.png "${THEME_DIR}/${name}"
|
|
fi
|
|
|
|
# Set theme
|
|
prompt -i "\n Setting ${name} as default..."
|
|
|
|
# Backup grub config
|
|
cp -an /etc/default/grub /etc/default/grub.bak
|
|
|
|
if grep "GRUB_THEME=" /etc/default/grub 2>&1 >/dev/null; then
|
|
#Replace GRUB_THEME
|
|
sed -i "s|.*GRUB_THEME=.*|GRUB_THEME=\"${THEME_DIR}/${name}/theme.txt\"|" /etc/default/grub
|
|
else
|
|
#Append GRUB_THEME
|
|
echo "GRUB_THEME=\"${THEME_DIR}/${name}/theme.txt\"" >> /etc/default/grub
|
|
fi
|
|
|
|
# Make sure the right resolution for grub is set
|
|
if [[ ${screen} == '1080p' ]]; then
|
|
gfxmode="GRUB_GFXMODE=1920x1080,auto"
|
|
elif [[ ${screen} == '1080p_21:9' ]]; then
|
|
gfxmode="GRUB_GFXMODE=2560x1080,auto"
|
|
elif [[ ${screen} == '4k' ]]; then
|
|
gfxmode="GRUB_GFXMODE=3840x2160,auto"
|
|
elif [[ ${screen} == '2k' ]]; then
|
|
gfxmode="GRUB_GFXMODE=2560x1440,auto"
|
|
fi
|
|
|
|
if grep "GRUB_GFXMODE=" /etc/default/grub 2>&1 >/dev/null; then
|
|
#Replace GRUB_GFXMODE
|
|
sed -i "s|.*GRUB_GFXMODE=.*|${gfxmode}|" /etc/default/grub
|
|
else
|
|
#Append GRUB_GFXMODE
|
|
echo "${gfxmode}" >> /etc/default/grub
|
|
fi
|
|
|
|
if grep "GRUB_TERMINAL=console" /etc/default/grub 2>&1 >/dev/null || grep "GRUB_TERMINAL=\"console\"" /etc/default/grub 2>&1 >/dev/null; then
|
|
#Replace GRUB_TERMINAL
|
|
sed -i "s|.*GRUB_TERMINAL=.*|#GRUB_TERMINAL=console|" /etc/default/grub
|
|
fi
|
|
|
|
if grep "GRUB_TERMINAL_OUTPUT=console" /etc/default/grub 2>&1 >/dev/null || grep "GRUB_TERMINAL_OUTPUT=\"console\"" /etc/default/grub 2>&1 >/dev/null; then
|
|
#Replace GRUB_TERMINAL_OUTPUT
|
|
sed -i "s|.*GRUB_TERMINAL_OUTPUT=.*|#GRUB_TERMINAL_OUTPUT=console|" /etc/default/grub
|
|
fi
|
|
|
|
# Update grub config
|
|
prompt -i "\n Updating grub config...\n"
|
|
|
|
updating_grub
|
|
prompt -w "\n * At the next restart of your computer you will see your new Grub theme: '$theme' "
|
|
else
|
|
#Check if password is cached (if cache timestamp not expired yet)
|
|
sudo -n true 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#No need to ask for password
|
|
sudo "$0" --${theme} --${icon} --${screen}
|
|
|
|
else
|
|
#Ask for password
|
|
|
|
if [[ -n ${tui_root_login} ]] ; then
|
|
if [[ -n "${theme}" && -n "${screen}" ]]; then
|
|
sudo -S $0 --${theme} --${icon} --${screen} <<< ${tui_root_login}
|
|
fi
|
|
else
|
|
prompt -e "\n [ Error! ] -> Run me as root! "
|
|
read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
|
|
|
|
sudo -S echo <<< $REPLY 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#Correct password, use with sudo's stdin
|
|
sudo -S "$0" --${theme} --${icon} --${screen} <<< ${REPLY}
|
|
else
|
|
#block for 3 seconds before allowing another attempt
|
|
sleep 3
|
|
prompt -e "\n [ Error! ] -> Incorrect password!\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
fi
|
|
fi
|
|
}
|
|
|
|
run_dialog() {
|
|
if [[ -x /usr/bin/dialog ]]; then
|
|
if [[ "$UID" -ne "$ROOT_UID" ]]; then
|
|
#Check if password is cached (if cache timestamp not expired yet)
|
|
sudo -n true 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#No need to ask for password
|
|
sudo $0
|
|
else
|
|
#Ask for password
|
|
tui_root_login=$(dialog --backtitle ${Project_Name} \
|
|
--title "ROOT LOGIN" \
|
|
--insecure \
|
|
--passwordbox "require root permission" 8 50 \
|
|
--output-fd 1 )
|
|
|
|
sudo -S echo <<< $tui_root_login 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#Correct password, use with sudo's stdin
|
|
sudo -S "$0" <<< $tui_root_login
|
|
else
|
|
#block for 3 seconds before allowing another attempt
|
|
sleep 3
|
|
clear
|
|
prompt -e "\n [ Error! ] -> Incorrect password!\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
tui=$(dialog --backtitle ${Project_Name} \
|
|
--radiolist "Choose your Grub theme : " 15 40 5 \
|
|
1 "Vimix Theme" off \
|
|
2 "Tela Theme" on \
|
|
3 "Stylish Theme" off \
|
|
4 "Slaze Theme" off --output-fd 1 )
|
|
case "$tui" in
|
|
1) theme="vimix" ;;
|
|
2) theme="tela" ;;
|
|
3) theme="stylish" ;;
|
|
4) theme="slaze" ;;
|
|
*) operation_canceled ;;
|
|
esac
|
|
|
|
tui=$(dialog --backtitle ${Project_Name} \
|
|
--radiolist "Choose icon style : " 15 40 5 \
|
|
1 "white" off \
|
|
2 "color" on --output-fd 1 )
|
|
case "$tui" in
|
|
1) icon="white" ;;
|
|
2) icon="color" ;;
|
|
*) operation_canceled ;;
|
|
esac
|
|
|
|
tui=$(dialog --backtitle ${Project_Name} \
|
|
--radiolist "Choose your Display Resolution : " 15 40 5 \
|
|
1 "1080p (1920x1080)" on \
|
|
2 "1080p ultrawide (2560x1080)" off \
|
|
3 "2k (2560x1440)" off \
|
|
4 "4k (3840x2160)" off --output-fd 1 )
|
|
case "$tui" in
|
|
1) screen="1080p" ;;
|
|
2) screen="1080p_21:9" ;;
|
|
3) screen="2k" ;;
|
|
4) screen="4k" ;;
|
|
*) operation_canceled ;;
|
|
esac
|
|
fi
|
|
}
|
|
|
|
operation_canceled() {
|
|
clear
|
|
prompt -i "\n Operation canceled by user, Bye!"
|
|
exit 1
|
|
}
|
|
|
|
updating_grub() {
|
|
if has_command update-grub; then
|
|
update-grub
|
|
elif has_command grub-mkconfig; then
|
|
grub-mkconfig -o /boot/grub/grub.cfg
|
|
elif has_command zypper; then
|
|
grub2-mkconfig -o /boot/grub2/grub.cfg
|
|
elif has_command dnf; then
|
|
grub2-mkconfig -o /boot/efi/EFI/fedora/grub.cfg
|
|
fi
|
|
|
|
# Success message
|
|
prompt -s "\n * All done!"
|
|
}
|
|
|
|
install_dialog() {
|
|
if [ ! "$(which dialog 2> /dev/null)" ]; then
|
|
prompt -i "\n 'dialog' needs to be installed for this shell"
|
|
if has_command zypper; then
|
|
sudo zypper in dialog
|
|
elif has_command apt-get; then
|
|
sudo apt-get install dialog
|
|
elif has_command dnf; then
|
|
sudo dnf install -y dialog
|
|
elif has_command yum; then
|
|
sudo yum install dialog
|
|
elif has_command pacman; then
|
|
sudo pacman -S --noconfirm dialog
|
|
fi
|
|
fi
|
|
}
|
|
|
|
remove() {
|
|
if [[ ${theme} == 'slaze' ]]; then
|
|
local name="Slaze"
|
|
elif [[ ${theme} == 'stylish' ]]; then
|
|
local name="Stylish"
|
|
elif [[ ${theme} == 'tela' ]]; then
|
|
local name="Tela"
|
|
elif [[ ${theme} == 'vimix' ]]; then
|
|
local name="Vimix"
|
|
else
|
|
prompt -i "\n Run ./install.sh -h for help!"
|
|
exit 0
|
|
fi
|
|
|
|
# Check for root access and proceed if it is present
|
|
if [ "$UID" -eq "$ROOT_UID" ]; then
|
|
echo -e "\n Checking for the existence of themes directory..."
|
|
if [[ -d "${THEME_DIR}/${name}" ]]; then
|
|
rm -rf "${THEME_DIR}/${name}"
|
|
else
|
|
prompt -i "\n ${name} grub theme not exist!"
|
|
exit 0
|
|
fi
|
|
|
|
# Backup grub config
|
|
if [[ -f /etc/default/grub.bak ]]; then
|
|
rm -rf /etc/default/grub && mv /etc/default/grub.bak /etc/default/grub
|
|
else
|
|
prompt -i "\n grub.bak not exist!"
|
|
exit 0
|
|
fi
|
|
|
|
# Update grub config
|
|
prompt -i "\n Resetting grub theme...\n"
|
|
updating_grub
|
|
|
|
else
|
|
#Check if password is cached (if cache timestamp not expired yet)
|
|
sudo -n true 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#No need to ask for password
|
|
sudo "$0" "${PROG_ARGS[@]}"
|
|
else
|
|
#Ask for password
|
|
prompt -e "\n [ Error! ] -> Run me as root! "
|
|
read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
|
|
|
|
sudo -S echo <<< $REPLY 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#Correct password, use with sudo's stdin
|
|
sudo -S "$0" "${PROG_ARGS[@]}" <<< $REPLY
|
|
else
|
|
#block for 3 seconds before allowing another attempt
|
|
sleep 3
|
|
clear
|
|
prompt -e "\n [ Error! ] -> Incorrect password!\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
fi
|
|
}
|
|
|
|
# Show terminal user interface for better use
|
|
if [[ $# -lt 1 ]] && [[ $UID -ne $ROOT_UID ]] && [[ -x /usr/bin/dialog ]] ; then
|
|
run_dialog
|
|
fi
|
|
|
|
if [[ $# -lt 1 ]] && [[ $UID -ne $ROOT_UID ]] && [[ ! -x /usr/bin/dialog ]] ; then
|
|
#Check if password is cached (if cache timestamp not expired yet)
|
|
sudo -n true 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#No need to ask for password
|
|
exec sudo $0
|
|
else
|
|
#Ask for password
|
|
prompt -e "\n [ Error! ] -> Run me as root! "
|
|
read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
|
|
|
|
sudo -S echo <<< $REPLY 2> /dev/null && echo
|
|
|
|
if [[ $? == 0 ]]; then
|
|
#Correct password, use with sudo's stdin
|
|
sudo $0 <<< $REPLY
|
|
else
|
|
#block for 3 seconds before allowing another attempt
|
|
sleep 3
|
|
prompt -e "\n [ Error! ] -> Incorrect password!\n"
|
|
exit 1
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
while [[ $# -ge 1 ]]; do
|
|
PROG_ARGS+=("${1}")
|
|
case "${1}" in
|
|
-b|--boot)
|
|
THEME_DIR="/boot/grub/themes"
|
|
;;
|
|
-l|--slaze)
|
|
theme='slaze'
|
|
;;
|
|
-s|--stylish)
|
|
theme='stylish'
|
|
;;
|
|
-t|--tela)
|
|
theme='tela'
|
|
;;
|
|
-v|--vimix)
|
|
theme='vimix'
|
|
;;
|
|
-w|--white)
|
|
icon='white'
|
|
;;
|
|
-c|--color)
|
|
icon='color'
|
|
;;
|
|
-1|--1080p)
|
|
screen='1080p'
|
|
;;
|
|
-2|--2k)
|
|
screen='2k'
|
|
;;
|
|
-4|--4k)
|
|
screen='4k'
|
|
;;
|
|
-u|--ultrawide|--1080p_21:9)
|
|
screen='1080p_21:9'
|
|
;;
|
|
-C|--custom-background|--custom)
|
|
custom_background='custom-background'
|
|
;;
|
|
-r|--remove)
|
|
remove='true'
|
|
;;
|
|
-h|--help)
|
|
usage
|
|
exit 0
|
|
;;
|
|
*)
|
|
prompt -e "\n ERROR: Unrecognized installation option '$1'."
|
|
prompt -i "Try '$0 --help' for more information."
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
if [[ "${remove:-}" != 'true' ]]; then
|
|
install
|
|
elif [[ "${remove:-}" == 'true' ]]; then
|
|
remove
|
|
fi
|
|
|
|
exit 0
|