using static PKHeX.Core.Ball;

namespace PKHeX.Core;

internal static class BallUseLegality
{
    internal static ulong GetWildBalls(int generation, GameVersion game) => generation switch
    {
        1 => WildPokeBalls1,
        2 => WildPokeBalls2,
        3 => WildPokeBalls3,
        4 => GameVersion.HGSS.Contains(game) ? WildPokeBalls4_HGSS : WildPokeBalls4_DPPt,
        5 => WildPokeBalls5,
        6 => WildPokeballs6,
        7 => GameVersion.Gen7b.Contains(game) ? WildPokeballs7b : WildPokeballs7,
        8 when GameVersion.BDSP.Contains(game) => WildPokeBalls4_HGSS,
        8 when GameVersion.PLA == game => WildPokeBalls8a,
        8 => GameVersion.GO == game ? WildPokeballs8g : WildPokeballs8,
        9 => WildPokeballs9,
        _ => default,
    };

    private const ulong WildPokeRegular = (1 << (int)Master)
                                        | (1 << (int)Ultra)
                                        | (1 << (int)Great)
                                        | (1 << (int)Poke);

    private const ulong WildPokeKurt4 = (1 << (int)Fast)
                                      | (1 << (int)Level)
                                      | (1 << (int)Lure)
                                      | (1 << (int)Heavy)
                                      | (1 << (int)Love)
                                      | (1 << (int)Friend)
                                      | (1 << (int)Moon);

    private const ulong WildPokeEnhance3 = (1 << (int)Net)
                                         | (1 << (int)Dive)
                                         | (1 << (int)Nest)
                                         | (1 << (int)Repeat)
                                         | (1 << (int)Timer)
                                         | (1 << (int)Luxury)
                                         | (1 << (int)Premier);

    private const ulong WildPokeEnhance4 = (1 << (int)Dusk) | (1 << (int)Heal) | (1 << (int)Quick);
    private const ulong WildPokeEnhance5 = (1 << (int)Dream);
    private const ulong WildPokeEnhance7 = (1 << (int)Beast);
    private const ulong WildPokeEnhance8 = (1 << (int)Dream) | (1 << (int)Safari) | (1 << (int)Sport);

    private const ulong WildPokeBalls8a =
      (1ul << (int)LAPoke)
    | (1ul << (int)LAGreat)
    | (1ul << (int)LAUltra)
    | (1ul << (int)LAFeather)
    | (1ul << (int)LAWing)
    | (1ul << (int)LAJet)
    | (1ul << (int)LAHeavy)
    | (1ul << (int)LALeaden)
    | (1ul << (int)LAGigaton);

    private const ulong WildPokeBalls1 = 1 << (int)Poke;
    private const ulong WildPokeBalls2 = WildPokeBalls1;
    private const ulong WildPokeBalls3 = WildPokeRegular | WildPokeEnhance3;
    private const ulong WildPokeBalls4_DPPt = WildPokeBalls3 | WildPokeEnhance4;
    private const ulong WildPokeBalls4_HGSS = WildPokeBalls4_DPPt | WildPokeKurt4;
    private const ulong WildPokeBalls5 = WildPokeBalls4_DPPt;

    internal const ulong DreamWorldBalls = WildPokeBalls5 | WildPokeEnhance5;
    internal const ulong WildPokeballs6 = WildPokeBalls5; // Same as Gen5
    internal const ulong WildPokeballs7 = WildPokeBalls4_HGSS | WildPokeEnhance7; // Same as HGSS + Beast
    internal const ulong WildPokeballs8 = WildPokeballs7 | WildPokeEnhance8;

    private const ulong WildPokeballs7b = WildPokeRegular | (1 << (int)Premier);
    private const ulong WildPokeballs8g = WildPokeballs7b & ~(1ul << (int)Master); // Ultra Great Poke Premier, no Master

    internal const ulong WildPokeballs9 = WildPokeballs7 | WildPokeEnhance5; // Same as Gen7 + Dream
}