fix: add GetAttributesCore, modify GetAttributes and HasAttribute

riok · Jul 10, 2023 · b910c18 · b910c18
1 parent 5419f34
commit b910c18
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 22 deletions.
diff --git a/src/Riok.Mapperly/Configuration/AttributeDataAccessor.cs b/src/Riok.Mapperly/Configuration/AttributeDataAccessor.cs
@@ -44,11 +44,8 @@ public IEnumerable<TData> Access<TAttribute, TData>(ISymbol symbol)
     {
         var attrType = typeof(TAttribute);
         var dataType = typeof(TData);
-        var attrSymbol = _types.Get($"{attrType.Namespace}.{attrType.Name}");
 
-        var attrDatas = _symbolAccessor
-            .GetAttributes(symbol)
-            .Where(x => SymbolEqualityComparer.Default.Equals(x.AttributeClass?.ConstructedFrom ?? x.AttributeClass, attrSymbol));
+        var attrDatas = _symbolAccessor.GetAttributes<TAttribute>(symbol);
 
         foreach (var attrData in attrDatas)
         {

diff --git a/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs b/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs
@@ -18,26 +18,17 @@ public SymbolAccessor(WellKnownTypes types)
         _types = types;
     }
 
-    internal ImmutableArray<AttributeData> GetAttributes(ISymbol symbol)
-    {
-        if (_attributes.TryGetValue(symbol, out var attributes))
-        {
-            return attributes;
-        }
-
-        attributes = symbol.GetAttributes();
-        _attributes.Add(symbol, attributes);
-
-        return attributes;
-    }
-
-    internal bool HasAttribute<T>(ISymbol symbol)
+    internal IEnumerable<AttributeData> GetAttributes<T>(ISymbol symbol)
         where T : Attribute
     {
         var attributeSymbol = _types.Get(typeof(T));
-        return GetAttributes(symbol).Any(a => SymbolEqualityComparer.Default.Equals(a.AttributeClass, attributeSymbol));
+        return GetAttributesCore(symbol)
+            .Where(x => SymbolEqualityComparer.Default.Equals(x.AttributeClass?.ConstructedFrom ?? x.AttributeClass, attributeSymbol));
     }
 
+    internal bool HasAttribute<T>(ISymbol symbol)
+        where T : Attribute => GetAttributes<T>(symbol).Any();
+
     internal IEnumerable<IMethodSymbol> GetAllMethods(ITypeSymbol symbol) => GetAllMembers(symbol).OfType<IMethodSymbol>();
 
     internal IEnumerable<IMethodSymbol> GetAllMethods(ITypeSymbol symbol, string name) =>
@@ -81,6 +72,19 @@ internal IEnumerable<IMappableMember> GetMappableMembers(ITypeSymbol symbol, str
 
     private IEnumerable<ISymbol> GetAllMembers(ITypeSymbol symbol, string name) => GetAllMembers(symbol).Where(x => name.Equals(x.Name));
 
+    private ImmutableArray<AttributeData> GetAttributesCore(ISymbol symbol)
+    {
+        if (_attributes.TryGetValue(symbol, out var attributes))
+        {
+            return attributes;
+        }
+
+        attributes = symbol.GetAttributes();
+        _attributes.Add(symbol, attributes);
+
+        return attributes;
+    }
+
     private IEnumerable<ISymbol> GetAllMembersCore(ITypeSymbol symbol)
     {
         var members = symbol.GetMembers();
@@ -97,7 +101,7 @@ private IEnumerable<ISymbol> GetAllMembersCore(ITypeSymbol symbol)
     private IEnumerable<IMappableMember> GetAllAccessibleMappableMembersCore(ITypeSymbol symbol)
     {
         return GetAllMembers(symbol)
-            .Where(x => !x.IsStatic && x.IsAccessible() && x.Kind is SymbolKind.Property or SymbolKind.Field)
+            .Where(x => x is { IsStatic: false, Kind: SymbolKind.Property or SymbolKind.Field } && x.IsAccessible())
             .DistinctBy(x => x.Name)
             .Select(MappableMember.Create)
             .WhereNotNull();

diff --git a/src/Riok.Mapperly/Descriptors/WellKnownTypes.cs b/src/Riok.Mapperly/Descriptors/WellKnownTypes.cs
@@ -21,10 +21,10 @@ internal WellKnownTypes(Compilation compilation)
     public ITypeSymbol GetArrayType(ITypeSymbol type) =>
         _compilation.CreateArrayTypeSymbol(type, elementNullableAnnotation: type.NullableAnnotation).NonNullable();
 
-    public INamedTypeSymbol Get<T>() => Get(typeof(T).FullName);
+    public INamedTypeSymbol Get<T>() => Get(typeof(T));
 
     public INamedTypeSymbol Get(Type type) =>
-        Get(type.FullName ?? throw new InvalidOperationException("Could not get name of type " + type));
+        Get($"{type.Namespace}.{type.Name}" ?? throw new InvalidOperationException("Could not get name of type " + type));
 
     public INamedTypeSymbol Get(string typeFullName) =>
         TryGet(typeFullName) ?? throw new InvalidOperationException("Could not get type " + typeFullName);