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($wordToComplete,
        $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($wordToComplete,
        $fakeBoundParameters)
    $Session = Get-CompleteSession $fakeBoundParameters.F5Session
    if ($Session) {
        Get-HealthMonitorType -F5Session $Session | Where-Object { $_ -like "$wordToComplete*" }
    }
}
function CompleteNodeAddress {
    param($wordToComplete,
        $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($wordToComplete,
        $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($wordToComplete,
        $fakeBoundParameters)
    $Session = Get-CompleteSession $fakeBoundParameters.F5Session
    if ($Session) {
        Get-BIGIPPartition -F5Session $Session | Where-Object { $_ -like "$wordToComplete*" }
    }
}
function CompletePoolMemberAddress {
    param($wordToComplete,
        $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($wordToComplete,
        $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($wordToComplete,
        $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($wordToComplete,
        $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($wordToComplete,
        $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($wordToComplete,
        $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)"

}