server-master/srcs/_plugins/WingsEmu.Plugins.BasicImplementation/BCards/Handlers/BCardDrainAndStealHandler.cs

// WingsEmu
// 
// Developed by NosWings Team

using WingsEmu.DTOs.BCards;
using WingsEmu.Game;
using WingsEmu.Game.Battle;
using WingsEmu.Game.Buffs;
using WingsEmu.Game.Characters;
using WingsEmu.Game.Characters.Events;
using WingsEmu.Game.Entities;
using WingsEmu.Game.Extensions;
using WingsEmu.Packets.Enums;

namespace WingsEmu.Plugins.BasicImplementations.BCards.Handlers;

public class BCardDrainAndStealHandler : IBCardEffectAsyncHandler
{
    private readonly IRandomGenerator _randomGenerator;

    public BCardDrainAndStealHandler(IRandomGenerator randomGenerator) => _randomGenerator = randomGenerator;

    public BCardType HandledType => BCardType.DrainAndSteal;

    public void Execute(IBCardEffectContext ctx)
    {
        IBattleEntity target = ctx.Target;
        IBattleEntity sender = ctx.Sender;
        BCardDTO bCard = ctx.BCard;
        byte subType = ctx.BCard.SubType;

        int firstDataValue = bCard.FirstDataValue(sender.Level);
        int secondDataValue = bCard.SecondDataValue(sender.Level);

        if (!target.IsAlive() || !sender.IsAlive())
        {
            return;
        }

        if (target.IsMateTrainer())
        {
            return;
        }

        switch ((AdditionalTypes.DrainAndSteal)subType)
        {
            case AdditionalTypes.DrainAndSteal.GiveEnemyHP:
                break;
            case AdditionalTypes.DrainAndSteal.LeechEnemyHP:
                if (_randomGenerator.RandomNumber() > firstDataValue)
                {
                    return;
                }

                if (target.Hp - secondDataValue > 0)
                {
                    target.BroadcastDamage(secondDataValue);
                    target.Hp -= secondDataValue;
                }

                if (sender.Hp + secondDataValue < sender.MaxHp)
                {
                    if (target.Hp - secondDataValue <= 0)
                    {
                        sender.EmitEvent(new BattleEntityHealEvent
                        {
                            Entity = sender,
                            HpHeal = target.Hp
                        });

                        target.Hp = 1;
                    }
                    else
                    {
                        sender.EmitEvent(new BattleEntityHealEvent
                        {
                            Entity = sender,
                            HpHeal = secondDataValue
                        });
                    }
                }
                else
                {
                    if (target.Hp - secondDataValue <= 0)
                    {
                        target.Hp = 1;
                    }

                    sender.EmitEvent(new BattleEntityHealEvent
                    {
                        Entity = sender,
                        HpHeal = sender.MaxHp
                    });
                }

                (target as IPlayerEntity)?.Session.RefreshStat();
                (sender as IPlayerEntity)?.Session.RefreshStat();
                break;
            case AdditionalTypes.DrainAndSteal.GiveEnemyMP:
                break;
            case AdditionalTypes.DrainAndSteal.LeechEnemyMP:
                if (_randomGenerator.RandomNumber() > firstDataValue)
                {
                    return;
                }

                if (target.Mp - secondDataValue > 0)
                {
                    target.Mp -= secondDataValue;
                }

                if (sender.Mp + secondDataValue < sender.MaxMp)
                {
                    if (target.Mp - secondDataValue <= 0)
                    {
                        target.Mp = 1;
                    }

                    sender.Mp += secondDataValue;
                }
                else
                {
                    if (target.Mp - secondDataValue <= 0)
                    {
                        target.Mp = 1;
                    }

                    sender.Mp = sender.MaxMp;
                }

                (target as IPlayerEntity)?.Session.RefreshStat();
                (sender as IPlayerEntity)?.Session.RefreshStat();
                break;
            case AdditionalTypes.DrainAndSteal.ConvertEnemyMPToHP:
                break;
            case AdditionalTypes.DrainAndSteal.ConvertEnemyHPToMP:

                int toRemoveAndAdd = firstDataValue;

                if (sender.Mp + toRemoveAndAdd > sender.MaxMp)
                {
                    if (target.Hp - toRemoveAndAdd <= 0)
                    {
                        target.BroadcastDamage(target.Hp - 1);
                        target.Hp = 1;
                    }
                    else
                    {
                        target.BroadcastDamage(toRemoveAndAdd);
                        target.Hp -= toRemoveAndAdd;
                        sender.EmitEvent(new BattleEntityHealEvent
                        {
                            Entity = sender,
                            MpHeal = sender.MaxMp
                        });
                    }
                }
                else
                {
                    if (target.Hp - toRemoveAndAdd <= 0)
                    {
                        toRemoveAndAdd = target.Hp - 1;
                        target.BroadcastDamage(toRemoveAndAdd);
                        target.Hp = 1;
                    }
                    else
                    {
                        target.BroadcastDamage(toRemoveAndAdd);
                        target.Hp -= toRemoveAndAdd;
                    }

                    sender.EmitEvent(new BattleEntityHealEvent
                    {
                        Entity = sender,
                        MpHeal = toRemoveAndAdd
                    });
                }

                (target as IPlayerEntity)?.Session.RefreshStat();
                (sender as IPlayerEntity)?.Session.RefreshStat();

                break;
        }
    }
}