如何修复Oh-My-Posh瞬态提示符在PSReadLine AcceptLine()下失效问题

问题分析 #

你通过自定义Tab键的PSReadLine处理逻辑，实现了按下Tab时调用历史命令并自动执行，但执行后瞬态提示符未正常切换——这是因为直接调用AcceptLine()会绕过PowerShell默认的提示符更新流程，没有触发prompt函数的重新执行，而瞬态提示符的切换逻辑正是在prompt函数中定义的。

解决方案 #

修改你的按键处理脚本，在AcceptLine()后手动触发提示符的重新渲染：

Set-PSReadLineKeyHandler -Key Tab -ScriptBlock {
    [Microsoft.PowerShell.PSConsoleReadLine]::NextHistory()
    [Microsoft.PowerShell.PSConsoleReadLine]::AcceptLine()
    
    # 手动调用prompt函数获取瞬态提示符内容
    $transientPrompt = & $function:prompt
    # 清除当前行残留内容
    [Microsoft.PowerShell.PSConsoleReadLine]::ClearLine()
    # 写入瞬态提示符
    $host.ui.Write($transientPrompt)
    # 将光标定位到提示符末尾
    [Microsoft.PowerShell.PSConsoleReadLine]::SetCursorPosition($transientPrompt.Length)
}

原理说明 #

1. 调用& $function:prompt直接执行你的自定义提示符函数，获取瞬态状态下的提示符内容；
2. ClearLine()清除当前行可能残留的旧提示符或命令内容；
3. 通过$host.ui.Write()写入新的瞬态提示符；
4. SetCursorPosition()将光标移动到提示符末尾，确保后续输入正常。

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