From 98e5a7d0ae7f5a36ee6f96cb1cfcc5f90ba95a1f Mon Sep 17 00:00:00 2001
From: Jeevitha Kannan K S <ksjeevithakannan123@gmail.com>
Date: Thu, 10 Oct 2024 21:11:05 +0530
Subject: [PATCH] Add arch nvidia script

---
 core/tabs/system-setup/arch/nvidia-drivers.sh | 113 ++++++++++++++++++
 core/tabs/system-setup/tab_data.toml          |   7 ++
 2 files changed, 120 insertions(+)
 create mode 100755 core/tabs/system-setup/arch/nvidia-drivers.sh

diff --git a/core/tabs/system-setup/arch/nvidia-drivers.sh b/core/tabs/system-setup/arch/nvidia-drivers.sh
new file mode 100755
index 00000000..a6f2588b
--- /dev/null
+++ b/core/tabs/system-setup/arch/nvidia-drivers.sh
@@ -0,0 +1,113 @@
+#!/bin/sh -e
+
+. ../../common-script.sh
+
+LIBVA_DIR="$HOME/linuxtoolbox/libva"
+
+installDeps() {
+    "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm base-devel dkms ninja meson
+    
+    installed_kernels=$(pacman -Q | grep -E '^linux(| |-rt|-rt-lts|-hardened|-zen|-lts)[^-headers]' | cut -d ' ' -f 1)
+
+    for kernel in $installed_kernels; do
+        header="${kernel}-headers"
+        printf "%b\n" "${CYAN}Installing headers for $kernel...${RC}"
+        "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm "$header"
+    done
+
+}
+
+checkHardware() {
+    model=$(lspci -k | grep -A 2 -E "(VGA|3D)" | grep controller | cut -d ' ' -f 7 |  cut -c 1-2 )
+    case "$model" in
+        GM|GP|GV) return 1 ;;
+        TU|GA|AD) return 0 ;;
+        *) printf "%b\n" "${RED}Your hardware is not supported." && exit 1 ;;
+    esac
+}
+
+promptUser() {
+    printf "%b" "Do you want to $1 ? [y/N]:"
+    read -r confirm
+    [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]
+}
+
+enableNvidiaModeset() {
+    PARAMETER="nvidia-drm.modeset=1"
+
+    if grep -q "$PARAMETER" /etc/default/grub; then
+        printf "%b\n" "${YELLOW}NVIDIA modesetting is already enabled in GRUB.${RC}"
+    else
+        "$ESCALATION_TOOL" sed -i "/^GRUB_CMDLINE_LINUX_DEFAULT=/ s/\"$/ $PARAMETER\"/" /etc/default/grub
+        printf "%b\n" "${CYAN}Added $PARAMETER to /etc/default/grub.${RC}"
+
+        "$ESCALATION_TOOL" grub-mkconfig -o /boot/grub/grub.cfg
+    fi
+}
+
+setupHardwareAccelration() {
+    modeset=$("$ESCALATION_TOOL" cat /sys/module/nvidia_drm/parameters/modeset)
+    if [ ! "$modeset" = "Y" ]; then
+        enableNvidiaModeset
+    fi
+
+    "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm libva-nvidia-driver
+
+    if pacman -Q | grep -q 'libva '; then
+        "$ESCALATION_TOOL" "$PACKAGER" -Rdd libva
+    fi
+
+    mkdir -p "$HOME/linuxtoolbox"
+    if [ -d "$LIBVA_DIR" ]; then
+        rm -rf "$LIBVA_DIR"
+    fi
+
+    printf "%b\n" "${YELLOW}Cloning libva from https://github.com/intel/libva in ${LIBVA_DIR}${RC}"
+    git clone https://github.com/intel/libva "$LIBVA_DIR"
+
+    mkdir -p "$LIBVA_DIR/build"
+    cd "$LIBVA_DIR/build" && arch-meson .. -Dwith_legacy=nvctrl && ninja
+    "$ESCALATION_TOOL" ninja install
+
+    "$ESCALATION_TOOL" sed -i '/^MOZ_DISABLE_RDD_SANDBOX/d' "/etc/environment"
+    "$ESCALATION_TOOL" sed -i '/^LIBVA_DRIVER_NAME/d' "/etc/environment"
+
+    printf "LIBVA_DRIVER_NAME=nvidia\nMOZ_DISABLE_RDD_SANDBOX=1" | "$ESCALATION_TOOL" tee -a /etc/environment > /dev/null
+
+    printf "%b\n" "${GREEN}Hardware Accelration setup completed successfully.${RC}"
+    
+    if promptUser "enable hardware accelration in mpv player"; then
+        if [ -f "$HOME/.config/mpv/mpv.conf" ];then
+            sed -i '/^hwdec/d' "$HOME/.config/mpv/mpv.conf"
+        fi
+        printf "hwdec=auto" | tee -a "$HOME/.config/mpv/mpv.conf" > /dev/null
+        printf "%b\n" "${GREEN}MPV Hardware Accelration enabled successfully.${RC}"
+    fi
+}
+
+installDriver() {
+    if checkHardware && promptUser "install nvidia's open source drivers"; then
+        printf "%b\n" "${YELLOW}Installing nvidia open source driver...${RC}"
+        installDeps
+        "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm nvidia-open-dkms
+    else
+        printf "%b\n" "${YELLOW}Installing nvidia proprietary driver...${RC}"
+        installDeps
+        "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm nvidia-dkms
+    fi
+
+    if echo "$XDG_CURRENT_DESKTOP" | grep -q "GNOME"; then
+        "$ESCALATION_TOOL" systemctl enable nvidia-suspend.service nvidia-hibernate.service nvidia-resume.service
+    fi
+
+    printf "%b\n" "${GREEN}Driver installed successfully.${RC}"
+    if command_exists grub-mkconfig && promptUser "setup hardware accelration"; then
+        setupHardwareAccelration
+    fi
+
+    printf "%b\n" "${GREEN}Please reboot your system for the changes to take effect.${RC}"
+}
+
+checkEnv
+checkEscalationTool
+installDriver
\ No newline at end of file
diff --git a/core/tabs/system-setup/tab_data.toml b/core/tabs/system-setup/tab_data.toml
index 080c06d9..80dcc33c 100644
--- a/core/tabs/system-setup/tab_data.toml
+++ b/core/tabs/system-setup/tab_data.toml
@@ -15,6 +15,13 @@ description = "This command installs a minimal arch server setup under 5 minutes
 script = "arch/server-setup.sh"
 task_list = "SI D"
 
+[[data.entries]]
+name = "Nvidia Drivers && Hardware Accelration"
+description = "This script installs and configures nvidia drivers with Hardware Accelration"
+script = "arch/nvidia-drivers.sh"
+task_list = "I FM RP SS"
+
+
 [[data.entries]]
 name = "Paru AUR Helper"
 description = "Paru is your standard pacman wrapping AUR helper with lots of features and minimal interaction.\nTo know more about AUR helpers visit: https://wiki.archlinux.org/title/AUR_helpers"