-
Notifications
You must be signed in to change notification settings - Fork 755
/
MagicProtectionModifier.java
83 lines (75 loc) · 3.83 KB
/
MagicProtectionModifier.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package slimeknights.tconstruct.tools.modifiers.defense;
import net.minecraft.network.chat.Component;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.TooltipFlag;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.entity.living.MobEffectEvent;
import net.minecraftforge.eventbus.api.EventPriority;
import slimeknights.mantle.client.TooltipKey;
import slimeknights.tconstruct.TConstruct;
import slimeknights.tconstruct.library.modifiers.ModifierEntry;
import slimeknights.tconstruct.library.modifiers.ModifierHooks;
import slimeknights.tconstruct.library.modifiers.data.ModifierMaxLevel;
import slimeknights.tconstruct.library.modifiers.hook.armor.ProtectionModifierHook;
import slimeknights.tconstruct.library.modifiers.hook.display.TooltipModifierHook;
import slimeknights.tconstruct.library.modifiers.modules.armor.ProtectionModule;
import slimeknights.tconstruct.library.module.ModuleHookMap.Builder;
import slimeknights.tconstruct.library.tools.capability.TinkerDataCapability;
import slimeknights.tconstruct.library.tools.capability.TinkerDataCapability.TinkerDataKey;
import slimeknights.tconstruct.library.tools.context.EquipmentChangeContext;
import slimeknights.tconstruct.library.tools.context.EquipmentContext;
import slimeknights.tconstruct.library.tools.nbt.IToolStackView;
import javax.annotation.Nullable;
import java.util.List;
public class MagicProtectionModifier extends AbstractProtectionModifier<ModifierMaxLevel> implements ProtectionModifierHook, TooltipModifierHook {
/** Entity data key for the data associated with this modifier */
private static final TinkerDataKey<ModifierMaxLevel> MAGIC_DATA = TConstruct.createKey("magic_protection");
public MagicProtectionModifier() {
super(MAGIC_DATA);
MinecraftForge.EVENT_BUS.addListener(EventPriority.NORMAL, false, MobEffectEvent.Added.class, MagicProtectionModifier::onPotionStart);
}
@Override
protected void registerHooks(Builder hookBuilder) {
super.registerHooks(hookBuilder);
hookBuilder.addHook(this, ModifierHooks.PROTECTION, ModifierHooks.TOOLTIP);
}
@Override
public float getProtectionModifier(IToolStackView tool, ModifierEntry modifier, EquipmentContext context, EquipmentSlot slotType, DamageSource source, float modifierValue) {
if (!source.isBypassMagic() && !source.isBypassInvul() && source.isMagic()) {
modifierValue += modifier.getEffectiveLevel() * 2.5f;
}
return modifierValue;
}
@Override
public void addTooltip(IToolStackView tool, ModifierEntry modifier, @Nullable Player player, List<Component> tooltip, TooltipKey tooltipKey, TooltipFlag tooltipFlag) {
ProtectionModule.addResistanceTooltip(tool, this, modifier.getEffectiveLevel() * 2.5f, player, tooltip);
}
@Override
protected ModifierMaxLevel createData(EquipmentChangeContext context) {
return new ModifierMaxLevel();
}
private static void onPotionStart(MobEffectEvent.Added event) {
MobEffectInstance newEffect = event.getEffectInstance();
if (!newEffect.getEffect().isBeneficial() && !newEffect.getCurativeItems().isEmpty()) {
LivingEntity living = event.getEntity();
living.getCapability(TinkerDataCapability.CAPABILITY).ifPresent(data -> {
ModifierMaxLevel magicData = data.get(MAGIC_DATA);
if (magicData != null) {
float max = magicData.getMax();
if (max > 0) {
// decrease duration by 5% per level
int duration = (int)(newEffect.getDuration() * (1 - (max * 0.05f)));
if (duration < 0) {
duration = 0;
}
newEffect.duration = duration;
}
}
});
}
}
}