nadekobot/src/NadekoBot/Modules/Games/Trivia/TriviaQuestionPool.cs

#nullable disable
namespace NadekoBot.Modules.Games.Common.Trivia;

public class TriviaQuestionPool
{
    private TriviaQuestion[] Pool
        => _cache.LocalData.TriviaQuestions;

    private IReadOnlyDictionary<int, string> Map
        => _cache.LocalData.PokemonMap;

    private readonly IDataCache _cache;
    private readonly int maxPokemonId;

    private readonly NadekoRandom _rng = new();

    public TriviaQuestionPool(IDataCache cache)
    {
        _cache = cache;
        maxPokemonId = 721; //xd
    }

    public TriviaQuestion GetRandomQuestion(HashSet<TriviaQuestion> exclude, bool isPokemon)
    {
        if (Pool.Length == 0)
            return null;

        if (isPokemon)
        {
            var num = _rng.Next(1, maxPokemonId + 1);
            return new("Who's That Pokémon?",
                Map[num].ToTitleCase(),
                "Pokemon",
                $@"https://nadeko.bot/images/pokemon/shadows/{num}.png",
                $@"https://nadeko.bot/images/pokemon/real/{num}.png");
        }

        TriviaQuestion randomQuestion;
        while (exclude.Contains(randomQuestion = Pool[_rng.Next(0, Pool.Length)])) ;

        return randomQuestion;
    }
}