Preload assemblies on PS5 (#21780)

Author: isra-fel

src/Compute/Compute/ChangeLog.md

@@ -21,6 +21,7 @@
 -->
 ## Upcoming Release
 * Added new switch parameter `OSImageScheduledEventEnabled` and string parameter `OSImageScheduledEventNotBeforeTimeoutInMinutes` to the cmdlets `New-AzVmssConfig` and `Update-AzVmss`.
+* Fixed an issue that `Add-AzVhd` throws `FileNotFoundException` on Windows PowerShell. [#21321]
 
 ## Version 5.7.1
 * Added a breaking change warning to the `Get-AzVM` cmdlet to show that the `NextLink` parameter and parameter set will be removed in June 2023. The parameter has been non-functional for a long time. 

tools/AzureRM.Example.psm1

@@ -15,7 +15,7 @@ function Test-DotNet
     {
         if ((Get-PSDrive 'HKLM' -ErrorAction Ignore) -and (-not (Get-ChildItem 'HKLM:\SOFTWARE\Microsoft\NET Framework Setup\NDP\v4\Full\' -ErrorAction Stop | Get-ItemPropertyValue -ErrorAction Stop -Name Release | Where-Object { $_ -ge 461808 })))
         {
-            throw ".NET Framework versions lower than 4.7.2 are not supported in Az.  Please upgrade to .NET Framework 4.7.2 or higher."
+            throw ".NET Framework versions lower than 4.7.2 are not supported in Az. Please upgrade to .NET Framework 4.7.2 or higher."
         }
     }
     catch [System.Management.Automation.DriveNotFoundException]
@@ -66,6 +66,13 @@ if (Test-Path -Path "$PSScriptRoot\StartupScripts" -ErrorAction Ignore)
     }
 }
 
+# [windows powershell] preload assemblies
+%PRELOAD-ASSEMBLY%
+
+# [windows powershell] preload module alc assemblies
+$preloadPath = (Join-Path $PSScriptRoot -ChildPath "ModuleAlcAssemblies")
+Preload-Assembly -AssemblyDirectory $preloadPath
+
 if (Get-Module %AZORAZURERM%.profile -ErrorAction Ignore)
 {
     Write-Warning ("%AZORAZURERM%.Profile already loaded. Az and AzureRM modules cannot be imported in the same session or used in the same script or runbook. If you are running PowerShell in an environment you control you can use the 'Uninstall-AzureRm' cmdlet to remove all AzureRm modules from your machine. " +

tools/UpdateModules.psm1

@@ -92,11 +92,30 @@ function New-ModulePsm1 {
             }
         }
 
+        # Scripts to preload dependency assemblies on Windows PowerShell
+        # https://stackoverflow.com/a/60068470
+        $preloadAssemblies = ""
+        $isAzAccounts = $file.BaseName -ieq 'Az.Accounts'
+        if ($isAzAccounts) {
+            $preloadAssemblies += 'if ($PSEdition -eq "Desktop") {
+    [Microsoft.Azure.PowerShell.AssemblyLoading.ConditionalAssemblyProvider]::GetAssemblies().Values | ForEach-Object {
+        $path = $$_.Item1
+        try {
+            Add-Type -Path $path -ErrorAction Ignore | Out-Null
+        }
+        catch {
+            Write-Verbose "Could not preload $path"
+        }
+    }
+}'
+        }
+
         # Grab the template and replace with information.
         $template = Get-Content -Path $TemplatePath
         $template = $template -replace "%MODULE-NAME%", $file.BaseName
         $template = $template -replace "%DATE%", [string](Get-Date)
         $template = $template -replace "%IMPORTED-DEPENDENCIES%", $importedModules
+        $template = $template -replace "%PRELOAD-ASSEMBLY%", $preloadAssemblies
 
         #Az.Storage is using Azure.Core, so need to check PS version
         if ($IsNetcore)
@@ -107,7 +126,7 @@ function New-ModulePsm1 {
             }
             elseif($file.BaseName -ieq 'Az.Accounts')
             {
-                $template = $template -replace "%AZURECOREPREREQUISITE%", 
+                $template = $template -replace "%AZURECOREPREREQUISITE%",
 @"
 if (%ISAZMODULE% -and (`$PSEdition -eq 'Core'))
 {
@@ -124,7 +143,7 @@ if (%ISAZMODULE% -and (`$PSEdition -eq 'Core'))
             }
             else
             {
-                $template = $template -replace "%AZURECOREPREREQUISITE%", 
+                $template = $template -replace "%AZURECOREPREREQUISITE%",
 @"
 if (%ISAZMODULE% -and (`$PSEdition -eq 'Core'))
 {
@@ -191,7 +210,7 @@ function Get-Cmdlets {
     $nestedModules = $ModuleMetadata.NestedModules
     $cmdlets = @()
     foreach ($module in $nestedModules) {
-        if('.dll' -ne [System.IO.Path]::GetExtension($module)) 
+        if('.dll' -ne [System.IO.Path]::GetExtension($module))
         {
             continue;
         }