Add module AutoEnchant (#314)

README.md

@@ -39,6 +39,7 @@
 - ArrowDmg (Ported from [Wurst](https://github.com/Wurst-Imperium/Wurst7/tree))
 - AutoBedTrap (Ported from [BleachHack-CupEdition](https://github.com/CUPZYY/BleachHack-CupEdition/blob/master/CupEdition-1.17/src/main/java/bleach/hack/module/mods/AutoBedtrap.java))
 - AutoCraft (More generalized version of [AutoBedCraft](https://github.com/Anticope/orion/blob/main/src/main/java/me/ghosttypes/orion/modules/main/AutoBedCraft.java) from orion)
+- AutoEnchant
 - AutoExtinguish
 - AutoFarm
 - AutoGrind

src/main/java/anticope/rejects/MeteorRejectsAddon.java

@@ -83,6 +83,7 @@ public void onInitialize() {
         modules.add(new SoundLocator());
         modules.add(new TreeAura());
         modules.add(new VehicleOneHit());
+        modules.add(new AutoEnchant());
 
         // Commands
         Commands.add(new CenterCommand());

src/main/java/anticope/rejects/modules/AutoEnchant.java

@@ -0,0 +1,135 @@
+package anticope.rejects.modules;
+
+import anticope.rejects.MeteorRejectsAddon;
+
+import meteordevelopment.meteorclient.events.game.OpenScreenEvent;
+import meteordevelopment.meteorclient.settings.*;
+import meteordevelopment.meteorclient.utils.network.MeteorExecutor;
+import meteordevelopment.meteorclient.utils.player.FindItemResult;
+import meteordevelopment.meteorclient.utils.player.InvUtils;
+import meteordevelopment.orbit.EventHandler;
+import net.minecraft.enchantment.EnchantmentHelper;
+import net.minecraft.item.Item;
+import net.minecraft.item.Items;
+import net.minecraft.screen.EnchantmentScreenHandler;
+import net.minecraft.screen.ScreenHandler;
+
+import java.util.List;
+import java.util.Objects;
+
+public class AutoEnchant extends meteordevelopment.meteorclient.systems.modules.Module {
+
+    public final SettingGroup sgGeneral = settings.getDefaultGroup();
+
+    private final Setting<Integer> delay = sgGeneral.add(new IntSetting.Builder()
+            .name("delay")
+            .description("The tick delay between enchanting items.")
+            .defaultValue(50)
+            .sliderMax(500)
+            .min(0)
+            .build()
+    );
+
+    private final Setting<Integer> level = sgGeneral.add(new IntSetting.Builder()
+            .name("level")
+            .description("Choose enchantment levels 1-3")
+            .defaultValue(3)
+            .max(3)
+            .min(1)
+            .build()
+    );
+
+    private final Setting<Boolean> drop = sgGeneral.add(new BoolSetting.Builder()
+            .name("drop")
+            .description("Automatically drops enchanted items (useful for when not enough inventory space)")
+            .defaultValue(false)
+            .build()
+    );
+
+    private final Setting<List<Item>> itemWhitelist = sgGeneral.add(new ItemListSetting.Builder()
+            .name("item-whitelist")
+            .description("Item that require enchantment.")
+            .defaultValue()
+            .filter(item -> item.equals(Items.BOOK) || item.isDamageable())
+            .build()
+    );
+
+    public AutoEnchant() {
+        super(MeteorRejectsAddon.CATEGORY, "auto-enchant", "Automatically enchanting items.");
+    }
+
+    @EventHandler
+    private void onOpenScreen(OpenScreenEvent event) {
+        if (!(Objects.requireNonNull(mc.player).currentScreenHandler instanceof EnchantmentScreenHandler))
+            return;
+        MeteorExecutor.execute(this::autoEnchant);
+    }
+
+    private void autoEnchant() {
+        if (!(Objects.requireNonNull(mc.player).currentScreenHandler instanceof EnchantmentScreenHandler handler))
+            return;
+        if (mc.player.experienceLevel < 30) {
+            info("You don't have enough experience levels");
+            return;
+        }
+        while (getEmptySlotCount(handler) > 2 || drop.get()) {
+            if (!(mc.player.currentScreenHandler instanceof EnchantmentScreenHandler)) {
+                info("Enchanting table is closed.");
+                break;
+            }
+            if (handler.getLapisCount() < level.get() && !fillLapisItem()) {
+                info("Lapis lazuli is not found.");
+                break;
+            }
+            if (!fillCanEnchantItem()) {
+                info("No items found to enchant.");
+                break;
+            }
+            Objects.requireNonNull(mc.interactionManager).clickButton(handler.syncId, level.get() - 1);
+            if (getEmptySlotCount(handler) > 2) {
+                InvUtils.shiftClick().slotId(0);
+            } else if (drop.get() && handler.getSlot(0).hasStack()) {
+                // I don't know why an exception LegacyRandomSource is thrown here,
+                // so I used the main thread to drop items.
+                mc.execute(() -> InvUtils.drop().slotId(0));
+            }
+
+            /*
+            Although the description here indicates that the tick is the unit,
+            the actual delay is not the tick unit,
+            but it does not affect the normal operation in the game.
+            Perhaps we can ignore it
+            */
+            try {
+                Thread.sleep(delay.get());
+            } catch (InterruptedException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
+    private boolean fillCanEnchantItem() {
+        FindItemResult res = InvUtils.find(stack -> itemWhitelist.get().contains(stack.getItem()) && EnchantmentHelper.get(stack).isEmpty());
+        if (!res.found()) return false;
+        InvUtils.shiftClick().slot(res.slot());
+        return true;
+    }
+
+    private boolean fillLapisItem() {
+        FindItemResult res = InvUtils.find(Items.LAPIS_LAZULI);
+        if (!res.found()) return false;
+        InvUtils.shiftClick().slot(res.slot());
+        return true;
+    }
+
+    private int getEmptySlotCount(ScreenHandler handler) {
+        int emptySlotCount = 0;
+        for (int i = 0; i < handler.slots.size(); i++) {
+            if (!handler.slots.get(i).getStack().getItem().equals(Items.AIR))
+                continue;
+            emptySlotCount++;
+        }
+        return emptySlotCount;
+    }
+
+}