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.resources.Identifier;

/**
 * Access widened by fabric-data-generation-api-v1 to extendable
 */
public class RegistryDumpReport implements DataProvider {
	private final PackOutput output;

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

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

	private static <T> JsonElement dumpRegistry(final Registry<T> registry) {
		JsonObject result = new JsonObject();
		if (registry instanceof DefaultedRegistry) {
			Identifier defaultKey = ((DefaultedRegistry)registry).getDefaultKey();
			result.addProperty("default", defaultKey.toString());
		}

		int registryId = BuiltInRegistries.REGISTRY.getId(registry);
		result.addProperty("protocol_id", registryId);
		JsonObject entries = new JsonObject();
		registry.listElements().forEach(holder -> {
			T value = (T)holder.value();
			int protocolId = registry.getId(value);
			JsonObject entry = new JsonObject();
			entry.addProperty("protocol_id", protocolId);
			entries.add(holder.key().identifier().toString(), entry);
		});
		result.add("entries", entries);
		return result;
	}

	/**
	 * Access widened by fabric-data-generation-api-v1 to extendable
	 */
	@Override
	public String getName() {
		return "Registry Dump";
	}
}