Add ML Modifiers with attribute

1 files changed, 25 insertions, 0 deletions

diff --git a/RhSolutions.MLModifiers/MLModifiersRegistration.cs b/RhSolutions.MLModifiers/MLModifiersRegistration.cs
new file mode 100644
index 0000000..57016ea
--- /dev/null
+++ b/RhSolutions.MLModifiers/MLModifiersRegistration.cs
@@ -0,0 +1,25 @@
+using Microsoft.Extensions.DependencyInjection;
+using System.Reflection;
+
+namespace RhSolutions.MLModifiers;
+
+public static class MLModifiersRegistration
+{
+	public static void AddModifiers(this IServiceCollection services)
+	{
+		var types = AppDomain.CurrentDomain.GetAssemblies()
+			.SelectMany(s => s.GetTypes())
+			.Where(p => p.IsDefined(typeof(MLModifierKey), true));
+
+		foreach (Type t in types)
+		{
+			string key = GetModifierKey(t);
+			services.AddKeyedTransient(typeof(IProductMLModifier), key, t);
+		}
+	}
+	
+	private static string GetModifierKey(Type t)	
+	{		
+		return t.GetCustomAttribute<MLModifierKey>()?.Value ?? string.Empty;
+	}
+}
\ No newline at end of file