#include "../Platform/stdafx.h"
#include "../Headers/net.minecraft.world.entity.h"
#include "../Headers/net.minecraft.world.effect.h"
#include "AbsoptionMobEffect.h"

AbsoptionMobEffect::AbsoptionMobEffect(int id, bool isHarmful,
                                       eMinecraftColour color)
    : MobEffect(id, isHarmful, color) {}

void AbsoptionMobEffect::removeAttributeModifiers(
    std::shared_ptr<LivingEntity> entity, BaseAttributeMap* attributes,
    int amplifier) {
    entity->setAbsorptionAmount(entity->getAbsorptionAmount() -
                                4 * (amplifier + 1));
    MobEffect::removeAttributeModifiers(entity, attributes, amplifier);
}

void AbsoptionMobEffect::addAttributeModifiers(
    std::shared_ptr<LivingEntity> entity, BaseAttributeMap* attributes,
    int amplifier) {
    entity->setAbsorptionAmount(entity->getAbsorptionAmount() +
                                4 * (amplifier + 1));
    MobEffect::addAttributeModifiers(entity, attributes, amplifier);
}