setup-main.sh

#!/bin/bash
# shellcheck disable=SC2034

# shellcheck disable=SC2128
scriptDir=$(dirname -- "$(readlink -f -- "$BASH_SOURCE")")
repoDir="$(dirname "$scriptDir")"
if [[ -d $repoDir/.git && -f $repoDir/scripts/setup-main.sh ]]; then
    BASE_REPO_LOCATION=$repoDir/
else
    BASE_REPO_LOCATION="https://raw.githubusercontent.com/krish-gh/linux-setup/main/"
fi

command_exists() {
    command -v "$1" >/dev/null 2>&1
}

DISTRO_TYPE=''
PKG_MGR=''
command_exists pacman && PKG_MGR=pacman && DISTRO_TYPE=arch
command_exists apt && PKG_MGR=apt && DISTRO_TYPE=debian
command_exists dnf && PKG_MGR=dnf && DISTRO_TYPE=fedora
command_exists zypper && PKG_MGR=zypper && DISTRO_TYPE=opensuse

if [[ $DISTRO_TYPE == '' ]]; then
    >&2 echo "You are not running supported Linux distrbution..."
    exit 1
fi

if ! command_exists curl; then
    >&2 echo "curl required, but not found..."
    exit 2
fi

DIST_ID=''
# shellcheck disable=SC1091
[[ -f /etc/os-release ]] && source /etc/os-release && DIST_ID=$ID

# shellcheck disable=SC2086
DESKTOP=$(echo ${XDG_CURRENT_DESKTOP##*:} | tr '[:upper:]' '[:lower:]' | sed 's/^x-//')
CURRENT_TERMINAL=$(ps -p $PPID -o comm= | sed 's/-$//')

echo -e "#################################################################"
echo -e "BASE_REPO_LOCATION=$BASE_REPO_LOCATION"
echo -e "DISTRO_TYPE=$DISTRO_TYPE"
echo -e "PACKAGE_MANAGER=$PKG_MGR"
echo -e "DESKTOP=$DESKTOP"
echo -e "TERMINAL=$CURRENT_TERMINAL"
echo -e "DISTRO_ID=$DIST_ID"
echo -e "#################################################################"
cat /etc/os-release
echo -e "#################################################################"

REFRESH_CMD=""        #override from DISTRO_TYPE specific script
UPDATE_CMD=""         #override from DISTRO_TYPE specific script
INSTALL_CMD=""        #override from DISTRO_TYPE specific script
UNINSTALL_CMD=""      #override from DISTRO_TYPE specific script
UNINSTALL_ONLY_CMD="" #override from DISTRO_TYPE specific script

FLATPAK_INSTALL_CMD="flatpak install --assumeyes flathub" #override from DISTRO_TYPE specific script
FLATPAK_UPDATE_CMD="flatpak update --assumeyes" #override from DISTRO_TYPE specific script

REQUIREMENTS=""                 #override from DISTRO_TYPE specific script
SYSTEM_PACKAGES_TO_INSTALL=""   #override from DISTRO_TYPE specific script
INTEL_PACKAGES_TO_INSTALL=""    #override from DISTRO_TYPE specific script
VMWARE_PACKAGES_TO_INSTALL=""   #override from DISTRO_TYPE specific script
VBOX_PACKAGES_TO_INSTALL=""     #override from DISTRO_TYPE specific script
HYPERV_PACKAGES_TO_INSTALL=""   #override from DISTRO_TYPE specific script
VIRT_PACKAGES_TO_INSTALL=""     #override from DISTRO_TYPE specific script
FONTS_TO_INSTALL=""             #override from DISTRO_TYPE specific script
TERM_PACKAGES_TO_INSTALL=""     #override from DISTRO_TYPE specific script
APP_PACKAGES_TO_INSTALL=""      #override from DISTRO_TYPE specific script
DEV_PACKAGES_TO_INSTALL=""      #override from DISTRO_TYPE specific script
GTK_PACKAGES_TO_INSTALL=""      #override from DISTRO_TYPE specific script
QT_PACKAGES_TO_INSTALL=""       #override from DISTRO_TYPE specific script
QT_PATCHES_TO_INSTALL=""        #override from DISTRO_TYPE specific script
GNOME_PACKAGES_TO_INSTALL=""    #override from DISTRO_TYPE specific script
GNOME_EXT_MGR_PKG=""            #override from DISTRO_TYPE specific script
KDE_PACKAGES_TO_INSTALL=""      #override from DISTRO_TYPE specific script
CINNAMON_PACKAGES_TO_INSTALL="" #override from DISTRO_TYPE specific script
XFCE_PACKAGES_TO_INSTALL=""     #override from DISTRO_TYPE specific script
XFCE_MENU_LOGO=""               #override from DISTRO_TYPE specific script
PACKAGES_TO_REMOVE=""           #override from DISTRO_TYPE specific script

TERMINAL_TO_INSTALL=none
GUI_TEXT_EDITOR="" #override from desktop specific script

TEMP_DIR=/tmp/linux-setup
mkdir -p $TEMP_DIR

# arg1 = destination path, arg2 = source path
copy_file() {
    if [[ $2 == http* ]]; then
        curl -f -o "$1" "$2?$(date +%s)"
        curl_exit_status=$?
        [[ $curl_exit_status != 0 ]] && >&2 echo -e "Error downloading $2"
    else
        cp -f "$2" "$1"
    fi
}

# arg1 = source path
copy_content() {
    if [[ $1 == http* ]]; then
        curl -f "$1?$(date +%s)"
        curl_exit_status=$?
        [[ $curl_exit_status != 0 ]] && >&2 echo -e "Error downloading $1"
    else
        cat "$1"
    fi
}

refresh_package_sources() {
    eval "$REFRESH_CMD"
}

update_packages() {
    eval "$UPDATE_CMD"
    command_exists flatpak && $FLATPAK_UPDATE_CMD
}

install_pkgs() {
    #doing in loop to avoid abort in case something is wrong
    # shellcheck disable=SC2207
    pkgs=($(eval echo "$1"))
    for i in "${pkgs[@]}"; do eval "$INSTALL_CMD $i"; done
}

uninstall_pkgs() {
    #doing in loop to avoid abort in case something is wrong
    # shellcheck disable=SC2207
    pkgs=($(eval echo "$1"))
    for i in "${pkgs[@]}"; do eval "$UNINSTALL_CMD $i"; done
}

uninstall_only_pkgs() {
    #doing in loop to avoid abort in case something is wrong
    # shellcheck disable=SC2207
    pkgs=($(eval echo "$1"))
    for i in "${pkgs[@]}"; do eval "$UNINSTALL_ONLY_CMD $i"; done
}

debloat_pkgs() {
    echo -e "Debloating..."
    copy_file $TEMP_DIR/$DISTRO_TYPE.txt "${BASE_REPO_LOCATION}"debloat/$DISTRO_TYPE.txt
    while read -r pkg; do
        uninstall_pkgs "$pkg"
    done <$TEMP_DIR/$DISTRO_TYPE.txt
    rm -f $TEMP_DIR/$DISTRO_TYPE.txt

    if [[ $PACKAGES_TO_REMOVE != "" ]]; then
        echo -e "Removing additional packages..."
        uninstall_only_pkgs "$PACKAGES_TO_REMOVE"
    fi
}

# override with DISTRO_TYPE specific stuffs
echo -e "Executing common $DISTRO_TYPE specific script..."
copy_file $TEMP_DIR/"$DISTRO_TYPE".sh "${BASE_REPO_LOCATION}"distros/"$DISTRO_TYPE".sh
if [[ ! -f $TEMP_DIR/"$DISTRO_TYPE".sh ]]; then
    >&2 echo "Error: $DISTRO_TYPE specific script not found!"
    exit 3
fi
# shellcheck disable=SC1090
source $TEMP_DIR/"$DISTRO_TYPE".sh
rm -f $TEMP_DIR/"$DISTRO_TYPE".sh

# desktop environment specific stuffs
copy_file $TEMP_DIR/"$DESKTOP".sh "${BASE_REPO_LOCATION}"desktop/"$DESKTOP".sh
# shellcheck disable=SC1090
[[ -f $TEMP_DIR/"$DESKTOP".sh ]] && source $TEMP_DIR/"$DESKTOP".sh
rm -f $TEMP_DIR/"$DESKTOP".sh

# execute exact distro specic stuffs if exists e.g. linux mint, ubuntu, manjaro etc. Optional.
if [[ $DIST_ID != '' ]]; then
    copy_file $TEMP_DIR/"$DIST_ID".sh "${BASE_REPO_LOCATION}"specific/"$DIST_ID".sh
    # shellcheck disable=SC1090
    [[ -f $TEMP_DIR/"$DIST_ID".sh ]] && source $TEMP_DIR/"$DIST_ID".sh
    rm -f $TEMP_DIR/"$DIST_ID".sh
fi

setup_system() {
    install_pkgs "virt-what"
    SYSTEM_TO_SETUP=$(sudo virt-what)

    if [[ $SYSTEM_TO_SETUP != '' ]]; then
        echo -e "SYSTEM=$SYSTEM_TO_SETUP"

        case $SYSTEM_TO_SETUP in

        vmware)
            install_pkgs "$VMWARE_PACKAGES_TO_INSTALL"
            sudo systemctl enable --now vmtoolsd.service
            #sudo systemctl disable --now vmware-vmblock-fuse.service
            ;;

        virtualbox)
            install_pkgs "$VBOX_PACKAGES_TO_INSTALL"
            sudo systemctl enable --now vboxservice.service
            ;;

        hyperv)
            install_pkgs "$HYPERV_PACKAGES_TO_INSTALL"
            sudo systemctl enable --now hv_{fcopy,kvp,vss}_daemon.service
            ;;

        qemu | kvm | xen | virt)
            install_pkgs "$VIRT_PACKAGES_TO_INSTALL"
            ;;

        *)
            echo "Ahh! Taking a note..."
            ;;
        esac
    else
        # TODO detect bare metal
        SYSTEM_TO_SETUP=intel   
        echo -e "SYSTEM=$SYSTEM_TO_SETUP"

        case $SYSTEM_TO_SETUP in

        intel)
            install_pkgs "$INTEL_PACKAGES_TO_INSTALL"
            ;;

        *)
            echo "Ahh! Taking a note..."
            ;;
        esac 
    fi

    install_pkgs "$SYSTEM_PACKAGES_TO_INSTALL"

    echo -e "Tweaking some system stuffs..."
    sudo mkdir -p /etc/sysctl.d /etc/systemd/{journald.conf.d,coredump.conf.d}
    copy_content "${BASE_REPO_LOCATION}"system/etc/sysctl.d/999-sysctl.conf | sudo tee /etc/sysctl.d/999-sysctl.conf
    copy_content "${BASE_REPO_LOCATION}"system/etc/systemd/journald.conf.d/00-journal-size.conf | sudo tee /etc/systemd/journald.conf.d/00-journal-size.conf
    sudo journalctl --rotate --vacuum-size=10M
    copy_content "${BASE_REPO_LOCATION}"system/etc/systemd/coredump.conf.d/custom.conf | sudo tee /etc/systemd/coredump.conf.d/custom.conf

    # env var
    mkdir -p ~/.config/environment.d
    copy_file ~/.config/environment.d/10-defaults.conf "${BASE_REPO_LOCATION}"home/.config/environment.d/10-defaults.conf

    #mkdir -p ~/.config/systemd/user/service.d
    #copy_file ~/.config/systemd/user/service.d/env.conf ${BASE_REPO_LOCATION}home/.config/systemd/user/service.d/env.conf

    profileAppend="$(
        grep "~custom-setup~" ~/.profile >/dev/null 2>&1
        echo $?
    )"
    if [[ "${profileAppend}" -ne 0 ]]; then
        copy_content "${BASE_REPO_LOCATION}"home/.profile >>~/.profile
    fi

    xprofileAppend="$(
        grep "~custom-setup~" ~/.xprofile >/dev/null 2>&1
        echo $?
    )"
    if [[ "${xprofileAppend}" -ne 0 ]]; then
        copy_content "${BASE_REPO_LOCATION}"home/.xprofile >>~/.xprofile
    fi

    xsessionrcAppend="$(
        grep "~custom-setup~" ~/.xsessionrc >/dev/null 2>&1
        echo $?
    )"
    if [[ "${xsessionrcAppend}" -ne 0 ]]; then
        copy_content "${BASE_REPO_LOCATION}"home/.xsessionrc >>~/.xsessionrc
    fi

    # DO NOT have .xinitrc for now due to some problems
    #xinitrcAppend="$(
    #    grep "~custom-setup~" ~/.xinitrc >/dev/null 2>&1
    #    echo $?
    #)"
    #if [[ "${xinitrcAppend}" -ne 0 ]]; then
    #    copy_content ${BASE_REPO_LOCATION}home/.xinitrc >>~/.xinitrc
    #fi

    echo -e "Setting up keyring..."
    mkdir -p ~/.local/share/keyrings/
    copy_file ~/.local/share/keyrings/Default_keyring.keyring "${BASE_REPO_LOCATION}"home/.local/share/keyrings/Default_keyring.keyring
    copy_file ~/.local/share/keyrings/default "${BASE_REPO_LOCATION}"home/.local/share/keyrings/default
    chmod og= ~/.local/share/keyrings/
    chmod og= ~/.local/share/keyrings/Default_keyring.keyring

    echo -e "Updating some sudo stuffs..."
    sudo mkdir -p /etc/sudoers.d
    echo -e Defaults:"$(whoami)" \!authenticate | sudo tee /etc/sudoers.d/99-custom

    # autologin capability
    sudo groupadd -r autologin
    sudo gpasswd -a "$(whoami)" autologin

    systemctl is-enabled casper-md5check.service && sudo systemctl disable casper-md5check.service
    sudo systemctl daemon-reload
}

setup_font() {
    echo -e "Installing fonts..."
    install_pkgs "$FONTS_TO_INSTALL"
    echo -e "Making font look better..."
    mkdir -p ~/.config/fontconfig/conf.d
    copy_file ~/.config/fontconfig/fonts.conf "${BASE_REPO_LOCATION}"home/.config/fontconfig/fonts.conf
    copy_file ~/.config/fontconfig/conf.d/20-no-embedded.conf "${BASE_REPO_LOCATION}"home/.config/fontconfig/conf.d/20-no-embedded.conf
    copy_file ~/.Xresources "${BASE_REPO_LOCATION}"home/.Xresources
    xrdb -merge ~/.Xresources
    [[ -f /etc/profile.d/freetype2.sh ]] && sudo sed -i '/export FREETYPE_PROPERTIES=/s/^#//g' /etc/profile.d/freetype2.sh
    sudo ln -s /usr/share/fontconfig/conf.avail/10-sub-pixel-rgb.conf /etc/fonts/conf.d/
    sudo ln -s /usr/share/fontconfig/conf.avail/10-hinting-slight.conf /etc/fonts/conf.d/
    sudo ln -s /usr/share/fontconfig/conf.avail/11-lcdfilter-default.conf /etc/fonts/conf.d/
    [[ -f /usr/share/fontconfig/conf.avail/10-nerd-font-symbols.conf ]] && sudo ln -s /usr/share/fontconfig/conf.avail/10-nerd-font-symbols.conf /etc/fonts/conf.d/
    if [[ $(fc-list | grep -i "JetBrainsMono Nerd") == "" ]]; then
        echo -e "Installing Nerd Font manually as not found..."
        mkdir -p ~/.local/bin
        curl -fs https://ohmyposh.dev/install.sh | bash -s -- -d ~/.local/bin
        ~/.local/bin/oh-my-posh font install JetBrainsMono
    fi
    sudo fc-cache -fv
    fc-cache -fv
}

setup_terminal() {
    echo -e "Configuring shell stuffs..."
    install_pkgs "$TERM_PACKAGES_TO_INSTALL"
    if ! command_exists starship; then
        mkdir -p ~/.local/bin
        curl -fsS https://starship.rs/install.sh | sh -s -- -y --bin-dir ~/.local/bin
    fi
    #starship preset no-nerd-font -o ~/.config/starship.toml
    copy_file ~/.aliases "${BASE_REPO_LOCATION}"distros/$DISTRO_TYPE.aliases
    bashrcAppend="$(
        grep "~custom-setup~" ~/.bashrc >/dev/null 2>&1
        echo $?
    )"
    if [[ "${bashrcAppend}" -ne 0 ]]; then
        copy_content "${BASE_REPO_LOCATION}"home/.bashrc >>~/.bashrc
    fi

    # nano
    mkdir -p ~/.config/nano
    copy_file ~/.config/nano/nanorc "${BASE_REPO_LOCATION}"home/.config/nano/nanorc
    if [[ -d /usr/share/nano-syntax-highlighting/ ]]; then
        nanorcAppend="$(
            grep "nano-syntax-highlighting" ~/.config/nano/nanorc >/dev/null 2>&1
            echo $?
        )"
        if [[ "${nanorcAppend}" -ne 0 ]]; then
            echo -e 'include "/usr/share/nano-syntax-highlighting/*.nanorc"' >>~/.config/nano/nanorc
        fi
    fi

    # if fastfetch not found at this point fallback to neofetch, otherwise remove neofetch
    if ! which fastfetch; then
        install_pkgs neofetch
    else
        uninstall_pkgs neofetch
    fi

    # fastfetch
    #mkdir p ~/.config/fastfetch
    #copy_file ~/.config/fastfetch/config.jsonc ${BASE_REPO_LOCATION}home/.config/fastfetch/config.jsonc

    echo -e "Installing terminal $TERMINAL_TO_INSTALL..."
    case $TERMINAL_TO_INSTALL in

    alacritty)
        install_pkgs $TERMINAL_TO_INSTALL
        mkdir -p ~/.config/alacritty
        copy_file ~/.config/alacritty/catppuccin-mocha.toml https://raw.githubusercontent.com/catppuccin/alacritty/main/catppuccin-mocha.toml
        copy_file ~/.config/alacritty/alacritty.toml "${BASE_REPO_LOCATION}"home/.config/alacritty/alacritty.toml
        ;;

    kitty)
        install_pkgs $TERMINAL_TO_INSTALL
        mkdir -p ~/.config/kitty
        copy_file ~/.config/kitty/mocha.conf https://raw.githubusercontent.com/catppuccin/kitty/main/themes/mocha.conf
        copy_file ~/.config/kitty/kitty.conf "${BASE_REPO_LOCATION}"home/.config/kitty/kitty.conf
        ;;

    wezterm)
        install_pkgs $TERMINAL_TO_INSTALL
        mkdir -p ~/.config/wezterm
        copy_file ~/.config/wezterm/wezterm.lua "${BASE_REPO_LOCATION}"home/.config/wezterm/wezterm.lua
        ;;

    *)
        echo -e "No additional terminal installed..."
        ;;
    esac

    # gnome terminal
    if command_exists gnome-terminal; then
        tprofileid=$(gsettings get org.gnome.Terminal.ProfilesList default | tr -d "'")
        copy_file $TEMP_DIR/gterm.dconf "${BASE_REPO_LOCATION}"desktop/gterm.dconf
        sed -i "s/DEFAULT_PROFILE/$tprofileid/g" $TEMP_DIR/gterm.dconf
        dconf load /org/gnome/terminal/ <$TEMP_DIR/gterm.dconf
        rm -f $TEMP_DIR/gterm.dconf
    fi

    #source ~/.bashrc
}

setup_common_ui() {
    install_pkgs "$GTK_PACKAGES_TO_INSTALL"
    install_pkgs "$QT_PACKAGES_TO_INSTALL"

    gtktheme=$(gsettings get org.gnome.desktop.interface gtk-theme | tr -d \'\")
    echo -e "CURRENT_GTK_THEME=$gtktheme"
    # shellcheck disable=SC2086
    # shellcheck disable=SC2143
    # make it dark
    if [[ $gtktheme != '' && ! $(echo $gtktheme | grep -i dark) ]]; then
        gsettings set org.gnome.desktop.interface gtk-theme "$gtktheme"-dark
    fi

    copy_file $TEMP_DIR/common.dconf "${BASE_REPO_LOCATION}"desktop/common.dconf
    dconf load / <$TEMP_DIR/common.dconf
    rm -f $TEMP_DIR/common.dconf

    mkdir -p ~/.config/gtk-{3,4}.0
    if [[ ! -f ~/.config/gtk-3.0/settings.ini ]]; then
        echo -e "[Settings]" >~/.config/gtk-3.0/settings.ini && echo -e "#gtk-application-prefer-dark-theme=true" >>~/.config/gtk-3.0/settings.ini
    fi
    if [[ ! -f ~/.config/gtk-4.0/settings.ini ]]; then
        cp ~/.config/gtk-3.0/settings.ini ~/.config/gtk-4.0/
        echo -e "gtk-hint-font-metrics=1" >>~/.config/gtk-4.0/settings.ini
    fi

    mkdir -p ~/.local/share/gtksourceview-{3.0,4,5}/styles
    copy_file ~/.local/share/gtksourceview-3.0/styles/mocha.xml https://raw.githubusercontent.com/catppuccin/xed/main/src/mocha.xml
    for i in ~/.local/share/gtksourceview-{4,5}/styles; do
        cp -s -f ~/.local/share/gtksourceview-3.0/styles/mocha.xml "$i"
    done

    echo -e "Setting up QT apps to look like GTK.."
    mkdir -p ~/.config/Kvantum ~/.config/qt{5,6}ct
    copy_file ~/.config/Kvantum/kvantum.kvconfig "${BASE_REPO_LOCATION}"home/.config/Kvantum/kvantum.kvconfig
    for i in 5 6; do
        copy_file ~/.config/qt${i}ct/qt${i}ct.conf "${BASE_REPO_LOCATION}"home/.config/qt${i}ct/qt${i}ct.conf
    done

    # wallpaper
    #mkdir -p ~/.local/share/backgrounds
    #copy_file ~/.local/share/backgrounds/wallpaper ${BASE_REPO_LOCATION}home/.local/share/backgrounds/wallpaper

    if [[ -d /etc/lightdm ]]; then
        echo -e "Configuring lightdm stuffs..."
        grep -rl greeter-hide-users /etc/lightdm /usr/share/lightdm \
        | xargs sudo sed -i "/greeter-hide-users=true/c\greeter-hide-users=false
            /greeter-hide-users = true/c\greeter-hide-users = false"
    fi
}

setup_apps() {
    echo -e "Installing some apps..."
    install_pkgs "$APP_PACKAGES_TO_INSTALL"
    echo -e "Installing some dev stuffs..."
    install_pkgs "$DEV_PACKAGES_TO_INSTALL"
    #python -m pip install --user --upgrade pip

    # vlc
    mkdir -p ~/.config/vlc
    copy_file ~/.config/vlc/vlcrc "${BASE_REPO_LOCATION}"home/.config/vlc/vlcrc

    if command_exists yad; then
        gsettings set yad.settings terminal "$CURRENT_TERMINAL"' -e "%s"'
    fi

    echo -e "Setting up file associations..."
    copy_file ~/.config/mimeapps.list "${BASE_REPO_LOCATION}"home/.config/mimeapps.list
    sed -i "s/DEFAULT_TEXT_EDITOR/$GUI_TEXT_EDITOR/g" ~/.config/mimeapps.list
    mkdir -p ~/.local/share/applications
    ln -sf ~/.config/mimeapps.list ~/.local/share/applications/mimeapps.list
}

debloat_pkgs
refresh_package_sources
echo -e "Installing some needed stuffs..."
install_pkgs "$REQUIREMENTS"
[[ $(type -t setup_"$DISTRO_TYPE") == function ]] && setup_"$DISTRO_TYPE"
install_pkgs crudini
if [[ $(type -t setup_specific_"$DIST_ID") == function ]]; then
    echo -e "Executing additional $DIST_ID specific script..."
    setup_specific_"$DIST_ID"
fi
update_packages
setup_system
setup_font
setup_apps
[[ $(type -t setup_"$DESKTOP") == function ]] && setup_"$DESKTOP"
setup_common_ui
if [[ $(type -t setup_"$DISTRO_TYPE"_"$DESKTOP") == function ]]; then
    echo -e "Executing additional $DISTRO_TYPE $DESKTOP specific script..."
    setup_"$DISTRO_TYPE"_"$DESKTOP"
fi
if [[ $(type -t setup_specific_"$DIST_ID"_"$DESKTOP") == function ]]; then
    echo -e "Executing additional $DIST_ID $DESKTOP specific script..."
    setup_specific_"$DIST_ID"_"$DESKTOP"
fi
setup_terminal
update_packages
alias reboot='sudo reboot'

echo -e ""
echo -e "Done...Reboot..."
#disown -ar