PKHeX/PKHeX.Core/Saves/Substructures/Gen3/EReaderBerrySettings.cs

using System.Collections.Generic;
using static PKHeX.Core.EReaderBerryMatch;

namespace PKHeX.Core
{
    /// <summary>
    /// Stores details about the Gen3 e-Reader Berry values.
    /// </summary>
    public static class EReaderBerrySettings
    {
        /// <summary> e-Reader Berry is Enigma or special berry (from e-Reader data)</summary>
        public static bool IsEnigma { get; set; } = true;

        /// <summary> e-Reader Berry Name </summary>
        private static string Name { get; set; } = string.Empty;

        /// <summary> e-Reader Berry Name formatted in Title Case </summary>
        public static string DisplayName => string.Format(LegalityCheckStrings.L_XEnigmaBerry_0, Util.ToTitleCase(Name));

        private static int Language { get; set; }

        /// <summary>
        /// Checks if the most recently loaded Generation 3 Save File has a proper Enigma Berry that matches known distributions.
        /// </summary>
        public static EReaderBerryMatch GetStatus()
        {
            if (IsEnigma) // no e-Reader Berry data provided, can't hold berry.
                return NoData;

            var name = Name;
            if (EReaderBerriesNames_USA.Contains(name))
            {
                return Language switch
                {
                    <= 0 => ValidAny,
                    not 1 => ValidUSA,
                    _ => InvalidUSA,
                };
            }
            if (EReaderBerriesNames_JP.Contains(name))
            {
                return Language switch
                {
                    <= 0 => ValidAny,
                    1 => ValidJPN,
                    _ => InvalidJPN,
                };
            }
            return NoMatch;
        }

        private static readonly HashSet<string> EReaderBerriesNames_USA = new()
        {
            // USA Series 1
            "PUMKIN",
            "DRASH",
            "EGGANT",
            "STRIB",
            "CHILAN",
            "NUTPEA",
        };

        private static readonly HashSet<string> EReaderBerriesNames_JP = new()
        {
            // JP Series 1
            "カチャ", // PUMKIN
            "ブ－カ", // DRASH
            "ビスナ", // EGGANT
            "エドマ", // STRIB
            "ホズ", // CHILAN
            "ラッカ", // NUTPEA
            "クオ", // KU
            // JP Series 2
            "ギネマ", // GINEMA
            "クオ", // KUO
            "ヤゴ", // YAGO
            "トウガ", // TOUGA
            "ニニク", // NINIKU
            "トポ", // TOPO
        };

        public static void LoadFrom(SAV3 sav3)
        {
            Language = sav3.Japanese ? (int)LanguageID.Japanese : (int)LanguageID.English;
            if (sav3.IsEBerryEngima)
            {
                IsEnigma = true;
                Name = string.Empty;
            }
            else
            {
                IsEnigma = false;
                Name = sav3.EBerryName;
            }
        }
    }

    /// <summary>
    /// For use in validating the current e-Reader Berry settings.
    /// </summary>
    public enum EReaderBerryMatch : byte
    {
        /// <summary> Invalid: Doesn't match either language </summary>
        NoMatch,
        /// <summary> Invalid: No data provided from a save file </summary>
        NoData,
        /// <summary> Invalid: USA berry name on JPN save file </summary>
        InvalidUSA,
        /// <summary> Invalid: JPN berry name on USA save file</summary>
        InvalidJPN,

        /// <summary> Valid: Berry name matches either USA/JPN (Ambiguous Save File Language) </summary>
        ValidAny,
        /// <summary> Valid: Berry name matches a USA berry name </summary>
        ValidUSA,
        /// <summary> Valid: Berry name matches a JPN berry name </summary>
        ValidJPN,
    }
}