Remove more workspace registration hookup code. (#78910)

Followup to #78909

Author: CyrusNajmabadi

src/EditorFeatures/Core/Suggestions/SuggestedActionsSource.State.cs

@@ -3,7 +3,6 @@
 // See the LICENSE file in the project root for more information.
 
 using System;
-using Microsoft.CodeAnalysis.Text;
 using Microsoft.VisualStudio.Text;
 using Microsoft.VisualStudio.Text.Editor;
 
@@ -20,9 +19,6 @@ private sealed class State : IDisposable
             public readonly SuggestedActionsSourceProvider Owner;
             public readonly ITextView TextView;
             public readonly ITextBuffer SubjectBuffer;
-            public readonly WorkspaceRegistration Registration;
-
-            public Workspace? Workspace => Registration.Workspace;
 
             public State(SuggestedActionsSource source, SuggestedActionsSourceProvider owner, ITextView textView, ITextBuffer textBuffer)
             {
@@ -31,7 +27,6 @@ public State(SuggestedActionsSource source, SuggestedActionsSourceProvider owner
                 Owner = owner;
                 TextView = textView;
                 SubjectBuffer = textBuffer;
-                Registration = Workspace.GetWorkspaceRegistration(textBuffer.AsTextContainer());
             }
 
             void IDisposable.Dispose()

src/EditorFeatures/Core/Suggestions/SuggestedActionsSource.cs

@@ -63,29 +63,10 @@ public bool TryGetTelemetryId(out Guid telemetryId)
 
             using var state = _state.TryAddReference();
             if (state is null)
-            {
-                return false;
-            }
-
-            var workspace = state.Target.Workspace;
-            if (workspace == null)
-            {
-                return false;
-            }
-
-            var documentId = workspace.GetDocumentIdInCurrentContext(state.Target.SubjectBuffer.AsTextContainer());
-            if (documentId == null)
-            {
-                return false;
-            }
-
-            var project = workspace.CurrentSolution.GetProject(documentId.ProjectId);
-            if (project == null)
-            {
                 return false;
-            }
 
-            switch (project.Language)
+            var document = state.Target.SubjectBuffer.CurrentSnapshot.GetOpenDocumentInCurrentContextWithChanges();
+            switch (document?.Project.Language)
             {
                 case LanguageNames.CSharp:
                     telemetryId = s_CSharpSourceGuid;
@@ -171,10 +152,6 @@ private void OnTextViewClosed(object sender, EventArgs e)
                 range.Snapshot.TextBuffer.Equals(state.Target.SubjectBuffer),
                 $"Invalid text buffer passed to {nameof(HasSuggestedActionsAsync)}");
 
-            var workspace = state.Target.Workspace;
-            if (workspace == null)
-                return null;
-
             using var asyncToken = state.Target.Owner.OperationListener.BeginAsyncOperation(nameof(GetSuggestedActionCategoriesAsync));
             var document = range.Snapshot.GetOpenTextDocumentInCurrentContextWithChanges();
             if (document == null)

src/EditorFeatures/Core/Suggestions/SuggestedActionsSource_Async.cs

@@ -73,18 +73,16 @@ private async Task GetSuggestedActionsWorkerAsync(
             if (state is null)
                 return;
 
-            var workspace = state.Target.Workspace;
-            if (workspace is null)
+            var document = range.Snapshot.GetOpenTextDocumentInCurrentContextWithChanges();
+            if (document is null)
                 return;
 
             var selection = TryGetCodeRefactoringSelection(state, range);
-            await workspace.Services.GetRequiredService<IWorkspaceStatusService>().WaitUntilFullyLoadedAsync(cancellationToken).ConfigureAwait(false);
+            await document.Project.Solution.Services.GetRequiredService<IWorkspaceStatusService>()
+                .WaitUntilFullyLoadedAsync(cancellationToken).ConfigureAwait(false);
 
             using (Logger.LogBlock(FunctionId.SuggestedActions_GetSuggestedActionsAsync, cancellationToken))
             {
-                var document = range.Snapshot.GetOpenTextDocumentInCurrentContextWithChanges();
-                if (document is null)
-                    return;
 
                 // Create a single keep-alive session as we process each lightbulb priority group.  We want to
                 // ensure that all calls to OOP will reuse the same solution-snapshot on the oop side (including

src/EditorFeatures/Core/Tagging/AbstractAsynchronousTaggerProvider.TagSource.cs

@@ -150,8 +150,6 @@ public TagSource(
             _nonFrozenComputationCancellationSeries = new(_disposalTokenSource.Token);
             _tagSpanSetPool = new ObjectPool<HashSet<TagSpan<TTag>>>(() => new HashSet<TagSpan<TTag>>(this), trimOnFree: false);
 
-            Workspace.GetWorkspaceRegistration(subjectBuffer.AsTextContainer());
-
             // PERF: Use AsyncBatchingWorkQueue<_, VoidResult> instead of AsyncBatchingWorkQueue<_> because the latter
             // has an async state machine that rethrows a very common cancellation exception.
             _eventChangeQueue = new AsyncBatchingWorkQueue<TagSourceQueueItem, VoidResult>(