using System.Collections.Generic;

namespace PkmnLib.Plugin.Gen7.Scripts.Side;

[Script(ScriptCategory.Side, "double_power_if_target_damaged_in_turn_data")]
public class DoublePowerIfTargetDamagedInTurnData : Script
{
    public readonly HashSet<IPokemon> HitPokemon = [];

    /// <param name="battle"></param>
    /// <inheritdoc />
    public override void OnEndTurn(IBattle battle)
    {
        RemoveSelf();
    }

    /// <inheritdoc />
    public override void OnDamage(IPokemon pokemon, DamageSource source, uint oldHealth, uint newHealth)
    {
        HitPokemon.Add(pokemon);
    }
}