LocationCompletion.ps1
|
# # .SYNOPSIS # # Auto-complete the -Location parameter value for Azure PowerShell cmdlets. (version 1.0) # # .NOTES # # Created by Stuart Leeks # http://blogs.msdn.com/stuartleeks # http://twitter.com/stuartleeks # function LocationCompleter { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter) ### Attempt to read Azure location details from the cache $CacheKey = 'AzureGeneral_LocationCache' $LocationCache = Get-CompletionPrivateData -Key $CacheKey ### If there is a valid cache for the Azure locations, then go ahead and return them immediately if ($LocationCache -and (Get-Date) -gt $LocationCache.ExpirationTime) { $ItemList = $LocationCache } else { ### Create fresh completion results for Azure locations $ItemList = Get-AzureRmResourceProvider | Select-Object -ExpandProperty ResourceTypes | Select-Object -ExpandProperty Locations -Unique | Sort-Object # Update the cache for Azure locations Set-CompletionPrivateData -Key $CacheKey -Value $ItemList } ### Return the fresh completion results $wordToCompleteWildcard = $wordToComplete.Trim("'") + "*" $results = $ItemList ` | Where-Object { $PSItem -like $wordToCompleteWildcard} ` | Foreach-Object { if($PSItem -ne $null -and $PSItem -ne ""){ $CompletionResult = @{ CompletionText = $PSItem ToolTip = $PSItem ListItemText = $PSItem CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue } New-CompletionResult @CompletionResult } } return $results } # "'" + [string]::Join("', '", (Get-Command -Module Azure* -ParameterName Location -ParameterType string)) + "'" | clip Register-ArgumentCompleter ` -Command ( 'Get-AzureRmResourceGroup', 'Get-AzureRmResourceProvider', 'Get-AzureRmVMExtensionImage', 'Get-AzureRmVMExtensionImageType', 'Get-AzureRmVMImage', 'Get-AzureRmVMImageOffer', 'Get-AzureRmVMImagePublisher', 'Get-AzureRmVMImageSku', 'Get-AzureRmVMSize', 'Get-AzureRmVMUsage', 'New-AzureRmAvailabilitySet', 'New-AzureRmContainerServiceConfig', 'New-AzureRmDiskConfig', 'New-AzureRmImageConfig', 'New-AzureRmResource', 'New-AzureRmResourceGroup', 'New-AzureRmSnapshotConfig', 'New-AzureRmVM', 'New-AzureRmVmssConfig', 'Set-AzureRmVMAccessExtension', 'Set-AzureRmVMADDomainExtension', 'Set-AzureRmVMBginfoExtension', 'Set-AzureRmVMChefExtension', 'Set-AzureRmVMCustomScriptExtension', 'Set-AzureRmVMDiagnosticsExtension', 'Set-AzureRmVMDscExtension', 'Set-AzureRmVMExtension', 'Set-AzureRmVMSqlServerExtension' ) ` -Parameter 'Location' ` -Description 'Complete the -Location parameter value for Azure cmdlets: New-AzureRmResourceGroup -Location <TAB>' ` -ScriptBlock $function:LocationCompleter |