package com.mojang.blaze3d.font;

import com.mojang.datafixers.util.Either;
import com.mojang.serialization.Codec;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
import it.unimi.dsi.fastutil.ints.IntSet;
import it.unimi.dsi.fastutil.ints.IntSets;
import java.util.Map;
import net.minecraft.client.gui.font.glyphs.EmptyGlyph;
import net.minecraft.client.gui.font.providers.GlyphProviderDefinition;
import net.minecraft.client.gui.font.providers.GlyphProviderType;
import net.minecraft.util.ExtraCodecs;
import org.jspecify.annotations.Nullable;

public class SpaceProvider implements GlyphProvider {
	private final Int2ObjectMap<EmptyGlyph> glyphs;

	public SpaceProvider(final Map<Integer, Float> advances) {
		this.glyphs = new Int2ObjectOpenHashMap<>(advances.size());
		advances.forEach((codepoint, advance) -> this.glyphs.put(codepoint.intValue(), new EmptyGlyph(advance)));
	}

	@Nullable
	@Override
	public UnbakedGlyph getGlyph(final int codepoint) {
		return this.glyphs.get(codepoint);
	}

	@Override
	public IntSet getSupportedGlyphs() {
		return IntSets.unmodifiable(this.glyphs.keySet());
	}

	public record Definition(Map<Integer, Float> advances) implements GlyphProviderDefinition {
		public static final MapCodec<SpaceProvider.Definition> CODEC = RecordCodecBuilder.mapCodec(
			i -> i.group(Codec.unboundedMap(ExtraCodecs.CODEPOINT, Codec.FLOAT).fieldOf("advances").forGetter(SpaceProvider.Definition::advances))
				.apply(i, SpaceProvider.Definition::new)
		);

		@Override
		public GlyphProviderType type() {
			return GlyphProviderType.SPACE;
		}

		@Override
		public Either<GlyphProviderDefinition.Loader, GlyphProviderDefinition.Reference> unpack() {
			GlyphProviderDefinition.Loader loader = resourceManager -> new SpaceProvider(this.advances);
			return Either.left(loader);
		}
	}
}