NuGet Restore Status Question - Visual Studio Extension

[william-courrau-in]

Hello,

I am unsure how the API signature for Task NominateProjectAsync(string projectUniqueName, IVsProjectRestoreInfo projectRestoreInfo, CancellationToken token); works.

Specifically, I'm unclear about how to manipulate the IVsProjectRestoreInfo projectRestoreInfo parameter from outside the code. I am currently using the NuGet package, but I don't know how to nominate a project for a NuGet restore, as there seems to be no accessible API to create an IVsProjectRestoreInfo.

What I really want to achieve is to know when the solution is ready to build after the NuGet restore has finished. Can this be accomplished by using another API?

my current code:

[Export(typeof(IWaitForNuGetRestore))]
public class WaitForNuGetRestore : IWaitForNuGetRestore
{
    private readonly SVsServiceProvider serviceProvider;
    private readonly IComponentModel componentModel;
    private readonly IVsSolutionRestoreService vsSolutionRestoreService;
    private readonly IVsSolutionRestoreStatusProvider vsSolutionRestoreStatus;

    [ImportingConstructor]
    public WaitForNuGetRestore(SVsServiceProvider serviceProvider)
    {
        this.serviceProvider = serviceProvider;
        componentModel = (IComponentModel)serviceProvider.GetService(typeof(SComponentModel)) ?? throw new InvalidOperationException("Service not found");
        vsSolutionRestoreService = componentModel.GetService<IVsSolutionRestoreService>();
        vsSolutionRestoreStatus = componentModel.GetService<IVsSolutionRestoreStatusProvider>();
    }

    public async Task<bool> IsNuGetRestoreFinishedAsync(CancellationToken cancellationToken)
    {
#pragma warning disable VSTHRD003 // Avoid awaiting foreign Tasks
        var buildReadyToStart = await vsSolutionRestoreService.CurrentRestoreOperation;
#pragma warning restore VSTHRD003 // Avoid awaiting foreign Tasks
        var nugetRestoreReady = await vsSolutionRestoreStatus.IsRestoreCompleteAsync(cancellationToken);

        vsSolutionRestoreService.NominateProjectAsync();

        return buildReadyToStart && nugetRestoreReady;
    }

    public async Task WaitForNuGetRestoreFinishedAsync(CancellationToken cancellationToken)
    {
        await Task.Run(
            async () =>
            {
                while (!cancellationToken.IsCancellationRequested)
                {
                    var restoreFinished = await IsNuGetRestoreFinishedAsync(cancellationToken); 
                    if (restoreFinished) { break; }
                    await Task.Delay(1500, cancellationToken);
                }
            }, 
            cancellationToken);
    }
}