ProxyAll

2021-07-23 21:30:01 +02:00
parent 42073eed3d
commit d129f52441
11 changed files with 108 additions and 96 deletions
@@ -0,0 +1,46 @@
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+
+namespace ProxyInterfaceSourceGenerator.SyntaxReceiver
+{
+    internal class ProxySyntaxReceiver : ISyntaxReceiver
+    {
+        public IDictionary<InterfaceDeclarationSyntax, ProxyData> CandidateInterfaces { get; } = new Dictionary<InterfaceDeclarationSyntax, ProxyData>();
+
+        public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
+        {
+            if (syntaxNode is InterfaceDeclarationSyntax interfaceDeclarationSyntax && TryGet(interfaceDeclarationSyntax, out var data))
+            {
+                CandidateInterfaces.Add(interfaceDeclarationSyntax, data);
+            }
+        }
+
+        private static bool TryGet(InterfaceDeclarationSyntax interfaceDeclarationSyntax, out ProxyData data)
+        {
+            data = new();
+
+            // TODO : how to check if the InterfaceDeclarationSyntax has 'partial' ?
+            var attributeLists = interfaceDeclarationSyntax.AttributeLists.FirstOrDefault(x => x.Attributes.Any(a => a.Name.ToString().Equals("ProxyInterfaceGenerator.Proxy")));
+            if (attributeLists is null)
+            {
+                return false;
+            }
+
+            var args = attributeLists.Attributes.FirstOrDefault()?.ArgumentList;
+            if (args is null)
+            {
+                return false;
+            }
+
+            data = new()
+            {
+                TypeName = args.Arguments[0].Expression.ChildNodes().First().GetText().ToString(),
+                ProxyAll = bool.Parse(args.Arguments[1].Expression.ChildNodes().First().GetText().ToString())
+            };
+
+            return true;
+        }
+    }
+}