package net.minecraft.client.gui.components;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.OptionInstance;
import net.minecraft.client.gui.Font;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.gui.narration.NarratedElementType;
import net.minecraft.client.gui.narration.NarrationElementOutput;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.ARGB;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class Checkbox extends AbstractButton {
	private static final ResourceLocation CHECKBOX_SELECTED_HIGHLIGHTED_SPRITE = ResourceLocation.withDefaultNamespace("widget/checkbox_selected_highlighted");
	private static final ResourceLocation CHECKBOX_SELECTED_SPRITE = ResourceLocation.withDefaultNamespace("widget/checkbox_selected");
	private static final ResourceLocation CHECKBOX_HIGHLIGHTED_SPRITE = ResourceLocation.withDefaultNamespace("widget/checkbox_highlighted");
	private static final ResourceLocation CHECKBOX_SPRITE = ResourceLocation.withDefaultNamespace("widget/checkbox");
	private static final int TEXT_COLOR = 14737632;
	private static final int SPACING = 4;
	private static final int BOX_PADDING = 8;
	private boolean selected;
	private final Checkbox.OnValueChange onValueChange;
	private final MultiLineTextWidget textWidget;

	Checkbox(int x, int y, int maxWidth, Component message, Font font, boolean selected, Checkbox.OnValueChange onValueChange) {
		super(x, y, 0, 0, message);
		this.width = this.getAdjustedWidth(maxWidth, message, font);
		this.textWidget = new MultiLineTextWidget(message, font).setMaxWidth(this.width).setColor(14737632);
		this.height = this.getAdjustedHeight(font);
		this.selected = selected;
		this.onValueChange = onValueChange;
	}

	private int getAdjustedWidth(int maxWidth, Component message, Font font) {
		return Math.min(getDefaultWidth(message, font), maxWidth);
	}

	private int getAdjustedHeight(Font font) {
		return Math.max(getBoxSize(font), this.textWidget.getHeight());
	}

	static int getDefaultWidth(Component message, Font font) {
		return getBoxSize(font) + 4 + font.width(message);
	}

	public static Checkbox.Builder builder(Component message, Font font) {
		return new Checkbox.Builder(message, font);
	}

	public static int getBoxSize(Font font) {
		return 9 + 8;
	}

	@Override
	public void onPress() {
		this.selected = !this.selected;
		this.onValueChange.onValueChange(this, this.selected);
	}

	public boolean selected() {
		return this.selected;
	}

	@Override
	public void updateWidgetNarration(NarrationElementOutput narrationElementOutput) {
		narrationElementOutput.add(NarratedElementType.TITLE, this.createNarrationMessage());
		if (this.active) {
			if (this.isFocused()) {
				narrationElementOutput.add(NarratedElementType.USAGE, Component.translatable("narration.checkbox.usage.focused"));
			} else {
				narrationElementOutput.add(NarratedElementType.USAGE, Component.translatable("narration.checkbox.usage.hovered"));
			}
		}
	}

	@Override
	public void renderWidget(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
		Minecraft minecraft = Minecraft.getInstance();
		Font font = minecraft.font;
		ResourceLocation resourceLocation;
		if (this.selected) {
			resourceLocation = this.isFocused() ? CHECKBOX_SELECTED_HIGHLIGHTED_SPRITE : CHECKBOX_SELECTED_SPRITE;
		} else {
			resourceLocation = this.isFocused() ? CHECKBOX_HIGHLIGHTED_SPRITE : CHECKBOX_SPRITE;
		}

		int i = getBoxSize(font);
		guiGraphics.blitSprite(RenderType::guiTextured, resourceLocation, this.getX(), this.getY(), i, i, ARGB.white(this.alpha));
		int j = this.getX() + i + 4;
		int k = this.getY() + i / 2 - this.textWidget.getHeight() / 2;
		this.textWidget.setPosition(j, k);
		this.textWidget.renderWidget(guiGraphics, mouseX, mouseY, partialTick);
	}

	@Environment(EnvType.CLIENT)
	public static class Builder {
		private final Component message;
		private final Font font;
		private int maxWidth;
		private int x = 0;
		private int y = 0;
		private Checkbox.OnValueChange onValueChange = Checkbox.OnValueChange.NOP;
		private boolean selected = false;
		@Nullable
		private OptionInstance<Boolean> option = null;
		@Nullable
		private Tooltip tooltip = null;

		Builder(Component message, Font font) {
			this.message = message;
			this.font = font;
			this.maxWidth = Checkbox.getDefaultWidth(message, font);
		}

		public Checkbox.Builder pos(int x, int y) {
			this.x = x;
			this.y = y;
			return this;
		}

		public Checkbox.Builder onValueChange(Checkbox.OnValueChange onValueChange) {
			this.onValueChange = onValueChange;
			return this;
		}

		public Checkbox.Builder selected(boolean selected) {
			this.selected = selected;
			this.option = null;
			return this;
		}

		public Checkbox.Builder selected(OptionInstance<Boolean> option) {
			this.option = option;
			this.selected = option.get();
			return this;
		}

		public Checkbox.Builder tooltip(Tooltip tooltip) {
			this.tooltip = tooltip;
			return this;
		}

		public Checkbox.Builder maxWidth(int maxWidth) {
			this.maxWidth = maxWidth;
			return this;
		}

		public Checkbox build() {
			Checkbox.OnValueChange onValueChange = this.option == null ? this.onValueChange : (checkboxx, bl) -> {
				this.option.set(bl);
				this.onValueChange.onValueChange(checkboxx, bl);
			};
			Checkbox checkbox = new Checkbox(this.x, this.y, this.maxWidth, this.message, this.font, this.selected, onValueChange);
			checkbox.setTooltip(this.tooltip);
			return checkbox;
		}
	}

	@Environment(EnvType.CLIENT)
	public interface OnValueChange {
		Checkbox.OnValueChange NOP = (checkbox, bl) -> {};

		void onValueChange(Checkbox checkbox, boolean bl);
	}
}