package net.minecraft.client.gui.screens.recipebook;

import com.mojang.blaze3d.systems.RenderSystem;
import java.util.List;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.ClientRecipeBook;
import net.minecraft.client.Minecraft;
import net.minecraft.client.RecipeBookCategories;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.components.StateSwitchingButton;
import net.minecraft.client.gui.components.WidgetSprites;
import net.minecraft.client.renderer.entity.ItemRenderer;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.inventory.RecipeBookMenu;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.crafting.RecipeHolder;

@Environment(EnvType.CLIENT)
public class RecipeBookTabButton extends StateSwitchingButton {
	private static final WidgetSprites SPRITES = new WidgetSprites(
		ResourceLocation.withDefaultNamespace("recipe_book/tab"), ResourceLocation.withDefaultNamespace("recipe_book/tab_selected")
	);
	private final RecipeBookCategories category;
	private static final float ANIMATION_TIME = 15.0F;
	private float animationTime;

	public RecipeBookTabButton(RecipeBookCategories category) {
		super(0, 0, 35, 27, false);
		this.category = category;
		this.initTextureValues(SPRITES);
	}

	public void startAnimation(Minecraft minecraft) {
		ClientRecipeBook clientRecipeBook = minecraft.player.getRecipeBook();
		List<RecipeCollection> list = clientRecipeBook.getCollection(this.category);
		if (minecraft.player.containerMenu instanceof RecipeBookMenu) {
			for (RecipeCollection recipeCollection : list) {
				for (RecipeHolder<?> recipeHolder : recipeCollection.getRecipes(clientRecipeBook.isFiltering((RecipeBookMenu<?, ?>)minecraft.player.containerMenu))) {
					if (clientRecipeBook.willHighlight(recipeHolder)) {
						this.animationTime = 15.0F;
						return;
					}
				}
			}
		}
	}

	@Override
	public void renderWidget(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		if (this.sprites != null) {
			if (this.animationTime > 0.0F) {
				float f = 1.0F + 0.1F * (float)Math.sin(this.animationTime / 15.0F * (float) Math.PI);
				guiGraphics.pose().pushPose();
				guiGraphics.pose().translate((float)(this.getX() + 8), (float)(this.getY() + 12), 0.0F);
				guiGraphics.pose().scale(1.0F, f, 1.0F);
				guiGraphics.pose().translate((float)(-(this.getX() + 8)), (float)(-(this.getY() + 12)), 0.0F);
			}

			Minecraft minecraft = Minecraft.getInstance();
			RenderSystem.disableDepthTest();
			ResourceLocation resourceLocation = this.sprites.get(true, this.isStateTriggered);
			int i = this.getX();
			if (this.isStateTriggered) {
				i -= 2;
			}

			guiGraphics.blitSprite(resourceLocation, i, this.getY(), this.width, this.height);
			RenderSystem.enableDepthTest();
			this.renderIcon(guiGraphics, minecraft.getItemRenderer());
			if (this.animationTime > 0.0F) {
				guiGraphics.pose().popPose();
				this.animationTime -= partialTick;
			}
		}
	}

	private void renderIcon(GuiGraphics guiGraphics, ItemRenderer itemRenderer) {
		List<ItemStack> list = this.category.getIconItems();
		int i = this.isStateTriggered ? -2 : 0;
		if (list.size() == 1) {
			guiGraphics.renderFakeItem((ItemStack)list.get(0), this.getX() + 9 + i, this.getY() + 5);
		} else if (list.size() == 2) {
			guiGraphics.renderFakeItem((ItemStack)list.get(0), this.getX() + 3 + i, this.getY() + 5);
			guiGraphics.renderFakeItem((ItemStack)list.get(1), this.getX() + 14 + i, this.getY() + 5);
		}
	}

	public RecipeBookCategories getCategory() {
		return this.category;
	}

	public boolean updateVisibility(ClientRecipeBook recipeBook) {
		List<RecipeCollection> list = recipeBook.getCollection(this.category);
		this.visible = false;
		if (list != null) {
			for (RecipeCollection recipeCollection : list) {
				if (recipeCollection.hasKnownRecipes() && recipeCollection.hasFitting()) {
					this.visible = true;
					break;
				}
			}
		}

		return this.visible;
	}
}