-
Notifications
You must be signed in to change notification settings - Fork 755
/
MithridatismModifier.java
34 lines (30 loc) · 1.61 KB
/
MithridatismModifier.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
package slimeknights.tconstruct.tools.modifiers.traits.skull;
import net.minecraft.world.effect.MobEffects;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.entity.living.MobEffectEvent.Applicable;
import net.minecraftforge.eventbus.api.Event.Result;
import net.minecraftforge.eventbus.api.EventPriority;
import slimeknights.tconstruct.TConstruct;
import slimeknights.tconstruct.common.TinkerTags;
import slimeknights.tconstruct.library.modifiers.impl.NoLevelsModifier;
import slimeknights.tconstruct.library.modifiers.modules.unserializable.ArmorLevelModule;
import slimeknights.tconstruct.library.modifiers.util.ModifierHookMap.Builder;
import slimeknights.tconstruct.library.tools.capability.TinkerDataCapability.TinkerDataKey;
public class MithridatismModifier extends NoLevelsModifier {
private static final TinkerDataKey<Integer> MITHRIDATISM = TConstruct.createKey("mithridatism");
public MithridatismModifier() {
// TODO: move this out of constructor to generalized logic
MinecraftForge.EVENT_BUS.addListener(EventPriority.NORMAL, false, Applicable.class, MithridatismModifier::isPotionApplicable);
}
@Override
protected void registerHooks(Builder hookBuilder) {
super.registerHooks(hookBuilder);
hookBuilder.addModule(new ArmorLevelModule(MITHRIDATISM, false, TinkerTags.Items.HELD_ARMOR));
}
/** Prevents poison on the entity */
private static void isPotionApplicable(Applicable event) {
if (event.getEffectInstance().getEffect() == MobEffects.POISON && ArmorLevelModule.getLevel(event.getEntity(), MITHRIDATISM) > 0) {
event.setResult(Result.DENY);
}
}
}