PKHeX/PKHeX.Core/Editing/Bulk/Suggestion/BatchModifications.cs

using System;
using System.Globalization;

namespace PKHeX.Core
{
    /// <summary>
    /// Modifications using <see cref="BatchInfo"/> legality.
    /// </summary>
    internal static class BatchModifications
    {
        private static bool IsAll(string p) => p.EndsWith("All", true, CultureInfo.CurrentCulture);
        private static bool IsNone(string p) => p.EndsWith("None", true, CultureInfo.CurrentCulture);

        public static ModifyResult SetSuggestedRelearnData(BatchInfo info, string propValue)
        {
            var pk = info.Entity;
            if (pk is ITechRecord8 t)
            {
                t.ClearRecordFlags();
                if (IsAll(propValue))
                    t.SetRecordFlags(); // all
                else if (!IsNone(propValue))
                    t.SetRecordFlags(pk.Moves); // whatever fit the current moves
            }

            pk.SetRelearnMoves(info.SuggestedRelearn);
            return ModifyResult.Modified;
        }

        public static ModifyResult SetSuggestedMasteryData(BatchInfo info, string propValue)
        {
            var pk = info.Entity;
            if (pk is not IMoveShop8Mastery t)
                return ModifyResult.Invalid;

            t.ClearMoveShopFlags();
            if (IsNone(propValue))
                return ModifyResult.Modified;

            var e = info.Legality.EncounterMatch;
            if (e is IMasteryInitialMoveShop8 enc)
                enc.SetInitialMastery(pk);
            if (IsAll(propValue))
                t.SetMoveShopFlagsAll(pk);
            else
                t.SetMoveShopFlags(pk);
            return ModifyResult.Modified;
        }

        public static ModifyResult SetSuggestedRibbons(BatchInfo info, string value)
        {
            var pk = info.Entity;
            if (IsNone(value))
                RibbonApplicator.RemoveAllValidRibbons(pk);
            else // All
                RibbonApplicator.SetAllValidRibbons(pk);
            return ModifyResult.Modified;
        }

        public static ModifyResult SetSuggestedMetData(BatchInfo info)
        {
            var pk = info.Entity;
            var encounter = EncounterSuggestion.GetSuggestedMetInfo(pk);
            if (encounter == null)
                return ModifyResult.Error;

            int level = encounter.LevelMin;
            int location = encounter.Location;
            int minimumLevel = EncounterSuggestion.GetLowestLevel(pk, encounter.LevelMin);

            pk.Met_Level = level;
            pk.Met_Location = location;
            pk.CurrentLevel = Math.Max(minimumLevel, level);

            return ModifyResult.Modified;
        }

        public static ModifyResult SetMinimumCurrentLevel(BatchInfo info)
        {
            var result = EncounterSuggestion.IterateMinimumCurrentLevel(info.Entity, info.Legal);
            return result ? ModifyResult.Modified : ModifyResult.Filtered;
        }

        /// <summary>
        /// Sets the provided moves in a random order.
        /// </summary>
        /// <param name="pk">Pokémon to modify.</param>
        /// <param name="moves">Moves to apply.</param>
        public static ModifyResult SetMoves(PKM pk, Span<int> moves)
        {
            pk.SetMoves(moves);
            pk.HealPP();
            return ModifyResult.Modified;
        }

        public static ModifyResult SetEVs(PKM pk)
        {
            Span<int> evs = stackalloc int[6];
            EffortValues.SetMax(evs, pk);
            pk.SetEVs(evs);
            return ModifyResult.Modified;
        }

        /// <summary>
        /// Sets the contests stats as requested.
        /// </summary>
        /// <param name="pk">Pokémon to modify.</param>
        /// <param name="la">Legality Information matched to.</param>
        /// <param name="option">Option to apply with</param>
        public static ModifyResult SetContestStats(PKM pk, LegalityAnalysis la, string option)
        {
            if (option.Length != 0 && option[BatchEditing.CONST_SUGGEST.Length..] is not "0")
                pk.SetMaxContestStats(la.EncounterMatch, la.Info.EvoChainsAllGens);
            else
                pk.SetSuggestedContestStats(la.EncounterMatch, la.Info.EvoChainsAllGens);
            return ModifyResult.Modified;
        }
    }
}