OpenSource/PaperMC
OpenSource/PaperMC
1
0
Fork 0
mirror of https://github.com/PaperMC/Paper.git synced 2024-12-28 23:38:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

Get entity default attributes

Browse source 
== AT ==
public net.minecraft.world.entity.ai.attributes.AttributeSupplier getAttributeInstance(Lnet/minecraft/core/Holder;)Lnet/minecraft/world/entity/ai/attributes/AttributeInstance;
This commit is contained in:
Jake Potrebic 2021-08-20 13:03:21 -07:00
parent 067c4e04bd
commit 4180e72054
4 changed files with 114 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
paper-server/src/main/java/io/papermc/paper/attribute/UnmodifiableAttributeInstance.java Normal file
View file

@ -0,0 +1,30 @@
package io.papermc.paper.attribute;
import net.minecraft.world.entity.ai.attributes.AttributeInstance;
import org.bukkit.attribute.Attribute;
import org.bukkit.attribute.AttributeModifier;
import org.bukkit.craftbukkit.attribute.CraftAttributeInstance;
import java.util.Collection;
public class UnmodifiableAttributeInstance extends CraftAttributeInstance {
public UnmodifiableAttributeInstance(AttributeInstance handle, Attribute attribute) {
super(handle, attribute);
}
@Override
public void setBaseValue(double d) {
throw new UnsupportedOperationException("Cannot modify default attributes");
}
@Override
public void addModifier(AttributeModifier modifier) {
throw new UnsupportedOperationException("Cannot modify default attributes");
}
@Override
public void removeModifier(AttributeModifier modifier) {
throw new UnsupportedOperationException("Cannot modify default attributes");
}
}

32
paper-server/src/main/java/io/papermc/paper/attribute/UnmodifiableAttributeMap.java Normal file
View file

@ -0,0 +1,32 @@
package io.papermc.paper.attribute;
import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
import org.bukkit.attribute.Attributable;
import org.bukkit.attribute.Attribute;
import org.bukkit.attribute.AttributeInstance;
import org.bukkit.craftbukkit.attribute.CraftAttribute;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
public class UnmodifiableAttributeMap implements Attributable {
private final AttributeSupplier handle;
public UnmodifiableAttributeMap(@NotNull AttributeSupplier handle) {
this.handle = handle;
}
@Override
public @Nullable AttributeInstance getAttribute(@NotNull Attribute attribute) {
net.minecraft.core.Holder<net.minecraft.world.entity.ai.attributes.Attribute> nmsAttribute = CraftAttribute.bukkitToMinecraftHolder(attribute);
if (!this.handle.hasAttribute(nmsAttribute)) {
return null;
}
return new UnmodifiableAttributeInstance(this.handle.getAttributeInstance(nmsAttribute), attribute);
}
@Override
public void registerAttribute(@NotNull Attribute attribute) {
throw new UnsupportedOperationException("Cannot register new attributes here");
}
}

12
paper-server/src/main/java/org/bukkit/craftbukkit/util/CraftMagicNumbers.java
View file

@ -560,6 +560,18 @@ public final class CraftMagicNumbers implements UnsafeValues {
}
return CraftItemStack.unwrap(itemToBeRepaired).isValidRepairItem(CraftItemStack.unwrap(repairMaterial));
}
@Override
public boolean hasDefaultEntityAttributes(NamespacedKey bukkitEntityKey) {
return net.minecraft.world.entity.ai.attributes.DefaultAttributes.hasSupplier(net.minecraft.core.registries.BuiltInRegistries.ENTITY_TYPE.getValue(CraftNamespacedKey.toMinecraft(bukkitEntityKey)));
}
@Override
public org.bukkit.attribute.Attributable getDefaultEntityAttributes(NamespacedKey bukkitEntityKey) {
Preconditions.checkArgument(hasDefaultEntityAttributes(bukkitEntityKey), bukkitEntityKey + " doesn't have default attributes");
var supplier = net.minecraft.world.entity.ai.attributes.DefaultAttributes.getSupplier((net.minecraft.world.entity.EntityType<? extends net.minecraft.world.entity.LivingEntity>) net.minecraft.core.registries.BuiltInRegistries.ENTITY_TYPE.getValue(CraftNamespacedKey.toMinecraft(bukkitEntityKey)));
return new io.papermc.paper.attribute.UnmodifiableAttributeMap(supplier);
}
// Paper end
/**

40
paper-server/src/test/java/io/papermc/paper/attribute/EntityTypeAttributesTest.java Normal file
View file

@ -0,0 +1,40 @@
package io.papermc.paper.attribute;
import org.bukkit.attribute.Attributable;
import org.bukkit.attribute.Attribute;
import org.bukkit.attribute.AttributeInstance;
import org.bukkit.attribute.AttributeModifier;
import org.bukkit.entity.EntityType;
import org.bukkit.support.environment.AllFeatures;
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
@AllFeatures
public class EntityTypeAttributesTest {
@Test
public void testIllegalEntity() {
assertFalse(EntityType.EGG.hasDefaultAttributes());
assertThrows(IllegalArgumentException.class, EntityType.EGG::getDefaultAttributes);
}
@Test
public void testLegalEntity() {
assertTrue(EntityType.ZOMBIE.hasDefaultAttributes());
EntityType.ZOMBIE.getDefaultAttributes();
}
@Test
public void testUnmodifiabilityOfAttributable() {
Attributable attributable = EntityType.ZOMBIE.getDefaultAttributes();
assertThrows(UnsupportedOperationException.class, () -> attributable.registerAttribute(Attribute.ATTACK_DAMAGE));
AttributeInstance instance = attributable.getAttribute(Attribute.FOLLOW_RANGE);
assertNotNull(instance);
assertThrows(UnsupportedOperationException.class, () -> instance.addModifier(new AttributeModifier("test", 3, AttributeModifier.Operation.ADD_NUMBER)));
assertThrows(UnsupportedOperationException.class, () -> instance.setBaseValue(3.2));
}
}