SampleCommand.cs

// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

namespace UserPromptSample;

using System.Diagnostics;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.VisualStudio.Extensibility;
using Microsoft.VisualStudio.Extensibility.Commands;
using Microsoft.VisualStudio.Extensibility.Shell;

[VisualStudioContribution]
public class SampleCommand : Command
{
    public enum TokenThemeResult
    {
        None,
        Solarized,
        OneDark,
        GruvBox,
    }

    /// <inheritdoc />
    public override CommandConfiguration CommandConfiguration => new("%UserPromptSample.SampleCommand.DisplayName%")
    {
        TooltipText = "%UserPromptSample.SampleCommand.ToolTip%",
        Placements = new[] { CommandPlacement.KnownPlacements.ToolsMenu },
    };

    /// <inheritdoc />
    public override async Task ExecuteCommandAsync(IClientContext context, CancellationToken ct)
    {
        var shell = this.Extensibility.Shell();

        // Asking the user to confirm an operation.
        if (!await shell.ShowPromptAsync("Continue with executing the command?", PromptOptions.OKCancel, ct))
        {
            return;
        }

        // Asking the user to confirm a dangerous operation.
        if (!await shell.ShowPromptAsync("Continue with executing the command?", PromptOptions.OKCancel.WithCancelAsDefault(), ct))
        {
            return;
        }

        // OK-only prompt
        await shell.ShowPromptAsync("The extension must reload.", PromptOptions.OK, ct);

        // Custom prompt
        var themeResult = await shell.ShowPromptAsync(
            "Which theme should be used for the generated output?",
            new PromptOptions<TokenThemeResult>
            {
                Choices =
                {
            { "Solarized Is Awesome", TokenThemeResult.Solarized },
            { "OneDark Is The Best", TokenThemeResult.OneDark },
            { "GruvBox Is Groovy", TokenThemeResult.GruvBox },
                },
                DismissedReturns = TokenThemeResult.None,
                DefaultChoiceIndex = 2,
            },
            ct);

        Debug.WriteLine($"Selected Token Theme: {themeResult}");
    }
}