98d4502d3a
* Support local addons path override in stress test deployment * Support username based deployId in local stress deployment * Support WhatIf in stress infrastructure provision script * Simplify stress user detection Co-authored-by: Wes Haggard <weshaggard@users.noreply.github.com> * Run helm plugin add with helper * Add WhatIf support to ps module install helper function Co-authored-by: Ben Broderick Phillips <bebroder@microsoft.com> Co-authored-by: Wes Haggard <weshaggard@users.noreply.github.com>
105 lines
4.0 KiB
PowerShell
105 lines
4.0 KiB
PowerShell
$DefaultPSRepositoryUrl = "https://www.powershellgallery.com/api/v2"
|
|
$global:CurrentUserModulePath = ""
|
|
|
|
function Update-PSModulePath()
|
|
{
|
|
# Information on PSModulePath taken from docs
|
|
# https://docs.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_psmodulepath
|
|
|
|
# Information on Az custom module paths on hosted agents taken from
|
|
# https://github.com/microsoft/azure-pipelines-tasks/blob/c9771bc064cd60f47587c68e5c871b7cd13f0f28/Tasks/AzurePowerShellV5/Utility.ps1
|
|
|
|
if ($IsWindows) {
|
|
$hostedAgentModulePath = $env:SystemDrive + "\Modules"
|
|
$moduleSeperator = ";"
|
|
} else {
|
|
$hostedAgentModulePath = "/usr/share"
|
|
$moduleSeperator = ":"
|
|
}
|
|
$modulePaths = $env:PSModulePath -split $moduleSeperator
|
|
|
|
# Remove any hosted agent paths (needed to remove old default azure/azurerm paths which cause conflicts)
|
|
$modulePaths = $modulePaths.Where({ !$_.StartsWith($hostedAgentModulePath) })
|
|
|
|
# Add any "az_" paths from the agent which is the lastest set of azure modules
|
|
$AzModuleCachPath = (Get-ChildItem "$hostedAgentModulePath/az_*" -Attributes Directory) -join $moduleSeperator
|
|
if ($AzModuleCachPath -and $env.PSModulePath -notcontains $AzModuleCachPath) {
|
|
$modulePaths += $AzModuleCachPath
|
|
}
|
|
|
|
$env:PSModulePath = $modulePaths -join $moduleSeperator
|
|
|
|
# Find the path that is under user home directory
|
|
$homeDirectories = $modulePaths.Where({ $_.StartsWith($home) })
|
|
if ($homeDirectories.Count -gt 0) {
|
|
$global:CurrentUserModulePath = $homeDirectories[0]
|
|
if ($homeDirectories.Count -gt 1) {
|
|
Write-Verbose "Found more then one module path starting with $home so selecting the first one $global:CurrentUserModulePath"
|
|
}
|
|
|
|
# In some cases the directory might not exist so we need to create it otherwise caching an empty directory will fail
|
|
if (!(Test-Path $global:CurrentUserModulePath)) {
|
|
New-Item $global:CurrentUserModulePath -ItemType Directory > $null
|
|
}
|
|
}
|
|
else {
|
|
Write-Error "Did not find a module path starting with $home to set up a user module path in $env:PSModulePath"
|
|
}
|
|
}
|
|
|
|
# If we want to use another default repository other then PSGallery we can update the default parameters
|
|
function Install-ModuleIfNotInstalled()
|
|
{
|
|
[CmdletBinding(SupportsShouldProcess = $true)]
|
|
param(
|
|
[string]$moduleName,
|
|
[string]$version,
|
|
[string]$repositoryUrl = $DefaultPSRepositoryUrl
|
|
)
|
|
|
|
# Check installed modules
|
|
$modules = (Get-Module -ListAvailable $moduleName)
|
|
if ($version -as [Version]) {
|
|
$modules = $modules.Where({ [Version]$_.Version -ge [Version]$version })
|
|
}
|
|
|
|
if ($modules.Count -eq 0)
|
|
{
|
|
$repositories = (Get-PSRepository).Where({ $_.SourceLocation -eq $repositoryUrl })
|
|
if ($repositories.Count -eq 0)
|
|
{
|
|
Register-PSRepository -Name $repositoryUrl -SourceLocation $repositoryUrl -InstallationPolicy Trusted
|
|
$repositories = (Get-PSRepository).Where({ $_.SourceLocation -eq $repositoryUrl })
|
|
if ($repositories.Count -eq 0) {
|
|
Write-Error "Failed to registory package repository $repositoryUrl."
|
|
return
|
|
}
|
|
}
|
|
$repository = $repositories[0]
|
|
|
|
if ($repository.InstallationPolicy -ne "Trusted") {
|
|
Set-PSRepository -Name $repository.Name -InstallationPolicy "Trusted"
|
|
}
|
|
|
|
Write-Host "Installing module $moduleName with min version $version from $repositoryUrl"
|
|
# Install under CurrentUser scope so that the end up under $CurrentUserModulePath for caching
|
|
Install-Module $moduleName -MinimumVersion $version -Repository $repository.Name -Scope CurrentUser -Force
|
|
|
|
# Ensure module installed
|
|
$modules = (Get-Module -ListAvailable $moduleName)
|
|
if ($version -as [Version]) {
|
|
$modules = $modules.Where({ [Version]$_.Version -ge [Version]$version })
|
|
}
|
|
|
|
if ($modules.Count -eq 0) {
|
|
Write-Error "Failed to install module $moduleName with version $version"
|
|
return
|
|
}
|
|
}
|
|
|
|
Write-Host "Using module $($modules[0].Name) with version $($modules[0].Version)."
|
|
return $modules[0]
|
|
}
|
|
|
|
Update-PSModulePath
|