namespace PkmnLib.Plugin.Gen7.Scripts.Abilities;

/// <summary>
/// Schooling is an ability that changes Wishiwashi's form in battle.
///
/// <see href="https://bulbapedia.bulbagarden.net/wiki/Schooling_(Ability)">Bulbapedia - Schooling</see>
/// </summary>
[Script(ScriptCategory.Ability, "schooling")]
public class Schooling : Script, IScriptOnEndTurn, IScriptOnSwitchIn
{
    private IBattlePokemon? _owningPokemon;

    /// <inheritdoc />
    public override void OnAddedToParent(IScriptSource source)
    {
        if (source is not IBattlePokemon pokemon)
            throw new ArgumentException("Schooling script must be added to a Pokemon.", nameof(source));
        _owningPokemon = pokemon;
    }

    /// <inheritdoc />
    public void OnSwitchIn(IBattlePokemon pokemon, byte position) => ChangeFormIfNeeded(pokemon);

    /// <inheritdoc />
    public void OnEndTurn(IScriptSource owner, IBattle battle) => ChangeFormIfNeeded(_owningPokemon);

    private static void ChangeFormIfNeeded(IBattlePokemon? pokemon)
    {
        if (pokemon is null)
            return;

        if (pokemon.Species.Name != SpeciesNames.Wishiwashi || pokemon.Battle == null)
            return;
        // If Wishiwashi has less than 25% health, change to Solo form
        if (pokemon.CurrentHealth < pokemon.MaxHealth / 4 && pokemon.Form.Name != FormNames.Default)
        {
            pokemon.ChangeForm(pokemon.Species.GetDefaultForm());
        }
        else if (pokemon.CurrentHealth >= pokemon.MaxHealth / 4 && pokemon.Form.Name != FormNames.School &&
                 pokemon.Species.TryGetForm(FormNames.School, out var schoolForm))
        {
            pokemon.ChangeForm(schoolForm);
        }
    }
}