package net.minecraft.data.advancements.packs;

import com.google.common.collect.BiMap;
import java.util.Comparator;
import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.stream.Stream;
import net.minecraft.advancements.Advancement;
import net.minecraft.advancements.AdvancementHolder;
import net.minecraft.advancements.AdvancementRewards;
import net.minecraft.advancements.AdvancementType;
import net.minecraft.advancements.AdvancementRequirements.Strategy;
import net.minecraft.advancements.critereon.BlockPredicate;
import net.minecraft.advancements.critereon.DataComponentMatchers;
import net.minecraft.advancements.critereon.EnchantmentPredicate;
import net.minecraft.advancements.critereon.EntityEquipmentPredicate;
import net.minecraft.advancements.critereon.EntityPredicate;
import net.minecraft.advancements.critereon.InventoryChangeTrigger;
import net.minecraft.advancements.critereon.ItemPredicate;
import net.minecraft.advancements.critereon.MinMaxBounds;
import net.minecraft.advancements.critereon.ConsumeItemTrigger.TriggerInstance;
import net.minecraft.advancements.critereon.LocationPredicate.Builder;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.Holder.Reference;
import net.minecraft.core.component.DataComponentExactPredicate;
import net.minecraft.core.component.DataComponents;
import net.minecraft.core.component.predicates.DataComponentPredicates;
import net.minecraft.core.component.predicates.EnchantmentsPredicate;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.advancements.AdvancementSubProvider;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.tags.BlockTags;
import net.minecraft.tags.EntityTypeTags;
import net.minecraft.tags.ItemTags;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.animal.CatVariant;
import net.minecraft.world.entity.animal.frog.FrogVariant;
import net.minecraft.world.entity.animal.wolf.WolfVariant;
import net.minecraft.world.item.HoneycombItem;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.Items;
import net.minecraft.world.item.enchantment.Enchantment;
import net.minecraft.world.item.enchantment.Enchantments;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;

public class VanillaHusbandryAdvancements implements AdvancementSubProvider {
	public static final List<EntityType<?>> BREEDABLE_ANIMALS = List.of(
		EntityType.HORSE,
		EntityType.DONKEY,
		EntityType.MULE,
		EntityType.SHEEP,
		EntityType.COW,
		EntityType.MOOSHROOM,
		EntityType.PIG,
		EntityType.CHICKEN,
		EntityType.WOLF,
		EntityType.OCELOT,
		EntityType.RABBIT,
		EntityType.LLAMA,
		EntityType.CAT,
		EntityType.PANDA,
		EntityType.FOX,
		EntityType.BEE,
		EntityType.HOGLIN,
		EntityType.STRIDER,
		EntityType.GOAT,
		EntityType.AXOLOTL,
		EntityType.CAMEL,
		EntityType.ARMADILLO
	);
	public static final List<EntityType<?>> INDIRECTLY_BREEDABLE_ANIMALS = List.of(EntityType.TURTLE, EntityType.FROG, EntityType.SNIFFER);
	private static final Item[] FISH = new Item[]{Items.COD, Items.TROPICAL_FISH, Items.PUFFERFISH, Items.SALMON};
	private static final Item[] FISH_BUCKETS = new Item[]{Items.COD_BUCKET, Items.TROPICAL_FISH_BUCKET, Items.PUFFERFISH_BUCKET, Items.SALMON_BUCKET};
	private static final Item[] EDIBLE_ITEMS = new Item[]{
		Items.APPLE,
		Items.MUSHROOM_STEW,
		Items.BREAD,
		Items.PORKCHOP,
		Items.COOKED_PORKCHOP,
		Items.GOLDEN_APPLE,
		Items.ENCHANTED_GOLDEN_APPLE,
		Items.COD,
		Items.SALMON,
		Items.TROPICAL_FISH,
		Items.PUFFERFISH,
		Items.COOKED_COD,
		Items.COOKED_SALMON,
		Items.COOKIE,
		Items.MELON_SLICE,
		Items.BEEF,
		Items.COOKED_BEEF,
		Items.CHICKEN,
		Items.COOKED_CHICKEN,
		Items.ROTTEN_FLESH,
		Items.SPIDER_EYE,
		Items.CARROT,
		Items.POTATO,
		Items.BAKED_POTATO,
		Items.POISONOUS_POTATO,
		Items.GOLDEN_CARROT,
		Items.PUMPKIN_PIE,
		Items.RABBIT,
		Items.COOKED_RABBIT,
		Items.RABBIT_STEW,
		Items.MUTTON,
		Items.COOKED_MUTTON,
		Items.CHORUS_FRUIT,
		Items.BEETROOT,
		Items.BEETROOT_SOUP,
		Items.DRIED_KELP,
		Items.SUSPICIOUS_STEW,
		Items.SWEET_BERRIES,
		Items.HONEY_BOTTLE,
		Items.GLOW_BERRIES
	};
	public static final Item[] WAX_SCRAPING_TOOLS = new Item[]{
		Items.WOODEN_AXE, Items.GOLDEN_AXE, Items.STONE_AXE, Items.IRON_AXE, Items.DIAMOND_AXE, Items.NETHERITE_AXE
	};
	private static final Comparator<Reference<?>> HOLDER_KEY_COMPARATOR = Comparator.comparing(reference -> reference.key().location());

	@Override
	public void generate(HolderLookup.Provider registries, Consumer<AdvancementHolder> writer) {
		HolderGetter<EntityType<?>> holderGetter = registries.lookupOrThrow(Registries.ENTITY_TYPE);
		HolderGetter<Item> holderGetter2 = registries.lookupOrThrow(Registries.ITEM);
		HolderGetter<Block> holderGetter3 = registries.lookupOrThrow(Registries.BLOCK);
		HolderLookup<FrogVariant> holderLookup = registries.lookupOrThrow(Registries.FROG_VARIANT);
		HolderLookup<CatVariant> holderLookup2 = registries.lookupOrThrow(Registries.CAT_VARIANT);
		HolderLookup<WolfVariant> holderLookup3 = registries.lookupOrThrow(Registries.WOLF_VARIANT);
		HolderLookup.RegistryLookup<Enchantment> registryLookup = registries.lookupOrThrow(Registries.ENCHANTMENT);
		AdvancementHolder advancementHolder = Advancement.Builder.advancement()
			.display(
				Blocks.HAY_BLOCK,
				Component.translatable("advancements.husbandry.root.title"),
				Component.translatable("advancements.husbandry.root.description"),
				ResourceLocation.withDefaultNamespace("gui/advancements/backgrounds/husbandry"),
				AdvancementType.TASK,
				false,
				false,
				false
			)
			.addCriterion("consumed_item", TriggerInstance.usedItem())
			.save(writer, "husbandry/root");
		AdvancementHolder advancementHolder2 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.WHEAT,
				Component.translatable("advancements.husbandry.plant_seed.title"),
				Component.translatable("advancements.husbandry.plant_seed.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.requirements(Strategy.OR)
			.addCriterion("wheat", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.WHEAT))
			.addCriterion("pumpkin_stem", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.PUMPKIN_STEM))
			.addCriterion("melon_stem", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.MELON_STEM))
			.addCriterion("beetroots", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.BEETROOTS))
			.addCriterion("nether_wart", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.NETHER_WART))
			.addCriterion("torchflower", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.TORCHFLOWER_CROP))
			.addCriterion("pitcher_pod", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.PITCHER_CROP))
			.save(writer, "husbandry/plant_seed");
		AdvancementHolder advancementHolder3 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.WHEAT,
				Component.translatable("advancements.husbandry.breed_an_animal.title"),
				Component.translatable("advancements.husbandry.breed_an_animal.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.requirements(Strategy.OR)
			.addCriterion("bred", net.minecraft.advancements.critereon.BredAnimalsTrigger.TriggerInstance.bredAnimals())
			.save(writer, "husbandry/breed_an_animal");
		createBreedAllAnimalsAdvancement(advancementHolder3, writer, holderGetter, BREEDABLE_ANIMALS.stream(), INDIRECTLY_BREEDABLE_ANIMALS.stream());
		addFood(Advancement.Builder.advancement(), holderGetter2)
			.parent(advancementHolder2)
			.display(
				Items.APPLE,
				Component.translatable("advancements.husbandry.balanced_diet.title"),
				Component.translatable("advancements.husbandry.balanced_diet.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.save(writer, "husbandry/balanced_diet");
		Advancement.Builder.advancement()
			.parent(advancementHolder2)
			.display(
				Items.NETHERITE_HOE,
				Component.translatable("advancements.husbandry.netherite_hoe.title"),
				Component.translatable("advancements.husbandry.netherite_hoe.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.addCriterion("netherite_hoe", InventoryChangeTrigger.TriggerInstance.hasItems(Items.NETHERITE_HOE))
			.save(writer, "husbandry/obtain_netherite_hoe");
		AdvancementHolder advancementHolder4 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.LEAD,
				Component.translatable("advancements.husbandry.tame_an_animal.title"),
				Component.translatable("advancements.husbandry.tame_an_animal.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("tamed_animal", net.minecraft.advancements.critereon.TameAnimalTrigger.TriggerInstance.tamedAnimal())
			.save(writer, "husbandry/tame_an_animal");
		AdvancementHolder advancementHolder5 = addFish(Advancement.Builder.advancement(), holderGetter2)
			.parent(advancementHolder)
			.requirements(Strategy.OR)
			.display(
				Items.FISHING_ROD,
				Component.translatable("advancements.husbandry.fishy_business.title"),
				Component.translatable("advancements.husbandry.fishy_business.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/fishy_business");
		AdvancementHolder advancementHolder6 = addFishBuckets(Advancement.Builder.advancement(), holderGetter2)
			.parent(advancementHolder5)
			.requirements(Strategy.OR)
			.display(
				Items.PUFFERFISH_BUCKET,
				Component.translatable("advancements.husbandry.tactical_fishing.title"),
				Component.translatable("advancements.husbandry.tactical_fishing.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/tactical_fishing");
		AdvancementHolder advancementHolder7 = Advancement.Builder.advancement()
			.parent(advancementHolder6)
			.requirements(Strategy.OR)
			.addCriterion(
				BuiltInRegistries.ITEM.getKey(Items.AXOLOTL_BUCKET).getPath(),
				net.minecraft.advancements.critereon.FilledBucketTrigger.TriggerInstance.filledBucket(ItemPredicate.Builder.item().of(holderGetter2, Items.AXOLOTL_BUCKET))
			)
			.display(
				Items.AXOLOTL_BUCKET,
				Component.translatable("advancements.husbandry.axolotl_in_a_bucket.title"),
				Component.translatable("advancements.husbandry.axolotl_in_a_bucket.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/axolotl_in_a_bucket");
		Advancement.Builder.advancement()
			.parent(advancementHolder7)
			.addCriterion(
				"kill_axolotl_target",
				net.minecraft.advancements.critereon.EffectsChangedTrigger.TriggerInstance.gotEffectsFrom(
					EntityPredicate.Builder.entity().of(holderGetter, EntityType.AXOLOTL)
				)
			)
			.display(
				Items.TROPICAL_FISH_BUCKET,
				Component.translatable("advancements.husbandry.kill_axolotl_target.title"),
				Component.translatable("advancements.husbandry.kill_axolotl_target.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/kill_axolotl_target");
		addCatVariants(Advancement.Builder.advancement(), holderLookup2)
			.parent(advancementHolder4)
			.display(
				Items.COD,
				Component.translatable("advancements.husbandry.complete_catalogue.title"),
				Component.translatable("advancements.husbandry.complete_catalogue.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(50))
			.save(writer, "husbandry/complete_catalogue");
		addTamedWolfVariants(Advancement.Builder.advancement(), holderLookup3)
			.parent(advancementHolder4)
			.display(
				Items.BONE,
				Component.translatable("advancements.husbandry.whole_pack.title"),
				Component.translatable("advancements.husbandry.whole_pack.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(50))
			.save(writer, "husbandry/whole_pack");
		AdvancementHolder advancementHolder8 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.addCriterion(
				"safely_harvest_honey",
				net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.itemUsedOnBlock(
					Builder.location().setBlock(BlockPredicate.Builder.block().of(holderGetter3, BlockTags.BEEHIVES)).setSmokey(true),
					ItemPredicate.Builder.item().of(holderGetter2, Items.GLASS_BOTTLE)
				)
			)
			.display(
				Items.HONEY_BOTTLE,
				Component.translatable("advancements.husbandry.safely_harvest_honey.title"),
				Component.translatable("advancements.husbandry.safely_harvest_honey.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/safely_harvest_honey");
		AdvancementHolder advancementHolder9 = Advancement.Builder.advancement()
			.parent(advancementHolder8)
			.display(
				Items.HONEYCOMB,
				Component.translatable("advancements.husbandry.wax_on.title"),
				Component.translatable("advancements.husbandry.wax_on.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"wax_on",
				net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.itemUsedOnBlock(
					Builder.location().setBlock(BlockPredicate.Builder.block().of(holderGetter3, ((BiMap)HoneycombItem.WAXABLES.get()).keySet())),
					ItemPredicate.Builder.item().of(holderGetter2, Items.HONEYCOMB)
				)
			)
			.save(writer, "husbandry/wax_on");
		Advancement.Builder.advancement()
			.parent(advancementHolder9)
			.display(
				Items.STONE_AXE,
				Component.translatable("advancements.husbandry.wax_off.title"),
				Component.translatable("advancements.husbandry.wax_off.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"wax_off",
				net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.itemUsedOnBlock(
					Builder.location().setBlock(BlockPredicate.Builder.block().of(holderGetter3, ((BiMap)HoneycombItem.WAX_OFF_BY_BLOCK.get()).keySet())),
					ItemPredicate.Builder.item().of(holderGetter2, WAX_SCRAPING_TOOLS)
				)
			)
			.save(writer, "husbandry/wax_off");
		AdvancementHolder advancementHolder10 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.addCriterion(
				BuiltInRegistries.ITEM.getKey(Items.TADPOLE_BUCKET).getPath(),
				net.minecraft.advancements.critereon.FilledBucketTrigger.TriggerInstance.filledBucket(ItemPredicate.Builder.item().of(holderGetter2, Items.TADPOLE_BUCKET))
			)
			.display(
				Items.TADPOLE_BUCKET,
				Component.translatable("advancements.husbandry.tadpole_in_a_bucket.title"),
				Component.translatable("advancements.husbandry.tadpole_in_a_bucket.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/tadpole_in_a_bucket");
		AdvancementHolder advancementHolder11 = addLeashedFrogVariants(holderGetter, holderGetter2, holderLookup, Advancement.Builder.advancement())
			.parent(advancementHolder10)
			.display(
				Items.LEAD,
				Component.translatable("advancements.husbandry.leash_all_frog_variants.title"),
				Component.translatable("advancements.husbandry.leash_all_frog_variants.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/leash_all_frog_variants");
		Advancement.Builder.advancement()
			.parent(advancementHolder11)
			.display(
				Items.VERDANT_FROGLIGHT,
				Component.translatable("advancements.husbandry.froglights.title"),
				Component.translatable("advancements.husbandry.froglights.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.addCriterion("froglights", InventoryChangeTrigger.TriggerInstance.hasItems(Items.OCHRE_FROGLIGHT, Items.PEARLESCENT_FROGLIGHT, Items.VERDANT_FROGLIGHT))
			.save(writer, "husbandry/froglights");
		Advancement.Builder.advancement()
			.parent(advancementHolder)
			.addCriterion(
				"silk_touch_nest",
				net.minecraft.advancements.critereon.BeeNestDestroyedTrigger.TriggerInstance.destroyedBeeNest(
					Blocks.BEE_NEST,
					ItemPredicate.Builder.item()
						.withComponents(
							DataComponentMatchers.Builder.components()
								.partial(
									DataComponentPredicates.ENCHANTMENTS,
									EnchantmentsPredicate.enchantments(List.of(new EnchantmentPredicate(registryLookup.getOrThrow(Enchantments.SILK_TOUCH), MinMaxBounds.Ints.atLeast(1))))
								)
								.build()
						),
					MinMaxBounds.Ints.exactly(3)
				)
			)
			.display(
				Blocks.BEE_NEST,
				Component.translatable("advancements.husbandry.silk_touch_nest.title"),
				Component.translatable("advancements.husbandry.silk_touch_nest.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.save(writer, "husbandry/silk_touch_nest");
		Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.OAK_BOAT,
				Component.translatable("advancements.husbandry.ride_a_boat_with_a_goat.title"),
				Component.translatable("advancements.husbandry.ride_a_boat_with_a_goat.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"ride_a_boat_with_a_goat",
				net.minecraft.advancements.critereon.StartRidingTrigger.TriggerInstance.playerStartsRiding(
					EntityPredicate.Builder.entity()
						.vehicle(
							EntityPredicate.Builder.entity().of(holderGetter, EntityTypeTags.BOAT).passenger(EntityPredicate.Builder.entity().of(holderGetter, EntityType.GOAT))
						)
				)
			)
			.save(writer, "husbandry/ride_a_boat_with_a_goat");
		Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.GLOW_INK_SAC,
				Component.translatable("advancements.husbandry.make_a_sign_glow.title"),
				Component.translatable("advancements.husbandry.make_a_sign_glow.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"make_a_sign_glow",
				net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.itemUsedOnBlock(
					Builder.location().setBlock(BlockPredicate.Builder.block().of(holderGetter3, BlockTags.ALL_SIGNS)),
					ItemPredicate.Builder.item().of(holderGetter2, Items.GLOW_INK_SAC)
				)
			)
			.save(writer, "husbandry/make_a_sign_glow");
		AdvancementHolder advancementHolder12 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.COOKIE,
				Component.translatable("advancements.husbandry.allay_deliver_item_to_player.title"),
				Component.translatable("advancements.husbandry.allay_deliver_item_to_player.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				true
			)
			.addCriterion(
				"allay_deliver_item_to_player",
				net.minecraft.advancements.critereon.PickedUpItemTrigger.TriggerInstance.thrownItemPickedUpByPlayer(
					Optional.empty(), Optional.empty(), Optional.of(EntityPredicate.wrap(EntityPredicate.Builder.entity().of(holderGetter, EntityType.ALLAY)))
				)
			)
			.save(writer, "husbandry/allay_deliver_item_to_player");
		Advancement.Builder.advancement()
			.parent(advancementHolder12)
			.display(
				Items.NOTE_BLOCK,
				Component.translatable("advancements.husbandry.allay_deliver_cake_to_note_block.title"),
				Component.translatable("advancements.husbandry.allay_deliver_cake_to_note_block.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				true
			)
			.addCriterion(
				"allay_deliver_cake_to_note_block",
				net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.allayDropItemOnBlock(
					Builder.location().setBlock(BlockPredicate.Builder.block().of(holderGetter3, Blocks.NOTE_BLOCK)),
					ItemPredicate.Builder.item().of(holderGetter2, Items.CAKE)
				)
			)
			.save(writer, "husbandry/allay_deliver_cake_to_note_block");
		AdvancementHolder advancementHolder13 = Advancement.Builder.advancement()
			.parent(advancementHolder)
			.display(
				Items.SNIFFER_EGG,
				Component.translatable("advancements.husbandry.obtain_sniffer_egg.title"),
				Component.translatable("advancements.husbandry.obtain_sniffer_egg.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				true
			)
			.addCriterion("obtain_sniffer_egg", InventoryChangeTrigger.TriggerInstance.hasItems(Items.SNIFFER_EGG))
			.save(writer, "husbandry/obtain_sniffer_egg");
		AdvancementHolder advancementHolder14 = Advancement.Builder.advancement()
			.parent(advancementHolder13)
			.display(
				Items.TORCHFLOWER_SEEDS,
				Component.translatable("advancements.husbandry.feed_snifflet.title"),
				Component.translatable("advancements.husbandry.feed_snifflet.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				true
			)
			.addCriterion(
				"feed_snifflet",
				net.minecraft.advancements.critereon.PlayerInteractTrigger.TriggerInstance.itemUsedOnEntity(
					ItemPredicate.Builder.item().of(holderGetter2, ItemTags.SNIFFER_FOOD),
					Optional.of(
						EntityPredicate.wrap(
							EntityPredicate.Builder.entity()
								.of(holderGetter, EntityType.SNIFFER)
								.flags(net.minecraft.advancements.critereon.EntityFlagsPredicate.Builder.flags().setIsBaby(true))
						)
					)
				)
			)
			.save(writer, "husbandry/feed_snifflet");
		Advancement.Builder.advancement()
			.parent(advancementHolder14)
			.display(
				Items.PITCHER_POD,
				Component.translatable("advancements.husbandry.plant_any_sniffer_seed.title"),
				Component.translatable("advancements.husbandry.plant_any_sniffer_seed.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				true
			)
			.requirements(Strategy.OR)
			.addCriterion("torchflower", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.TORCHFLOWER_CROP))
			.addCriterion("pitcher_pod", net.minecraft.advancements.critereon.ItemUsedOnLocationTrigger.TriggerInstance.placedBlock(Blocks.PITCHER_CROP))
			.save(writer, "husbandry/plant_any_sniffer_seed");
		Advancement.Builder.advancement()
			.parent(advancementHolder4)
			.display(
				Items.SHEARS,
				Component.translatable("advancements.husbandry.remove_wolf_armor.title"),
				Component.translatable("advancements.husbandry.remove_wolf_armor.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"remove_wolf_armor",
				net.minecraft.advancements.critereon.PlayerInteractTrigger.TriggerInstance.itemUsedOnEntity(
					ItemPredicate.Builder.item().of(holderGetter2, Items.SHEARS),
					Optional.of(EntityPredicate.wrap(EntityPredicate.Builder.entity().of(holderGetter, EntityType.WOLF)))
				)
			)
			.save(writer, "husbandry/remove_wolf_armor");
		Advancement.Builder.advancement()
			.parent(advancementHolder4)
			.display(
				Items.WOLF_ARMOR,
				Component.translatable("advancements.husbandry.repair_wolf_armor.title"),
				Component.translatable("advancements.husbandry.repair_wolf_armor.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"repair_wolf_armor",
				net.minecraft.advancements.critereon.PlayerInteractTrigger.TriggerInstance.itemUsedOnEntity(
					ItemPredicate.Builder.item().of(holderGetter2, Items.ARMADILLO_SCUTE),
					Optional.of(
						EntityPredicate.wrap(
							EntityPredicate.Builder.entity()
								.of(holderGetter, EntityType.WOLF)
								.equipment(
									EntityEquipmentPredicate.Builder.equipment()
										.body(
											ItemPredicate.Builder.item()
												.of(holderGetter2, Items.WOLF_ARMOR)
												.withComponents(DataComponentMatchers.Builder.components().exact(DataComponentExactPredicate.expect(DataComponents.DAMAGE, 0)).build())
										)
								)
						)
					)
				)
			)
			.save(writer, "husbandry/repair_wolf_armor");
	}

	public static AdvancementHolder createBreedAllAnimalsAdvancement(
		AdvancementHolder parent,
		Consumer<AdvancementHolder> writer,
		HolderGetter<EntityType<?>> entityTypeRegistry,
		Stream<EntityType<?>> breedableAnimals,
		Stream<EntityType<?>> indirectlyBreedableAnimals
	) {
		return addBreedable(Advancement.Builder.advancement(), breedableAnimals, entityTypeRegistry, indirectlyBreedableAnimals)
			.parent(parent)
			.display(
				Items.GOLDEN_CARROT,
				Component.translatable("advancements.husbandry.breed_all_animals.title"),
				Component.translatable("advancements.husbandry.breed_all_animals.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.save(writer, "husbandry/bred_all_animals");
	}

	private static Advancement.Builder addLeashedFrogVariants(
		HolderGetter<EntityType<?>> entityTypeRegistry, HolderGetter<Item> itemRegistry, HolderLookup<FrogVariant> frogVariants, Advancement.Builder builder
	) {
		sortedVariants(frogVariants)
			.forEach(
				reference -> builder.addCriterion(
					reference.key().location().toString(),
					net.minecraft.advancements.critereon.PlayerInteractTrigger.TriggerInstance.itemUsedOnEntity(
						ItemPredicate.Builder.item().of(itemRegistry, Items.LEAD),
						Optional.of(
							EntityPredicate.wrap(
								EntityPredicate.Builder.entity()
									.of(entityTypeRegistry, EntityType.FROG)
									.components(DataComponentMatchers.Builder.components().exact(DataComponentExactPredicate.expect(DataComponents.FROG_VARIANT, reference)).build())
							)
						)
					)
				)
			);
		return builder;
	}

	private static <T> Stream<Reference<T>> sortedVariants(HolderLookup<T> variantRegistry) {
		return variantRegistry.listElements().sorted(HOLDER_KEY_COMPARATOR);
	}

	private static Advancement.Builder addFood(Advancement.Builder builder, HolderGetter<Item> food) {
		for (Item item : EDIBLE_ITEMS) {
			builder.addCriterion(BuiltInRegistries.ITEM.getKey(item).getPath(), TriggerInstance.usedItem(food, item));
		}

		return builder;
	}

	private static Advancement.Builder addBreedable(
		Advancement.Builder builder,
		Stream<EntityType<?>> breedableAnimals,
		HolderGetter<EntityType<?>> entityTypeRegistry,
		Stream<EntityType<?>> indirectlyBreedableAnimals
	) {
		breedableAnimals.forEach(
			entityType -> builder.addCriterion(
				EntityType.getKey(entityType).toString(),
				net.minecraft.advancements.critereon.BredAnimalsTrigger.TriggerInstance.bredAnimals(EntityPredicate.Builder.entity().of(entityTypeRegistry, entityType))
			)
		);
		indirectlyBreedableAnimals.forEach(
			entityType -> builder.addCriterion(
				EntityType.getKey(entityType).toString(),
				net.minecraft.advancements.critereon.BredAnimalsTrigger.TriggerInstance.bredAnimals(
					Optional.of(EntityPredicate.Builder.entity().of(entityTypeRegistry, entityType).build()),
					Optional.of(EntityPredicate.Builder.entity().of(entityTypeRegistry, entityType).build()),
					Optional.empty()
				)
			)
		);
		return builder;
	}

	private static Advancement.Builder addFishBuckets(Advancement.Builder builder, HolderGetter<Item> itemRegistry) {
		for (Item item : FISH_BUCKETS) {
			builder.addCriterion(
				BuiltInRegistries.ITEM.getKey(item).getPath(),
				net.minecraft.advancements.critereon.FilledBucketTrigger.TriggerInstance.filledBucket(ItemPredicate.Builder.item().of(itemRegistry, item))
			);
		}

		return builder;
	}

	private static Advancement.Builder addFish(Advancement.Builder builder, HolderGetter<Item> itemRegistry) {
		for (Item item : FISH) {
			builder.addCriterion(
				BuiltInRegistries.ITEM.getKey(item).getPath(),
				net.minecraft.advancements.critereon.FishingRodHookedTrigger.TriggerInstance.fishedItem(
					Optional.empty(), Optional.empty(), Optional.of(ItemPredicate.Builder.item().of(itemRegistry, item).build())
				)
			);
		}

		return builder;
	}

	private static Advancement.Builder addCatVariants(Advancement.Builder builder, HolderLookup<CatVariant> variantRegistry) {
		sortedVariants(variantRegistry)
			.forEach(
				reference -> builder.addCriterion(
					reference.key().location().toString(),
					net.minecraft.advancements.critereon.TameAnimalTrigger.TriggerInstance.tamedAnimal(
						EntityPredicate.Builder.entity()
							.components(DataComponentMatchers.Builder.components().exact(DataComponentExactPredicate.expect(DataComponents.CAT_VARIANT, reference)).build())
					)
				)
			);
		return builder;
	}

	private static Advancement.Builder addTamedWolfVariants(Advancement.Builder builder, HolderLookup<WolfVariant> variantRegistry) {
		sortedVariants(variantRegistry)
			.forEach(
				reference -> builder.addCriterion(
					reference.key().location().toString(),
					net.minecraft.advancements.critereon.TameAnimalTrigger.TriggerInstance.tamedAnimal(
						EntityPredicate.Builder.entity()
							.components(DataComponentMatchers.Builder.components().exact(DataComponentExactPredicate.expect(DataComponents.WOLF_VARIANT, reference)).build())
					)
				)
			);
		return builder;
	}
}