Enchantment Menu mixin, Tag API rename

src/main/java/ru/bclib/mixin/common/BoneMealItemMixin.java

@@ -16,7 +16,7 @@ import net.minecraft.world.level.biome.Biome.BiomeCategory;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.Blocks;
 import net.minecraft.world.level.block.state.BlockState;
-import ru.bclib.api.BCLibTags;
+import ru.bclib.api.TagAPI;
 import ru.bclib.api.BiomeAPI;
 import ru.bclib.api.BonemealAPI;
 import ru.bclib.util.BlocksHelper;
@@ -34,7 +34,7 @@ public class BoneMealItemMixin {
 			BlockPos offseted = blockPos.relative(context.getClickedFace());
 			boolean endBiome = world.getBiome(offseted).getBiomeCategory() == BiomeCategory.THEEND;
 			
-			if (world.getBlockState(blockPos).is(BCLibTags.END_GROUND)) {
+			if (world.getBlockState(blockPos).is(TagAPI.END_GROUND)) {
 				boolean consume = false;
 				if (world.getBlockState(blockPos).is(Blocks.END_STONE)) {
 					BlockState nylium = bclib_getNylium(world, blockPos);

src/main/java/ru/bclib/mixin/common/EnchantmentMenuMixin.java

@@ -0,0 +1,140 @@
+package ru.bclib.mixin.common;
+
+import java.util.List;
+import java.util.Random;
+
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+import net.minecraft.core.Registry;
+import net.minecraft.world.Container;
+import net.minecraft.world.inventory.AbstractContainerMenu;
+import net.minecraft.world.inventory.ContainerLevelAccess;
+import net.minecraft.world.inventory.DataSlot;
+import net.minecraft.world.inventory.EnchantmentMenu;
+import net.minecraft.world.inventory.MenuType;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.item.enchantment.EnchantmentHelper;
+import net.minecraft.world.item.enchantment.EnchantmentInstance;
+import ru.bclib.api.TagAPI;
+
+@Mixin(EnchantmentMenu.class)
+public abstract class EnchantmentMenuMixin extends AbstractContainerMenu {
+	@Final
+	@Shadow
+	private Container enchantSlots;
+
+	@Final
+	@Shadow
+	private ContainerLevelAccess access;
+
+	@Final
+	@Shadow
+	private Random random;
+
+	@Final
+	@Shadow
+	private DataSlot enchantmentSeed;
+
+	@Shadow
+	@Final
+	public int[] costs;
+
+	@Shadow
+	@Final
+	public int[] enchantClue;
+
+	@Shadow
+	@Final
+	public int[] levelClue;
+
+	protected EnchantmentMenuMixin(MenuType<?> type, int syncId) {
+		super(type, syncId);
+	}
+
+	@Inject(method = "slotsChanged", at = @At("HEAD"), cancellable = true)
+	private void be_slotsChanged(Container inventory, CallbackInfo info) {
+		if (inventory == this.enchantSlots) {
+			ItemStack itemStack = inventory.getItem(0);
+			if (!itemStack.isEmpty() && itemStack.isEnchantable()) {
+				this.access.execute((world, blockPos) -> {
+					int i = 0;
+
+					int j;
+					for (j = -1; j <= 1; ++j) {
+						for (int k = -1; k <= 1; ++k) {
+							if ((j != 0 || k != 0) && world.isEmptyBlock(blockPos.offset(k, 0, j)) && world.isEmptyBlock(blockPos.offset(k, 1, j))) {
+								if (world.getBlockState(blockPos.offset(k * 2, 0, j * 2)).is(TagAPI.BOOKSHELVES)) {
+									++i;
+								}
+
+								if (world.getBlockState(blockPos.offset(k * 2, 1, j * 2)).is(TagAPI.BOOKSHELVES)) {
+									++i;
+								}
+
+								if (k != 0 && j != 0) {
+									if (world.getBlockState(blockPos.offset(k * 2, 0, j)).is(TagAPI.BOOKSHELVES)) {
+										++i;
+									}
+
+									if (world.getBlockState(blockPos.offset(k * 2, 1, j)).is(TagAPI.BOOKSHELVES)) {
+										++i;
+									}
+
+									if (world.getBlockState(blockPos.offset(k, 0, j * 2)).is(TagAPI.BOOKSHELVES)) {
+										++i;
+									}
+
+									if (world.getBlockState(blockPos.offset(k, 1, j * 2)).is(TagAPI.BOOKSHELVES)) {
+										++i;
+									}
+								}
+							}
+						}
+					}
+
+					random.setSeed(enchantmentSeed.get());
+
+					for (j = 0; j < 3; ++j) {
+						costs[j] = EnchantmentHelper.getEnchantmentCost(this.random, j, i, itemStack);
+						enchantClue[j] = -1;
+						levelClue[j] = -1;
+						if (costs[j] < j + 1) {
+							costs[j] = 0;
+						}
+					}
+
+					for (j = 0; j < 3; ++j) {
+						if (this.costs[j] > 0) {
+							List<EnchantmentInstance> list = this.getEnchantmentList(itemStack, j, this.costs[j]);
+							if (list != null && !list.isEmpty()) {
+								EnchantmentInstance enchantmentLevelEntry = (EnchantmentInstance) list.get(this.random.nextInt(list.size()));
+								enchantClue[j] = Registry.ENCHANTMENT.getId(enchantmentLevelEntry.enchantment);
+								levelClue[j] = enchantmentLevelEntry.level;
+							}
+						}
+					}
+
+					broadcastChanges();
+				});
+			}
+			else {
+				for (int i = 0; i < 3; ++i) {
+					costs[i] = 0;
+					enchantClue[i] = -1;
+					levelClue[i] = -1;
+				}
+			}
+			info.cancel();
+		}
+	}
+
+	@Shadow
+	private List<EnchantmentInstance> getEnchantmentList(ItemStack stack, int slot, int level) {
+		return null;
+	}
+}