-
Notifications
You must be signed in to change notification settings - Fork 0
/
DependecyResolver.cs
116 lines (101 loc) · 3.6 KB
/
DependecyResolver.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
using System;
using System.Collections.Generic;
using System.Reflection;
using Serilog;
using System.Linq;
namespace TaskBot
{
[AttributeUsage(AttributeTargets.Property)]
public sealed class DependencyAttribute : Attribute { }
[AttributeUsage(AttributeTargets.Property)]
public sealed class StatocDependencyAttribute : Attribute { }
public class DependecyResolver
{
private readonly ILogger log;
private readonly Dictionary<Type, Func<object>> dependencies;
private readonly Dictionary<Type, object> staticDependecies;
public DependecyResolver(ILogger log)
{
this.log = log;
dependencies = new Dictionary<Type, Func<object>>();
staticDependecies = new Dictionary<Type, object>();
}
public void ResolveDependecy(object dependentObject, PropertyInfo prop)
{
var dependecyType = prop.PropertyType;
try
{
var dependecyValue = dependencies[dependecyType]();
prop.SetValue(dependentObject, dependecyValue);
}
catch(KeyNotFoundException)
{
log.Error(
"Can't find dependecy initialized for propery of type {dependecyType} in {objectType}.",
dependecyType,
dependentObject.GetType().Name);
}
}
public void ResolveStaticDependecy(object dependentObject, PropertyInfo prop)
{
var dependecyType = prop.PropertyType;
try
{
var dependecyValue = staticDependecies[dependecyType];
prop.SetValue(dependentObject, dependecyValue);
}
catch(KeyNotFoundException)
{
log.Fatal(
"Can't find static dependecy value for propery of type {dependecyType} in {objectType}.",
dependecyType,
dependentObject.GetType().Name);
}
}
public void Resolve(object dependentObject)
{
log.Debug("Resolving dependecies for {type}", dependentObject.GetType().Name);
var objectType = dependentObject.GetType();
var props = objectType.GetProperties(BindingFlags.Instance | BindingFlags.Public);
foreach (var prop in props)
{
if (prop.CanWrite && prop.GetCustomAttribute<DependencyAttribute>() != null)
{
ResolveDependecy(dependentObject, prop);
}
if (prop.CanWrite && prop.GetCustomAttribute<StatocDependencyAttribute>() != null)
{
ResolveStaticDependecy(dependentObject, prop);
}
}
}
public DependecyResolver Add<T>(Func<T> initFunc) where T : class
{
dependencies.Add(typeof(T), initFunc);
return this;
}
public DependecyResolver Add<T>() where T : class, new()
{
dependencies.Add(typeof(T), () => new T());
return this;
}
public DependecyResolver AddStatic<T>(T value)
{
staticDependecies.Add(typeof(T), value);
return this;
}
}
internal static class DI
{
private static DependecyResolver resolver;
public static void SetResolver(DependecyResolver resolver)
{
DI.resolver = resolver;
}
public static T Resolve<T>(T o)
{
resolver.Resolve(o);
return o;
}
}
}