Merge pull request #67650 from mavasani/FixFixAllInProjectSolutionForFSharp

Fix for FixAll in project and solution scopes for non-C#/VB projects

Author: mavasani

src/Workspaces/Core/Portable/CodeFixes/FixAllOccurrences/FixAllContextHelper.cs

@@ -130,9 +130,11 @@ private static async Task<ImmutableDictionary<Document, ImmutableArray<Diagnosti
             CancellationToken cancellationToken)
         {
             var builder = ImmutableDictionary.CreateBuilder<Document, ImmutableArray<Diagnostic>>();
-            foreach (var (document, diagnosticsForDocument) in diagnostics.GroupBy(d => solution.GetDocument(d.Location.SourceTree)))
+
+            // NOTE: We use 'GetTextDocumentForLocation' extension to ensure we also handle external location diagnostics in non-C#/VB languages.
+            foreach (var (textDocument, diagnosticsForDocument) in diagnostics.GroupBy(d => solution.GetTextDocumentForLocation(d.Location)))
             {
-                if (document is null)
+                if (textDocument is not Document document)
                     continue;
 
                 cancellationToken.ThrowIfCancellationRequested();

src/Workspaces/Core/Portable/Shared/Extensions/ISolutionExtensions.cs

@@ -4,6 +4,7 @@
 
 using System;
 using System.Collections.Immutable;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.CodeAnalysis.PooledObjects;
@@ -38,6 +39,20 @@ public static async Task<ImmutableArray<INamespaceSymbol>> GetGlobalNamespacesAs
         public static TextDocumentKind? GetDocumentKind(this Solution solution, DocumentId documentId)
             => solution.GetTextDocument(documentId)?.Kind;
 
+        internal static TextDocument? GetTextDocumentForLocation(this Solution solution, Location location)
+        {
+            switch (location.Kind)
+            {
+                case LocationKind.SourceFile:
+                    return solution.GetDocument(location.SourceTree);
+                case LocationKind.ExternalFile:
+                    var documentId = solution.GetDocumentIdsWithFilePath(location.GetLineSpan().Path).FirstOrDefault();
+                    return solution.GetTextDocument(documentId);
+                default:
+                    return null;
+            }
+        }
+
         public static Solution WithTextDocumentText(this Solution solution, DocumentId documentId, SourceText text, PreservationMode mode = PreservationMode.PreserveIdentity)
         {
             var documentKind = solution.GetDocumentKind(documentId);