Fix code fix construction

OmniSharp · Jul 25, 2024 · 57f5898 · 57f5898
1 parent 2dad815
commit 57f5898
Showing 1 changed file with 26 additions and 6 deletions.
diff --git a/src/OmniSharp.Roslyn.CSharp/Services/Refactoring/V2/CachingCodeFixProviderForProjects.cs b/src/OmniSharp.Roslyn.CSharp/Services/Refactoring/V2/CachingCodeFixProviderForProjects.cs
@@ -61,30 +61,30 @@ private ImmutableArray<CodeFixProvider> LoadFrom(Project project)
         {
             var codeFixesFromProjectReferences = project.AnalyzerReferences
                 .OfType<AnalyzerFileReference>()
-                .SelectMany(analyzerFileReference => analyzerFileReference.GetAssembly().DefinedTypes)
+                .SelectMany(analyzerFileReference => GetConcreteTypes(analyzerFileReference.GetAssembly()))
                 .Where(x => !x.IsAbstract && x.IsSubclassOf(typeof(CodeFixProvider)))
                 .Select(x =>
                 {
                     try
                     {
-                        var attribute = x.GetCustomAttribute<ExportCodeFixProviderAttribute>();
+                        var attribute = x.GetCustomAttribute<ExportCodeFixProviderAttribute>(inherit: false);
                         if (attribute == null)
                         {
-                            _logger.LogTrace($"Skipping code fix provider '{x.AsType()}' because it is missing the ExportCodeFixProviderAttribute.");
+                            _logger.LogTrace($"Skipping code fix provider '{x}' because it is missing the ExportCodeFixProviderAttribute.");
                             return null;
                         }
 
                         if (attribute.Languages == null || !attribute.Languages.Contains(project.Language))
                         {
-                            _logger.LogInformation($"Skipping code fix provider '{x.AsType()}' because its language '{attribute.Languages?.FirstOrDefault()}' doesn't match '{project.Language}'.");
+                            _logger.LogInformation($"Skipping code fix provider '{x}' because its language '{attribute.Languages?.FirstOrDefault()}' doesn't match '{project.Language}'.");
                             return null;
                         }
 
-                        return x.AsType().CreateInstance<CodeFixProvider>();
+                        return (CodeFixProvider)Activator.CreateInstance(x);
                     }
                     catch (Exception ex)
                     {
-                        _logger.LogError($"Creating instance of code fix provider '{x.AsType()}' failed, error: {ex}");
+                        _logger.LogError($"Creating instance of code fix provider '{x}' failed, error: {ex}");
                         return null;
                     }
                 })
@@ -98,5 +98,25 @@ private ImmutableArray<CodeFixProvider> LoadFrom(Project project)
 
             return allCodeFixes;
         }
+
+        private IEnumerable<Type> GetConcreteTypes(Assembly assembly)
+        {
+            try
+            {
+                var concreteTypes = assembly
+                    .GetTypes()
+                    .Where(type => !type.GetTypeInfo().IsInterface
+                        && !type.GetTypeInfo().IsAbstract
+                        && !type.GetTypeInfo().ContainsGenericParameters);
+
+                // Realize the collection to ensure exceptions are caught
+                return concreteTypes.ToList();
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError($"Getting concrete types from assembly '{assembly}' failed, error: {ex}");
+                return Type.EmptyTypes;
+            }
+        }
     }
 }