Implement enchantWithLevels API

This commit is contained in:
Jason Penilla 2022-03-16 20:35:21 -07:00
parent d18cd1513e
commit c60aaed623

View file

@ -307,4 +307,47 @@ public final class CraftItemFactory implements ItemFactory {
return eggItem == null ? null : new net.minecraft.world.item.ItemStack(eggItem).asBukkitMirror();
}
// Paper end - old getSpawnEgg API
// Paper start - enchantWithLevels API
@Override
public ItemStack enchantWithLevels(ItemStack itemStack, int levels, boolean allowTreasure, java.util.Random random) {
return enchantWithLevels(
itemStack,
levels,
allowTreasure
? Optional.empty()
// While IN_ENCHANTING_TABLE is not logically the same as all but TREASURE, the tag is defined as
// NON_TREASURE, which does contain all enchantments not in the treasure tag.
// Additionally, the allowTreasure boolean is more intended to configure this method to behave like
// an enchanting table.
: net.minecraft.server.MinecraftServer.getServer().registryAccess().lookupOrThrow(Registries.ENCHANTMENT).get(EnchantmentTags.IN_ENCHANTING_TABLE),
random
);
}
@SuppressWarnings("OptionalUsedAsFieldOrParameterType")
private ItemStack enchantWithLevels(
ItemStack itemStack,
int levels,
Optional<? extends net.minecraft.core.HolderSet<net.minecraft.world.item.enchantment.Enchantment>> possibleEnchantments,
java.util.Random random
) {
Preconditions.checkArgument(itemStack != null, "Argument 'itemStack' must not be null");
Preconditions.checkArgument(!itemStack.isEmpty(), "Argument 'itemStack' cannot be empty");
Preconditions.checkArgument(levels > 0 && levels <= 30, "Argument 'levels' must be in range [1, 30] (attempted " + levels + ")");
Preconditions.checkArgument(random != null, "Argument 'random' must not be null");
final net.minecraft.world.item.ItemStack internalStack = CraftItemStack.asNMSCopy(itemStack);
if (internalStack.isEnchanted()) {
internalStack.set(net.minecraft.core.component.DataComponents.ENCHANTMENTS, net.minecraft.world.item.enchantment.ItemEnchantments.EMPTY);
}
final net.minecraft.core.RegistryAccess registryAccess = net.minecraft.server.MinecraftServer.getServer().registryAccess();
final net.minecraft.world.item.ItemStack enchanted = net.minecraft.world.item.enchantment.EnchantmentHelper.enchantItem(
new org.bukkit.craftbukkit.util.RandomSourceWrapper(random),
internalStack,
levels,
registryAccess,
possibleEnchantments
);
return CraftItemStack.asCraftMirror(enchanted);
}
// Paper end - enchantWithLevels API
}