using System.Linq;
using pk3DS.Core.Structures.PersonalInfo;

namespace pk3DS.Core.Randomizers;

public class FormRandomizer(GameConfig game)
{
    public bool AllowMega = false;
    public bool AllowAlolanForm = true;

    public int GetRandomForme(int species, PersonalInfo[] stats = null)
    {
        stats ??= game.Personal.Table;
        if (stats[species].FormeCount <= 1)
            return 0;

        switch (species)
        {
            case 658 when !AllowMega:
                return 0;
            case 664:
            case 665:
            case 666: // Vivillon evo chain
                return 30; // save file specific
            case 774: // Minior
                return (int)(Util.Random32() % 7);
        }

        if (AllowAlolanForm && Legal.EvolveToAlolanForms.Contains(species))
            return (int)(Util.Random32() % 2);
        if (!Legal.BattleExclusiveForms.Contains(species) || AllowMega)
            return (int)(Util.Random32() % stats[species].FormeCount); // Slot-Random
        return 0;
    }
}