PkmnLib.NET/Plugins/PkmnLib.Plugin.Gen7/Scripts/Abilities/ArenaTrap.cs

namespace PkmnLib.Plugin.Gen7.Scripts.Abilities;

[Script(ScriptCategory.Ability, "arena_trap")]
public class ArenaTrap : Script
{
    private IPokemon? _owner;

    /// <inheritdoc />
    public override void OnAddedToParent(IScriptSource source)
    {
        if (source is not IPokemon pokemon)
            throw new InvalidOperationException("ArenaTrap can only be added to a Pokemon.");
        _owner = pokemon;
    }

    /// <inheritdoc />
    public override void PreventOpponentRunAway(IFleeChoice choice, ref bool prevent)
    {
        if (choice.User.IsFloating)
            return;
        if (_owner is not null)
            choice.User.BattleData?.Battle.EventHook.Invoke(new AbilityTriggerEvent(_owner));
        prevent = true;
    }

    /// <inheritdoc />
    public override void PreventOpponentSwitch(ISwitchChoice choice, ref bool prevent)
    {
        if (choice.User.IsFloating)
            return;
        if (_owner is not null)
            choice.User.BattleData?.Battle.EventHook.Invoke(new AbilityTriggerEvent(_owner));
        prevent = true;
    }
}