using System.ComponentModel;
using System.Reflection;
using Pf2e.Database.Model;
using Pf2e.Database.Model.Base;
using Pf2e.Database.Model.Enum;

namespace Pf2e.Database;

/// <summary>
/// Аттрибут, указывающий тип сущности.
/// </summary>
/// <param name="type">Тип сущности.</param>
[AttributeUsage(AttributeTargets.Class)]
public class EntityOfTypeAttribute(EntityType type) : Attribute
{
    /// <summary>
    /// Тип сущности.
    /// </summary>
    public EntityType Type { get; } = type;

    /// <summary>
    /// Метод для получения программного типа сущности по её типу.
    /// </summary>
    /// <param name="entity">Тип сущности.</param>
    /// <returns>Программный тип сущности.</returns>
    /// <exception cref="Exception">Если есть ошибка в структурах.</exception>
    /// <exception cref="InvalidEnumArgumentException">Тип сущности.</exception>
    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(
                    $"'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 InvalidEnumArgumentException(
            $"Can't resolve class associated with '{entity.Type}' EntityType."
        );
    }
}