Load Razor source generators from SDK

analyzers/tests/SonarAnalyzer.TestFramework.Test/Common/SourceGeneratorProviderTest.cs

@@ -25,20 +25,40 @@ namespace SonarAnalyzer.TestFramework.Test.Common;
 [TestClass]
 public class SourceGeneratorProviderTest
 {
+    private static AnalyzerFileReference RazorSourceGenerator =>
+        SourceGeneratorProvider.SourceGenerators.Single(x => x.FullPath.EndsWith("Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll"));
+
     [TestMethod]
     public void SourceGenerators_ContainsRazorSourceGenerator() =>
         SourceGeneratorProvider.SourceGenerators.Should().ContainSingle().Which.FullPath.Should().EndWith("Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll");
 
     [TestMethod]
     public void RazorSourceGenerator_HasCorrectPath() =>
-        RazorSourceGenerator.FullPath
-            .Should()
-            .Be(Path.Combine(Path.GetDirectoryName(typeof(SourceGeneratorProvider).Assembly.Location), "Dependencies", "Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll"));
+        RazorSourceGenerator.FullPath.Should().EndWith(Path.Combine("Sdks", "Microsoft.NET.Sdk.Razor", "source-generators", "Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll"));
+
+    [TestMethod]
+    public void RazorSourceGenerator_ExistsLocally() =>
+        File.Exists(RazorSourceGenerator.FullPath).Should().BeTrue();
 
     [TestMethod]
     public void RazorSourceGenerator_LoadsCorrectAssembly() =>
         RazorSourceGenerator.GetAssembly().GetName().Name.Should().Be("Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators");
 
-    private static AnalyzerFileReference RazorSourceGenerator =>
-        SourceGeneratorProvider.SourceGenerators.Single(x => x.FullPath.EndsWith("Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll"));
+    [TestMethod]
+    public void LatestSdkFolder_ReturnsCorrectPath()
+    {
+        var latestSdkFolder = SourceGeneratorProvider.LatestSdkFolder();
+        var latestSdkVersion = Version.Parse(Path.GetFileName(latestSdkFolder));
+
+        var parentDirectory = Directory.GetParent(latestSdkFolder);
+        parentDirectory.Name.Should().Be("sdk", "Parent directory of the latest SDK should be 'sdk'");
+
+        Directory.GetDirectories(parentDirectory.FullName, $"{typeof(object).Assembly.GetName().Version.Major}.*", SearchOption.TopDirectoryOnly)
+            .Any(x => IsHigherVersion(x, latestSdkVersion))
+            .Should()
+            .BeFalse("There should be no SDK folders with a higher version number than the latest SDK folder");
+    }
+
+    private static bool IsHigherVersion(string directory, Version referenceVersion) =>
+        Version.TryParse(new DirectoryInfo(directory).Name, out var version) && version > referenceVersion;
 }

analyzers/tests/SonarAnalyzer.TestFramework/Common/SourceGeneratorProvider.cs

@@ -26,13 +26,34 @@ namespace SonarAnalyzer.TestFramework.Common;
 public static class SourceGeneratorProvider
 {
     private static readonly string RazorSourceGeneratorPath =
-        Path.Combine(Path.GetDirectoryName(typeof(SourceGeneratorProvider).Assembly.Location), "Dependencies", "Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll");
+        Path.Combine(LatestSdkFolder(), "Sdks", "Microsoft.NET.Sdk.Razor", "source-generators", "Microsoft.CodeAnalysis.Razor.Compiler.SourceGenerators.dll");
 
     public static AnalyzerFileReference[] SourceGenerators { get; } =
     [
-        new(RazorSourceGeneratorPath, new AssemblyLoader())
+        new(CheckAndReturnRazorSourceGeneratorPath(), new AssemblyLoader())
     ];
 
+    public static string CheckAndReturnRazorSourceGeneratorPath() =>
+        File.Exists(RazorSourceGeneratorPath) ? RazorSourceGeneratorPath : throw new FileNotFoundException($"Razor sourcegenerator not found: {RazorSourceGeneratorPath}");
+
+    public static string LatestSdkFolder()
+    {
+        var objectAssembly = typeof(object).Assembly;
+        var objectAssemblyDirectory = Directory.GetParent(objectAssembly.Location);    // C:\Program Files\dotnet\shared\Microsoft.NETCore.App\8.0.4
+        var dotnetDirectory = objectAssemblyDirectory.Parent.Parent.Parent;            // C:\Program Files\dotnet
+        var sdkDirectory = Path.Combine(dotnetDirectory.FullName, "sdk");              // C:\Program Files\dotnet\sdk
+
+        if (!Directory.Exists(sdkDirectory))
+        {
+            throw new NotSupportedException($"The directory '{sdkDirectory}' does not exist. " +
+                $"This may be because you are not using .NET Core. " +
+                $"Please note that Razor analysis is only supported when using .NET Core.");
+        }
+        return Directory.GetDirectories(sdkDirectory, $"{objectAssembly.GetName().Version.Major}.*", SearchOption.TopDirectoryOnly) is { Length: > 0 } specificMajorVersionSdkDirectories
+            ? specificMajorVersionSdkDirectories.OrderByDescending(x => Version.Parse(new DirectoryInfo(x).Name)).First()
+            : throw new DirectoryNotFoundException($"SDK directory not found for version {objectAssembly.GetName().Version.Major}");
+    }
+
     private sealed class AssemblyLoader : IAnalyzerAssemblyLoader
     {
         public void AddDependencyLocation(string fullPath) { }