package net.minecraft.data.worldgen;

import com.google.common.collect.ImmutableList;
import com.mojang.datafixers.util.Pair;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.registries.Registries;
import net.minecraft.world.level.levelgen.structure.pools.StructurePoolElement;
import net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureProcessorList;

public class BastionTreasureRoomPools {
	public static void bootstrap(BootstrapContext<StructureTemplatePool> context) {
		HolderGetter<StructureProcessorList> holderGetter = context.lookup(Registries.PROCESSOR_LIST);
		Holder<StructureProcessorList> holder = holderGetter.getOrThrow(ProcessorLists.TREASURE_ROOMS);
		Holder<StructureProcessorList> holder2 = holderGetter.getOrThrow(ProcessorLists.HIGH_WALL);
		Holder<StructureProcessorList> holder3 = holderGetter.getOrThrow(ProcessorLists.BOTTOM_RAMPART);
		Holder<StructureProcessorList> holder4 = holderGetter.getOrThrow(ProcessorLists.HIGH_RAMPART);
		Holder<StructureProcessorList> holder5 = holderGetter.getOrThrow(ProcessorLists.ROOF);
		HolderGetter<StructureTemplatePool> holderGetter2 = context.lookup(Registries.TEMPLATE_POOL);
		Holder<StructureTemplatePool> holder6 = holderGetter2.getOrThrow(Pools.EMPTY);
		Pools.register(
			context,
			"bastion/treasure/bases",
			new StructureTemplatePool(
				holder6, ImmutableList.of(Pair.of(StructurePoolElement.single("bastion/treasure/bases/lava_basin", holder), 1)), StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/stairs",
			new StructureTemplatePool(
				holder6, ImmutableList.of(Pair.of(StructurePoolElement.single("bastion/treasure/stairs/lower_stairs", holder), 1)), StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/bases/centers",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/bases/centers/center_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/bases/centers/center_1", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/bases/centers/center_2", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/bases/centers/center_3", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/brains",
			new StructureTemplatePool(
				holder6, ImmutableList.of(Pair.of(StructurePoolElement.single("bastion/treasure/brains/center_brain", holder), 1)), StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/walls",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/lava_wall", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/entrance_wall", holder2), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/walls/outer",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/top_corner", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/mid_corner", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/bottom_corner", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/outer_wall", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/medium_outer_wall", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/outer/tall_outer_wall", holder2), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/walls/bottom",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/bottom/wall_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/bottom/wall_1", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/bottom/wall_2", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/bottom/wall_3", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/walls/mid",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/mid/wall_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/mid/wall_1", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/mid/wall_2", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/walls/top",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/top/main_entrance", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/top/wall_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/walls/top/wall_1", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/connectors",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/connectors/center_to_wall_middle", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/connectors/center_to_wall_top", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/connectors/center_to_wall_top_entrance", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/entrances",
			new StructureTemplatePool(
				holder6, ImmutableList.of(Pair.of(StructurePoolElement.single("bastion/treasure/entrances/entrance_0", holder), 1)), StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/ramparts",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/mid_wall_main", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/mid_wall_side", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/bottom_wall_0", holder3), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/top_wall", holder4), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/lava_basin_side", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/ramparts/lava_basin_main", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/corners/bottom",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/bottom/corner_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/bottom/corner_1", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/corners/edges",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/edges/bottom", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/edges/middle", holder2), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/edges/top", holder2), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/corners/middle",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/middle/corner_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/middle/corner_1", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/corners/top",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/top/corner_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/corners/top/corner_1", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/extensions/large_pool",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/empty", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/empty", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/fire_room", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/large_bridge_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/large_bridge_1", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/large_bridge_2", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/large_bridge_3", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/roofed_bridge", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/empty", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/extensions/small_pool",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/empty", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/fire_room", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/empty", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/small_bridge_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/small_bridge_1", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/small_bridge_2", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/small_bridge_3", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/extensions/houses",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/house_0", holder), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/extensions/house_1", holder), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
		Pools.register(
			context,
			"bastion/treasure/roofs",
			new StructureTemplatePool(
				holder6,
				ImmutableList.of(
					Pair.of(StructurePoolElement.single("bastion/treasure/roofs/wall_roof", holder5), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/roofs/corner_roof", holder5), 1),
					Pair.of(StructurePoolElement.single("bastion/treasure/roofs/center_roof", holder5), 1)
				),
				StructureTemplatePool.Projection.RIGID
			)
		);
	}
}