Host.ChoiceDescription need additional keypress for choosing option

[MrFly72]

Prerequisites

• I have written a descriptive issue title.
• I have searched all issues to ensure it has not already been reported.
• I have read the troubleshooting guide.
• I am sure this issue is with the extension itself and does not reproduce in a standalone PowerShell instance.
• I have verified that I am using the latest version of Visual Studio Code and the PowerShell extension.
• If this is a security issue, I have read the security issue reporting guidance.

Summary

When using System.Management.Automation.Host.ChoiceDescription in VSCode terminal, I have to press the key two times. First keypress will not be shown. The keypress will even be delivered to the next entry line and cloak it at the front.
It works in PSextension v2021.12.0 but NOT in current preview.

PowerShell Version

Name                           Value
----                           -----
PSVersion                      5.1.19041.1320
PSEdition                      Desktop
PSCompatibleVersions           {1.0, 2.0, 3.0, 4.0...}
BuildVersion                   10.0.19041.1320
CLRVersion                     4.0.30319.42000
WSManStackVersion              3.0
PSRemotingProtocolVersion      2.3
SerializationVersion           1.1.0.1

Visual Studio Code Version

1.65.2
c722ca6c7eed3d7987c0d5c3df5c45f6b15e77d1
x64

Extension Version

ms-vscode.powershell@2021.12.0       
ms-vscode.powershell-preview@2022.3.0
tobysmith568.run-in-powershell@1.1.1

Steps to Reproduce

Use this code:

    $AvailableEnvironments = @(
        @{Choice = '&IAT'; Value = "IAT"; Help = "Test" }
        @{Choice = '&Prod'; Value = "Prod"; Help = "Production" }
        @{Choice = '&Dev'; Value = "Dev"; Help = "Development" }
    )
    $title = "Please choose the environment"
    $msg = "Which environment do you want to use ?"
    $Choices = foreach ($Choice in $AvailableEnvironments) {
        New-Object System.Management.Automation.Host.ChoiceDescription $Choice.Choice, $Choice.Help
    }
    $options = $Choices
    $default = 0  # 0=Yes, 1=No
    $response = $Host.UI.PromptForChoice($title, $msg, $options, $default)
    $MainEnvironment = $AvailableEnvironments[$response].Value
    $MainEnvironment

Visuals

Pressed "P" as an answer and then hit enter like expected.
Behaviour: Default value will be used instead and the "P" will end up in the input-line:

Logs

No response

[andyleejordan]

Ah ha, this one repros. Curiously enough, only when run via F5. After everything is defined, executing the script with . repro.ps or manually pasting $response = $Host.UI.PromptForChoice($title, $msg, $options, $default) works just fine.

[andyleejordan]

Curious, @SeeminglyScience, making the change:

        public override int PromptForChoice(string caption, string message, Collection<ChoiceDescription> choices, int defaultChoice)
        {
            // return _consoleHostUI is not null
            //     ? _consoleHostUI.PromptForChoice(caption, message, choices, defaultChoice)
            //     : _underlyingHostUI.PromptForChoice(caption, message, choices, defaultChoice);
            return _underlyingHostUI.PromptForChoice(caption, message, choices, defaultChoice);
        }

Seems to resolve this...when are we supposed to use _consoleHostUI?

That change did not resolve this, it was just difficult to repro as I wasn't quite following that it's a double-press bug.

[andyleejordan]

@MrFly72 Can you test the build I just posted in this comment: #3876 (comment)? Throw the UI through everything you can.

[MrFly72]

@andschwa Still the same problem on F5/F8, dropping my entry and presenting it later on in the normal prompt. Choosing the Default instead of the letter pressed (which is even the worst in this)
There is a difference between F5 and F8:
F5 will take the default, will not print the letter I pressed AND will not show the letter in the prompt....
F8 will take the default, will not print the letter I pressed BUT will put the letter I pressed in the following PS-Entryline
In both cases pressing the letter two times then chooses the choice of the letter, eg. type "pp". The behaviour for the letter to show up is the same as above, the letter will also only show up once in F8-case (not two times, how someone might expect)

[andyleejordan]

Ah ha...ok so this doesn't repro at all on macOS, only on Windows. As @SeeminglyScience astutely pointed out, it's due to a workaround of ReadKey on Windows so we can host PSReadLine, our custom ReadKey is buffering that first letter, but since the prompt is using the default from PowerShell, it's not being given that consumed (and buffered) letter after the PSReadLine task is canceled. I can repro your issue precisely on Windows...and we're not sure if they'll be a clean fix. This explains why before the pipeline rewrite we had essentially rewritten all the prompt and Read-Host logic, which we really don't want to have to do (again).