public Shell_OutOfProc(VisualStudioInstance visualStudioInstance)
: base(visualStudioInstance)
{
_inProc = CreateInProcComponent <Shell_InProc>(visualStudioInstance);
}
private static async Task <IEnumerable <SuggestedActionSet>?> TryWaitForItemsAsync(ILightBulbBroker broker, IWpfTextView view, CancellationToken cancellationToken)
{
var activeSession = broker.GetSession(view);
if (activeSession == null)
{
var bufferType = view.TextBuffer.ContentType.DisplayName;
throw new InvalidOperationException($"No expanded light bulb session found after View.ShowSmartTag. Buffer content type={bufferType}");
}
var asyncSession = (IAsyncLightBulbSession)activeSession;
var tcs = new TaskCompletionSource <List <SuggestedActionSet> >();
EventHandler <SuggestedActionsUpdatedArgs>?handler = null;
handler = (s, e) =>
{
// ignore these. we care about when the lightbulb items are all completed.
if (e.Status == QuerySuggestedActionCompletionStatus.InProgress)
{
return;
}
if (e.Status == QuerySuggestedActionCompletionStatus.Completed)
{
tcs.SetResult(e.ActionSets.ToList());
}
else if (e.Status == QuerySuggestedActionCompletionStatus.CompletedWithoutData)
{
tcs.SetResult(new List <SuggestedActionSet>());
}
else if (e.Status == QuerySuggestedActionCompletionStatus.Canceled)
{
tcs.TrySetCanceled();
}
else
{
tcs.TrySetException(new InvalidOperationException($"Light bulb transitioned to non-complete state: {e.Status}"));
}
asyncSession.SuggestedActionsUpdated -= handler;
};
asyncSession.SuggestedActionsUpdated += handler;
asyncSession.Dismissed += (_, _) => tcs.TrySetCanceled(new CancellationToken(true));
if (asyncSession.IsDismissed)
{
tcs.TrySetCanceled(new CancellationToken(true));
}
// Calling PopulateWithData ensures the underlying session will call SuggestedActionsUpdated at least once
// with the latest data computed. This is needed so that if the lightbulb computation is already complete
// that we hear about the results.
asyncSession.PopulateWithData(overrideRequestedActionCategories: null, operationContext: null);
try
{
return(await tcs.Task.WithCancellation(cancellationToken));
}
catch (OperationCanceledException) when(!cancellationToken.IsCancellationRequested)
{
var shell = Shell_InProc.Create();
var version = shell.GetVersion();
if (Version.Parse("17.2.32427.441") >= version)
{
// Unexpected cancellation can occur when the editor dismisses the light bulb without request
return(null);
}
throw new OperationCanceledException($"IDE version '{version}' unexpectedly dismissed the light bulb.");
}
}
