From 6882fb4519491c8e28ba9d154e1cc35237e88d13 Mon Sep 17 00:00:00 2001
From: Electroblob77 <35599699+Electroblob77@users.noreply.github.com>
Date: Wed, 15 Apr 2020 23:34:41 +0100
Subject: [PATCH] Prevent imbuements from showing up on enchanted books in JEI

---
 .../wizardry/integration/jei/WizardryJEIPlugin.java  | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/main/java/electroblob/wizardry/integration/jei/WizardryJEIPlugin.java b/src/main/java/electroblob/wizardry/integration/jei/WizardryJEIPlugin.java
index a5b050f2..ad70fb64 100644
--- a/src/main/java/electroblob/wizardry/integration/jei/WizardryJEIPlugin.java
+++ b/src/main/java/electroblob/wizardry/integration/jei/WizardryJEIPlugin.java
@@ -1,6 +1,7 @@
 package electroblob.wizardry.integration.jei;
 
 import electroblob.wizardry.Wizardry;
+import electroblob.wizardry.enchantment.Imbuement;
 import electroblob.wizardry.item.IConjuredItem;
 import electroblob.wizardry.registry.WizardryBlocks;
 import mezz.jei.api.IModPlugin;
@@ -9,6 +10,8 @@ import mezz.jei.api.ISubtypeRegistry;
 import mezz.jei.api.JEIPlugin;
 import mezz.jei.api.ingredients.IIngredientBlacklist;
 import mezz.jei.api.recipe.IRecipeCategoryRegistration;
+import net.minecraft.enchantment.Enchantment;
+import net.minecraft.enchantment.EnchantmentData;
 import net.minecraft.item.Item;
 import net.minecraft.item.ItemStack;
 
@@ -41,6 +44,15 @@ public class WizardryJEIPlugin implements IModPlugin {
 			if(item instanceof IConjuredItem) blacklist.addIngredientToBlacklist(new ItemStack(item));
 		}
 
+		// Hide imbuements from JEI
+		for(Enchantment enchantment : Enchantment.REGISTRY){
+			if(enchantment instanceof Imbuement){
+				for(int level = enchantment.getMinLevel(); level <= enchantment.getMaxLevel(); level++){
+					blacklist.addIngredientToBlacklist(new EnchantmentData(enchantment, level));
+				}
+			}
+		}
+
 	}
 
 	@Override