50 lines
2 KiB
Java
50 lines
2 KiB
Java
package ru.betterend.item;
|
|
|
|
import java.util.UUID;
|
|
import net.minecraft.world.entity.EquipmentSlot;
|
|
import net.minecraft.world.entity.ai.attributes.Attribute;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeModifier;
|
|
import net.minecraft.world.entity.ai.attributes.Attributes;
|
|
import net.minecraft.world.item.ArmorItem;
|
|
import net.minecraft.world.item.ArmorMaterial;
|
|
import net.minecraft.world.item.Item;
|
|
import com.google.common.collect.ImmutableMultimap;
|
|
import com.google.common.collect.Multimap;
|
|
import ru.betterend.mixin.common.ArmorItemAccessor;
|
|
import ru.betterend.patterns.Patterned;
|
|
import ru.betterend.patterns.Patterns;
|
|
|
|
public class EndArmorItem extends ArmorItem implements Patterned {
|
|
public EndArmorItem(ArmorMaterial material, EquipmentSlot slot, Item.Properties settings) {
|
|
super(material, slot, settings);
|
|
|
|
addKnockbackResistance((ArmorItemAccessor) this, slot, this.knockbackResistance);
|
|
}
|
|
|
|
/** Ensures knockback resistance is actually applied */
|
|
private static void addKnockbackResistance(ArmorItemAccessor accessor, EquipmentSlot slot, double knockbackResistance) {
|
|
if (knockbackResistance == 0) {
|
|
return;
|
|
}
|
|
|
|
Multimap<Attribute, AttributeModifier> attributeModifiers = accessor.getDefaultModifiers();
|
|
|
|
// In case Mojang or anyone else decided to fix this
|
|
if (attributeModifiers.keys().contains(Attributes.KNOCKBACK_RESISTANCE)) {
|
|
return;
|
|
}
|
|
|
|
UUID uuid = accessor.getModifiers()[slot.getIndex()];
|
|
|
|
// Rebuild attributeModifiers to include knockback resistance
|
|
ImmutableMultimap.Builder<Attribute, AttributeModifier> builder = ImmutableMultimap.builder();
|
|
builder.putAll(attributeModifiers);
|
|
builder.put(Attributes.KNOCKBACK_RESISTANCE, new AttributeModifier(uuid, "Armor knockback resistance", knockbackResistance, AttributeModifier.Operation.ADDITION));
|
|
accessor.setDefaultModifiers(builder.build());
|
|
}
|
|
|
|
@Override
|
|
public String getModelPattern(String name) {
|
|
return Patterns.createJson(Patterns.ITEM_GENERATED, name);
|
|
}
|
|
}
|