using System.Collections.Immutable;
using System.Reflection;
using Yuanshen.Common.Data.Excel.Attributes;
using Yuanshen.Common.Data.Provider;

namespace Yuanshen.Common.Data.Excel;

public sealed class ExcelTableCollection
{
    private readonly ImmutableDictionary<ExcelType, ExcelTable> _tables;

    public ExcelTableCollection(IAssetProvider assetProvider)
    {
        ImmutableDictionary<ExcelType, ExcelTable>.Builder builder =
            ImmutableDictionary.CreateBuilder<ExcelType, ExcelTable>();

        IEnumerable<Type> types = Assembly.GetExecutingAssembly().GetTypes()
            .Where(type => type.GetCustomAttribute<ExcelAttribute>() != null);

        foreach (Type type in types)
        {
            ExcelAttribute attribute = type.GetCustomAttribute<ExcelAttribute>()!;
            builder.Add(attribute.Type, new ExcelTable(assetProvider.GetExcelTableJson(attribute.AssetName), type));
        }

        _tables = builder.ToImmutable();
    }

    public ExcelTable GetTable(ExcelType type) => _tables[type];

    public TExcel? GetExcel<TExcel>(ExcelType type, uint id) where TExcel : ExcelItem
        => _tables[type].GetById<TExcel>(id);
}