using System.Collections.Immutable;
using System.Linq;
using Dalamud.Plugin.Services;
using Lumina.Excel.Sheets;

namespace Questionable.Data;

internal sealed class AetherCurrentData
{
    private readonly ImmutableDictionary<ushort, ImmutableList<uint>> _overworldCurrents;

    public AetherCurrentData(IDataManager dataManager)
    {
        _overworldCurrents = dataManager.GetExcelSheet<AetherCurrentCompFlgSet>()
            .Where(x => x.RowId > 0)
            .Where(x => x.Territory.IsValid)
            .ToImmutableDictionary(
                x => (ushort)x.Territory.RowId,
                x => x.AetherCurrents
                    .Where(y => y.RowId > 0 && y.Value.Quest.RowId == 0)
                    .Select(y => y.RowId)
                    .ToImmutableList());
    }

    public bool IsValidAetherCurrent(ushort territoryId, uint aetherCurrentId)
    {
        return _overworldCurrents.TryGetValue(territoryId, out ImmutableList<uint>? currentIds) &&
               currentIds.Contains(aetherCurrentId);
    }
}