package com.example.enchantmentrevealer.mixin; import net.minecraft.client.gui.DrawContext; import net.minecraft.client.gui.screen.ingame.EnchantmentScreen; import net.minecraft.client.gui.screen.ingame.HandledScreen; import net.minecraft.client.font.TextRenderer; import net.minecraft.enchantment.Enchantment; import net.minecraft.enchantment.EnchantmentHelper; import net.minecraft.enchantment.EnchantmentLevelEntry; import net.minecraft.entity.player.PlayerInventory; import net.minecraft.item.ItemStack; import net.minecraft.registry.RegistryKeys; import net.minecraft.registry.entry.RegistryEntry; import net.minecraft.screen.EnchantmentScreenHandler; import net.minecraft.text.Text; import net.minecraft.util.Formatting; import net.minecraft.util.math.random.Random; import org.spongepowered.asm.mixin.Mixin; import org.spongepowered.asm.mixin.Unique; import org.spongepowered.asm.mixin.injection.At; import org.spongepowered.asm.mixin.injection.Redirect; import java.util.ArrayList; import java.util.List; import java.util.stream.Stream; @Mixin(EnchantmentScreen.class) public abstract class EnchantmentScreenMixin extends HandledScreen { public EnchantmentScreenMixin(EnchantmentScreenHandler handler, PlayerInventory inventory, Text title) { super(handler, inventory, title); } @Redirect(method = "render", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/DrawContext;drawTooltip(Lnet/minecraft/client/font/TextRenderer;Ljava/util/List;II)V")) private void onDrawTooltip(DrawContext instance, TextRenderer textRenderer, List text, int mouseX, int mouseY) { if (this.handler.getSlot(0).hasStack() && !text.isEmpty()) { int hoveredSlot = -1; for (int i = 0; i < 3; ++i) { int levelReq = this.handler.enchantmentPower[i]; if (levelReq == 0) continue; int btnX = this.x + 60; int btnY = this.y + 14 + 19 * i; if (mouseX >= btnX && mouseX < btnX + 108 && mouseY >= btnY && mouseY < btnY + 19) { hoveredSlot = i; break; } } if (hoveredSlot != -1) { List newTooltip = this.getEnchantmentList(hoveredSlot); if (newTooltip != null && !newTooltip.isEmpty()) { instance.drawTooltip(textRenderer, newTooltip, mouseX, mouseY); return; } } } instance.drawTooltip(textRenderer, text, mouseX, mouseY); } @Unique private List getEnchantmentList(int slotIndex) { ItemStack itemStack = this.handler.getSlot(0).getStack(); if (itemStack.isEmpty() || !itemStack.isEnchantable()) { return null; } int seed = this.handler.getSeed(); int power = this.handler.enchantmentPower[slotIndex]; if (power == 0) return null; var registryManager = this.client.world.getRegistryManager(); Random random = Random.create(); random.setSeed((long)seed + slotIndex); Stream> possibleEnchantments = registryManager .getOrThrow(RegistryKeys.ENCHANTMENT) .streamEntries() .filter(entry -> entry.value().isPrimaryItem(itemStack)); List entries = EnchantmentHelper.generateEnchantments( random, itemStack, power, possibleEnchantments ); List tooltip = new ArrayList<>(); tooltip.add(Text.translatable("container.enchant.clue", "").append(Text.literal(" (Cost: " + power + ")").formatted(Formatting.YELLOW))); if (entries == null || entries.isEmpty()) { tooltip.add(Text.literal("No enchantments").formatted(Formatting.RED)); } else { for (EnchantmentLevelEntry entry : entries) { Text name = Enchantment.getName(entry.enchantment, entry.level); tooltip.add(name.copy().formatted(Formatting.GREEN)); } } return tooltip; } }