Browse Source

Scan base types during DI injection

tags/1.0.0-rc2
RogueException 8 years ago
parent
commit
33cd1a6c9f
1 changed files with 20 additions and 2 deletions
  1. +20
    -2
      src/Discord.Net.Commands/Utilities/ReflectionUtils.cs

+ 20
- 2
src/Discord.Net.Commands/Utilities/ReflectionUtils.cs View File

@@ -1,14 +1,32 @@
using System; using System;
using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Reflection; using System.Reflection;


namespace Discord.Commands namespace Discord.Commands
{ {
internal class ReflectionUtils
internal static class ReflectionUtils
{ {
private static readonly TypeInfo objectTypeInfo = typeof(object).GetTypeInfo();

internal static T CreateObject<T>(TypeInfo typeInfo, CommandService service, IDependencyMap map = null) internal static T CreateObject<T>(TypeInfo typeInfo, CommandService service, IDependencyMap map = null)
=> CreateBuilder<T>(typeInfo, service)(map); => CreateBuilder<T>(typeInfo, service)(map);


private static System.Reflection.PropertyInfo[] GetProperties(TypeInfo typeInfo)
{
var result = new List<System.Reflection.PropertyInfo>();
while (typeInfo != objectTypeInfo)
{
foreach (var prop in typeInfo.DeclaredProperties)
{
if (prop.SetMethod?.IsPublic == true && prop.GetCustomAttribute<DontInjectAttribute>() == null)
result.Add(prop);
}
typeInfo = typeInfo.BaseType.GetTypeInfo();
}
return result.ToArray();
}

internal static Func<IDependencyMap, T> CreateBuilder<T>(TypeInfo typeInfo, CommandService service) internal static Func<IDependencyMap, T> CreateBuilder<T>(TypeInfo typeInfo, CommandService service)
{ {
var constructors = typeInfo.DeclaredConstructors.Where(x => !x.IsStatic).ToArray(); var constructors = typeInfo.DeclaredConstructors.Where(x => !x.IsStatic).ToArray();
@@ -19,7 +37,7 @@ namespace Discord.Commands


var constructor = constructors[0]; var constructor = constructors[0];
System.Reflection.ParameterInfo[] parameters = constructor.GetParameters(); System.Reflection.ParameterInfo[] parameters = constructor.GetParameters();
System.Reflection.PropertyInfo[] properties = typeInfo.DeclaredProperties
System.Reflection.PropertyInfo[] properties = GetProperties(typeInfo)
.Where(p => p.SetMethod?.IsPublic == true && p.GetCustomAttribute<DontInjectAttribute>() == null) .Where(p => p.SetMethod?.IsPublic == true && p.GetCustomAttribute<DontInjectAttribute>() == null)
.ToArray(); .ToArray();




Loading…
Cancel
Save