package net.minecraft.gizmos;

import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.phys.AABB;
import net.minecraft.world.phys.Vec3;
import org.jspecify.annotations.Nullable;

public class Gizmos {
	private static final ThreadLocal<GizmoCollector> collector = new ThreadLocal();

	private Gizmos() {
	}

	public static Gizmos.TemporaryCollection withCollector(final GizmoCollector collector) {
		Gizmos.TemporaryCollection result = new Gizmos.TemporaryCollection();
		Gizmos.collector.set(collector);
		return result;
	}

	public static GizmoProperties addGizmo(final Gizmo gizmo) {
		GizmoCollector collector = (GizmoCollector)Gizmos.collector.get();
		if (collector == null) {
			throw new IllegalStateException("Gizmos cannot be created here! No GizmoCollector has been registered.");
		} else {
			return collector.add(gizmo);
		}
	}

	public static GizmoProperties cuboid(final AABB aabb, final GizmoStyle style) {
		return cuboid(aabb, style, false);
	}

	public static GizmoProperties cuboid(final AABB aabb, final GizmoStyle style, final boolean coloredCorner) {
		return addGizmo(new CuboidGizmo(aabb, style, coloredCorner));
	}

	public static GizmoProperties cuboid(final BlockPos blockPos, final GizmoStyle style) {
		return cuboid(new AABB(blockPos), style);
	}

	public static GizmoProperties cuboid(final BlockPos blockPos, final float padding, final GizmoStyle style) {
		return cuboid(new AABB(blockPos).inflate(padding), style);
	}

	public static GizmoProperties circle(final Vec3 pos, final float radius, final GizmoStyle style) {
		return addGizmo(new CircleGizmo(pos, radius, style));
	}

	public static GizmoProperties line(final Vec3 start, final Vec3 end, final int argb) {
		return addGizmo(new LineGizmo(start, end, argb, 3.0F));
	}

	public static GizmoProperties line(final Vec3 start, final Vec3 end, final int argb, final float width) {
		return addGizmo(new LineGizmo(start, end, argb, width));
	}

	public static GizmoProperties arrow(final Vec3 start, final Vec3 end, final int argb) {
		return addGizmo(new ArrowGizmo(start, end, argb, 2.5F));
	}

	public static GizmoProperties arrow(final Vec3 start, final Vec3 end, final int argb, final float width) {
		return addGizmo(new ArrowGizmo(start, end, argb, width));
	}

	public static GizmoProperties rect(final Vec3 cuboidCornerA, final Vec3 cuboidCornerB, final Direction face, final GizmoStyle style) {
		return addGizmo(RectGizmo.fromCuboidFace(cuboidCornerA, cuboidCornerB, face, style));
	}

	public static GizmoProperties rect(final Vec3 cornerA, final Vec3 cornerB, final Vec3 cornerC, final Vec3 cornerD, final GizmoStyle style) {
		return addGizmo(new RectGizmo(cornerA, cornerB, cornerC, cornerD, style));
	}

	public static GizmoProperties point(final Vec3 position, final int argb, final float size) {
		return addGizmo(new PointGizmo(position, argb, size));
	}

	public static GizmoProperties billboardTextOverBlock(final String text, final BlockPos pos, final int row, final int color, final float scale) {
		double firstRowStartPosition = 1.3;
		double rowHeight = 0.2;
		GizmoProperties properties = billboardText(
			text, Vec3.atLowerCornerWithOffset(pos, 0.5, 1.3 + row * 0.2, 0.5), TextGizmo.Style.forColorAndCentered(color).withScale(scale)
		);
		properties.setAlwaysOnTop();
		return properties;
	}

	public static GizmoProperties billboardTextOverMob(final Entity entity, final int row, final String text, final int color, final float scale) {
		double firstRowStartPosition = 2.4;
		double rowHeight = 0.25;
		double x = entity.getBlockX() + 0.5;
		double y = entity.getY() + 2.4 + row * 0.25;
		double z = entity.getBlockZ() + 0.5;
		float textAdjustLeft = 0.5F;
		GizmoProperties properties = billboardText(text, new Vec3(x, y, z), TextGizmo.Style.forColor(color).withScale(scale).withLeftAlignment(0.5F));
		properties.setAlwaysOnTop();
		return properties;
	}

	public static GizmoProperties billboardText(final String name, final Vec3 pos, final TextGizmo.Style style) {
		return addGizmo(new TextGizmo(pos, name, style));
	}

	public static class TemporaryCollection implements AutoCloseable {
		@Nullable
		private final GizmoCollector old = (GizmoCollector)Gizmos.collector.get();
		private boolean closed;

		private TemporaryCollection() {
		}

		public void close() {
			if (!this.closed) {
				this.closed = true;
				Gizmos.collector.set(this.old);
			}
		}
	}
}