PonyMirrors/Unicopia
1
0
Fork 0
mirror of https://github.com/Sollace/Unicopia.git synced 2024-12-18 07:57:59 +01:00
Code Issues Projects Releases Packages Wiki Activity

Fix ##520

Browse source
This commit is contained in:
Sollace 2024-12-10 23:10:36 +01:00
parent f1ea3cacb6
commit 43fab8916b
No known key found for this signature in database
GPG key ID: E52FACE7B5C773DB
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/main/java/com/minelittlepony/unicopia/mixin/MixinEnchantmentHelper.java
View file

@ -8,12 +8,14 @@ import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
import com.minelittlepony.unicopia.entity.player.Pony;
import com.minelittlepony.unicopia.item.enchantment.SimpleEnchantment;
import net.minecraft.enchantment.Enchantment;
import net.minecraft.enchantment.EnchantmentHelper;
import net.minecraft.enchantment.EnchantmentLevelEntry;
import net.minecraft.entity.LivingEntity;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
@Mixin(EnchantmentHelper.class)
abstract class MixinEnchantmentHelper {
@ -30,6 +32,8 @@ abstract class MixinEnchantmentHelper {
@Inject(method = "getPossibleEntries", at = @At("RETURN"))
private static void onGetPossibleEntries(int power, ItemStack stack, boolean treasureAllowed, CallbackInfoReturnable<List<EnchantmentLevelEntry>> info) {
info.getReturnValue().removeIf(entry -> !entry.enchantment.isAcceptableItem(stack));
if (!stack.isOf(Items.BOOK)) {
info.getReturnValue().removeIf(entry -> entry.enchantment instanceof SimpleEnchantment && !entry.enchantment.isAcceptableItem(stack));
}
}
}