Allow for weird Uris as file paths (#12155)

Part of fixing https://github.com/dotnet/razor/issues/10437 along with
dotnet/roslyn#80068

Author: davidwengier

src/Compiler/Microsoft.CodeAnalysis.Razor.Compiler/src/SourceGenerators/RazorSourceGenerator.cs

@@ -8,6 +8,7 @@
 using Microsoft.AspNetCore.Razor;
 using Microsoft.AspNetCore.Razor.Language;
 using Microsoft.AspNetCore.Razor.PooledObjects;
+using Microsoft.AspNetCore.Razor.Utilities;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 
@@ -54,7 +55,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                 .ReportDiagnostics(context);
 
             var sourceItems = additionalTexts
-                .Where(static (file) => file.Path.EndsWith(".razor", StringComparison.OrdinalIgnoreCase) || file.Path.EndsWith(".cshtml", StringComparison.OrdinalIgnoreCase))
+                .Where(static (file) => FileUtilities.IsAnyRazorFilePath(file.Path, StringComparison.OrdinalIgnoreCase))
                 .Combine(analyzerConfigOptions)
                 .Select(ComputeProjectItems)
                 .ReportDiagnostics(context);
@@ -79,7 +80,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                 return false;
             });
 
-            var componentFiles = sourceItems.Where(static file => file.FilePath.EndsWith(".razor", StringComparison.OrdinalIgnoreCase));
+            var componentFiles = sourceItems.Where(static file => FileUtilities.IsRazorComponentFilePath(file.FilePath, StringComparison.OrdinalIgnoreCase));
 
             var generatedDeclarationText = componentFiles
                 .Combine(importFiles.Collect())

src/Razor/src/Microsoft.CodeAnalysis.Razor.Workspaces/Extensions/StringExtensions.cs

@@ -2,20 +2,17 @@
 // The .NET Foundation licenses this file to you under the MIT license.
 
 using Microsoft.AspNetCore.Razor;
+using Microsoft.AspNetCore.Razor.Utilities;
 
 namespace Microsoft.CodeAnalysis.Razor;
 
 internal static class StringExtensions
 {
-    private const string RazorExtension = ".razor";
-    private const string CSHtmlExtension = ".cshtml";
-
     public static bool IsRazorFilePath(this string filePath)
     {
         var comparison = PathUtilities.OSSpecificPathComparison;
 
-        return filePath.EndsWith(RazorExtension, comparison) ||
-               filePath.EndsWith(CSHtmlExtension, comparison);
+        return FileUtilities.IsAnyRazorFilePath(filePath, comparison);
     }
 
     public static int? GetFirstNonWhitespaceOffset(this string line)

src/Razor/test/Microsoft.VisualStudio.LanguageServices.Razor.Test/Cohost/CohostFoldingRangeEndpointTest.cs

@@ -17,6 +17,30 @@ namespace Microsoft.VisualStudio.Razor.LanguageClient.Cohost;
 
 public class CohostFoldingRangeEndpointTest(ITestOutputHelper testOutputHelper) : CohostEndpointTestBase(testOutputHelper)
 {
+    [Fact]
+    public Task BadLooseFileUri()
+       => VerifyFoldingRangesAsync("""
+            <div>
+              @if (true) {[|
+                <div>
+                  Hello World
+                </div>
+              }
+            |]</div>
+
+            @if (true) {[|
+              <div>
+                Hello World
+              </div>
+            }
+            |]
+            @if (true) {[|
+            }|]
+            """,
+           fileKind: RazorFileKind.Legacy,
+           miscellaneousFile: true,
+           razorFilePath: "git:/c:/Users/dawengie/source/repos/razor01/Pages/Index.cshtml?%7B%22path%22:%22c:%5C%5CUsers%5C%5Cdawengie%5C%5Csource%5C%5Crepos%5C%5Crazor01%5C%5CPages%5C%5CIndex.cshtml%22,%22ref%22:%22~%22%7D");
+
     [Theory]
     [CombinatorialData]
     public Task IfStatements(bool miscellaneousFile)
@@ -215,10 +239,10 @@ public void M() {[|
             }|]
             """);
 
-    private async Task VerifyFoldingRangesAsync(string input, RazorFileKind? fileKind = null, bool miscellaneousFile = false)
+    private async Task VerifyFoldingRangesAsync(string input, RazorFileKind? fileKind = null, bool miscellaneousFile = false, string? razorFilePath = null)
     {
         TestFileMarkupParser.GetSpans(input, out var source, out ImmutableDictionary<string, ImmutableArray<TextSpan>> spans);
-        var document = CreateProjectAndRazorDocument(source, fileKind, miscellaneousFile: miscellaneousFile);
+        var document = CreateProjectAndRazorDocument(source, fileKind, miscellaneousFile: miscellaneousFile, documentFilePath: razorFilePath);
         var inputText = await document.GetTextAsync(DisposalToken);
 
         var htmlSpans = spans.GetValueOrDefault("html").NullToEmpty();

src/Shared/Microsoft.AspNetCore.Razor.Utilities.Shared/FileUtilities.cs

@@ -1,13 +1,52 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 
+using System;
 using System.Globalization;
 using System.IO;
 
 namespace Microsoft.AspNetCore.Razor.Utilities;
 
 internal static class FileUtilities
 {
+    private const string RazorExtension = ".razor";
+    private const string CSHtmlExtension = ".cshtml";
+
+    public static bool IsAnyRazorFilePath(ReadOnlySpan<char> filePath, StringComparison comparison)
+    {
+        return IsRazorComponentFilePath(filePath, comparison) ||
+               IsMvcFilePath(filePath, comparison);
+    }
+
+    public static bool IsRazorComponentFilePath(ReadOnlySpan<char> filePath, StringComparison comparison)
+    {
+        return AdjustToUsableFilePath(filePath).EndsWith(RazorExtension, comparison);
+    }
+
+    public static bool IsMvcFilePath(ReadOnlySpan<char> filePath, StringComparison comparison)
+    {
+        return AdjustToUsableFilePath(filePath).EndsWith(CSHtmlExtension, comparison);
+    }
+
+    private static ReadOnlySpan<char> AdjustToUsableFilePath(ReadOnlySpan<char> filePath)
+    {
+        // In VS Code we sometimes get odd uris with query string components as the file path, for example on the left side of
+        // a diff view. In those cases Roslyn will create a document and the file path will be set to the full raw Uri send via
+        // VS Code. When trying to find out the file extension for those Uris, we need to strip off the query string.
+        //
+        // For example we get:
+        //   git:/c:/Users/dawengie/source/repos/razor01/Pages/Index.cshtml?%7B%22path%22:%22c:%5C%5CUsers%5C%5Cdawengie%5C%5Csource%5C%5Crepos%5C%5Crazor01%5C%5CPages%5C%5CIndex.cshtml%22,%22ref%22:%22~%22%7D
+        //
+        // Given colons and question marks are unlikely, or illegal, file path characters the risk of false positives here is hopefully low.
+        if (filePath.IndexOf(":/") > 0 &&
+           filePath.IndexOf('?') is int realPathEnd and > 0)
+        {
+            return filePath[..realPathEnd];
+        }
+
+        return filePath;
+    }
+
     /// <summary>
     /// Generate a file path adjacent to the input path that has the
     /// specified file extension, using numbers to differentiate for