auto hit swap

Author: AutumnVN

src/client/java/autumnvn/autumn/Options.java

@@ -52,7 +52,7 @@ public Options() {
         options.put("autoAttack", autoAttack);
         ignorePlayer = SimpleOption.ofBoolean("Ignore Player", value -> Tooltip.of(Text.of("Auto Attack will ignore player")), false);
         options.put("ignorePlayer", ignorePlayer);
-        autoHitSwap = SimpleOption.ofBoolean("Auto Hit Swap", value -> Tooltip.of(Text.of("Swap item in main hand to another item in hotbar when attacking, then swap back after attack so it uses base damage & cooldown of original item and enchantments & ability to disable shield of swapped item\nPriority: axe when target is using shield, enchanted sword")), true);
+        autoHitSwap = SimpleOption.ofBoolean("Auto Hit Swap", value -> Tooltip.of(Text.of("Swap item in main hand to another item in hotbar when attacking, then swap back after attack so it uses base damage & cooldown of original item and enchantments & ability to disable shield of swapped item\nPriority:\nAxe if target is using shield\n\nSmite sword if target is undead\nBane of arthropods sword if target is arthropod\nImpaling trident if target is aquatic\nEnchanted sword\nNon-weapon if holding unenchanted sword/non-sharpness axe/other weapons")), true);
         options.put("autoHitSwap", autoHitSwap);
         autoSprint = SimpleOption.ofBoolean("Auto Sprint", value -> Tooltip.of(Text.of("Automatically sprint when moving forward")), true);
         options.put("autoSprint", autoSprint);

src/client/java/autumnvn/autumn/mixin/client/ClientPlayerInteractionManagerMixin.java

@@ -3,7 +3,11 @@
 import autumnvn.autumn.AutumnClient;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.network.ClientPlayerInteractionManager;
+import net.minecraft.enchantment.EnchantmentHelper;
+import net.minecraft.enchantment.Enchantments;
 import net.minecraft.entity.Entity;
+import net.minecraft.entity.EntityGroup;
+import net.minecraft.entity.LivingEntity;
 import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.item.*;
 import org.spongepowered.asm.mixin.Final;
@@ -53,13 +57,34 @@ private void attackEntity2(PlayerEntity player, Entity target, CallbackInfo ci)
             slot = player.getInventory().selectedSlot;
             ItemStack stack = player.getMainHandStack();
             Item item = stack.getItem();
-            boolean targetIsUsingShield = target instanceof PlayerEntity playerEntity && playerEntity.getActiveItem().isOf(Items.SHIELD);
-            if (!(item instanceof AxeItem) && targetIsUsingShield && getAxeHotbarSlot(player) != -1) {
+
+            if (getAxeHotbarSlot(player) != -1 && target instanceof PlayerEntity playerEntity && playerEntity.getActiveItem().isOf(Items.SHIELD)) {
                 player.getInventory().selectedSlot = getAxeHotbarSlot(player);
-            } else if (!(item instanceof AxeItem) || !targetIsUsingShield) {
-                if (!(item instanceof SwordItem && !stack.getEnchantments().isEmpty()) && getEnchantedSwordHotbarSlot(player) != -1) {
-                    player.getInventory().selectedSlot = getEnchantedSwordHotbarSlot(player);
-                }
+                return;
+            }
+
+            if (getSmiteSwordHotbarSlot(player) != -1 && target instanceof LivingEntity livingEntity && livingEntity.getGroup() == EntityGroup.UNDEAD) {
+                player.getInventory().selectedSlot = getSmiteSwordHotbarSlot(player);
+                return;
+            }
+
+            if (getBaneOfArthropodsSwordHotbarSlot(player) != -1 && target instanceof LivingEntity livingEntity && livingEntity.getGroup() == EntityGroup.ARTHROPOD) {
+                player.getInventory().selectedSlot = getBaneOfArthropodsSwordHotbarSlot(player);
+                return;
+            }
+
+            if (getImpalingTridentHotbarSlot(player) != -1 && target instanceof LivingEntity livingEntity && livingEntity.getGroup() == EntityGroup.AQUATIC) {
+                player.getInventory().selectedSlot = getImpalingTridentHotbarSlot(player);
+                return;
+            }
+
+            if (getEnchantedSwordHotbarSlot(player) != -1) {
+                player.getInventory().selectedSlot = getEnchantedSwordHotbarSlot(player);
+                return;
+            }
+
+            if (getNonWeaponHotbarSlot(player) != -1 && ((item instanceof SwordItem && stack.getEnchantments().isEmpty()) || (item instanceof AxeItem && EnchantmentHelper.getLevel(Enchantments.SHARPNESS, stack) > 0) || item instanceof PickaxeItem || item instanceof ShovelItem || item instanceof HoeItem || item instanceof TridentItem)) {
+                player.getInventory().selectedSlot = getNonWeaponHotbarSlot(player);
             }
         }
     }
@@ -83,6 +108,42 @@ int getAxeHotbarSlot(PlayerEntity player) {
         return -1;
     }
 
+    @Unique
+    int getSmiteSwordHotbarSlot(PlayerEntity player) {
+        for (int i = 0; i < 9; i++) {
+            ItemStack stack = player.getInventory().getStack(i);
+            Item item = stack.getItem();
+            if (item instanceof SwordItem && EnchantmentHelper.getLevel(Enchantments.SMITE, stack) > 0) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
+    @Unique
+    int getBaneOfArthropodsSwordHotbarSlot(PlayerEntity player) {
+        for (int i = 0; i < 9; i++) {
+            ItemStack stack = player.getInventory().getStack(i);
+            Item item = stack.getItem();
+            if (item instanceof SwordItem && EnchantmentHelper.getLevel(Enchantments.BANE_OF_ARTHROPODS, stack) > 0) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
+    @Unique
+    int getImpalingTridentHotbarSlot(PlayerEntity player) {
+        for (int i = 0; i < 9; i++) {
+            ItemStack stack = player.getInventory().getStack(i);
+            Item item = stack.getItem();
+            if (item instanceof TridentItem && EnchantmentHelper.getLevel(Enchantments.IMPALING, stack) > 0) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
     @Unique
     int getEnchantedSwordHotbarSlot(PlayerEntity player) {
         for (int i = 0; i < 9; i++) {
@@ -94,4 +155,16 @@ int getEnchantedSwordHotbarSlot(PlayerEntity player) {
         }
         return -1;
     }
+
+    @Unique
+    int getNonWeaponHotbarSlot(PlayerEntity player) {
+        for (int i = 0; i < 9; i++) {
+            ItemStack stack = player.getInventory().getStack(i);
+            Item item = stack.getItem();
+            if (!(item instanceof SwordItem) && !(item instanceof AxeItem) && !(item instanceof PickaxeItem) && !(item instanceof ShovelItem) && !(item instanceof HoeItem) && !(item instanceof TridentItem)) {
+                return i;
+            }
+        }
+        return -1;
+    }
 }