package net.minecraft.world.level.levelgen.structure.templatesystem.loader;

import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.datafixers.DataFixer;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.StandardCharsets;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.stream.Stream;
import net.minecraft.core.HolderGetter;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.NbtAccounter;
import net.minecraft.nbt.NbtIo;
import net.minecraft.nbt.NbtUtils;
import net.minecraft.resources.Identifier;
import net.minecraft.server.packs.resources.IoSupplier;
import net.minecraft.util.FastBufferedInputStream;
import net.minecraft.util.datafix.DataFixTypes;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplate;

public abstract class TemplateSource {
	private final DataFixer fixerUpper;
	private final HolderGetter<Block> blockLookup;

	protected TemplateSource(final DataFixer fixerUpper, final HolderGetter<Block> blockLookup) {
		this.fixerUpper = fixerUpper;
		this.blockLookup = blockLookup;
	}

	public abstract Optional<StructureTemplate> load(Identifier id);

	public abstract Stream<Identifier> list();

	protected Optional<StructureTemplate> load(final IoSupplier<InputStream> opener, final boolean asText, final Consumer<Throwable> onError) {
		try {
			InputStream rawInput = opener.get();

			Optional var7;
			try {
				InputStream input = new FastBufferedInputStream(rawInput);

				try {
					CompoundTag structureTag;
					if (asText) {
						structureTag = readTextStructure(input);
					} else {
						structureTag = readStructure(input);
					}

					var7 = Optional.of(this.readStructure(structureTag));
				} catch (Throwable var10) {
					try {
						input.close();
					} catch (Throwable var9) {
						var10.addSuppressed(var9);
					}

					throw var10;
				}

				input.close();
			} catch (Throwable var11) {
				if (rawInput != null) {
					try {
						rawInput.close();
					} catch (Throwable var8) {
						var11.addSuppressed(var8);
					}
				}

				throw var11;
			}

			if (rawInput != null) {
				rawInput.close();
			}

			return var7;
		} catch (FileNotFoundException var12) {
			return Optional.empty();
		} catch (Throwable var13) {
			onError.accept(var13);
			return Optional.empty();
		}
	}

	private static CompoundTag readStructure(final InputStream input) throws IOException {
		return NbtIo.readCompressed(input, NbtAccounter.unlimitedHeap());
	}

	private static CompoundTag readTextStructure(final InputStream input) throws IOException, CommandSyntaxException {
		Reader reader = new InputStreamReader(input, StandardCharsets.UTF_8);

		CompoundTag var3;
		try {
			String contents = reader.readAllAsString();
			var3 = NbtUtils.snbtToStructure(contents);
		} catch (Throwable var5) {
			try {
				reader.close();
			} catch (Throwable var4) {
				var5.addSuppressed(var4);
			}

			throw var5;
		}

		reader.close();
		return var3;
	}

	private StructureTemplate readStructure(final CompoundTag tag) {
		StructureTemplate structureTemplate = new StructureTemplate();
		int version = NbtUtils.getDataVersion(tag, 500);
		structureTemplate.load(this.blockLookup, DataFixTypes.STRUCTURE.updateToCurrentVersion(this.fixerUpper, tag, version));
		return structureTemplate;
	}
}