package com.kasetoatz.jukeboxboat.mixin;

import com.llamalad7.mixinextras.injector.ModifyReturnValue;
import com.llamalad7.mixinextras.sugar.Local;
import net.minecraft.component.DataComponentTypes;
import net.minecraft.item.ItemStack;
import net.minecraft.recipe.ShapelessRecipe;
import net.minecraft.recipe.input.CraftingRecipeInput;
import net.minecraft.text.Text;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;

@Mixin(ShapelessRecipe.class)
public abstract class ShapelessRecipeMixin
{
    @Shadow public abstract String getGroup();

    @ModifyReturnValue(method="craft(Lnet/minecraft/recipe/input/CraftingRecipeInput;Lnet/minecraft/registry/RegistryWrapper$WrapperLookup;)Lnet/minecraft/item/ItemStack;", at=@At("RETURN"))
    public ItemStack craft(ItemStack original, @Local(argsOnly = true) CraftingRecipeInput input)
    {
        if (!original.isEmpty() && input.getStacks().stream().anyMatch(stack -> stack.getName().getString().equals("Boat with Jukebox")))
        {
            return ItemStack.EMPTY;
        }
        if (getGroup().equals("jukebox_boat"))
        {
            original.set(DataComponentTypes.ITEM_NAME, Text.of("Boat with Jukebox"));
        }
        return original;
    }
}