PKHeX/PKHeX.Core/Legality/Encounters/EncounterStatic/EncounterStatic7.cs

using System;
using System.Collections.Generic;

namespace PKHeX.Core
{
    /// <summary>
    /// Generation 7 Static Encounter
    /// </summary>
    /// <inheritdoc cref="EncounterStatic"/>
    public sealed record EncounterStatic7(GameVersion Version) : EncounterStatic(Version), IRelearn
    {
        public override int Generation => 7;
        public IReadOnlyList<int> Relearn { get; init; } = Array.Empty<int>();

        public bool IsTotem => FormInfo.IsTotemForm(Species, Form);
        public bool IsTotemNoTransfer => Legal.Totem_NoTransfer.Contains(Species);
        public int GetTotemBaseForm() => FormInfo.GetTotemBaseForm(Species, Form);

        protected override bool IsMatchLocation(PKM pkm)
        {
            if (EggLocation == Locations.Daycare5 && Relearn.Count == 0 && pkm.RelearnMove1 != 0) // Gift Eevee edge case
                return false;
            return base.IsMatchLocation(pkm);
        }

        protected override bool IsMatchEggLocation(PKM pkm)
        {
            if (!EggEncounter)
                return base.IsMatchEggLocation(pkm);

            var eggloc = pkm.Egg_Location;
            if (!pkm.IsEgg) // hatched
                return eggloc == EggLocation || eggloc == Locations.LinkTrade6;

            // Unhatched:
            if (eggloc != EggLocation)
                return false;
            if (pkm.Met_Location is not (0 or Locations.LinkTrade6))
                return false;
            return true;
        }

        protected override bool IsMatchForm(PKM pkm, EvoCriteria evo)
        {
            if (IsTotem)
            {
                var expectForm = pkm.Format == 7 ? Form : FormInfo.GetTotemBaseForm(Species, Form);
                return expectForm == evo.Form;
            }
            return base.IsMatchForm(pkm, evo);
        }

        protected override void ApplyDetails(ITrainerInfo sav, EncounterCriteria criteria, PKM pk)
        {
            base.ApplyDetails(sav, criteria, pk);
            if (Species == (int)Core.Species.Magearna && pk is IRibbonSetEvent4 e4)
                e4.RibbonWishing = true;
            if (Form == FormVivillon && pk is PK7 pk7)
                pk.Form = Vivillon3DS.GetPattern(pk7.Country, pk7.Region);
            pk.SetRandomEC();
        }

        internal static EncounterStatic7 GetVC1(int species, byte metLevel)
        {
            bool mew = species == (int)Core.Species.Mew;
            return new EncounterStatic7(GameVersion.RBY)
            {
                Species = species,
                Gift = true, // Forces Poké Ball
                Ability = Legal.TransferSpeciesDefaultAbilityGen1(species) ? AbilityPermission.OnlyFirst : AbilityPermission.OnlyHidden, // Hidden by default, else first
                Shiny = mew ? Shiny.Never : Shiny.Random,
                Fateful = mew,
                Location = Locations.Transfer1,
                Level = metLevel,
                FlawlessIVCount = mew ? (byte)5 : (byte)3,
            };
        }

        internal static EncounterStatic7 GetVC2(int species, byte metLevel)
        {
            bool mew = species == (int)Core.Species.Mew;
            bool fateful = mew || species == (int)Core.Species.Celebi;
            return new EncounterStatic7(GameVersion.GSC)
            {
                Species = species,
                Gift = true, // Forces Poké Ball
                Ability = Legal.TransferSpeciesDefaultAbilityGen2(species) ? AbilityPermission.OnlyFirst : AbilityPermission.OnlyHidden, // Hidden by default, else first
                Shiny = mew ? Shiny.Never : Shiny.Random,
                Fateful = fateful,
                Location = Locations.Transfer2,
                Level = metLevel,
                FlawlessIVCount = fateful ? (byte)5 : (byte)3,
            };
        }
    }
}