From bd502d57d4c3698d608f3cc76cd5e67de35dffd0 Mon Sep 17 00:00:00 2001
From: Jai Luthra <me@jailuthra.in>
Date: Thu, 12 Oct 2023 16:33:21 +0530
Subject: [PATCH] actions: Fix workflow for studio release

Signed-off-by: Jai Luthra <me@jailuthra.in>
---
 .github/workflows/gen_win_patches.yml | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/gen_win_patches.yml b/.github/workflows/gen_win_patches.yml
index ffb7d4c..1502fa5 100644
--- a/.github/workflows/gen_win_patches.yml
+++ b/.github/workflows/gen_win_patches.yml
@@ -34,6 +34,8 @@ jobs:
 
             if [ "$variant" == "dch" ]; then
               variant="DCH"
+            elif [ "$variant" == "studio" ]; then
+              variant="Studio Driver"
             fi
 
             echo "OS=$os" >> $GITHUB_ENV
@@ -57,22 +59,34 @@ jobs:
       - name: Delete Existing Files
         run: |
           echo "Deleting existing files if they exist"
-          rm -f "${{ github.workspace }}/win/win10_x64/${{ env.VERSION }}/nvencodeapi64.1337"
-          rm -f "${{ github.workspace }}/win/win10_x64/${{ env.VERSION }}/nvencodeapi.1337"
+          dir=${{ env.VERSION }}
+
+          if [ "${{ env.VARIANT }}" == "Studio Driver" ]; then
+            dir=nsd_${{ env.VERSION }}
+          fi
+
+          rm -f "${{ github.workspace }}/win/win10_x64/$dir/nvencodeapi64.1337"
+          rm -f "${{ github.workspace }}/win/win10_x64/$dir/nvencodeapi.1337"
           echo "Existing files deleted successfully"
 
       - name: Run autopatch.py
         run: |
-          echo "Running autopatch.py with version ${{ env.VERSION }}"
+          echo "Running autopatch.py with version ${{ env.VERSION }} and variant ${{ env.VARIANT }}"
           cd "${{ github.workspace }}/win/tools/autopatch"
-          python autopatch.py ${{ env.VERSION }}
+
+          if [ "${{ env.VARIANT }}" == "DCH" ]; then
+            python autopatch.py ${{ env.VERSION }}
+          elif [ "${{ env.VARIANT }}" == "Studio Driver" ]; then
+            python autopatch.py https://international.download.nvidia.com/Windows/${{ env.VERSION }}/${{ env.VERSION }}-desktop-win10-win11-64bit-international-nsd-dch-whql.exe
+          fi
+
           echo "autopatch.py executed successfully"
 
       - name: Run add_driver.py
         run: |
           echo "Running add_driver.py with variant ${{ env.VARIANT }} and version ${{ env.VERSION }}"
           cd "${{ github.workspace }}/tools/readme-autogen"
-          python add_driver.py -W -P GeForce --variant ${{ env.VARIANT }} -w win10 ${{ env.VERSION }}
+          python add_driver.py -W -P GeForce --variant "${{ env.VARIANT }}" -w win10 ${{ env.VERSION }}
           echo "add_driver.py executed successfully"
 
       - name: Run readme_autogen.py