package net.radsteve.islanders.mixin;

import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.screen.AnvilScreenHandler;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

@Mixin(AnvilScreenHandler.class)
public class AnvilScreenHandlerMixin {
  @Inject(method = "onTakeOutput", at = @At("HEAD"), cancellable = true)
  private void islanders$onTakeOutput(PlayerEntity player, ItemStack stack, CallbackInfo ci) {
    if (stack.isOf(Items.POPPED_CHORUS_FRUIT) || stack.isOf(Items.MACE)) {
      ci.cancel();
    }
  }
}