PKHeX/PKHeX.Core/Legality/Verifiers/TransferVerifier.cs

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

namespace PKHeX.Core;

/// <summary>
/// Verifies the transfer data for a <see cref="PKM"/> that has been irreversibly transferred forward.
/// </summary>
public sealed class TransferVerifier : Verifier
{
    protected override CheckIdentifier Identifier => CheckIdentifier.Encounter;

    public override void Verify(LegalityAnalysis data)
    {
        throw new Exception("Don't call via this.");
    }

    public void VerifyTransferLegalityG12(LegalityAnalysis data)
    {
        VerifyVCOTGender(data);
        VerifyVCNatureEXP(data);
        VerifyVCShinyXorIfShiny(data);
        VerifyVCGeolocation(data);
    }

    private void VerifyVCOTGender(LegalityAnalysis data)
    {
        var pk = data.Entity;
        if (pk.OT_Gender == 1 && pk.Version != (int)GameVersion.C)
            data.AddLine(GetInvalid(LG2OTGender));
    }

    private void VerifyVCNatureEXP(LegalityAnalysis data)
    {
        var pk = data.Entity;
        var met = pk.Met_Level;

        if (met == 100) // check for precise match, can't receive EXP after transfer.
        {
            var nature = Experience.GetNatureVC(pk.EXP);
            if (nature != pk.Nature)
                data.AddLine(GetInvalid(LTransferNature));
            return;
        }
        if (met <= 2) // Not enough EXP to have every nature -- check for exclusions!
        {
            var pi = pk.PersonalInfo;
            var growth = pi.EXPGrowth;
            var nature = pk.Nature;
            bool valid = VerifyVCNature(growth, nature);
            if (!valid)
                data.AddLine(GetInvalid(LTransferNature));
        }
    }

    private static bool VerifyVCNature(int growth, int nature) => growth switch
    {
        // exp % 25 with a limited amount of EXP does not allow for every nature
        0 => (0x01FFFF03 & (1 << nature)) != 0, // MediumFast -- Can't be Brave, Adamant, Naughty, Bold, Docile, or Relaxed
        4 => (0x001FFFC0 & (1 << nature)) != 0, // Fast -- Can't be Gentle, Sassy, Careful, Quirky, Hardy, Lonely, Brave, Adamant, Naughty, or Bold
        5 => (0x01FFFCFF & (1 << nature)) != 0, // Slow -- Can't be Impish or Lax
        _ => true,
    };

    private static void VerifyVCShinyXorIfShiny(LegalityAnalysis data)
    {
        // Star, not square. Requires transferring a shiny and having the initially random PID to already be a Star shiny.
        // (15:65536, ~1:4096) odds on a given shiny transfer!
        var xor = data.Entity.ShinyXor;
        if (xor is <= 15 and not 0)
            data.AddLine(Get(LEncStaticPIDShiny, ParseSettings.Gen7TransferStarPID, CheckIdentifier.PID));
    }

    private static void VerifyVCGeolocation(LegalityAnalysis data)
    {
        if (data.Entity is not PK7 pk7)
            return;

        // VC Games were region locked to the Console, meaning not all language games are available.
        var within = Locale3DS.IsRegionLockedLanguageValidVC(pk7.ConsoleRegion, pk7.Language);
        if (!within)
            data.AddLine(GetInvalid(string.Format(LOTLanguage, $"!={(LanguageID)pk7.Language}", ((LanguageID)pk7.Language).ToString()), CheckIdentifier.Language));
    }

    public void VerifyTransferLegalityG3(LegalityAnalysis data)
    {
        var pk = data.Entity;
        if (pk.Format == 4) // Pal Park (3->4)
        {
            if (pk.Met_Location != Locations.Transfer3)
                data.AddLine(GetInvalid(LEggLocationPalPark));
        }
        else // Transporter (4->5)
        {
            if (pk.Met_Location != Locations.Transfer4)
                data.AddLine(GetInvalid(LTransferEggLocationTransporter));
        }
    }

    public void VerifyTransferLegalityG4(LegalityAnalysis data)
    {
        var pk = data.Entity;
        int loc = pk.Met_Location;
        if (loc == Locations.Transfer4)
            return;

        // Crown met location must be present if transferred via lock capsule
        switch (pk.Species)
        {
            case (int)Species.Celebi:
                if (loc is not (Locations.Transfer4_CelebiUnused or Locations.Transfer4_CelebiUsed))
                    data.AddLine(GetInvalid(LTransferMet));
                break;
            case (int)Species.Raikou or (int)Species.Entei or (int)Species.Suicune:
                if (loc is not (Locations.Transfer4_CrownUnused or Locations.Transfer4_CrownUsed))
                    data.AddLine(GetInvalid(LTransferMet));
                break;
            default:
                data.AddLine(GetInvalid(LTransferEggLocationTransporter));
                break;
        }
    }

    public void VerifyTransferLegalityG8(LegalityAnalysis data)
    {
        var pk = data.Entity;
        var enc = data.EncounterMatch;
        bool native = enc.Generation == 8 && pk.IsNative;
        if (!native || IsHOMETrackerRequired(enc))
            VerifyHOMETracker(data, pk);

        if (enc.Generation < 8)
        {
            VerifyHOMETransfer(data, pk);
            // Check for impossible 7->8 transfers
            if (enc is EncounterStatic7 { IsTotem: true } s)
            {
                if (s.IsTotemNoTransfer)
                    data.AddLine(GetInvalid(LTransferBad));
                else if (pk.Form != s.GetTotemBaseForm())
                    data.AddLine(GetInvalid(LTransferBad));
            }
        }

        // Starting in Generation 8, games have a selective amount of species/forms from prior games.
        var pt = pk switch
        {
            PA8 => PersonalTable.LA,
            PB8 => PersonalTable.BDSP,
            _ => PersonalTable.SWSH,
        };
        if (!pt.IsPresentInGame(pk.Species, pk.Form))
            data.AddLine(GetInvalid(LTransferBad));
    }

    // Encounters that originate in HOME -> transfer to save data
    private static bool IsHOMETrackerRequired(IEncounterTemplate enc) => enc switch
    {
        EncounterSlot8GO => true,
        WC8 { IsHOMEGift: true } => true,
        WB8 { IsHOMEGift: true } => true,
        WA8 { IsHOMEGift: true } => true,
        _ => enc.Generation < 8,
    };

    private void VerifyHOMETransfer(LegalityAnalysis data, PKM pk)
    {
        if (pk is not IScaledSize s)
            return;

        if (pk.LGPE || pk.GO)
            return; // can have any size value
        if (s.HeightScalar != 0)
            data.AddLine(GetInvalid(LTransferBad));
        if (s.WeightScalar != 0)
            data.AddLine(GetInvalid(LTransferBad));
    }

    private void VerifyHOMETracker(LegalityAnalysis data, PKM pk)
    {
        // Tracker value is set via Transfer across HOME.
        // Can't validate the actual values (we aren't the server), so we can only check against zero.
        if (pk is IHomeTrack {Tracker: 0})
        {
            data.AddLine(Get(LTransferTrackerMissing, ParseSettings.Gen8TransferTrackerNotPresent));
            // To the reader: It seems like the best course of action for setting a tracker is:
            // - Transfer a 0-Tracker pk to HOME to get assigned a valid Tracker
            // - Don't make one up.
        }
    }

    public IEnumerable<CheckResult> VerifyVCEncounter(PKM pk, IEncounterTemplate encounter, ILocation transfer, IReadOnlyList<CheckMoveResult> Moves)
    {
        if (pk.Met_Location != transfer.Location)
            yield return GetInvalid(LTransferMetLocation);

        var expecteEgg = pk is PB8 ? Locations.Default8bNone : transfer.EggLocation;
        if (pk.Egg_Location != expecteEgg)
            yield return GetInvalid(LEggLocationNone);

        // Flag Moves that cannot be transferred
        if (encounter is EncounterStatic2Odd) // Dizzy Punch Gifts
            FlagIncompatibleTransferMove(pk, Moves, 146, 2); // can't have Dizzy Punch at all

        bool checkShiny = pk.VC2 || (pk.VC1 && GBRestrictions.IsTimeCapsuleTransferred(pk, Moves, encounter).WasTimeCapsuleTransferred());
        if (!checkShiny)
            yield break;

        if (pk.Gender == 1) // female
        {
            if (pk.PersonalInfo.Gender == 31 && pk.IsShiny) // impossible gender-shiny
                yield return GetInvalid(LEncStaticPIDShiny, CheckIdentifier.PID);
        }
        else if (pk.Species == (int)Species.Unown)
        {
            if (pk.Form is not (8 or 21) && pk.IsShiny) // impossibly form-shiny (not I or V)
                yield return GetInvalid(LEncStaticPIDShiny, CheckIdentifier.PID);
        }
    }

    private static void FlagIncompatibleTransferMove(PKM pk, IReadOnlyList<CheckMoveResult> Moves, int move, int gen)
    {
        int index = pk.GetMoveIndex(move);
        if (index < 0)
            return; // doesn't have move

        var chk = Moves[index];
        if (chk.Generation == gen) // not obtained from a future gen
            Moves[index].FlagIllegal(LTransferMove, CheckIdentifier.CurrentMove);
    }
}