fix bug with reader generation

loresoft · Sep 14, 2023 · 8b8a981 · 8b8a981
1 parent 64c8065
commit 8b8a981
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 8 deletions.
diff --git a/src/FluentCommand.Generators/DataReaderFactoryGenerator.cs b/src/FluentCommand.Generators/DataReaderFactoryGenerator.cs
@@ -57,11 +57,11 @@ private static void Execute(SourceProductionContext context, EntityClass entityC
     private static bool SyntacticPredicate(SyntaxNode syntaxNode, CancellationToken cancellationToken)
     {
         return syntaxNode is ClassDeclarationSyntax
-               { AttributeLists.Count: > 0 } classDeclaration
+        { AttributeLists.Count: > 0 } classDeclaration
                && !classDeclaration.Modifiers.Any(SyntaxKind.AbstractKeyword)
                && !classDeclaration.Modifiers.Any(SyntaxKind.StaticKeyword)
             || syntaxNode is RecordDeclarationSyntax
-               { AttributeLists.Count: > 0 } recordDeclaration
+            { AttributeLists.Count: > 0 } recordDeclaration
                && !recordDeclaration.Modifiers.Any(SyntaxKind.AbstractKeyword)
                && !recordDeclaration.Modifiers.Any(SyntaxKind.StaticKeyword);
     }
@@ -78,12 +78,7 @@ private static EntityContext SemanticTransform(GeneratorAttributeSyntaxContext c
             ? InitializationMode.ObjectInitializer
             : InitializationMode.Constructor;
 
-        var propertySymbols = targetSymbol
-            .GetMembers()
-            .Where(m => m.Kind == SymbolKind.Property)
-            .OfType<IPropertySymbol>()
-            .Where(IsIncluded)
-            .ToList();
+        var propertySymbols = GetProperties(targetSymbol);
 
         if (mode == InitializationMode.ObjectInitializer)
         {
@@ -143,6 +138,31 @@ private static EntityContext SemanticTransform(GeneratorAttributeSyntaxContext c
         return new EntityContext(entityClass, diagnostics);
     }
 
+    private static List<IPropertySymbol> GetProperties(INamedTypeSymbol targetSymbol)
+    {
+        var properties = new Dictionary<string, IPropertySymbol>();
+
+        var currentSymbol = targetSymbol;
+
+        // get nested properties
+        while (currentSymbol != null)
+        {
+            var propertySymbols = currentSymbol
+                .GetMembers()
+                .Where(m => m.Kind == SymbolKind.Property)
+                .OfType<IPropertySymbol>()
+                .Where(IsIncluded)
+                .Where(p => !properties.ContainsKey(p.Name));
+
+            foreach (var propertySymbol in propertySymbols)
+                properties.Add(propertySymbol.Name, propertySymbol);
+
+            currentSymbol = currentSymbol.BaseType;
+        }
+
+        return properties.Values.ToList();
+    }
+
     private static EntityProperty CreateProperty(IPropertySymbol propertySymbol, string parameterName = null)
     {
         var propertyType = propertySymbol.Type.ToDisplayString();

diff --git a/test/FluentCommand.Tests/Models/Truck.cs b/test/FluentCommand.Tests/Models/Truck.cs
@@ -0,0 +1,10 @@
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace FluentCommand.Tests.Models;
+
+[Table(nameof(Truck))]
+public class Truck : Vehicle
+{
+    public string Color { get; set; }
+    public override int Type { get; set; }
+}
diff --git a/test/FluentCommand.Tests/Models/Vehicle.cs b/test/FluentCommand.Tests/Models/Vehicle.cs
@@ -0,0 +1,10 @@
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace FluentCommand.Tests.Models;
+
+[Table(nameof(Vehicle))]
+public class Vehicle
+{
+    public string Name { get; set; }
+    public virtual int Type { get; set; }
+}