Initial commit

src/main/java/xyz/nuark/predench/mixin/MixinEnchantmentMenu.java

@@ -0,0 +1,96 @@
+package xyz.nuark.predench.mixin;
+
+import com.google.gson.Gson;
+import net.minecraft.core.BlockPos;
+import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.inventory.DataSlot;
+import net.minecraft.world.inventory.EnchantmentMenu;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.item.enchantment.EnchantmentHelper;
+import net.minecraft.world.item.enchantment.EnchantmentInstance;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.level.block.EnchantmentTableBlock;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import xyz.nuark.predench.EnchantmentData;
+import xyz.nuark.predench.Predench;
+import xyz.nuark.predench.PredenchEnchantmentDataExtractor;
+import xyz.nuark.predench.PredenchSeedUpdater;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.OptionalInt;
+import java.util.Random;
+
+@Mixin(EnchantmentMenu.class)
+public abstract class MixinEnchantmentMenu implements PredenchSeedUpdater, PredenchEnchantmentDataExtractor {
+    @Shadow protected abstract List<EnchantmentInstance> getEnchantmentList(ItemStack p_39472_, int p_39473_, int p_39474_);
+
+    public OptionalInt updatePlayerEnchantmentSeed(Player player) {
+        try {
+            player.onEnchantmentPerformed(null, 0);
+            int seed = player.getEnchantmentSeed();
+            enchantmentSeed.set(seed);
+            return OptionalInt.of(seed);
+        } catch (Exception e) {
+            Predench.LOGGER.error("Failed to update enchantment seed for player " + player.getName().getString(), e);
+        }
+        return OptionalInt.empty();
+    }
+
+    @Override
+    public Optional<EnchantmentData> getEnchantmentData(Level level, BlockPos tablePos, ItemStack itemstack) {
+        try {
+            int[] costs = new int[3];
+            float j = 0;
+
+            for(BlockPos blockpos : EnchantmentTableBlock.BOOKSHELF_OFFSETS) {
+                if (EnchantmentTableBlock.isValidBookShelf(level, tablePos, blockpos)) {
+                    j += level.getBlockState(tablePos.offset(blockpos)).getEnchantPowerBonus(level, tablePos.offset(blockpos));
+                }
+            }
+
+            Random random = new Random();
+            random.setSeed(this.enchantmentSeed.get());
+
+            for(int k = 0; k < 3; ++k) {
+                costs[k] = EnchantmentHelper.getEnchantmentCost(random, k, (int)j, itemstack);
+                if (costs[k] < k + 1) {
+                    costs[k] = 0;
+                }
+                costs[k] = net.minecraftforge.event.ForgeEventFactory.onEnchantmentLevelSet(level, tablePos, k, (int)j, itemstack, costs[k]);
+            }
+
+            EnchantmentData enchantmentData = new EnchantmentData();
+
+            for(int l = 0; l < 3; ++l) {
+                if (costs[l] > 0) {
+                    List<EnchantmentInstance> list = this.getEnchantmentList(itemstack, l, costs[l]);
+                    if (list != null && !list.isEmpty()) {
+                        for (EnchantmentInstance enchantmentInstance : list) {
+                            enchantmentData.addToTier(l, new EnchantmentData.Enchantment(
+                                    enchantmentInstance.enchantment.getFullname(enchantmentInstance.level).getString(),
+                                    enchantmentInstance.level,
+                                    costs[l]
+                            ));
+                        }
+                    }
+                }
+            }
+
+            Predench.LOGGER.info("Enchantment data: " + new Gson().toJson(enchantmentData));
+
+            return Optional.of(enchantmentData);
+        } catch (Exception e) {
+            Predench.LOGGER.error("Failed to get enchantment data for itemstack " + itemstack.toString(), e);
+        }
+
+        return Optional.empty();
+    }
+
+    @Final
+    @Shadow
+    private final DataSlot enchantmentSeed = DataSlot.standalone();
+}
+