70 lines
2.4 KiB
C#
70 lines
2.4 KiB
C#
using Microsoft.CodeAnalysis;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
namespace ProxyInterfaceSourceGenerator.Utils
|
|
{
|
|
internal static class MemberHelper
|
|
{
|
|
private static string[] _excludedMethods = new string[] { "ToString", "GetHashCode" };
|
|
|
|
public static IEnumerable<IPropertySymbol> GetPublicProperties(INamedTypeSymbol classSymbol, params Func<IPropertySymbol, bool>[] filters)
|
|
{
|
|
var allFilters = new List<Func<IPropertySymbol, bool>>(filters);
|
|
allFilters.Add(p => p.Kind == SymbolKind.Property);
|
|
|
|
return GetPublicMembers(classSymbol, allFilters.ToArray());
|
|
}
|
|
|
|
public static IEnumerable<IMethodSymbol> GetPublicMethods(INamedTypeSymbol classSymbol, Func<IMethodSymbol, bool>? filter = null)
|
|
{
|
|
if (filter is null)
|
|
{
|
|
filter = _ => true;
|
|
}
|
|
|
|
return GetPublicMembers(classSymbol,
|
|
m => m.Kind == SymbolKind.Method,
|
|
m => m.MethodKind == MethodKind.Ordinary,
|
|
m => !_excludedMethods.Contains(m.Name),
|
|
filter);
|
|
}
|
|
|
|
// TODO : do we need also to check for "SanitizedName()" here?
|
|
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;
|
|
}
|
|
}
|
|
} |