Fix #23, allow using projects with .Tools NuGet installed instead of …

….Design
JetBrains · Dec 5, 2021 · 1e6e7a8 · 1e6e7a8
1 parent 2cacf27
commit 1e6e7a8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/dotnet/Rider.Plugins.EfCore/EfCoreHelper.cs b/src/dotnet/Rider.Plugins.EfCore/EfCoreHelper.cs
@@ -8,15 +8,17 @@ namespace Rider.Plugins.EfCore
 {
     public static class EfCoreHelper
     {
-        private static readonly string EfCoreToolsNugetId = "Microsoft.EntityFrameworkCore.Design";
+        private static readonly string EfCoreDesignNugetId = "Microsoft.EntityFrameworkCore.Design";
+        private static readonly string EfCoreToolsNugetId = "Microsoft.EntityFrameworkCore.Tools";
 
         public static IEnumerable<IProject> GetSupportedMigrationProjects(ISolution solution) =>
             GetSupportedDotnetProjects(solution)
                 .Where(project => project.ProjectFileLocation.ExtensionNoDot == "csproj");
 
         public static IEnumerable<IProject> GetSupportedStartupProjects(ISolution solution) =>
             GetSupportedDotnetProjects(solution)
-                .Where(project => project.GetInstalledPackage(EfCoreToolsNugetId) != default);
+                .Where(project => project.GetInstalledPackage(EfCoreToolsNugetId) != default
+                                  || project.GetInstalledPackage(EfCoreDesignNugetId) != default);
 
         private static IEnumerable<IProject> GetSupportedDotnetProjects(ISolution solution) =>
             solution