package net.minecraft.client.renderer.texture.atlas;

import com.mojang.blaze3d.platform.NativeImage;
import com.mojang.logging.LogUtils;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.texture.SpriteContents;
import net.minecraft.client.resources.metadata.animation.AnimationMetadataSection;
import net.minecraft.client.resources.metadata.animation.FrameSize;
import net.minecraft.client.resources.metadata.texture.TextureMetadataSection;
import net.minecraft.resources.Identifier;
import net.minecraft.server.packs.metadata.MetadataSectionType;
import net.minecraft.server.packs.resources.Resource;
import net.minecraft.server.packs.resources.ResourceMetadata;
import net.minecraft.util.Mth;
import org.jspecify.annotations.Nullable;
import org.slf4j.Logger;

@FunctionalInterface
@Environment(EnvType.CLIENT)
public interface SpriteResourceLoader {
	Logger LOGGER = LogUtils.getLogger();

	static SpriteResourceLoader create(final Set<MetadataSectionType<?>> additionalMetadataSections) {
		return (spriteLocation, resource) -> {
			Optional<AnimationMetadataSection> animationInfo;
			Optional<TextureMetadataSection> textureInfo;
			List<MetadataSectionType.WithValue<?>> additionalMetadata;
			try {
				ResourceMetadata metadata = resource.metadata();
				animationInfo = metadata.getSection(AnimationMetadataSection.TYPE);
				textureInfo = metadata.getSection(TextureMetadataSection.TYPE);
				additionalMetadata = metadata.getTypedSections(additionalMetadataSections);
			} catch (Exception var11) {
				LOGGER.error("Unable to parse metadata from {}", spriteLocation, var11);
				return null;
			}

			NativeImage image;
			try {
				InputStream is = resource.open();

				try {
					image = NativeImage.read(is);
				} catch (Throwable var12) {
					if (is != null) {
						try {
							is.close();
						} catch (Throwable var10) {
							var12.addSuppressed(var10);
						}
					}

					throw var12;
				}

				if (is != null) {
					is.close();
				}
			} catch (IOException var13) {
				LOGGER.error("Using missing texture, unable to load {}", spriteLocation, var13);
				return null;
			}

			FrameSize frameSize;
			if (animationInfo.isPresent()) {
				frameSize = ((AnimationMetadataSection)animationInfo.get()).calculateFrameSize(image.getWidth(), image.getHeight());
				if (!Mth.isMultipleOf(image.getWidth(), frameSize.width()) || !Mth.isMultipleOf(image.getHeight(), frameSize.height())) {
					LOGGER.error(
						"Image {} size {},{} is not multiple of frame size {},{}", spriteLocation, image.getWidth(), image.getHeight(), frameSize.width(), frameSize.height()
					);
					image.close();
					return null;
				}
			} else {
				frameSize = new FrameSize(image.getWidth(), image.getHeight());
			}

			return new SpriteContents(spriteLocation, frameSize, image, animationInfo, additionalMetadata, textureInfo);
		};
	}

	@Nullable
	SpriteContents loadSprite(Identifier spriteLocation, Resource resource);
}