package net.minecraft.data.registries;

import java.util.Optional;
import java.util.concurrent.CompletableFuture;
import net.minecraft.core.Cloner;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.RegistryAccess;
import net.minecraft.core.RegistrySetBuilder;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.core.registries.Registries;
import net.minecraft.resources.RegistryDataLoader;
import net.minecraft.world.level.biome.Biome;
import net.minecraft.world.level.levelgen.placement.PlacedFeature;

public class RegistryPatchGenerator {
	public static CompletableFuture<RegistrySetBuilder.PatchedRegistries> createLookup(
		CompletableFuture<HolderLookup.Provider> lookup, RegistrySetBuilder registrySetBuilder
	) {
		return lookup.thenApply(
			provider -> {
				RegistryAccess.Frozen frozen = RegistryAccess.fromRegistryOfRegistries(BuiltInRegistries.REGISTRY);
				Cloner.Factory factory = new Cloner.Factory();
				RegistryDataLoader.WORLDGEN_REGISTRIES.forEach(registryData -> registryData.runWithArguments(factory::addCodec));
				RegistrySetBuilder.PatchedRegistries patchedRegistries = registrySetBuilder.buildPatch(frozen, provider, factory);
				HolderLookup.Provider provider2 = patchedRegistries.full();
				Optional<HolderLookup.RegistryLookup<Biome>> optional = provider2.lookup(Registries.BIOME);
				Optional<HolderLookup.RegistryLookup<PlacedFeature>> optional2 = provider2.lookup(Registries.PLACED_FEATURE);
				if (optional.isPresent() || optional2.isPresent()) {
					VanillaRegistries.validateThatAllBiomeFeaturesHaveBiomeFilter(
						(HolderGetter<PlacedFeature>)optional2.orElseGet(() -> provider.lookupOrThrow(Registries.PLACED_FEATURE)),
						(HolderLookup<Biome>)optional.orElseGet(() -> provider.lookupOrThrow(Registries.BIOME))
					);
				}

				return patchedRegistries;
			}
		);
	}
}