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.gui.ComponentPath;
import net.minecraft.client.gui.narration.NarratedElementType;
import net.minecraft.client.gui.narration.NarrationElementOutput;
import net.minecraft.client.gui.narration.NarrationSupplier;
import net.minecraft.client.gui.navigation.FocusNavigationEvent;
import net.minecraft.client.input.MouseButtonEvent;
import net.minecraft.network.chat.Component;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public abstract class ObjectSelectionList<E extends ObjectSelectionList.Entry<E>> extends AbstractSelectionList<E> {
	private static final Component USAGE_NARRATION = Component.translatable("narration.selection.usage");

	public ObjectSelectionList(final Minecraft minecraft, final int width, final int height, final int y, final int itemHeight) {
		super(minecraft, width, height, y, itemHeight);
	}

	@Nullable
	@Override
	public ComponentPath nextFocusPath(final FocusNavigationEvent navigationEvent) {
		if (this.getItemCount() == 0) {
			return null;
		} else if (this.isFocused() && navigationEvent instanceof FocusNavigationEvent.ArrowNavigation arrowNavigation) {
			E entry = this.nextEntry(arrowNavigation.direction());
			if (entry != null) {
				return ComponentPath.path(this, ComponentPath.leaf(entry));
			} else {
				this.setFocused(null);
				this.setSelected(null);
				return null;
			}
		} else if (!this.isFocused()) {
			E entry = this.getSelected();
			if (entry == null) {
				entry = this.nextEntry(navigationEvent.getVerticalDirectionForInitialFocus());
			}

			return entry == null ? null : ComponentPath.path(this, ComponentPath.leaf(entry));
		} else {
			return null;
		}
	}

	@Override
	public void updateWidgetNarration(final NarrationElementOutput output) {
		E hovered = this.getHovered();
		if (hovered != null) {
			this.narrateListElementPosition(output.nest(), hovered);
			hovered.updateNarration(output);
		} else {
			E selected = this.getSelected();
			if (selected != null) {
				this.narrateListElementPosition(output.nest(), selected);
				selected.updateNarration(output);
			}
		}

		if (this.isFocused()) {
			output.add(NarratedElementType.USAGE, USAGE_NARRATION);
		}
	}

	@Environment(EnvType.CLIENT)
	public abstract static class Entry<E extends ObjectSelectionList.Entry<E>> extends AbstractSelectionList.Entry<E> implements NarrationSupplier {
		public abstract Component getNarration();

		@Override
		public boolean mouseClicked(final MouseButtonEvent event, final boolean doubleClick) {
			return true;
		}

		@Override
		public void updateNarration(final NarrationElementOutput output) {
			output.add(NarratedElementType.TITLE, this.getNarration());
		}
	}
}