package net.minecraft.data.info;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.mojang.serialization.JsonOps;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.component.DataComponentPatch;
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;
import net.minecraft.resources.RegistryOps;

/**
 * Access widened by fabric-data-generation-api-v1 to extendable
 */
public class RegistryComponentsReport implements DataProvider {
	private final PackOutput output;
	private final CompletableFuture<HolderLookup.Provider> registries;

	public RegistryComponentsReport(final PackOutput output, final CompletableFuture<HolderLookup.Provider> registries) {
		this.output = output;
		this.registries = registries;
	}

	@Override
	public CompletableFuture<?> run(final CachedOutput cache) {
		return this.registries
			.thenCompose(
				registries -> {
					RegistryOps<JsonElement> registryOps = registries.createSerializationContext(JsonOps.INSTANCE);
					List<CompletableFuture<?>> writes = new ArrayList();
					BuiltInRegistries.DATA_COMPONENT_INITIALIZERS
						.build(registries)
						.forEach(
							pendingComponents -> {
								PackOutput.PathProvider registryPathProvider = this.output.createRegistryComponentPathProvider(pendingComponents.key());
								pendingComponents.forEach(
									(element, components) -> {
										if (!components.isEmpty()) {
											Identifier elementId = element.key().identifier();
											Path elementPath = registryPathProvider.json(elementId);
											DataComponentPatch patch = DataComponentPatch.builder().set(components).build();
											JsonObject root = new JsonObject();
											root.add(
												"components",
												DataComponentPatch.CODEC
													.encodeStart(registryOps, patch)
													.getOrThrow(err -> new IllegalStateException("Failed to encode components for item " + elementId + ": " + err))
											);
											writes.add(DataProvider.saveStable(cache, root, elementPath));
										}
									}
								);
							}
						);
					return CompletableFuture.allOf((CompletableFuture[])writes.toArray(CompletableFuture[]::new));
				}
			);
	}

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