Unicopia/src/main/java/com/minelittlepony/unicopia/mixin/MixinBlockItem.java

package com.minelittlepony.unicopia.mixin;

import org.spongepowered.asm.mixin.Mixin;

import com.minelittlepony.unicopia.item.toxin.ToxicHolder;

import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.item.BlockItem;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.util.Hand;
import net.minecraft.util.TypedActionResult;
import net.minecraft.util.UseAction;
import net.minecraft.world.World;

@Mixin(BlockItem.class)
abstract class MixinBlockItem extends Item implements ToxicHolder {
    MixinBlockItem() {super(null); }

    @Override
    public UseAction getUseAction(ItemStack stack) {
        return getToxic(stack).useAction().orElseGet(() -> super.getUseAction(stack));
    }

    @Override
    public TypedActionResult<ItemStack> use(World world, PlayerEntity player, Hand hand) {
        return getToxic(player.getStackInHand(hand)).ailment().get(player)
                .map(t -> t.use(world, player, hand))
                .orElseGet(() -> super.use(world, player, hand));
    }
}