using System;
using static PKHeX.Core.GameVersion;

namespace PKHeX.Core;

public static class GameData
{
    public static Learnset[] GetLearnsets(GameVersion game) => Learnsets(game);
    public static IPersonalTable GetPersonal(GameVersion game) => Personal(game);

    public static Learnset GetLearnset(GameVersion game, int species, int form)
    {
        var pt = Personal(game);
        var index = pt.GetFormIndex(species, form);
        var sets = Learnsets(game);
        return sets[index];
    }

    private static Learnset[] Learnsets(GameVersion game) => game switch
    {
        RD or GN or BU or RB => Legal.LevelUpRB,
        YW or RBY => Legal.LevelUpY,
        GD or SI or GS => Legal.LevelUpGS,
        C or GSC => Legal.LevelUpC,

        R or S or RS or RSE => Legal.LevelUpRS,
        E or COLO or XD or FRLG or CXD => Legal.LevelUpE,
        FR => Legal.LevelUpFR,
        LG => Legal.LevelUpLG,

        D or P or DP => Legal.LevelUpDP,
        Pt or DPPt => Legal.LevelUpPt,
        HG or SS or HGSS => Legal.LevelUpHGSS,

        B or W or BW => Legal.LevelUpBW,
        B2 or W2 or B2W2 => Legal.LevelUpB2W2,

        X or Y or XY => Legal.LevelUpXY,
        AS or OR or ORAS => Legal.LevelUpAO,

        SN or MN or SM => Legal.LevelUpSM,
        US or UM or USUM => Legal.LevelUpUSUM,
        GO or GP or GE or GG => Legal.LevelUpGG,

        SW or SH or SWSH => Legal.LevelUpSWSH,
        BD or SP or BDSP => Legal.LevelUpBDSP,
        PLA => Legal.LevelUpLA,

        Gen1 => Legal.LevelUpY,
        Gen2 => Legal.LevelUpC,
        Gen3 => Legal.LevelUpE,
        Gen4 => Legal.LevelUpHGSS,
        Gen5 => Legal.LevelUpB2W2,
        Gen6 => Legal.LevelUpAO,
        Gen7 => Legal.LevelUpSM,
        Gen7b => Legal.LevelUpGG,
        Gen8 => Legal.LevelUpSWSH,

        Stadium => Legal.LevelUpY,
        Stadium2 => Legal.LevelUpGS,

        _ => throw new ArgumentOutOfRangeException(nameof(game), $"{game} is not a valid entry in the expression."),
    };

    private static IPersonalTable Personal(GameVersion game) => game switch
    {
        RD or GN or BU or RB => PersonalTable.RB,
        YW or RBY => PersonalTable.Y,
        GD or SI or GS => PersonalTable.GS,
        C or GSC => PersonalTable.C,

        R or S or RS or RSE => PersonalTable.RS,
        E or COLO or XD or FRLG or CXD => PersonalTable.E,
        FR => PersonalTable.FR,
        LG => PersonalTable.LG,

        D or P or DP => PersonalTable.DP,
        Pt or DPPt => PersonalTable.Pt,
        HG or SS or HGSS => PersonalTable.HGSS,

        B or W or BW => PersonalTable.BW,
        B2 or W2 or B2W2 => PersonalTable.B2W2,

        X or Y or XY => PersonalTable.XY,
        AS or OR or ORAS => PersonalTable.AO,

        SN or MN or SM => PersonalTable.SM,
        US or UM or USUM => PersonalTable.USUM,
        GO or GP or GE or GG => PersonalTable.GG,

        SW or SH or SWSH => PersonalTable.SWSH,
        BD or SP or BDSP => PersonalTable.BDSP,
        PLA => PersonalTable.LA,

        Gen1 => PersonalTable.Y,
        Gen2 => PersonalTable.C,
        Gen3 => PersonalTable.E,
        Gen4 => PersonalTable.HGSS,
        Gen5 => PersonalTable.B2W2,
        Gen6 => PersonalTable.AO,
        Gen7 => PersonalTable.USUM,
        Gen7b => PersonalTable.GG,
        Gen8 => PersonalTable.SWSH,

        Stadium => PersonalTable.Y,
        Stadium2 => PersonalTable.GS,

        _ => throw new ArgumentOutOfRangeException(nameof(game), $"{game} is not a valid entry in the expression."),
    };
}