using PkmnLib.Plugin.Gen7.Scripts.Utils;

namespace PkmnLib.Plugin.Gen7.Scripts.Pokemon;

[Script(ScriptCategory.Pokemon, "dig")]
public class DigEffect : Script, IScriptForceTurnSelection, IScriptChangeIncomingMoveDamage, IScriptBlockIncomingHit,
    IScriptOnAfterMoveChoice
{
    private readonly IBattlePokemon _owner;

    /// <inheritdoc />
    public override bool IsSemiInvulnerableTurn => true;

    public DigEffect(IBattlePokemon owner)
    {
        _owner = owner;
    }

    /// <inheritdoc />
    public void ForceTurnSelection(IBattle battle, byte sideIndex, byte position, ref ITurnChoice? choice)
    {
        var opposingSideIndex = (byte)(_owner.SideIndex == 0 ? 1 : 0);
        choice = TurnChoiceHelper.CreateMoveChoice(_owner, "dig", opposingSideIndex, position);
    }

    /// <inheritdoc />
    public void BlockIncomingHit(IExecutingMove executingMove, IBattlePokemon target, byte hitIndex, ref bool block)
    {
        if (!executingMove.UseMove.HasFlag(MoveFlags.HitUnderground))
            block = true;
    }

    /// <inheritdoc />
    public void ChangeIncomingMoveDamage(IExecutingMove move, IBattlePokemon target, byte hit, ref uint damage)
    {
        if (move.UseMove.HasFlag(MoveFlags.EffectiveAgainstUnderground))
            damage *= 2;
    }

    /// <inheritdoc />
    public void OnAfterMoveChoice(IMoveChoice moveChoice)
    {
        if (moveChoice.AdditionalData?.ContainsKey("dig_charge") != true)
        {
            RemoveSelf();
        }
    }
}