package net.minecraft.util.datafix.schemas;

import com.google.common.collect.ImmutableMap;
import com.mojang.datafixers.DSL;
import com.mojang.datafixers.schemas.Schema;
import com.mojang.datafixers.types.templates.TypeTemplate;
import java.util.Map;
import java.util.function.Supplier;
import net.minecraft.util.datafix.fixes.References;

public class V2551 extends NamespacedSchema {
	public V2551(int i, Schema schema) {
		super(i, schema);
	}

	@Override
	public void registerTypes(Schema schema, Map<String, Supplier<TypeTemplate>> map, Map<String, Supplier<TypeTemplate>> map2) {
		super.registerTypes(schema, map, map2);
		schema.registerType(
			false,
			References.WORLD_GEN_SETTINGS,
			() -> DSL.fields(
				"dimensions",
				DSL.compoundList(
					DSL.constType(namespacedString()),
					DSL.fields(
						"generator",
						DSL.taggedChoiceLazy(
							"type",
							DSL.string(),
							ImmutableMap.of(
								"minecraft:debug",
								DSL::remainder,
								"minecraft:flat",
								() -> DSL.optionalFields(
									"settings",
									DSL.optionalFields("biome", References.BIOME.in(schema), "layers", DSL.list(DSL.optionalFields("block", References.BLOCK_NAME.in(schema))))
								),
								"minecraft:noise",
								() -> DSL.optionalFields(
									"biome_source",
									DSL.taggedChoiceLazy(
										"type",
										DSL.string(),
										ImmutableMap.of(
											"minecraft:fixed",
											() -> DSL.fields("biome", References.BIOME.in(schema)),
											"minecraft:multi_noise",
											() -> DSL.list(DSL.fields("biome", References.BIOME.in(schema))),
											"minecraft:checkerboard",
											() -> DSL.fields("biomes", DSL.list(References.BIOME.in(schema))),
											"minecraft:vanilla_layered",
											DSL::remainder,
											"minecraft:the_end",
											DSL::remainder
										)
									),
									"settings",
									DSL.or(
										DSL.constType(DSL.string()), DSL.optionalFields("default_block", References.BLOCK_NAME.in(schema), "default_fluid", References.BLOCK_NAME.in(schema))
									)
								)
							)
						)
					)
				)
			)
		);
	}
}