From 605d0ddd9fc994496f742a4d0d8e9b1f9bd94446 Mon Sep 17 00:00:00 2001
From: Afroz Mohammed <afrozam@amazon.com>
Date: Tue, 13 Aug 2024 13:52:31 -0700
Subject: [PATCH] added removing sdk references for dryrun in Build.ps1, added
 keyscanner excludes for sdk artifacts

---
 buildtools/Build.ps1  | 24 +++++++++++++++++++++---
 buildtools/build.proj |  1 +
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/buildtools/Build.ps1 b/buildtools/Build.ps1
index 4507178c4..7829c3f2e 100644
--- a/buildtools/Build.ps1
+++ b/buildtools/Build.ps1
@@ -86,6 +86,14 @@ function DownloadSdkArtifacts {
   }
 }
 
+# same as msbuild task clean-sdk-references in build.proj
+function Remove-SdkReferences {
+  $sdkFolderPath = 'Include/sdk/assemblies'
+  if (Test-Path $sdkFolderPath) {
+    $null = Remove-Item $sdkFolderPath -Recurse -Force
+  }
+}
+
 $ErrorActionPreference = "Stop"
 $ProgressPreference = "SilentlyContinue"
 
@@ -130,8 +138,16 @@ try {
     # for Release, $SdkArtifactsUri is uri to _sdk-versions.json in github repo aws-sdk-net
     #	https://raw.githubusercontent.com/aws/aws-sdk-net/{path to_sdk-versions.json}
 
+    $cleanSDKReferenceInMsBuild = 'true'
+    if ($BuildType -eq 'DRY_RUN') {
+      Write-Host "Removing Sdk References for DRY_RUN build"
+      Remove-SdkReferences
+      # for dry run, the clean-sdk-references should be set to false
+      # so that the sdk dlls that are downloaded in Build.ps1 DownloadSdkArtifacts are not overwritten in the generator by downloading artifacts from github
+      $cleanSDKReferenceInMsBuild = 'false'
+    }
+    
     # for Dry run, $SdkArtifactsUri can be _sdk-versions.json or dry run artifacts from .NET build.
-
     if ($SdkArtifactsUri) {
       DownloadSdkArtifacts -SdkArtifactsUri $SdkArtifactsUri
     }
@@ -146,7 +162,8 @@ try {
     if ($RunAsStagingBuild -eq 'true') {
       msbuild ./buildtools/build.proj `
         /t:staging-build `
-        /p:Configuration=$Configuration
+        /p:Configuration=$Configuration `
+        /p:CleanSdkReferences=$cleanSDKReferenceInMsBuild
     }
     else {
       msbuild ./buildtools/build.proj `
@@ -157,7 +174,8 @@ try {
         /p:Configuration=$Configuration `
         /p:SignModules=$SignModules `
         /p:DraftReleaseNotes=true `
-        /p:SkipCmdletGeneration=false
+        /p:SkipCmdletGeneration=false `
+        /p:CleanSdkReferences=$cleanSDKReferenceInMsBuild
     }
     $BuildResult = $LASTEXITCODE
   }
diff --git a/buildtools/build.proj b/buildtools/build.proj
index 97e83cf99..168d98268 100644
--- a/buildtools/build.proj
+++ b/buildtools/build.proj
@@ -449,6 +449,7 @@
         <ItemGroup>
             <FolderException Include="Include/sdk/models"/>
             <FolderException Include="PSGalleryDownloads"/>
+            <FolderException Include="Include/sdk/assemblies"/>
         </ItemGroup>
 
         <KeyScannerTask Folder="$(MSBuildProjectDirectory)/.." FilePattern="**" PrivateKeyExceptions="@(PrivateKeyException)" FileExceptions="@(FileException)" FolderExceptions="@(FolderException)" ParallelScan="true" />