From 797c2ddfaf56d53f1f23211146642ac456371bc1 Mon Sep 17 00:00:00 2001
From: beech <beechmanbopp@gmail.com>
Date: Sun, 26 Jan 2025 03:03:41 -0600
Subject: [PATCH] nvidia

---
 install3.sh | 16 ++++++++++++++--
 nvidia.hook | 14 ++++++++++++++
 2 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 nvidia.hook

diff --git a/install3.sh b/install3.sh
index b1ed1ab..51c0e1e 100644
--- a/install3.sh
+++ b/install3.sh
@@ -28,8 +28,19 @@ setup_aur () {
     echo "Yay installed."
 }
 
+setup_nvidia () {
+    echo "Setting up Nvidia drivers..."
+    sudo pacman -S nvidia nvidia-utils lib32-nvidia-utils nvidia-settings --noconfirm --needed
+    sudo sed -i 's/^MODULES=().*/MODULES=(nvidia nvidia_modeset nvidia_uvm nvidia_drm)/' /etc/mkinitcpio.conf
+    sudo sed -i 's/\(HOOKS=.*\) kms/\1/' /etc/mkinitcpio.conf
+    sudo mkinitcpio -P
+    sudo mkdir -p /etc/pacman.d/hooks/ && sudo mv ./nvidia.hook /etc/pacman.d/hooks/
+    sudo sh -c 'echo " nvidia-drm.modeset=1 nvidia_drm.fbdev=1 nvidia-drm.ForceCompositionPipeline=1 nvidia.NVreg_EnableGpuFirmware=0" >> /boot/loader/entries/arch.conf'
+    echo "Nvidia driver setup finished."
+}
+
 while true; do
-    options=("Install GNOME + SDDM" "Install KDE + SDDM" "Install Software" "Setup Yay" "Exit")
+    options=("Install GNOME + SDDM" "Install KDE + SDDM" "Install Software" "Setup Yay" "Setup Nvidia Drivers" "Exit")
 
     echo "Debian Server Setup: "
     select opt in "${options[@]}"; do
@@ -38,7 +49,8 @@ while true; do
             2) install_kde; break ;;
             3) install_software; break ;;
             4) setup_aur; break ;;
-            5) break 2 ;;
+            5) setup_nvidia; break ;;
+            6) break 2 ;;
             *) echo "Invalid" >&2
         esac
     done
diff --git a/nvidia.hook b/nvidia.hook
new file mode 100644
index 0000000..51f360a
--- /dev/null
+++ b/nvidia.hook
@@ -0,0 +1,14 @@
+[Trigger]
+Operation=Install
+Operation=Upgrade
+Operation=Remove
+Type=Package
+Target=nvidia
+Target=linux
+
+[Action]
+Description=Update Nvidia module in initcpio
+Depends=mkinitcpio
+When=PostTransaction
+NeedsTargets
+Exec=/bin/sh -c 'while read -r trg; do case $trg in linux) exit 0; esac; done; /usr/bin/mkinitcpio -P'
\ No newline at end of file