package net.minecraft.data.info;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import java.nio.file.Path;
import java.util.concurrent.CompletableFuture;
import net.minecraft.core.DefaultedRegistry;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.data.CachedOutput;
import net.minecraft.data.DataProvider;
import net.minecraft.data.PackOutput;
import net.minecraft.data.PackOutput.Target;
import net.minecraft.resources.ResourceLocation;

public class RegistryDumpReport implements DataProvider {
	private final PackOutput output;

	public RegistryDumpReport(PackOutput output) {
		this.output = output;
	}

	@Override
	public CompletableFuture<?> run(CachedOutput output) {
		JsonObject jsonObject = new JsonObject();
		BuiltInRegistries.REGISTRY
			.listElements()
			.forEach(reference -> jsonObject.add(reference.key().location().toString(), dumpRegistry((Registry)reference.value())));
		Path path = this.output.getOutputFolder(Target.REPORTS).resolve("registries.json");
		return DataProvider.saveStable(output, jsonObject, path);
	}

	private static <T> JsonElement dumpRegistry(Registry<T> registry) {
		JsonObject jsonObject = new JsonObject();
		if (registry instanceof DefaultedRegistry) {
			ResourceLocation resourceLocation = ((DefaultedRegistry)registry).getDefaultKey();
			jsonObject.addProperty("default", resourceLocation.toString());
		}

		int i = BuiltInRegistries.REGISTRY.getId(registry);
		jsonObject.addProperty("protocol_id", i);
		JsonObject jsonObject2 = new JsonObject();
		registry.listElements().forEach(reference -> {
			T object = (T)reference.value();
			int ix = registry.getId(object);
			JsonObject jsonObject2x = new JsonObject();
			jsonObject2x.addProperty("protocol_id", ix);
			jsonObject2.add(reference.key().location().toString(), jsonObject2x);
		});
		jsonObject.add("entries", jsonObject2);
		return jsonObject;
	}

	@Override
	public final String getName() {
		return "Registry Dump";
	}
}