Fix #10. Change receivers logic.

Author: ChebanovDD

src/UnityUxmlGenerator.UnityPackage/Assets/Plugins/UnityUxmlGenerator/Runtime/UnityUxmlGenerator.dll

@@ -4,9 +4,9 @@ namespace UnityUxmlGenerator.Captures;
 
 internal sealed class UxmlFactoryCapture : BaseCapture
 {
-    public UxmlFactoryCapture(ClassDeclarationSyntax @class, AttributeSyntax attribute) : base(@class)
+    public UxmlFactoryCapture((ClassDeclarationSyntax Class, AttributeSyntax Attribute) data) : base(data.Class)
     {
-        Attribute = attribute;
+        Attribute = data.Attribute;
     }
 
     public override string ClassTag => "UxmlFactory";

src/UnityUxmlGenerator/Captures/UxmlFactoryCapture.cs

@@ -5,26 +5,38 @@ namespace UnityUxmlGenerator.Extensions;
 
 internal static class SyntaxNodeExtensions
 {
-    public static bool IsAttributeWithName(this SyntaxNode syntaxNode, string attributeName,
-        out AttributeSyntax? attribute)
+    public static bool IsMemberHasAttribute<TMember>(this SyntaxNode syntaxNode, string attributeName,
+        out (TMember Member, AttributeSyntax Attribute) result) where TMember : MemberDeclarationSyntax
     {
-        if (syntaxNode is not AttributeSyntax attributeSyntax)
+        if (syntaxNode is not TMember memberSyntax)
         {
-            attribute = default;
+            result = default;
             return false;
         }
 
-        attribute = attributeSyntax;
+        result.Member = memberSyntax;
 
-        switch (attributeSyntax.Name)
+        for (var i = 0; i < memberSyntax.AttributeLists.Count; i++)
         {
-            case IdentifierNameSyntax identifierNameSyntax
-                when identifierNameSyntax.Identifier.Text.Contains(attributeName):
-            case QualifiedNameSyntax qualifiedNameSyntax
-                when qualifiedNameSyntax.Right.Identifier.Text.Contains(attributeName):
-                return true;
+            var attributeList = memberSyntax.AttributeLists[i];
+            for (var j = 0; j < attributeList.Attributes.Count; j++)
+            {
+                var attributeSyntax = attributeList.Attributes[j];
+                switch (attributeSyntax.Name)
+                {
+                    case IdentifierNameSyntax identifierNameSyntax
+                        when identifierNameSyntax.Identifier.Text.Contains(attributeName):
+                    case QualifiedNameSyntax qualifiedNameSyntax
+                        when qualifiedNameSyntax.Right.Identifier.Text.Contains(attributeName):
+                    {
+                        result.Attribute = attributeSyntax;
+                        return true;
+                    }
+                }
+            }
         }
 
+        result = default;
         return false;
     }
 

src/UnityUxmlGenerator/Extensions/SyntaxNodeExtensions.cs

@@ -15,20 +15,19 @@ internal sealed class UxmlFactoryReceiver : BaseReceiver
 
     public override void OnVisitSyntaxNode(SyntaxNode syntaxNode)
     {
-        if (syntaxNode.IsAttributeWithName(AttributeName, out var attribute) == false)
+        if (syntaxNode.IsMemberHasAttribute<ClassDeclarationSyntax>(AttributeName,
+                out (ClassDeclarationSyntax Class, AttributeSyntax Attribute) result) == false)
         {
             return;
         }
 
-        var @class = attribute!.GetParent<ClassDeclarationSyntax>();
-
-        if (@class.InheritsFromAnyType())
+        if (result.Class.InheritsFromAnyType())
         {
-            _captures.Add(new UxmlFactoryCapture(@class!, attribute!));
+            _captures.Add(new UxmlFactoryCapture(result));
         }
-        else if (@class is not null)
+        else if (result.Class is not null)
         {
-            RegisterDiagnostic(ClassHasNoBaseClassError, @class.GetLocation(), @class.Identifier.Text);
+            RegisterDiagnostic(ClassHasNoBaseClassError, result.Class.GetLocation(), result.Class.Identifier.Text);
         }
     }
 }

src/UnityUxmlGenerator/SyntaxReceivers/UxmlFactoryReceiver.cs

@@ -16,18 +16,15 @@ internal sealed class UxmlTraitsReceiver : BaseReceiver
 
     public override void OnVisitSyntaxNode(SyntaxNode syntaxNode)
     {
-        if (syntaxNode.IsAttributeWithName(AttributeName, out var attribute) == false)
+        if (syntaxNode.IsMemberHasAttribute<PropertyDeclarationSyntax>(AttributeName,
+                out (PropertyDeclarationSyntax Property, AttributeSyntax Attribute) result) == false)
         {
             return;
         }
 
-        var property = attribute!.GetParent<PropertyDeclarationSyntax>();
-        if (property is null)
-        {
-            return;
-        }
+        var (property, attribute) = result;
 
-        if (attribute!.ArgumentList is not null && attribute.ArgumentList.Arguments.Any())
+        if (attribute.ArgumentList is not null && attribute.ArgumentList.Arguments.Any())
         {
             if (HasSameType(property, attribute.ArgumentList.Arguments.First()) == false)
             {
@@ -58,7 +55,7 @@ public override void OnVisitSyntaxNode(SyntaxNode syntaxNode)
             _captures.Add(uxmlTraits.ClassName, uxmlTraits);
         }
 
-        uxmlTraits.Properties.Add((property, attribute));
+        uxmlTraits.Properties.Add(result);
     }
 
     private static bool HasSameType(BasePropertyDeclarationSyntax property, AttributeArgumentSyntax attributeArgument)