AzureExt

0.1.28

Completers/Microsoft.Azure.ArgumentCompleters.ps1

#
# .SYNOPSIS
#
# Auto-complete the -StorageAccountName parameter value for Azure PowerShell cmdlets.
#
# .NOTES
#
# Created by Trevor Sullivan <trevor@trevorsullivan.net>
# http://trevorsullivan.net
#
function StorageAccount_StorageAccountNameCompleter
{
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    #Write-Verbose -Message ('Called Azure StorageAccountName completer at {0}' -f (Get-Date))

    $CacheKey = 'StorageAccount_StorageAccountNameCache'
    $StorageAccountNameCache = Get-CompletionPrivateData -Key $CacheKey

    ### Return the cached value if it has not expired
    if ($StorageAccountNameCache) {
        return $StorageAccountNameCache
    }

    $StorageAccountList = Get-AzureStorageAccount -WarningAction SilentlyContinue | Where-Object {$PSItem.StorageAccountName -match ${wordToComplete} } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = $PSItem.StorageAccountName
            ToolTip = 'Storage Account "{0}" in "{1}" region.' -f $PSItem.StorageAccountName, $PSItem.Location
            ListItemText = '{0} ({1})' -f $PSItem.StorageAccountName, $PSItem.Location
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue
            }
        New-CompletionResult @CompletionResult
    }

    Set-CompletionPrivateData -Key $CacheKey -Value $StorageAccountList
    return $StorageAccountList
}

#
# .SYNOPSIS
#
# Auto-complete the -Name parameter value for Azure PowerShell storage container cmdlets.
#
# .NOTES
#
# Created by Trevor Sullivan <trevor@trevorsullivan.net>
# http://trevorsullivan.net
#
function AzureStorage_StorageContainerNameCompleter
{
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    $CacheKey = 'AzureStorage_ContainerNameCache'
    $ContainerNameCache = Get-CompletionPrivateData -Key $CacheKey

    ### Return the cached value if it has not expired
    if ($ContainerNameCache) {
        return $ContainerNameCache
    }

    $ContainerList = Get-AzureStorageContainer -Context $fakeBoundParameter['Context'] | Where-Object -FilterScript { $PSItem.Name -match ${wordToComplete} } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = $PSItem.Name
            ToolTip = 'Storage Container "{0}" in "{1}" Storage Account.' -f $PSItem.Name, $fakeBoundParameter['Context'].StorageAccountName
            ListItemText = '{0} ({1})' -f $PSItem.Name, $fakeBoundParameter['Context'].StorageAccountName
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue
            }
        New-CompletionResult @CompletionResult
    }

    Set-CompletionPrivateData -Key $CacheKey -Value $ContainerList
    return $ContainerList
}

#
# .SYNOPSIS
#
# Auto-complete the -ServiceName parameter value for Azure PowerShell cmdlets.
#
# .NOTES
#
# Created by Trevor Sullivan <trevor@trevorsullivan.net>
# http://trevorsullivan.net
#
function CloudService_ServiceNameCompleter
{
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    #Write-Verbose -Message ('Called Azure ServiceName completer at {0}' -f (Get-Date))
    $CacheKey = 'CloudService_ServiceNameCache'
    $ServiceNameCache = Get-CompletionPrivateData -Key $CacheKey
    if ($ServiceNameCache) {
        return $ServiceNameCache
    }

    $ItemList = Get-AzureService | Where-Object { $PSItem.ServiceName -match $wordToComplete } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = $PSItem.ServiceName
            ToolTip = 'Cloud Service in "{0}" region.' -f $PSItem.ExtendedProperties.ResourceLocation
            ListItemText = '{0} ({1})' -f $PSItem.ServiceName, $PSItem.ExtendedProperties.ResourceLocation
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue
            }
        New-CompletionResult @CompletionResult
    }
    Set-CompletionPrivateData -Key $CacheKey -Value $ItemList

    return $ItemList
}

#
# .SYNOPSIS
#
# Auto-complete the -SubscriptionName parameter value for Azure PowerShell cmdlets.
#
# .NOTES
#
# Created by Trevor Sullivan <trevor@trevorsullivan.net>
# http://trevorsullivan.net
#
function Subscription_SubscriptionNameCompleter
{
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    #Write-Verbose -Message ('Called Azure SubscriptionName completer at {0}' -f (Get-Date))

    ### Attempt to read Azure subscription details from the cache
    $CacheKey = 'AzureSubscription_SubscriptionNameCache'
    $SubscriptionNameCache = Get-CompletionPrivateData -Key $CacheKey

    ### If there is a valid cache for the Azure subscription names, then go ahead and return them immediately
    if ($SubscriptionNameCache) {
        return $SubscriptionNameCache
    }

    ### Create fresh completion results for Azure subscriptions
    $ItemList = Get-AzureSubscription | Where-Object { $PSItem.SubscriptionName -match ${wordToComplete} } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = $PSItem.SubscriptionName
            ToolTip = 'Azure subscription "{0}" with ID {1}.' -f $PSItem.SubscriptionName, $PSItem.SubscriptionId
            ListItemText = '{0} ({1})' -f $PSItem.SubscriptionName, $PSItem.SubscriptionId
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue
            }
        New-CompletionResult @CompletionResult
    }
    
    ### Update the cache for Azure subscription names
    Set-CompletionPrivateData -Key $CacheKey -Value $ItemList

    ### Return the fresh completion results
    return $ItemList
}

#
# .SYNOPSIS
#
# Auto-complete the -Name parameter value for Azure PowerShell virtual machine cmdlets.
#
# .NOTES
#
# Created by Trevor Sullivan <trevor@trevorsullivan.net>
# http://trevorsullivan.net
# http://twitter.com/pcgeek86
#
function AzureVirtualMachine_NameCompleter
{
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    #Write-Verbose -Message ('Called Azure Virtual Machine Name completer at {0}' -f (Get-Date))

    ### Attempt to read Azure virtual machine details from the cache
    $CacheKey = 'AzureVirtualMachine_NameCache'
    $VirtualMachineNameCache = Get-CompletionPrivateData -Key $CacheKey

    ### If there is a valid cache for the Azure virtual machine names, then go ahead and return them immediately
    if ($VirtualMachineNameCache -and (Get-Date) -gt $VirtualMachineNameCache.ExpirationTime) {
        return $VirtualMachineNameCache
    }

    ### Create fresh completion results for Azure virtual machines
    $ItemList = Get-AzureVM | Where-Object { $PSItem.Name -match $wordToComplete } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = '{0} -ServiceName {1}' -f $PSItem.Name, $PSItem.ServiceName
            ToolTip = 'Azure VM {0}/{1} in state {2}.' -f $PSItem.ServiceName, $PSItem.Name, $PSItem.Status
            ListItemText = '{0}/{1}' -f $PSItem.ServiceName, $PSItem.Name
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue
            NoQuotes = $true
            }
        New-CompletionResult @CompletionResult
    }
    
    ### Update the cache for Azure virtual machines
    Set-CompletionPrivateData -Key $CacheKey -Value $ItemList

    ### Return the fresh completion results
    return $ItemList
}

function Azure_LocationCompleter {
    <#
    .SYNOPSIS
    Auto-complete the -Location parameter value for Azure cmdlets.
    
    .NOTES
    Created by Trevor Sullivan <trevor@trevorsullivan.net>
    http://trevorsullivan.net
    http://twitter.com/pcgeek86
    #>
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)

    function Get-AllAzureLocations {
        <#
        .Synopsis
        Retrieves a list of Azure locations, using the AzureResourceManager PowerShell module.

        .Description

        This function may be deprecated, or require modification, if the Get-AzureLocation command is changed
        from returning Azure platform features, to returning Azure locations. As of this implementation, the
        ARM Get-AzureLocation command returns a list of Azure Resource Providers as the top-level object.
        #>
        [CmdletBinding()]
        param ()

        $LocationList = (AzureResourceManager\Get-AzureLocation).Locations | Select-Object -Unique | Sort-Object;
        Write-Output -InputObject $LocationList;
    }

    ### Attempt to read Azure virtual machine details from the cache
    $CacheKey = 'Azure_LocationCache';
    $Cache = Get-CompletionPrivateData -Key $CacheKey;

    ### If there is a valid cache for the Azure virtual machine names, then go ahead and return them immediately
    if ($Cache -and (Get-Date) -gt $Cache.ExpirationTime) {
        return $Cache;
    }

    ### Create fresh completion results for Azure virtual machines
    $ItemList = Get-AllAzureLocations | Where-Object { $PSItem.Name -match $wordToComplete } | ForEach-Object {
        $CompletionResult = @{
            CompletionText = $PSItem;
            ToolTip = $PSItem;
            ListItemText = $PSItem;
            CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue;
            NoQuotes = $false;
            }
        New-CompletionResult @CompletionResult;
    }
    
    ### Update the cache for Azure virtual machines
    Set-CompletionPrivateData -Key $CacheKey -Value $ItemList;

    ### Return the fresh completion results
    return $ItemList;
}

$ArgumentCompleterList = @(
    @{
        Command = @('Enable-WAPackWebsiteApplicationDiagnositc', 
                    'Add-AzureHDInsightStorage', 
                    'Enable-AzureWebsiteApplicationDiagnostic', 
                    'Get-AzureStorageAccount', 
                    'Get-AzureStorageKey', 
                    'New-AzureMediaServicesAccount', 
                    'New-AzureStorageAccount', 
                    'New-AzureStorageContext', 
                    'New-AzureStorageKey', 
                    'Publish-AzureServiceProject', 
                    'Remove-AzureStorageAccount', 
                    'Set-AzureHDInsightDefaultStorage', 
                    'Set-AzureStorageAccount', 
                    'Set-AzureVMCustomScriptExtension' );
        Parameter = 'StorageAccountName';
        Description = 'Complete the -StorageAccountName parameter value for Azure cmdlets: Get-AzureStorageAccount -StorageAccountName <TAB>';
        ScriptBlock = $function:StorageAccount_StorageAccountNameCompleter;
    }
    @{
        Command = @('Get-AzureStorageContainerAcl', 
                    'Get-AzureSiteRecoveryProtectionContainer', 
                    'Get-AzureStorageContainer', 
                    'New-AzureStorageContainer', 
                    'New-AzureStorageContainerSASToken', 
                    'Remove-AzureStorageContainer', 
                    'Set-AzureStorageContainerAcl');
        Parameter = 'Name';
        Description = 'Complete the -Name parameter value for Azure cmdlets: Get-AzureStorageContainer -Context $Context -Name <TAB>';
        ScriptBlock = $function:AzureStorage_StorageContainerNameCompleter;
    }
    @{
        Command = @('Add-AzureCertificate', 
                    'Add-AzureDns', 
                    'Add-AzureInternalLoadBalancer', 
                    'Export-AzureVM', 
                    'Get-AzureCertificate', 
                    'Get-AzureDeployment', 
                    'Get-AzureDeploymentEvent', 
                    'Get-AzureInternalLoadBalancer', 
                    'Get-AzureRemoteDesktopFile', 
                    'Get-AzureRole', 
                    'Get-AzureService', 
                    'Get-AzureServiceADDomainExtension', 
                    'Get-AzureServiceAntimalwareConfig', 'Get-AzureServiceDiagnosticsExtension', 'Get-AzureServiceExtension', 'Get-AzureServiceRemoteDesktopExtension', 'Get-AzureVM', 'Get-AzureWinRMUri', 'Move-AzureDeployment', 'New-AzureDeployment', 'New-AzureQuickVM', 'New-AzureService', 'New-AzureServiceProject', 'New-AzureVM', 'Publish-AzureServiceProject', 'Remove-AzureCertificate', 'Remove-AzureDeployment', 'Remove-AzureDns', 'Remove-AzureInternalLoadBalancer', 'Remove-AzureService', 'Remove-AzureServiceADDomainExtension', 'Remove-AzureServiceAntimalwareExtension', 'Remove-AzureServiceDiagnosticsExtension', 'Remove-AzureServiceExtension', 'Remove-AzureServiceRemoteDesktopExtension', 'Remove-AzureVM', 'Reset-AzureRoleInstance', 'Restart-AzureVM', 'Save-AzureVMImage', 'Set-AzureDeployment', 'Set-AzureDns', 'Set-AzureInternalLoadBalancer', 'Set-AzureLoadBalancedEndpoint', 'Set-AzureRole', 'Set-AzureService', 'Set-AzureServiceADDomainExtension', 'Set-AzureServiceAntimalwareExtension', 'Set-AzureServiceDiagnosticsExtension', 'Set-AzureServiceExtension', 'Set-AzureServiceRemoteDesktopExtension', 'Set-AzureWalkUpgradeDomain', 'Start-AzureService', 'Start-AzureVM', 'Stop-AzureService', 'Stop-AzureVM', 'Update-AzureVM');
        Parameter = 'ServiceName';
        Description = 'Complete the -ServiceName parameter value for Azure cmdlets: Get-AzureService -ServiceName <TAB>';
        ScriptBlock = $function:CloudService_ServiceNameCompleter;
    }
    @{
        Command = @('Get-WAPackSubscription', 'Remove-WAPackSubscription', 'Select-WAPackSubscription', 'Set-WAPackSubscription', 'Get-AzureSubscription', 'New-AzureSqlDatabaseServerContext', 'Remove-AzureSubscription', 'Select-AzureSubscription', 'Set-AzureSubscription');
        Parameter = 'SubscriptionName';
        Description = 'Complete the -SubscriptionName parameter value for Azure cmdlets: Select-AzureSubscription -SubscriptionName <TAB>';
        ScriptBlock = $function:Subscription_SubscriptionNameCompleter;
    }
    @{
        Command = @('Export-AzureVM', 
                    'Get-AzureVM', 
                    'Remove-AzureVM', 
                    'Restart-AzureVM', 
                    'Start-AzureVM', 
                    'Stop-AzureVM', 
                    'Update-AzureVM',
                    'Get-AzureRemoteDesktopFile');
        Parameter = 'Name';
        Description = 'Complete the -Name parameter value for Azure virtual machine cmdlets: Stop-AzureVM -Name <TAB>';
        ScriptBlock = $function:AzureVirtualMachine_NameCompleter;
    }
    @{
        Command = @(
                'Add-AlertRule',
                'Add-AutoscaleSetting',
                'Add-AzureApiManagementRegion',
                'Get-AzureHDInsightProperties',
                'Get-AzureStreamAnalyticsQuota',
                'Get-AzureVMExtensionImage',
                'Get-AzureVMExtensionImageType',
                'Get-AzureVMImage',
                'Get-AzureVMImageOffer',
                'Get-AzureVMImagePublisher',
                'Get-AzureVMImageSku',
                'Get-AzureVMSize',
                'Get-AzureVMUsage',
                'New-AzureApiManagement',
                'New-AzureApiManagementVirtualNetwork',
                'New-AzureApplicationGateway',
                'New-AzureAppServicePlan',
                'New-AzureAutomationAccount',
                'New-AzureAvailabilitySet',
                'New-AzureBatchAccount',
                'New-AzureDataFactory',
                'New-AzureHDInsightCluster',
                'New-AzureKeyVault',
                'New-AzureLoadBalancer',
                'New-AzureLocalNetworkGateway',
                'New-AzureNetworkInterface',
                'New-AzureNetworkSecurityGroup',
                'New-AzureOperationalInsightsWorkspace',
                'New-AzurePublicIpAddress',
                'New-AzureRedisCache',
                'New-AzureResource',
                'New-AzureResourceGroup',
                'New-AzureRouteTable',
                'New-AzureSiteRecoveryVault',
                'New-AzureSqlServer',
                'New-AzureStorageAccount',
                'New-AzureVirtualNetwork',
                'New-AzureVirtualNetworkGateway',
                'New-AzureVirtualNetworkGatewayConnection',
                'New-AzureVM',
                'New-AzureWebApp',
                'Remove-AzureApiManagementRegion',
                'Set-AzureAppServicePlan',
                'Set-AzureVMAccessExtension',
                'Set-AzureVMCustomScriptExtension',
                'Set-AzureVMDiagnosticsExtension',
                'Set-AzureVMDscExtension',
                'Set-AzureVMExtension',
                'Set-AzureVMSqlServerExtension',
                'Test-AzureDnsAvailability',
                'Update-AzureApiManagementDeployment',
                'Update-AzureApiManagementRegion'
        );
        Parameter = 'Location';
        Description = 'Complete the -Location parameter value for Azure Resource Manager cmdlets: New-AzureResource -Location <TAB>';
        ScriptBlock = $function:Azure_LocationCompleter;
    }
);

foreach ($ArgumentCompleter in $ArgumentCompleterList) {
    TabExpansion++\Register-ArgumentCompleter @ArgumentCompleter;
}