riok · latonz · Apr 20, 2025 · Apr 20, 2025
diff --git a/src/Riok.Mapperly/Descriptors/ObjectFactories/ObjectFactoryBuilder.cs b/src/Riok.Mapperly/Descriptors/ObjectFactories/ObjectFactoryBuilder.cs
@@ -9,9 +9,8 @@ public static class ObjectFactoryBuilder
 {
     public static ObjectFactoryCollection ExtractObjectFactories(SimpleMappingBuilderContext ctx, ITypeSymbol mapperSymbol, bool isStatic)
     {
-        var objectFactories = mapperSymbol
-            .GetMembers()
-            .OfType<IMethodSymbol>()
+        var objectFactories = ctx
+            .SymbolAccessor.GetAllDirectlyAccessibleMethods(mapperSymbol)
             .Where(m => ctx.SymbolAccessor.HasAttribute<ObjectFactoryAttribute>(m))
             .Select(x => BuildObjectFactory(ctx, x, isStatic))
             .WhereNotNull()

diff --git a/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs b/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs
@@ -256,6 +256,11 @@ internal Dictionary<object, IFieldSymbol> GetEnumFieldsByValue(ITypeSymbol symbo
     internal HashSet<IFieldSymbol> GetFieldsExcept(ITypeSymbol symbol, ISet<IFieldSymbol> ignoredMembers) =>
         GetAllFields(symbol).Where(x => !ignoredMembers.Remove(x)).ToHashSet(SymbolTypeEqualityComparer.FieldDefault);
 
+    internal IEnumerable<IMethodSymbol> GetAllDirectlyAccessibleMethods(ITypeSymbol symbol)
+    {
+        return GetAllMembers(symbol).OfType<IMethodSymbol>().Where(IsDirectlyAccessible);
+    }
+
     internal IEnumerable<IMethodSymbol> GetAllDirectlyAccessibleMethods(ITypeSymbol symbol, string name)
     {
         return GetAllMembers(symbol)

diff --git a/test/Riok.Mapperly.Tests/Mapping/ObjectFactoryTest.cs b/test/Riok.Mapperly.Tests/Mapping/ObjectFactoryTest.cs
@@ -28,6 +28,34 @@ public void ShouldUseSimpleObjectFactory()
             );
     }
 
+    [Fact]
+    public Task ShouldUseSimpleObjectFactoryOfBaseClass()
+    {
+        var source = TestSourceBuilder.CSharp(
+            """
+            using System;
+            using Riok.Mapperly.Abstractions;
+
+            public abstract class BaseMapper
+            {
+                [ObjectFactory]
+                protected B CreateB() => new B();
+            }
+
+            [Mapper]
+            public partial class Mapper : BaseMapper
+            {
+                public partial B Map(A source);
+            }
+
+            public record A(int Value);
+            public record B { public int Value { get; set; } };
+            """
+        );
+
+        return TestHelper.VerifyGenerator(source);
+    }
+
     [Fact]
     public void ShouldUseSimpleObjectFactoryForMultipleMaps()
     {

diff --git a/..._snapshots/ObjectFactoryTest.ShouldUseSimpleObjectFactoryOfBaseClass#Mapper.g.verified.cs b/..._snapshots/ObjectFactoryTest.ShouldUseSimpleObjectFactoryOfBaseClass#Mapper.g.verified.cs
@@ -0,0 +1,13 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    [global::System.CodeDom.Compiler.GeneratedCode("Riok.Mapperly", "0.0.1.0")]
+    public partial global::B Map(global::A source)
+    {
+        var target = CreateB();
+        target.Value = source.Value;
+        return target;
+    }
+}