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

SteveDunn · Dec 29, 2022 · b11b363 · b11b363
1 parent 7c75050
commit b11b363
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 @@ internal static class VoFilter
  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;
  }
-}
+}