mirror of https://github.com/PaperMC/Paper.git
97 lines
4.8 KiB
Diff
97 lines
4.8 KiB
Diff
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
|
|
From: Jake Potrebic <jake.m.potrebic@gmail.com>
|
|
Date: Thu, 27 May 2021 21:58:24 -0700
|
|
Subject: [PATCH] More PotionEffectType API
|
|
|
|
== AT ==
|
|
public net.minecraft.world.effect.MobEffect attributeModifiers
|
|
public net.minecraft.world.effect.MobEffect$AttributeTemplate
|
|
|
|
diff --git a/src/main/java/org/bukkit/craftbukkit/potion/CraftPotionEffectType.java b/src/main/java/org/bukkit/craftbukkit/potion/CraftPotionEffectType.java
|
|
index 8dffef71c8b193c0fc84b65a592c93827e341bf7..e15d83ac2668ebb0da9e22c15b9fd902689d5522 100644
|
|
--- a/src/main/java/org/bukkit/craftbukkit/potion/CraftPotionEffectType.java
|
|
+++ b/src/main/java/org/bukkit/craftbukkit/potion/CraftPotionEffectType.java
|
|
@@ -123,6 +123,48 @@ public class CraftPotionEffectType extends PotionEffectType implements Handleabl
|
|
return this.handle.getDescriptionId();
|
|
}
|
|
|
|
+ // Paper start
|
|
+ @Override
|
|
+ public java.util.Map<org.bukkit.attribute.Attribute, org.bukkit.attribute.AttributeModifier> getEffectAttributes() {
|
|
+ // re-create map each time because a nms MobEffect can have its attributes modified
|
|
+ final java.util.Map<org.bukkit.attribute.Attribute, org.bukkit.attribute.AttributeModifier> attributeMap = new java.util.HashMap<>();
|
|
+ this.handle.attributeModifiers.forEach((attribute, attributeModifier) -> {
|
|
+ attributeMap.put(
|
|
+ org.bukkit.craftbukkit.attribute.CraftAttribute.minecraftHolderToBukkit(attribute),
|
|
+ // use zero as amplifier to get the base amount, as it is amount = base * (amplifier + 1)
|
|
+ org.bukkit.craftbukkit.attribute.CraftAttributeInstance.convert(attributeModifier.create(this.handle.getDescriptionId(), 0))
|
|
+ );
|
|
+ });
|
|
+ return java.util.Map.copyOf(attributeMap);
|
|
+ }
|
|
+
|
|
+ @Override
|
|
+ public double getAttributeModifierAmount(org.bukkit.attribute.Attribute attribute, int effectAmplifier) {
|
|
+ com.google.common.base.Preconditions.checkArgument(effectAmplifier >= 0, "effectAmplifier must be greater than or equal to 0");
|
|
+ Holder<net.minecraft.world.entity.ai.attributes.Attribute> nmsAttribute = org.bukkit.craftbukkit.attribute.CraftAttribute.bukkitToMinecraftHolder(attribute);
|
|
+ com.google.common.base.Preconditions.checkArgument(this.handle.attributeModifiers.containsKey(nmsAttribute), attribute + " is not present on " + this.getKey());
|
|
+ return this.handle.attributeModifiers.get(nmsAttribute).create(this.handle.getDescriptionId(), effectAmplifier).amount();
|
|
+ }
|
|
+
|
|
+ @Override
|
|
+ public PotionEffectType.Category getEffectCategory() {
|
|
+ return fromNMS(handle.getCategory());
|
|
+ }
|
|
+
|
|
+ @Override
|
|
+ public String translationKey() {
|
|
+ return this.handle.getDescriptionId();
|
|
+ }
|
|
+
|
|
+ public static PotionEffectType.Category fromNMS(net.minecraft.world.effect.MobEffectCategory mobEffectInfo) {
|
|
+ return switch (mobEffectInfo) {
|
|
+ case BENEFICIAL -> PotionEffectType.Category.BENEFICIAL;
|
|
+ case HARMFUL -> PotionEffectType.Category.HARMFUL;
|
|
+ case NEUTRAL -> PotionEffectType.Category.NEUTRAL;
|
|
+ };
|
|
+ }
|
|
+ // Paper end
|
|
+
|
|
@Override
|
|
public boolean equals(Object other) {
|
|
if (this == other) {
|
|
diff --git a/src/test/java/io/papermc/paper/effects/EffectCategoryTest.java b/src/test/java/io/papermc/paper/effects/EffectCategoryTest.java
|
|
new file mode 100644
|
|
index 0000000000000000000000000000000000000000..6262598f85bd7d9af5546cc0a96531b2f4baf64d
|
|
--- /dev/null
|
|
+++ b/src/test/java/io/papermc/paper/effects/EffectCategoryTest.java
|
|
@@ -0,0 +1,28 @@
|
|
+package io.papermc.paper.effects;
|
|
+
|
|
+import io.papermc.paper.adventure.PaperAdventure;
|
|
+import net.minecraft.world.effect.MobEffectCategory;
|
|
+import org.bukkit.craftbukkit.potion.CraftPotionEffectType;
|
|
+import org.bukkit.potion.PotionEffectType;
|
|
+import org.junit.jupiter.api.Test;
|
|
+
|
|
+import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
+import static org.junit.jupiter.api.Assertions.assertNotNull;
|
|
+
|
|
+public class EffectCategoryTest {
|
|
+
|
|
+ @Test
|
|
+ public void testEffectCategoriesExist() {
|
|
+ for (MobEffectCategory mobEffectInfo : MobEffectCategory.values()) {
|
|
+ assertNotNull(CraftPotionEffectType.fromNMS(mobEffectInfo), mobEffectInfo + " is missing a bukkit equivalent");
|
|
+ }
|
|
+ }
|
|
+
|
|
+ @Test
|
|
+ public void testCategoryHasEquivalentColors() {
|
|
+ for (MobEffectCategory mobEffectInfo : MobEffectCategory.values()) {
|
|
+ PotionEffectType.Category bukkitEffectCategory = CraftPotionEffectType.fromNMS(mobEffectInfo);
|
|
+ assertEquals(bukkitEffectCategory.getColor(), PaperAdventure.asAdventure(mobEffectInfo.getTooltipFormatting()), mobEffectInfo.getTooltipFormatting().name() + " doesn't equal " + bukkitEffectCategory.getColor());
|
|
+ }
|
|
+ }
|
|
+}
|