Программа просмотра базы данных

Pf2eModel/EntityOfTypeAttribute.cs

@@ -0,0 +1,37 @@
+using Pf2eModel.Model.Base;
+using Pf2eModel.Model;
+using Pf2eModel.Model.Enum;
+using System.Reflection;
+
+namespace Pf2eModel;
+
+[AttributeUsage(AttributeTargets.Class)]
+public class EntityOfTypeAttribute(EntityType type) : Attribute
+{
+    public EntityType Type { get; } = type;
+
+    public static Type Resolve(Entity entity)
+    {
+        foreach (Type type in Assembly.GetExecutingAssembly().GetTypes())
+        {
+            if (type.GetCustomAttributes(typeof(EntityOfTypeAttribute), true).Length <= 0)
+            {
+                continue;
+            }
+
+            if (!type.IsAssignableTo(typeof(IEntity)))
+            {
+                throw new Exception($"'EntityOfTypeAttribute' attribute should only be applied to children of 'IEntity' class. '{type.FullName}' is not.");
+            }
+
+            EntityOfTypeAttribute attribute = type.GetCustomAttribute<EntityOfTypeAttribute>()!;
+
+            if (attribute.Type == entity.Type)
+            {
+                return type;
+            }
+        }
+
+        throw new Exception($"Can't resolve class associated with '{entity.Type}' EntityType.");
+    }
+}