Private/ArgumentCompletion.ps1
|
# Retrieve ALL ~/Public/*.ps1 Command names $Script:F5LTMPublicCommands = Get-ChildItem -Path ($PSScriptRoot -replace 'Private','Public') -Filter '*.ps1' -Recurse | Select-Object -ExpandProperty BaseName function Get-F5Command { <# .SYNOPSIS (Get-Command -Module F5-LTM) adversely affects module import performance. This is a much faster alternative without resorting to static command names for Register-ArgumentCompleter -Command parameters. #> param( [Parameter(ValueFromPipeline=$true)] [string[]]$Filter ) process{ foreach ($f in $filter) { $Script:F5LTMPublicCommands -like $f } } } function Get-CompleteSession { param($boundSession) Invoke-NullCoalescing {$boundSession} {$Script:F5Session} } function CompleteMonitorName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-HealthMonitor -F5Session $Session -Partition $fakeBoundParameters.Partition -Type $fakeBoundParameters.Type | Where-Object { $_.name -like "$wordToComplete*" } | ForEach-Object { if ($fakeBoundParameters.Partition) { $_.name } else { $_.fullPath } } } } function CompleteMonitorType { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-HealthMonitorType -F5Session $Session | Where-Object { $_ -like "$wordToComplete*" } } } function CompleteNodeAddress { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-Node -F5Session $Session -Partition $fakeBoundParameters.Partition | Where-Object { $_.address -like "$wordToComplete*" } | Select-Object -ExpandProperty address } } function CompleteNodeName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-Node -F5Session $Session -Partition $fakeBoundParameters.Partition | Where-Object { $_.name -like "$wordToComplete*" } | ForEach-Object { if ($fakeBoundParameters.Partition) { $_.name } else { $_.fullPath } } } } function CompleteBIGIPPartition { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-BIGIPPartition -F5Session $Session | Where-Object { $_ -like "$wordToComplete*" } } } function CompletePoolMemberAddress { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) if ($fakeBoundParameters.PoolName) { $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-PoolMember -F5Session $Session -PoolName $fakeBoundParameters.PoolName -Partition $fakeBoundParameters.Partition -Name (Invoke-NullCoalescing {$fakeBoundParameters.Name} {'*'}) | Where-Object { $_.address -like "$wordToComplete*" } | Select-Object -ExpandProperty address } } } function CompletePoolMemberName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) if ($fakeBoundParameters.PoolName) { $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-PoolMember -F5Session $Session -PoolName $fakeBoundParameters.PoolName -Partition $fakeBoundParameters.Partition -Address (Invoke-NullCoalescing {$fakeBoundParameters.Address} {'*'}) | Where-Object { $_.name -like "$wordToComplete*" } | Select-Object -ExpandProperty name } } } function CompletePoolMonitorName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) if ($fakeBoundParameters.PoolName) { $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-PoolMonitor -F5Session $Session -PoolName $fakeBoundParameters.PoolName -Partition $fakeBoundParameters.Partition | Where-Object { $_.name -match "\b$wordToComplete*" } | Select-Object -ExpandProperty name } } } function CompletePoolName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-Pool -F5Session $Session -Partition $fakeBoundParameters.Partition | Where-Object { $_.name -like "$wordToComplete*" -or $_.fullPath -like "$wordToComplete*" } | ForEach-Object { if ($fakeBoundParameters.Partition) { $_.name } else { $_.fullPath } } } } function CompleteRuleName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-iRule -F5Session $Session -Partition $fakeBoundParameters.Partition | Where-Object { $_.name -like "$wordToComplete*" -or $_.fullPath -like "$wordToComplete" } | ForEach-Object { if ($fakeBoundParameters.Partition) { $_.name } else { $_.fullPath } } } } function CompleteVirtualServerName { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $Session = Get-CompleteSession $fakeBoundParameters.F5Session if ($Session) { Get-VirtualServer -F5Session $Session -Partition $fakeBoundParameters.Partition | Where-Object { $_.name -like "$wordToComplete*" -or $_.fullPath -like "$wordToComplete" } | ForEach-Object { if ($fakeBoundParameters.Partition) { $_.name } else { $_.fullPath } } } } if (Get-Command Register-ArgumentCompleter -ErrorAction Ignore) { Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-PoolMember') ` -ParameterName Address ` -ScriptBlock $function:CompletePoolMemberAddress Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-PoolMember') ` -ParameterName Name ` -ScriptBlock $function:CompletePoolMemberName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-PoolMonitor') ` -ParameterName Name ` -ScriptBlock $function:CompletePoolMonitorName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-HealthMonitor') ` -ParameterName Name ` -ScriptBlock $function:CompleteMonitorName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-HealthMonitor') ` -ParameterName Type ` -ScriptBlock $function:CompleteMonitorType Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-Node') ` -ParameterName Address ` -ScriptBlock $function:CompleteNodeAddress Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-Node') ` -ParameterName Name ` -ScriptBlock $function:CompleteNodeName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-Pool') ` -ParameterName Name ` -ScriptBlock $function:CompletePoolName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*') ` -ParameterName Partition ` -ScriptBlock $function:CompletePartition Register-ArgumentCompleter ` -CommandName 'Get-BIGIPPartition' ` -ParameterName Name ` -ScriptBlock $function:CompleteBIGIPPartition Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-Pool*') ` -ParameterName PoolName ` -ScriptBlock $function:CompletePoolName Register-ArgumentCompleter ` -CommandName 'Get-iRule' ` -ParameterName Name ` -ScriptBlock $function:CompleteRuleName Register-ArgumentCompleter ` -CommandName @(Get-F5Command '*-VirtualServer') ` -ParameterName Name ` -ScriptBlock $function:CompleteVirtualServerName } Else { Write-Verbose "The Register-ArgumentCompleter cmdlet requires either PowerShell v5+ or the installation of the TabExpansionPlusPlus module (https://github.com/lzybkr/TabExpansionPlusPlus)" } |