package net.minecraft.client.gui.components;

import net.minecraft.client.gui.ActiveTextCollector;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.client.gui.components.Button.CreateNarration;
import net.minecraft.client.gui.components.Button.OnPress;
import net.minecraft.client.input.InputWithModifiers;
import net.minecraft.client.renderer.RenderPipelines;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.util.ARGB;
import org.jspecify.annotations.Nullable;

public abstract class SpriteIconButton extends Button {
	private static final Identifier BUTTON_DISABLED_SPRITE = Identifier.withDefaultNamespace("widget/button_disabled");
	private static final Identifier LOADING_SPRITE = Identifier.withDefaultNamespace("friends/loading");
	private static final Tooltip LOADING_TOOLTIP = Tooltip.create(Component.translatable("gui.friends.button.loading"));
	private static final int LOADING_SPRITE_W = 5;
	private static final int LOADING_SPRITE_H = 2;
	protected final WidgetSprites sprite;
	protected final int spriteWidth;
	protected final int spriteHeight;
	protected final int spriteOffsetX;
	protected final int spriteOffsetY;
	private final boolean switchToLoadingAfterPress;
	@Nullable
	private final Tooltip defaultTooltip;
	private boolean loading;

	private SpriteIconButton(
		final int width,
		final int height,
		final Component message,
		final int spriteWidth,
		final int spriteHeight,
		final int spriteOffsetX,
		final int spriteOffsetY,
		final WidgetSprites sprite,
		final OnPress onPress,
		@Nullable final Component tooltip,
		@Nullable final CreateNarration narration,
		final boolean switchToLoadingAfterPress
	) {
		super(0, 0, width, height, message, onPress, narration == null ? DEFAULT_NARRATION : narration);
		this.defaultTooltip = tooltip != null ? Tooltip.create(tooltip) : null;
		this.setTooltip(this.defaultTooltip);
		this.spriteWidth = spriteWidth;
		this.spriteHeight = spriteHeight;
		this.spriteOffsetX = spriteOffsetX;
		this.spriteOffsetY = spriteOffsetY;
		this.sprite = sprite;
		this.switchToLoadingAfterPress = switchToLoadingAfterPress;
	}

	public void setLoading(final boolean loading) {
		this.setLoading(loading, LOADING_TOOLTIP);
	}

	public void setLoading(final boolean loading, final Tooltip loadingTooltip) {
		this.loading = loading;
		if (loading) {
			this.setTooltip(loadingTooltip);
		} else {
			this.setTooltip(this.defaultTooltip);
		}
	}

	@Override
	public boolean isActive() {
		return super.isActive() && !this.loading;
	}

	@Override
	public void onPress(final InputWithModifiers input) {
		if (this.switchToLoadingAfterPress) {
			this.setLoading(true);
		}

		super.onPress(input);
	}

	protected void extractSprite(final GuiGraphicsExtractor graphics, final int x, final int y) {
		graphics.blitSprite(
			RenderPipelines.GUI_TEXTURED, this.sprite.get(this.isActive(), this.isHoveredOrFocused()), x, y, this.spriteWidth, this.spriteHeight, this.alpha
		);
	}

	protected boolean extractLoadingStateIfLoading(final GuiGraphicsExtractor graphics) {
		if (!this.loading) {
			return false;
		} else {
			graphics.blitSprite(
				RenderPipelines.GUI_TEXTURED, BUTTON_DISABLED_SPRITE, this.getX(), this.getY(), this.getWidth(), this.getHeight(), ARGB.white(this.alpha)
			);
			graphics.blitSprite(
				RenderPipelines.GUI_TEXTURED,
				LOADING_SPRITE,
				this.getX() + (this.getWidth() - 5) / 2,
				this.getY() + (this.getHeight() - 2) / 2,
				5,
				2,
				ARGB.white(this.alpha)
			);
			return true;
		}
	}

	public static SpriteIconButton.Builder builder(final Component message, final OnPress onPress, final boolean iconOnly) {
		return new SpriteIconButton.Builder(message, onPress, iconOnly);
	}

	public static class Builder {
		private final Component message;
		private final OnPress onPress;
		private final boolean iconOnly;
		private int width = 150;
		private int height = 20;
		@Nullable
		private WidgetSprites sprite;
		private int spriteWidth;
		private int spriteHeight;
		private int spriteOffsetX;
		private int spriteOffsetY;
		@Nullable
		private Component tooltip;
		@Nullable
		private CreateNarration narration;
		private boolean switchToLoadingAfterPress;

		private Builder(final Component message, final OnPress onPress, final boolean iconOnly) {
			this.message = message;
			this.onPress = onPress;
			this.iconOnly = iconOnly;
		}

		public SpriteIconButton.Builder width(final int width) {
			this.width = width;
			return this;
		}

		public SpriteIconButton.Builder size(final int width, final int height) {
			this.width = width;
			this.height = height;
			return this;
		}

		public SpriteIconButton.Builder sprite(final Identifier sprite, final int spriteWidth, final int spriteHeight) {
			this.sprite = new WidgetSprites(sprite);
			this.spriteWidth = spriteWidth;
			this.spriteHeight = spriteHeight;
			return this;
		}

		public SpriteIconButton.Builder sprite(final WidgetSprites sprite, final int spriteWidth, final int spriteHeight) {
			this.sprite = sprite;
			this.spriteWidth = spriteWidth;
			this.spriteHeight = spriteHeight;
			return this;
		}

		public SpriteIconButton.Builder spriteOffset(final int spriteOffsetX, final int spriteOffsetY) {
			this.spriteOffsetX = spriteOffsetX;
			this.spriteOffsetY = spriteOffsetY;
			return this;
		}

		public SpriteIconButton.Builder tooltip(final Component tooltip) {
			this.tooltip = tooltip;
			return this;
		}

		public SpriteIconButton.Builder withTootip() {
			this.tooltip = this.message;
			return this;
		}

		public SpriteIconButton.Builder narration(final CreateNarration narration) {
			this.narration = narration;
			return this;
		}

		public SpriteIconButton.Builder switchToLoadingAfterPress() {
			this.switchToLoadingAfterPress = true;
			return this;
		}

		public SpriteIconButton build() {
			if (this.sprite == null) {
				throw new IllegalStateException("Sprite not set");
			} else {
				return (SpriteIconButton)(this.iconOnly
					? new SpriteIconButton.CenteredIcon(
						this.width,
						this.height,
						this.message,
						this.spriteWidth,
						this.spriteHeight,
						this.spriteOffsetX,
						this.spriteOffsetY,
						this.sprite,
						this.onPress,
						this.tooltip,
						this.narration,
						this.switchToLoadingAfterPress
					)
					: new SpriteIconButton.TextAndIcon(
						this.width,
						this.height,
						this.message,
						this.spriteWidth,
						this.spriteHeight,
						this.spriteOffsetX,
						this.spriteOffsetY,
						this.sprite,
						this.onPress,
						this.tooltip,
						this.narration,
						this.switchToLoadingAfterPress
					));
			}
		}
	}

	public static class CenteredIcon extends SpriteIconButton {
		protected CenteredIcon(
			final int width,
			final int height,
			final Component message,
			final int spriteWidth,
			final int spriteHeight,
			final int spriteOffsetX,
			final int spriteOffsetY,
			final WidgetSprites sprite,
			final OnPress onPress,
			@Nullable final Component tooltip,
			@Nullable final CreateNarration narration,
			final boolean switchToLoadingAfterPress
		) {
			super(width, height, message, spriteWidth, spriteHeight, spriteOffsetX, spriteOffsetY, sprite, onPress, tooltip, narration, switchToLoadingAfterPress);
		}

		@Override
		public void extractContents(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
			if (!this.extractLoadingStateIfLoading(graphics)) {
				this.extractDefaultSprite(graphics);
				int x = this.spriteOffsetX + this.getX() + this.getWidth() / 2 - this.spriteWidth / 2;
				int y = this.spriteOffsetY + this.getY() + this.getHeight() / 2 - this.spriteHeight / 2;
				this.extractSprite(graphics, x, y);
			}
		}
	}

	public static class TextAndIcon extends SpriteIconButton {
		protected TextAndIcon(
			final int width,
			final int height,
			final Component message,
			final int spriteWidth,
			final int spriteHeight,
			final int spriteOffsetX,
			final int spriteOffsetY,
			final WidgetSprites sprite,
			final OnPress onPress,
			@Nullable final Component tooltip,
			@Nullable final CreateNarration narration,
			final boolean switchToLoadingAfterPress
		) {
			super(width, height, message, spriteWidth, spriteHeight, spriteOffsetX, spriteOffsetY, sprite, onPress, tooltip, narration, switchToLoadingAfterPress);
		}

		@Override
		public void extractContents(final GuiGraphicsExtractor graphics, final int mouseX, final int mouseY, final float a) {
			if (!this.extractLoadingStateIfLoading(graphics)) {
				this.extractDefaultSprite(graphics);
				int left = this.getX() + 2;
				int right = this.getX() + this.getWidth() - this.spriteWidth - 4;
				int centerX = this.getX() + this.getWidth() / 2;
				ActiveTextCollector output = graphics.textRendererForWidget(this, GuiGraphicsExtractor.HoveredTextEffects.NONE);
				output.acceptScrolling(this.getMessage(), centerX, left, right, this.getY(), this.getY() + this.getHeight());
				int x = this.spriteOffsetX + this.getX() + this.getWidth() - this.spriteWidth - 2;
				int y = this.spriteOffsetY + this.getY() + this.getHeight() / 2 - this.spriteHeight / 2;
				this.extractSprite(graphics, x, y);
			}
		}
	}
}