using System.Collections.Generic;
using PkmnLib.Static.Utils;

namespace PkmnLib.Plugin.Gen7.Scripts.Pokemon;

[Script(ScriptCategory.Pokemon, "roost_effect")]
public class RoostEffect : Script
{
    /// <inheritdoc />
    public override void ChangeTypesForIncomingMove(IExecutingMove executingMove, IPokemon target, byte hitIndex,
        IList<TypeIdentifier> types)
    {
        types.RemoveAll(x => x.Name == "flying");
    }

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