Making it a little bit more robust

codecentric · Aug 7, 2024 · 18fc53f · 18fc53f
1 parent 61bcacb
commit 18fc53f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs b/AutomaticInterface/AutomaticInterface/AutomaticInterfaceGenerator.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Immutable;
 using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
 
 namespace AutomaticInterface;
 
@@ -18,10 +19,9 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
         var classes = context
             .SyntaxProvider.ForAttributeWithMetadataName(
                 $"AutomaticInterface.{DefaultAttributeName}Attribute",
-                (_, _) => true,
+                (node, _) => node is ClassDeclarationSyntax,
                 (context, _) => (ITypeSymbol)context.TargetSymbol
             )
-            .Where(type => type is not null)
             .Collect();
 
         context.RegisterSourceOutput(classes, GenerateCode);