From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001 From: Jake Potrebic Date: Thu, 6 May 2021 19:57:58 -0700 Subject: [PATCH] More Enchantment API == AT == public net.minecraft.world.item.enchantment.Enchantment definition Co-authored-by: Luis diff --git a/src/main/java/org/bukkit/craftbukkit/enchantments/CraftEnchantment.java b/src/main/java/org/bukkit/craftbukkit/enchantments/CraftEnchantment.java index faf0f726453aa957d17b75dfd8de5b40d984e69c..3157cac1905728ba0f51af412a5bba57578352b3 100644 --- a/src/main/java/org/bukkit/craftbukkit/enchantments/CraftEnchantment.java +++ b/src/main/java/org/bukkit/craftbukkit/enchantments/CraftEnchantment.java @@ -94,7 +94,7 @@ public class CraftEnchantment extends Enchantment implements Handleable guessEntityTypeFromEnchantmentCategory( + final org.bukkit.entity.EntityCategory entityCategory + ) { + final net.minecraft.tags.TagKey> tag = switch (entityCategory) { + case ARTHROPOD -> net.minecraft.tags.EntityTypeTags.SENSITIVE_TO_BANE_OF_ARTHROPODS; + case UNDEAD -> net.minecraft.tags.EntityTypeTags.SENSITIVE_TO_SMITE; + case WATER -> net.minecraft.tags.EntityTypeTags.SENSITIVE_TO_IMPALING; + default -> null; + }; + if (tag == null) return null; + + return net.minecraft.core.registries.BuiltInRegistries.ENTITY_TYPE.getTag(tag) + .map(e -> e.size() > 0 ? e.get(0).value() : null) + .orElse(null); + } + + @Override + public java.util.Set getActiveSlots() { + return java.util.stream.Stream.of(this.handle.definition.slots()).map(org.bukkit.craftbukkit.CraftEquipmentSlot::getSlot).collect(java.util.stream.Collectors.toSet()); + } // Paper end @Override