1 files changed, 45 insertions, 0 deletions
diff --git a/patches/server/0681-Implement-enchantWithLevels-API.patch b/patches/server/0681-Implement-enchantWithLevels-API.patch
new file mode 100644
index 0000000000..7bf61f04ba
--- /dev/null
+++ b/patches/server/0681-Implement-enchantWithLevels-API.patch
@@ -0,0 +1,45 @@
+From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
+From: Jason Penilla <[email protected]>
+Date: Wed, 16 Mar 2022 20:35:21 -0700
+Subject: [PATCH] Implement enchantWithLevels API
+
+
+diff --git a/src/main/java/org/bukkit/craftbukkit/inventory/CraftItemFactory.java b/src/main/java/org/bukkit/craftbukkit/inventory/CraftItemFactory.java
+index 803a19063c03627dbea79cb1c395ae35aaef2834..317af98f8bd166a5a534f74e6ba177be9afb6a8f 100644
+--- a/src/main/java/org/bukkit/craftbukkit/inventory/CraftItemFactory.java
++++ b/src/main/java/org/bukkit/craftbukkit/inventory/CraftItemFactory.java
+@@ -302,4 +302,34 @@ 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) {
++        Preconditions.checkArgument(itemStack != null, "Argument 'itemStack' must not be null");
++        Preconditions.checkArgument(itemStack.getType() != Material.AIR, "Argument 'itemStack' must not be of type AIR");
++        Preconditions.checkArgument(itemStack.getAmount() > 0, "Argument 'itemStack' amount must be greater than 0");
++        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, null);
++        }
++        final net.minecraft.core.RegistryAccess registryAccess = net.minecraft.server.MinecraftServer.getServer().registryAccess();
++        final net.minecraft.core.Registry<Enchantment> enchantments = registryAccess.registryOrThrow(Registries.ENCHANTMENT);
++        final net.minecraft.world.item.ItemStack enchanted = net.minecraft.world.item.enchantment.EnchantmentHelper.enchantItem(
++            new org.bukkit.craftbukkit.util.RandomSourceWrapper(random),
++            internalStack,
++            levels,
++            registryAccess,
++            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.
++                : registryAccess.registryOrThrow(Registries.ENCHANTMENT).getTag(EnchantmentTags.IN_ENCHANTING_TABLE)
++        );
++        return CraftItemStack.asCraftMirror(enchanted);
++    }
++    // Paper end - enchantWithLevels API
+ }