initial code

2021-07-23 16:28:11 +02:00
parent af0a3e0a5d
commit 7c68455a8e
15 changed files with 527 additions and 0 deletions
@@ -0,0 +1,66 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.CodeAnalysis;
+
+namespace ProxyInterfaceSourceGenerator.Utils
+{
+    internal static class MemberHelper
+    {
+        private static string[] ExcludedMethods = new string[] { "ToString", "GetHashCode" };
+
+        public static IEnumerable<IPropertySymbol> GetPublicProperties(INamedTypeSymbol classSymbol, Func<IPropertySymbol, bool> filter)
+        {
+            return GetPublicMembers(classSymbol, p => p.Kind == SymbolKind.Property, filter);
+        }
+
+        public static IEnumerable<IMethodSymbol> GetPublicMethods(INamedTypeSymbol classSymbol, Func<IMethodSymbol, bool> filter = null)
+        {
+            if (filter == null)
+            {
+                filter = _ => true;
+            }
+
+            return GetPublicMembers(classSymbol,
+                m => m.Kind == SymbolKind.Method,
+                m => m.MethodKind == MethodKind.Ordinary,
+                m => !ExcludedMethods.Contains(m.Name),
+                filter);
+        }
+
+        private static IEnumerable<T> GetPublicMembers<T>(INamedTypeSymbol classSymbol, params Func<T, bool>[] filters) where T : ISymbol
+        {
+            var membersQuery = classSymbol.GetMembers().OfType<T>()
+                .Where(m => m.DeclaredAccessibility == Accessibility.Public);
+
+            foreach (var filter in filters)
+            {
+                membersQuery = membersQuery.Where(filter);
+            }
+
+            var members = membersQuery.ToList();
+
+            var propertyNames = membersQuery.Select(x => x.Name);
+
+            var baseType = classSymbol.BaseType;
+
+            while (baseType != null)
+            {
+                var baseMembers = baseType.GetMembers().OfType<T>()
+                    .Where(m => m.DeclaredAccessibility == Accessibility.Public)
+                    .Where(x => !propertyNames.Contains(x.Name));
+
+                foreach (var filter in filters)
+                {
+                    baseMembers = baseMembers.Where(filter);
+                }
+
+                members.AddRange(baseMembers);
+
+                baseType = baseType.BaseType;
+            }
+
+            return membersQuery;
+        }
+    }
+}