From bfc31b6d359289fa3d5b48ef6d4343921e088ba6 Mon Sep 17 00:00:00 2001
From: Jeevitha Kannan K S <ksjeevithakannan123@gmail.com>
Date: Sat, 12 Oct 2024 15:18:24 +0530
Subject: [PATCH] Update model fetching

---
 core/tabs/system-setup/arch/nvidia-drivers.sh | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/core/tabs/system-setup/arch/nvidia-drivers.sh b/core/tabs/system-setup/arch/nvidia-drivers.sh
index 52cc6fe5..2dd62043 100755
--- a/core/tabs/system-setup/arch/nvidia-drivers.sh
+++ b/core/tabs/system-setup/arch/nvidia-drivers.sh
@@ -16,9 +16,9 @@ installDeps() {
     done
 }
 
-checkHardware() {
+checkNvidiaHardware() {
     # Refer https://nouveau.freedesktop.org/CodeNames.html for model code names
-    model=$(lspci -k | grep -A 2 -E "(VGA|3D)" | grep controller | cut -d ' ' -f 7 |  cut -c 1-2 )
+    model=$(lspci -k | grep -A 2 -E "(VGA|3D)" | grep NVIDIA | sed 's/.*Corporation //;s/ .*//' | cut -c 1-2)
     case "$model" in
         GM|GP|GV) return 1 ;;
         TU|GA|AD) return 0 ;;
@@ -26,6 +26,11 @@ checkHardware() {
     esac
 }
 
+checkIntelHardware() {
+    model=$(grep "model name" /proc/cpuinfo | head -n 1 | cut -d ':' -f 2 | cut -c 2-3)
+    [ "$model" -ge 11 ]
+}
+
 promptUser() {
     printf "%b" "Do you want to $1 ? [y/N]:"
     read -r confirm
@@ -82,7 +87,7 @@ setupHardwareAcceleration() {
 
 installDriver() {
     # Refer https://wiki.archlinux.org/title/NVIDIA for open-dkms or dkms driver selection
-    if checkHardware && promptUser "install nvidia's open source drivers"; then
+    if checkNvidiaHardware && 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
@@ -92,6 +97,10 @@ installDriver() {
         "$ESCALATION_TOOL" "$PACKAGER" -S --needed --noconfirm nvidia-dkms
     fi
 
+    if checkIntelHardware; then
+        setKernelParam "ibt=off"
+    fi
+
     # Refer https://wiki.archlinux.org/title/NVIDIA/Tips_and_tricks#Preserve_video_memory_after_suspend
     setKernelParam "nvidia.NVreg_PreserveVideoMemoryAllocations=1"
     "$ESCALATION_TOOL" systemctl enable nvidia-suspend.service nvidia-hibernate.service nvidia-resume.service