Tell web tools which LSP server to send WrapWithTag requests to

Author: davidwengier

src/Razor/src/Microsoft.CodeAnalysis.Razor.CohostingShared/RazorLSPConstants.cs

@@ -5,10 +5,14 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient;
 
 internal static class RazorLSPConstants
 {
+    public const string WebToolsWrapWithTagServerNameProperty = "_wrap_with_tag_lsp_server_name";
+
     public const string RazorCSharpLanguageServerName = "Razor C# Language Server Client";
 
     public const string RazorLanguageServerName = "Razor Language Server Client";
 
+    public const string RoslynLanguageServerName = "Roslyn Language Server Client";
+
     public const string CohostLanguageServerName = "Cohosted Razor Language Server Client";
 
     public const string HtmlLanguageServerName = "HtmlDelegationLanguageServerClient";

src/Razor/src/Microsoft.VisualStudio.LanguageServices.Razor/RazorLSPTextViewConnectionListener.cs

@@ -11,6 +11,7 @@
 using Microsoft.VisualStudio.Language.Intellisense;
 using Microsoft.VisualStudio.OLE.Interop;
 using Microsoft.VisualStudio.Razor.Extensions;
+using Microsoft.VisualStudio.Razor.LanguageClient;
 using Microsoft.VisualStudio.Shell;
 using Microsoft.VisualStudio.Text;
 using Microsoft.VisualStudio.Text.Editor;
@@ -128,6 +129,15 @@ public void SubjectBuffersConnected(ITextView textView, ConnectionReason reason,
         {
             _activeTextViews.Add(textView);
 
+            if (!textView.TextBuffer.Properties.ContainsProperty(RazorLSPConstants.WebToolsWrapWithTagServerNameProperty))
+            {
+                // We have to tell web tools which language server to send requests to for this buffer, but that changes
+                // if cohosting is enabled.
+                textView.TextBuffer.Properties[RazorLSPConstants.WebToolsWrapWithTagServerNameProperty] = _featureOptions.UseRazorCohostServer
+                    ? RazorLSPConstants.RoslynLanguageServerName
+                    : RazorLSPConstants.RazorLanguageServerName;
+            }
+
             // Initialize the user's options and start listening for changes.
             // We only want to attach the option changed event once so we don't receive multiple
             // notifications if there is more than one TextView active.