PkmnLib.NET/Plugins/PkmnLib.Plugin.Gen7/Scripts/Battle/FairyLockEffect.cs

namespace PkmnLib.Plugin.Gen7.Scripts.Battle;

[Script(ScriptCategory.Battle, "fairy_lock")]
public class FairyLockEffect : Script
{
    private int _turns = 1;

    /// <inheritdoc />
    public override void PreventSelfRunAway(IFleeChoice choice, ref bool prevent)
    {
        prevent = true;
    }

    /// <inheritdoc />
    public override void PreventSelfSwitch(ISwitchChoice choice, ref bool prevent)
    {
        prevent = true;
    }

    /// <inheritdoc />
    public override void OnEndTurn(IBattle battle)
    {
        if (_turns <= 0)
            RemoveSelf();
        _turns--;
    }
}