feat: make despoil enchantment applicable to items with the knives tag

Elenterius · Jun 16, 2023 · 69d8ca5 · 69d8ca5
1 parent ffaee5e
commit 69d8ca5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/com/github/elenterius/biomancy/init/ModEnchantments.java b/src/main/java/com/github/elenterius/biomancy/init/ModEnchantments.java
@@ -3,6 +3,7 @@
 import com.github.elenterius.biomancy.BiomancyMod;
 import com.github.elenterius.biomancy.enchantment.AnestheticEnchantment;
 import com.github.elenterius.biomancy.enchantment.DespoilEnchantment;
+import com.github.elenterius.biomancy.init.tags.ModItemTags;
 import com.github.elenterius.biomancy.item.BioExtractorItem;
 import com.github.elenterius.biomancy.item.InjectorItem;
 import com.github.elenterius.biomancy.item.weapon.ClawsItem;
@@ -19,11 +20,11 @@ public final class ModEnchantments {
 
 	public static final EnchantmentCategory SYRINGE_CATEGORY = EnchantmentCategory.create("biomancy_syringe", item -> item instanceof BioExtractorItem || item instanceof InjectorItem);
 	public static final EnchantmentCategory GUN_CATEGORY = EnchantmentCategory.create("biomancy_gun", IGun.class::isInstance);
-	public static final EnchantmentCategory WEAPON_CATEGORY = EnchantmentCategory.create("biomancy_weapon", item -> EnchantmentCategory.WEAPON.canEnchant(item) || item instanceof ClawsItem || item instanceof AxeItem);
+	public static final EnchantmentCategory WEAPON_CATEGORY = EnchantmentCategory.create("biomancy_weapon", item -> EnchantmentCategory.WEAPON.canEnchant(item) || item instanceof ClawsItem || item instanceof AxeItem || item.builtInRegistryHolder().is(ModItemTags.FORGE_TOOLS_KNIVES));
 
 	public static final DeferredRegister<Enchantment> ENCHANTMENTS = DeferredRegister.create(ForgeRegistries.ENCHANTMENTS, BiomancyMod.MOD_ID);
 
-	public static final RegistryObject<DespoilEnchantment> DESPOIL = ENCHANTMENTS.register("despoil", () -> new DespoilEnchantment(Enchantment.Rarity.RARE, WEAPON_CATEGORY, EquipmentSlot.MAINHAND));
+	public static final RegistryObject<DespoilEnchantment> DESPOIL = ENCHANTMENTS.register("despoil", () -> new DespoilEnchantment(Enchantment.Rarity.RARE, WEAPON_CATEGORY, EquipmentSlot.MAINHAND, EquipmentSlot.OFFHAND));
 	public static final RegistryObject<AnestheticEnchantment> ANESTHETIC = ENCHANTMENTS.register("anesthetic", () -> new AnestheticEnchantment(Enchantment.Rarity.RARE, SYRINGE_CATEGORY, EquipmentSlot.MAINHAND));
 	//	public static final RegistryObject<QuickShotEnchantment> QUICK_SHOT = ENCHANTMENTS.register("quick_shot", () -> new QuickShotEnchantment(Enchantment.Rarity.VERY_RARE, EquipmentSlot.MAINHAND));
 	//	public static final RegistryObject<MaxAmmoEnchantment> MAX_AMMO = ENCHANTMENTS.register("max_ammo", () -> new MaxAmmoEnchantment(Enchantment.Rarity.VERY_RARE, EquipmentSlot.MAINHAND));