如何在自定义PowerShell函数中重现Select-Object的管道属性Tab补全？

现象对比 #

1. Select-Object的正常Tab补全 #

在命令行输入以下命令后，按下Tab键可以补全目标对象的属性名称——除默认属性外，自定义的name和description属性也能被识别并自动补全：

[pscustomobject]@{ name = 'test'; description='doing test' } | Select-Object

2. 自定义函数的Tab补全缺失 #

使用如下自定义函数时：

function pipeobject {
    param(
        [Parameter(ValueFromPipeline)]
        $InputObject,

        [Parameter(Position=0)]
        $Property
    )

    process {
        $Property | Foreach-Object { $InputObject.$_ }
    }
}

输入命令后，无法通过Tab补全-Property参数对应的输入对象属性：

[pscustomobject]@{ name = 'test'; description='doing test' } | pipeobject 

需求说明 #

本次需求针对泛型pscustomobject的属性解析，要求在自定义函数中重现Select-Object的管道属性Tab补全行为。已知通过声明输出类型的方案无法满足本次场景需求。

解决方案 #

通过为$Property参数添加[ArgumentCompleter]属性，在补全逻辑中动态解析管道输入对象的属性，实现Tab补全：

function pipeobject {
    param(
        [Parameter(ValueFromPipeline)]
        $InputObject,

        [Parameter(Position=0)]
        [ArgumentCompleter({
            param($wordToComplete, $commandAst, $cursorPosition)
            # 解析命令语法树，获取管道输入的第一个对象
            $pipelineInput = $commandAst.FindAll({ $args[0] -is [System.Management.Automation.Language.PipelineAst] }, $true) | 
                ForEach-Object { $_.PipelineElements[0].Invoke() } | 
                Select-Object -First 1
            
            if ($pipelineInput) {
                # 提取对象属性名，按输入前缀过滤后返回补全候选
                $pipelineInput.PSObject.Properties.Name | 
                    Where-Object { $_ -like "$wordToComplete*" } | 
                    ForEach-Object { "'$_'" }
            }
        })]
        $Property
    )

    process {
        $Property | Foreach-Object { $InputObject.$_ }
    }
}

原理说明 #

1. ArgumentCompleter脚本块会在用户按下Tab时触发执行
2. 通过解析命令的抽象语法树（AST）定位管道输入部分，执行并获取第一个输入对象
3. 提取该对象的所有属性名称，结合用户已输入的前缀进行过滤，返回匹配的属性名作为补全选项

测试验证 #

输入原命令后按下Tab键，即可看到name和description属性出现在补全候选中，实现与Select-Object一致的Tab补全效果。

内容的提问来源于stack exchange，提问作者Blaisem