From f3640195de311751db1aaa8bd8c08f3c3f7e81ee Mon Sep 17 00:00:00 2001
From: Kevin Mittman <kmittman@nvidia.com>
Date: Thu, 8 Apr 2021 21:33:19 -0700
Subject: [PATCH] Update example build script for RHEL7

Signed-off-by: Kevin Mittman <kmittman@nvidia.com>
---
 build.sh | 69 +++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 63 insertions(+), 6 deletions(-)

diff --git a/build.sh b/build.sh
index 1c7d4f9..7e40e31 100755
--- a/build.sh
+++ b/build.sh
@@ -1,18 +1,25 @@
 #!/usr/bin/env bash
 
 runfile="$1"
+distro="$2"
 topdir="$HOME/dkms-nvidia"
-arch="x86_64"
 epoch="3"
 
+[[ -n $distro ]] ||
+distro=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)
+[[ $distro == "main" ]] && distro="rhel8"
+
 drvname=$(basename "$runfile")
-version=$(echo "$drvname" | sed -e "s|NVIDIA\-Linux\-${arch}\-||" -e 's|\.run$||')
+arch=$(echo "$drvname" | awk -F "-" '{print $3}')
+version=$(echo "$drvname" | sed -e "s|NVIDIA\-Linux\-${arch}\-||" -e 's|\.run$||' -e 's|\-grid$||')
+drvbranch=$(echo "$version" | awk -F "." '{print $1}')
 
-tarball=nvidia-kmod-${version}-${arch}
+tarball="nvidia-kmod-${version}-${arch}"
 unpackDir="unpack"
 
 err() { echo; echo "ERROR: $*"; exit 1; }
 kmd() { echo; echo ">>> $*" | fold -s; eval "$*" || err "at line \`$*\`"; }
+dep() { type -p "$1" >/dev/null || err "missing dependency $1"; }
 
 generate_tarballs()
 {
@@ -23,40 +30,90 @@ generate_tarballs()
     tar --remove-files -cJf "${tarball}.tar.xz" "${tarball}"
 }
 
-build_rpm()
+build_dnf_rpm()
+{
+    mkdir -p "$topdir"
+    (cd "$topdir" && mkdir -p BUILD BUILDROOT RPMS SRPMS SOURCES SPECS)
+
+    cp -v -- *conf* "$topdir/SOURCES/"
+    cp -v -- *.tar* "$topdir/SOURCES/"
+    cp -v -- *.spec "$topdir/SPECS/"
+    cd "$topdir" || err "Unable to cd into $topdir"
+
+    kmd rpmbuild \
+        --define "'%_topdir $(pwd)'" \
+        --define "'debug_package %{nil}'" \
+        --define "'version $version'" \
+        --define "'epoch $epoch'" \
+        --target "$arch" \
+        -v -bb SPECS/dkms-nvidia.spec
+
+    cd - || err "Unable to cd into $OLDPWD"
+}
+
+build_yum_rpm()
 {
     mkdir -p "$topdir"
     (cd "$topdir" && mkdir -p BUILD BUILDROOT RPMS SRPMS SOURCES SPECS)
 
     cp -v -- *conf* "$topdir/SOURCES/"
-    cp -v -- *tar* "$topdir/SOURCES/"
+    cp -v -- *.tar* "$topdir/SOURCES/"
     cp -v -- *.spec "$topdir/SPECS/"
+
+    #
+    # NOTE: to build kmod-nvidia-branch-XXX and kmod-nvidia-latest packages
+    #       see https://github.com/NVIDIA/yum-packaging-precompiled-kmod
+    flavor="latest-dkms"
+    is_latest=1
+    is_dkms=1
+
     cd "$topdir" || err "Unable to cd into $topdir"
+    echo -e "\n:: flavor $flavor [$is_latest] [$is_dkms]"
 
     kmd rpmbuild \
         --define "'%_topdir $(pwd)'" \
         --define "'debug_package %{nil}'" \
         --define "'version $version'" \
+        --define "'driver_branch $flavor'" \
+        --define "'is_dkms $is_dkms'" \
+        --define "'is_latest $is_latest'" \
         --define "'epoch $epoch'" \
+        --target "$arch" \
         -v -bb SPECS/dkms-nvidia.spec
 
     cd - || err "Unable to cd into $OLDPWD"
 }
 
+build_wrapper()
+{
+    echo ":: Building $distro packages"
+    if [[ $distro == "rhel7" ]]; then
+        build_yum_rpm
+    else
+        build_dnf_rpm
+    fi
+}
+
+
 # Create tarball from runfile contents
 if [[ -f ${tarball}.tar.xz ]]; then
     echo "[SKIP] generate_tarballs()"
 else
     echo "==> generate_tarballs()"
+    [[ -f $runfile ]] || err "Usage: $0 /path/to/NVIDIA-Linux-*.run"
     generate_tarballs
 fi
 
+# Sanity check
+[[ -n $version ]] || err "version could not be determined"
+
 # Build RPMs
 empty=$(find "$topdir/RPMS" -maxdepth 0 -type d -empty 2>/dev/null)
 found=$(find "$topdir/RPMS" -mindepth 2 -maxdepth 2 -type f -name "*${version}*" 2>/dev/null)
 if [[ ! -d "$topdir/RPMS" ]] || [[ $empty ]] || [[ ! $found ]]; then
     echo "==> build_rpm(${version})"
-    build_rpm
+    dep rpmbuild
+    build_wrapper
 else
     echo "[SKIP] build_rpm(${version})"
 fi