fix(filter): check 2 layers of the attribute class' BaseType

SteveDunn · Jan 6, 2023 · 282dfc9 · 282dfc9
1 parent 2b62a1d
commit 282dfc9
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/Vogen/VoFilter.cs b/src/Vogen/VoFilter.cs
@@ -22,10 +22,11 @@ public static bool IsTarget(SyntaxNode syntaxNode) =>
     public static IEnumerable<AttributeData> TryGetValueObjectAttributes(INamedTypeSymbol voSymbolInformation)
     {
         var attrs = voSymbolInformation.GetAttributes();
-        
+
         return attrs.Where(
-            a => a.AttributeClass?.FullName() == "Vogen.ValueObjectAttribute"
-                 || a.AttributeClass?.BaseType?.FullName() == "Vogen.ValueObjectAttribute");
+                a => a.AttributeClass?.FullName() == "Vogen.ValueObjectAttribute"
+                     || a.AttributeClass?.BaseType?.FullName() == "Vogen.ValueObjectAttribute"
+                     || a.AttributeClass?.BaseType?.BaseType?.FullName() == "Vogen.ValueObjectAttribute");
     }
 
     // This is stage 2 in the pipeline - we filter down to just 1 target
@@ -34,16 +35,16 @@ public static IEnumerable<AttributeData> TryGetValueObjectAttributes(INamedTypeS
         var voSyntaxInformation = (TypeDeclarationSyntax) context.Node;
 
         var semanticModel = context.SemanticModel;
-        
+
         var voSymbolInformation = (INamedTypeSymbol) semanticModel.GetDeclaredSymbol(context.Node)!;
 
         var attributeData = TryGetValueObjectAttributes(voSymbolInformation).ToImmutableArray();
-        
+
         if(attributeData.Length > 0)
         {
             return new VoTarget(
                 semanticModel,
-                voSyntaxInformation, 
+                voSyntaxInformation,
                 semanticModel.GetDeclaredSymbol(context.Node)!.ContainingType,
                 voSymbolInformation,
                 attributeData);
@@ -71,4 +72,4 @@ public static bool IsTarget(INamedTypeSymbol? voClass)
 
         return voAttribute is not null;
     }
-}
+}