From eb6acd68d319066479abbbd6a3bc2eba8a2c33c5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Igor=20Pe=C4=8Dovnik?=
 <igorpecovnik@users.noreply.github.com>
Date: Tue, 29 Mar 2022 21:16:45 +0200
Subject: [PATCH] Add Nvidia driver to the x86 desktop images (#3572)

* Add Nvidia driver to the x86 desktop images
* Enable this driver on uefi x86 not on arm64
* Enable nvidia driver extension only on desktop builds
---
 config/sources/families/uefi-x86.conf |  1 +
 extensions/nvidia.sh                  | 13 +++++++++++++
 2 files changed, 14 insertions(+)
 create mode 100644 extensions/nvidia.sh

diff --git a/config/sources/families/uefi-x86.conf b/config/sources/families/uefi-x86.conf
index 36ca9caca..55923fd7b 100644
--- a/config/sources/families/uefi-x86.conf
+++ b/config/sources/families/uefi-x86.conf
@@ -1,4 +1,5 @@
 # Important: LINUXFAMILY and ARCH are defined _before_ including the common family include
+[[ "$BUILD_DESKTOP" == yes ]] && enable_extension "nvidia"
 export LINUXFAMILY="x86"
 export ARCH="amd64"
 source "${BASH_SOURCE%/*}/include/uefi_common.inc"
diff --git a/extensions/nvidia.sh b/extensions/nvidia.sh
new file mode 100644
index 000000000..2e3c55487
--- /dev/null
+++ b/extensions/nvidia.sh
@@ -0,0 +1,13 @@
+function pre_install_kernel_debs__build_nvidia_kernel_module() {
+
+	export INSTALL_HEADERS="yes"
+
+}
+
+function post_install_kernel_debs__build_nvidia_kernel_module() {
+
+	display_alert "Build kernel module" "${EXTENSION}" "info"
+	chroot "${SDCARD}" /bin/bash -c "apt -y -qq install nvidia-dkms-510 nvidia-driver-510 nvidia-settings nvidia-common" >>"$DEST"/"${LOG_SUBPATH}"/install.log 2>&1 || {
+		exit_with_error "${install_grub_cmdline} failed!"
+	}
+}