Skip to content

Commit 43fab89

Browse files
committed
Fix ##520
1 parent f1ea3ca commit 43fab89

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/main/java/com/minelittlepony/unicopia/mixin/MixinEnchantmentHelper.java

+5-1
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,14 @@
88
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
99

1010
import com.minelittlepony.unicopia.entity.player.Pony;
11+
import com.minelittlepony.unicopia.item.enchantment.SimpleEnchantment;
1112

1213
import net.minecraft.enchantment.Enchantment;
1314
import net.minecraft.enchantment.EnchantmentHelper;
1415
import net.minecraft.enchantment.EnchantmentLevelEntry;
1516
import net.minecraft.entity.LivingEntity;
1617
import net.minecraft.item.ItemStack;
18+
import net.minecraft.item.Items;
1719

1820
@Mixin(EnchantmentHelper.class)
1921
abstract class MixinEnchantmentHelper {
@@ -30,6 +32,8 @@ private static void getEquipmentLevel(Enchantment enchantment, LivingEntity enti
3032

3133
@Inject(method = "getPossibleEntries", at = @At("RETURN"))
3234
private static void onGetPossibleEntries(int power, ItemStack stack, boolean treasureAllowed, CallbackInfoReturnable<List<EnchantmentLevelEntry>> info) {
33-
info.getReturnValue().removeIf(entry -> !entry.enchantment.isAcceptableItem(stack));
35+
if (!stack.isOf(Items.BOOK)) {
36+
info.getReturnValue().removeIf(entry -> entry.enchantment instanceof SimpleEnchantment && !entry.enchantment.isAcceptableItem(stack));
37+
}
3438
}
3539
}

0 commit comments

Comments
 (0)