package net.minecraft.client.gui.components.debug;

import java.util.ArrayList;
import java.util.List;
import net.minecraft.ChatFormatting;
import net.minecraft.SharedConstants;
import net.minecraft.client.Minecraft;
import net.minecraft.core.BlockPos;
import net.minecraft.core.TypedInstance;
import net.minecraft.resources.Identifier;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.StateHolder;
import net.minecraft.world.level.block.state.properties.Property.Value;
import net.minecraft.world.level.chunk.LevelChunk;
import net.minecraft.world.level.material.Fluid;
import net.minecraft.world.level.material.FluidState;
import net.minecraft.world.phys.BlockHitResult;
import net.minecraft.world.phys.HitResult;
import net.minecraft.world.phys.HitResult.Type;
import org.jspecify.annotations.Nullable;

public abstract class DebugEntryLookingAt implements DebugScreenEntry {
	private static final int RANGE = 20;
	private static final Identifier BLOCK_GROUP = Identifier.withDefaultNamespace("looking_at_block");
	private static final Identifier FLUID_GROUP = Identifier.withDefaultNamespace("looking_at_fluid");

	@Override
	public void display(
		final DebugScreenDisplayer displayer,
		@Nullable final Level serverOrClientLevel,
		@Nullable final LevelChunk clientChunk,
		@Nullable final LevelChunk serverChunk
	) {
		Entity cameraEntity = Minecraft.getInstance().getCameraEntity();
		Level clientOrServerLevel = (Level)(SharedConstants.DEBUG_SHOW_SERVER_DEBUG_VALUES ? serverOrClientLevel : Minecraft.getInstance().level);
		if (cameraEntity != null && clientOrServerLevel != null) {
			HitResult block = this.getHitResult(cameraEntity);
			List<String> result = new ArrayList();
			if (block.getType() == Type.BLOCK) {
				BlockPos pos = ((BlockHitResult)block).getBlockPos();
				this.extractInfo(result, clientOrServerLevel, pos);
			}

			displayer.addToGroup(this.group(), result);
		}
	}

	public abstract HitResult getHitResult(final Entity cameraEntity);

	public abstract void extractInfo(List<String> result, Level level, BlockPos pos);

	public abstract Identifier group();

	public static void addTagEntries(final List<String> result, final TypedInstance<?> instance) {
		instance.tags().map(e -> "#" + e.location()).forEach(result::add);
	}

	public static class BlockStateInfo extends DebugEntryLookingAt.DebugEntryLookingAtState<Block, BlockState> {
		protected BlockStateInfo() {
			super("Targeted Block");
		}

		@Override
		public HitResult getHitResult(final Entity cameraEntity) {
			return cameraEntity.pick(20.0, 0.0F, false);
		}

		public BlockState getInstance(final Level level, final BlockPos pos) {
			return level.getBlockState(pos);
		}

		@Override
		public Identifier group() {
			return DebugEntryLookingAt.BLOCK_GROUP;
		}
	}

	public static class BlockTagInfo extends DebugEntryLookingAt.DebugEntryLookingAtTags<BlockState> {
		@Override
		public HitResult getHitResult(final Entity cameraEntity) {
			return cameraEntity.pick(20.0, 0.0F, false);
		}

		public BlockState getInstance(final Level level, final BlockPos pos) {
			return level.getBlockState(pos);
		}

		@Override
		public Identifier group() {
			return DebugEntryLookingAt.BLOCK_GROUP;
		}
	}

	public abstract static class DebugEntryLookingAtState<OwnerType, StateType extends StateHolder<OwnerType, StateType> & TypedInstance<OwnerType>>
		extends DebugEntryLookingAt {
		private final String prefix;

		protected DebugEntryLookingAtState(final String prefix) {
			this.prefix = prefix;
		}

		protected abstract StateType getInstance(Level level, BlockPos pos);

		@Override
		public void extractInfo(final List<String> result, final Level level, final BlockPos pos) {
			StateType stateInstance = this.getInstance(level, pos);
			result.add(ChatFormatting.UNDERLINE + this.prefix + ": " + pos.getX() + ", " + pos.getY() + ", " + pos.getZ());
			result.add(stateInstance.typeHolder().getRegisteredName());
			addStateProperties(result, stateInstance);
		}

		private static void addStateProperties(final List<String> result, final StateHolder<?, ?> stateHolder) {
			stateHolder.getValues().forEach(entry -> result.add(getPropertyValueString(entry)));
		}

		private static String getPropertyValueString(final Value<?> entry) {
			String valueString = entry.valueName();
			if (Boolean.TRUE.equals(entry.value())) {
				valueString = ChatFormatting.GREEN + valueString;
			} else if (Boolean.FALSE.equals(entry.value())) {
				valueString = ChatFormatting.RED + valueString;
			}

			return entry.property().getName() + ": " + valueString;
		}
	}

	public abstract static class DebugEntryLookingAtTags<T extends TypedInstance<?>> extends DebugEntryLookingAt {
		protected abstract T getInstance(Level level, BlockPos pos);

		@Override
		public void extractInfo(final List<String> result, final Level level, final BlockPos pos) {
			T instance = this.getInstance(level, pos);
			addTagEntries(result, instance);
		}
	}

	public static class FluidStateInfo extends DebugEntryLookingAt.DebugEntryLookingAtState<Fluid, FluidState> {
		protected FluidStateInfo() {
			super("Targeted Fluid");
		}

		@Override
		public HitResult getHitResult(final Entity cameraEntity) {
			return cameraEntity.pick(20.0, 0.0F, true);
		}

		public FluidState getInstance(final Level level, final BlockPos pos) {
			return level.getFluidState(pos);
		}

		@Override
		public Identifier group() {
			return DebugEntryLookingAt.FLUID_GROUP;
		}
	}

	public static class FluidTagInfo extends DebugEntryLookingAt.DebugEntryLookingAtTags<FluidState> {
		@Override
		public HitResult getHitResult(final Entity cameraEntity) {
			return cameraEntity.pick(20.0, 0.0F, true);
		}

		public FluidState getInstance(final Level level, final BlockPos pos) {
			return level.getFluidState(pos);
		}

		@Override
		public Identifier group() {
			return DebugEntryLookingAt.FLUID_GROUP;
		}
	}
}