如何检测PowerShell中传入的多行命令？解决Prompt阻塞问题

你的问题核心是直接使用[console]::KeyAvailable会和PowerShell宿主的输入处理机制冲突：当你拦截控制台输入缓冲区的按键时，用户提交的有效输入（比如第二次回车、粘贴的多行命令）会被你的循环吃掉，导致PowerShell无法接收到这些输入，从而出现无响应或阻塞。

解决方案：区分输入类型，将有效输入放回缓冲区 #

不要直接消费所有按键，而是读取按键后判断是否是PowerShell需要处理的指令（比如回车、换行），这类输入需要写回控制台输入缓冲区，让PowerShell后续处理。使用$host.UI.RawUI的相关方法替代[console]类，能更好地适配PowerShell宿主环境。

示例修改后的空闲检测逻辑：

function Prompt {
    # 自定义提示符内容
    Write-Host "PS> " -NoNewline -ForegroundColor Cyan

    # 空闲检测循环
    while (-not $host.UI.RawUI.KeyAvailable) {
        # 空闲时执行的操作（示例：每隔2秒输出一次空闲提示）
        Start-Sleep -Seconds 2
        Write-Host "`n[空闲中]..." -ForegroundColor Gray
        # 重新输出提示符，避免覆盖输入
        Write-Host "`nPS> " -NoNewline -ForegroundColor Cyan
    }

    # 读取按键
    $key = $host.UI.RawUI.ReadKey("NoEcho, IncludeKeyDown")
    
    # 判断是否是需要PowerShell处理的有效输入（回车、换行）
    if ($key.VirtualKeyCode -eq 13 -or $key.VirtualKeyCode -eq 10) {
        # 将回车写回输入缓冲区，让PowerShell处理命令提交
        $host.UI.RawUI.WriteInput($key.Character)
    }

    # 返回空字符串让PowerShell处理输入（Prompt函数需返回提示符文本，这里我们已经自定义输出）
    return ""
}

处理多行粘贴的额外优化 #

如果要处理粘贴多行的场景，需要检查输入缓冲区中是否有多个字符，而不是只检测单个按键。可以通过以下方式读取所有待处理的输入并写回：

# 在循环退出后，读取所有待输入的内容
$inputBuffer = @()
while ($host.UI.RawUI.KeyAvailable) {
    $key = $host.UI.RawUI.ReadKey("NoEcho, IncludeKeyDown")
    $inputBuffer += $key.Character
}
# 将读取到的内容写回缓冲区
if ($inputBuffer.Count -gt 0) {
    $host.UI.RawUI.WriteInput(-join $inputBuffer)
}

核心原理 #

• $host.UI.RawUI是PowerShell宿主提供的控制台交互接口，比直接调用[console]类更适配PowerShell的输入处理流程。
• 通过WriteInput方法将有效输入放回缓冲区，确保PowerShell能正常接收用户的命令提交（回车）或粘贴的多行内容。
• 避免在Prompt函数中消费所有按键，只拦截空闲检测的触发按键，将用户的有效操作交还给PowerShell处理。

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