PKHeX/PKHeX.Core/Legality/Moves/Breeding/EggSource.cs

using System;
using static PKHeX.Core.LegalityCheckStrings;

namespace PKHeX.Core
{
    /// <summary>
    /// Egg Moveset Building Order for Generation 2
    /// </summary>
    public enum EggSource2 : byte
    {
        None,
        /// <summary> Initial moveset for the egg's level. </summary>
        Base,
        /// <summary> Egg move inherited from the Father. </summary>
        FatherEgg,
        /// <summary> Technical Machine move inherited from the Father. </summary>
        FatherTM,
        /// <summary> Level Up move inherited from a parent. </summary>
        ParentLevelUp,
        /// <summary> Tutor move (Elemental Beam) inherited from a parent. </summary>
        Tutor,

        Max,
    }

    /// <summary>
    /// Egg Moveset Building Order for Generation 3 &amp; 4
    /// </summary>
    public enum EggSource34 : byte
    {
        None,
        /// <summary> Initial moveset for the egg's level. </summary>
        Base,
        /// <summary> Egg move inherited from the Father. </summary>
        FatherEgg,
        /// <summary> Technical Machine move inherited from the Father. </summary>
        FatherTM,
        /// <summary> Level Up move inherited from a parent. </summary>
        ParentLevelUp,

        Max,

        /// <summary> Special Move applied at the end if certain conditions are satisfied. </summary>
        VoltTackle,
    }

    /// <summary>
    /// Egg Moveset Building Order for Generation 5
    /// </summary>
    public enum EggSource5 : byte
    {
        None,
        /// <summary> Initial moveset for the egg's level. </summary>
        Base,
        /// <summary> Egg move inherited from the Father. </summary>
        FatherEgg,
        /// <summary> Technical Machine move inherited from the Father. </summary>
        ParentLevelUp,
        /// <summary> Technical Machine move inherited from the Father. </summary>
        /// <remarks> After level up, unlike Gen3/4! </remarks>
        FatherTM,

        Max,

        /// <summary> Special Move applied at the end if certain conditions are satisfied. </summary>
        VoltTackle,
    }

    /// <summary>
    /// Egg Moveset Building Order for Generation 6+
    /// </summary>
    public enum EggSource6 : byte
    {
        None,
        /// <summary> Initial moveset for the egg's level. </summary>
        Base,
        /// <summary> Level Up move inherited from a parent. </summary>
        ParentLevelUp,
        /// <summary> Egg move inherited from a parent. </summary>
        ParentEgg,

        Max,

        /// <summary> Special Move applied at the end if certain conditions are satisfied. </summary>
        VoltTackle,
    }

    /// <summary>
    /// Utility logic for converting a <see cref="MoveBreed"/> move result into a user friendly string.
    /// </summary>
    public static class EggSourceUtil
    {
        /// <summary>
        /// Unboxes the parse result and returns a user friendly string for the move result.
        /// </summary>
        public static string GetSource(object parse, int generation, int index)
        {
            static string GetLine<T>(T[] arr, Func<T, string> act, int i)
            {
                if (i >= arr.Length)
                    return LMoveSourceEmpty;
                return act(arr[i]);
            }

            return generation switch
            {
                2      => GetLine((EggSource2[]) parse, GetSource, index),
                3 or 4 => GetLine((EggSource34[])parse, GetSource, index),
                5      => GetLine((EggSource5[]) parse, GetSource, index),
                >= 6   => GetLine((EggSource6[]) parse, GetSource, index),
                _      => LMoveSourceEmpty,
            };
        }

        public static string GetSource(this EggSource2 source) => source switch
        {
            EggSource2.Base => LMoveRelearnEgg,
            EggSource2.FatherEgg => LMoveEggLevelUp,
            EggSource2.FatherTM => LMoveEggTMHM,
            EggSource2.ParentLevelUp => LMoveEggInherited,
            EggSource2.Tutor => LMoveEggInheritedTutor,
            EggSource2.Max => "Any",
            _ => LMoveEggInvalid,
        };

        public static string GetSource(this EggSource34 source) => source switch
        {
            EggSource34.Base => LMoveRelearnEgg,
            EggSource34.FatherEgg => LMoveEggLevelUp,
            EggSource34.FatherTM => LMoveEggTMHM,
            EggSource34.ParentLevelUp => LMoveEggInherited,
            EggSource34.VoltTackle => LMoveSourceSpecial,
            EggSource34.Max => "Any",
            _ => LMoveEggInvalid,
        };

        public static string GetSource(this EggSource5 source) => source switch
        {
            EggSource5.Base => LMoveRelearnEgg,
            EggSource5.FatherEgg => LMoveEggLevelUp,
            EggSource5.FatherTM => LMoveEggTMHM,
            EggSource5.ParentLevelUp => LMoveEggInherited,
            EggSource5.VoltTackle => LMoveSourceSpecial,
            EggSource5.Max => "Any",
            _ => LMoveEggInvalid,
        };

        public static string GetSource(this EggSource6 source) => source switch
        {
            EggSource6.Base => LMoveRelearnEgg,
            EggSource6.ParentEgg => LMoveEggLevelUp,
            EggSource6.ParentLevelUp => LMoveEggInherited,
            EggSource6.VoltTackle => LMoveSourceSpecial,
            EggSource6.Max => "Any",
            _ => LMoveEggInvalid,
        };
    }
}