package net.minecraft.data.advancements.packs;

import java.util.Optional;
import java.util.function.Consumer;
import net.minecraft.advancements.Advancement;
import net.minecraft.advancements.AdvancementHolder;
import net.minecraft.advancements.AdvancementRequirements;
import net.minecraft.advancements.AdvancementRewards;
import net.minecraft.advancements.AdvancementType;
import net.minecraft.advancements.predicates.BlockPredicate;
import net.minecraft.advancements.predicates.ContextAwarePredicate;
import net.minecraft.advancements.predicates.DamageSourcePredicate;
import net.minecraft.advancements.predicates.DistancePredicate;
import net.minecraft.advancements.predicates.ItemPredicate;
import net.minecraft.advancements.predicates.LocationPredicate;
import net.minecraft.advancements.predicates.MinMaxBounds;
import net.minecraft.advancements.predicates.MobEffectsPredicate;
import net.minecraft.advancements.predicates.StatePropertiesPredicate;
import net.minecraft.advancements.predicates.TagPredicate;
import net.minecraft.advancements.predicates.entity.EntityEquipmentPredicate;
import net.minecraft.advancements.predicates.entity.EntityFlagsPredicate;
import net.minecraft.advancements.predicates.entity.EntityPredicate;
import net.minecraft.advancements.triggers.BrewedPotionTrigger;
import net.minecraft.advancements.triggers.ChangeDimensionTrigger;
import net.minecraft.advancements.triggers.ConstructBeaconTrigger;
import net.minecraft.advancements.triggers.DistanceTrigger;
import net.minecraft.advancements.triggers.EffectsChangedTrigger;
import net.minecraft.advancements.triggers.InventoryChangeTrigger;
import net.minecraft.advancements.triggers.ItemDurabilityTrigger;
import net.minecraft.advancements.triggers.ItemUsedOnLocationTrigger;
import net.minecraft.advancements.triggers.KilledTrigger;
import net.minecraft.advancements.triggers.LootTableTrigger;
import net.minecraft.advancements.triggers.PickedUpItemTrigger;
import net.minecraft.advancements.triggers.PlayerInteractTrigger;
import net.minecraft.advancements.triggers.PlayerTrigger;
import net.minecraft.advancements.triggers.SummonedEntityTrigger;
import net.minecraft.core.HolderGetter;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.advancements.AdvancementSubProvider;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.Identifier;
import net.minecraft.tags.DamageTypeTags;
import net.minecraft.tags.ItemTags;
import net.minecraft.world.effect.MobEffects;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.EntityTypes;
import net.minecraft.world.entity.monster.piglin.PiglinAi;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.Items;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.biome.MultiNoiseBiomeSourceParameterList;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.RespawnAnchorBlock;
import net.minecraft.world.level.levelgen.structure.BuiltinStructures;
import net.minecraft.world.level.storage.loot.BuiltInLootTables;
import net.minecraft.world.level.storage.loot.LootContext;
import net.minecraft.world.level.storage.loot.predicates.LootItemEntityPropertyCondition;

public class VanillaNetherAdvancements implements AdvancementSubProvider {
	@Override
	public void generate(final HolderLookup.Provider registries, final Consumer<AdvancementHolder> output) {
		HolderGetter<EntityType<?>> entityTypes = registries.lookupOrThrow(Registries.ENTITY_TYPE);
		HolderGetter<Item> items = registries.lookupOrThrow(Registries.ITEM);
		HolderGetter<Block> blocks = registries.lookupOrThrow(Registries.BLOCK);
		AdvancementHolder root = Advancement.Builder.advancement()
			.display(
				Blocks.RED_NETHER_BRICKS,
				Component.translatable("advancements.nether.root.title"),
				Component.translatable("advancements.nether.root.description"),
				Identifier.withDefaultNamespace("gui/advancements/backgrounds/nether"),
				AdvancementType.TASK,
				false,
				false,
				false
			)
			.addCriterion("entered_nether", ChangeDimensionTrigger.TriggerInstance.changedDimensionTo(Level.NETHER))
			.save(output, "nether/root");
		AdvancementHolder returnToSender = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.FIRE_CHARGE,
				Component.translatable("advancements.nether.return_to_sender.title"),
				Component.translatable("advancements.nether.return_to_sender.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(50))
			.addCriterion(
				"killed_ghast",
				KilledTrigger.TriggerInstance.playerKilledEntity(
					EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.GHAST),
					DamageSourcePredicate.Builder.damageType()
						.tag(TagPredicate.is(DamageTypeTags.IS_PROJECTILE))
						.direct(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.FIREBALL))
				)
			)
			.save(output, "nether/return_to_sender");
		AdvancementHolder findFortress = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Blocks.NETHER_BRICKS,
				Component.translatable("advancements.nether.find_fortress.title"),
				Component.translatable("advancements.nether.find_fortress.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"fortress",
				PlayerTrigger.TriggerInstance.located(
					LocationPredicate.Builder.inStructure(registries.lookupOrThrow(Registries.STRUCTURE).getOrThrow(BuiltinStructures.FORTRESS))
				)
			)
			.save(output, "nether/find_fortress");
		Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.MAP,
				Component.translatable("advancements.nether.fast_travel.title"),
				Component.translatable("advancements.nether.fast_travel.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.addCriterion("travelled", DistanceTrigger.TriggerInstance.travelledThroughNether(DistancePredicate.horizontal(MinMaxBounds.Doubles.atLeast(7000.0))))
			.save(output, "nether/fast_travel");
		Advancement.Builder.advancement()
			.parent(returnToSender)
			.display(
				Items.GHAST_TEAR,
				Component.translatable("advancements.nether.uneasy_alliance.title"),
				Component.translatable("advancements.nether.uneasy_alliance.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.addCriterion(
				"killed_ghast",
				KilledTrigger.TriggerInstance.playerKilledEntity(
					EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.GHAST).located(LocationPredicate.Builder.inDimension(Level.OVERWORLD))
				)
			)
			.save(output, "nether/uneasy_alliance");
		AdvancementHolder getWitherSkull = Advancement.Builder.advancement()
			.parent(findFortress)
			.display(
				Blocks.WITHER_SKELETON_SKULL,
				Component.translatable("advancements.nether.get_wither_skull.title"),
				Component.translatable("advancements.nether.get_wither_skull.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("wither_skull", InventoryChangeTrigger.TriggerInstance.hasItems(Blocks.WITHER_SKELETON_SKULL))
			.save(output, "nether/get_wither_skull");
		AdvancementHolder summonWither = Advancement.Builder.advancement()
			.parent(getWitherSkull)
			.display(
				Items.NETHER_STAR,
				Component.translatable("advancements.nether.summon_wither.title"),
				Component.translatable("advancements.nether.summon_wither.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("summoned", SummonedEntityTrigger.TriggerInstance.summonedEntity(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.WITHER)))
			.save(output, "nether/summon_wither");
		AdvancementHolder obtainBlazeRod = Advancement.Builder.advancement()
			.parent(findFortress)
			.display(
				Items.BLAZE_ROD,
				Component.translatable("advancements.nether.obtain_blaze_rod.title"),
				Component.translatable("advancements.nether.obtain_blaze_rod.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("blaze_rod", InventoryChangeTrigger.TriggerInstance.hasItems(Items.BLAZE_ROD))
			.save(output, "nether/obtain_blaze_rod");
		AdvancementHolder createBeacon = Advancement.Builder.advancement()
			.parent(summonWither)
			.display(
				Blocks.BEACON,
				Component.translatable("advancements.nether.create_beacon.title"),
				Component.translatable("advancements.nether.create_beacon.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("beacon", ConstructBeaconTrigger.TriggerInstance.constructedBeacon(MinMaxBounds.Ints.atLeast(1)))
			.save(output, "nether/create_beacon");
		Advancement.Builder.advancement()
			.parent(createBeacon)
			.display(
				Blocks.BEACON,
				Component.translatable("advancements.nether.create_full_beacon.title"),
				Component.translatable("advancements.nether.create_full_beacon.description"),
				null,
				AdvancementType.GOAL,
				true,
				true,
				false
			)
			.addCriterion("beacon", ConstructBeaconTrigger.TriggerInstance.constructedBeacon(MinMaxBounds.Ints.exactly(4)))
			.save(output, "nether/create_full_beacon");
		AdvancementHolder brewPotion = Advancement.Builder.advancement()
			.parent(obtainBlazeRod)
			.display(
				Items.POTION,
				Component.translatable("advancements.nether.brew_potion.title"),
				Component.translatable("advancements.nether.brew_potion.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("potion", BrewedPotionTrigger.TriggerInstance.brewedPotion())
			.save(output, "nether/brew_potion");
		AdvancementHolder allPotions = Advancement.Builder.advancement()
			.parent(brewPotion)
			.display(
				Items.MILK_BUCKET,
				Component.translatable("advancements.nether.all_potions.title"),
				Component.translatable("advancements.nether.all_potions.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.addCriterion(
				"all_effects",
				EffectsChangedTrigger.TriggerInstance.hasEffects(
					MobEffectsPredicate.Builder.effects()
						.and(MobEffects.SPEED)
						.and(MobEffects.SLOWNESS)
						.and(MobEffects.STRENGTH)
						.and(MobEffects.JUMP_BOOST)
						.and(MobEffects.REGENERATION)
						.and(MobEffects.FIRE_RESISTANCE)
						.and(MobEffects.WATER_BREATHING)
						.and(MobEffects.INVISIBILITY)
						.and(MobEffects.NIGHT_VISION)
						.and(MobEffects.WEAKNESS)
						.and(MobEffects.POISON)
						.and(MobEffects.SLOW_FALLING)
						.and(MobEffects.RESISTANCE)
						.and(MobEffects.OOZING)
						.and(MobEffects.INFESTED)
						.and(MobEffects.WIND_CHARGED)
						.and(MobEffects.WEAVING)
				)
			)
			.save(output, "nether/all_potions");
		Advancement.Builder.advancement()
			.parent(allPotions)
			.display(
				Items.BUCKET,
				Component.translatable("advancements.nether.all_effects.title"),
				Component.translatable("advancements.nether.all_effects.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				true
			)
			.rewards(AdvancementRewards.Builder.experience(1000))
			.addCriterion(
				"all_effects",
				EffectsChangedTrigger.TriggerInstance.hasEffects(
					MobEffectsPredicate.Builder.effects()
						.and(MobEffects.SPEED)
						.and(MobEffects.SLOWNESS)
						.and(MobEffects.STRENGTH)
						.and(MobEffects.JUMP_BOOST)
						.and(MobEffects.REGENERATION)
						.and(MobEffects.FIRE_RESISTANCE)
						.and(MobEffects.WATER_BREATHING)
						.and(MobEffects.INVISIBILITY)
						.and(MobEffects.NIGHT_VISION)
						.and(MobEffects.WEAKNESS)
						.and(MobEffects.POISON)
						.and(MobEffects.WITHER)
						.and(MobEffects.HASTE)
						.and(MobEffects.MINING_FATIGUE)
						.and(MobEffects.LEVITATION)
						.and(MobEffects.GLOWING)
						.and(MobEffects.ABSORPTION)
						.and(MobEffects.HUNGER)
						.and(MobEffects.NAUSEA)
						.and(MobEffects.RESISTANCE)
						.and(MobEffects.SLOW_FALLING)
						.and(MobEffects.CONDUIT_POWER)
						.and(MobEffects.DOLPHINS_GRACE)
						.and(MobEffects.BLINDNESS)
						.and(MobEffects.BAD_OMEN)
						.and(MobEffects.HERO_OF_THE_VILLAGE)
						.and(MobEffects.DARKNESS)
						.and(MobEffects.OOZING)
						.and(MobEffects.INFESTED)
						.and(MobEffects.WIND_CHARGED)
						.and(MobEffects.WEAVING)
						.and(MobEffects.TRIAL_OMEN)
						.and(MobEffects.RAID_OMEN)
						.and(MobEffects.BREATH_OF_THE_NAUTILUS)
				)
			)
			.save(output, "nether/all_effects");
		AdvancementHolder obtainAncientDebris = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.ANCIENT_DEBRIS,
				Component.translatable("advancements.nether.obtain_ancient_debris.title"),
				Component.translatable("advancements.nether.obtain_ancient_debris.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("ancient_debris", InventoryChangeTrigger.TriggerInstance.hasItems(Items.ANCIENT_DEBRIS))
			.save(output, "nether/obtain_ancient_debris");
		Advancement.Builder.advancement()
			.parent(obtainAncientDebris)
			.display(
				Items.NETHERITE_CHESTPLATE,
				Component.translatable("advancements.nether.netherite_armor.title"),
				Component.translatable("advancements.nether.netherite_armor.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(100))
			.addCriterion(
				"netherite_armor",
				InventoryChangeTrigger.TriggerInstance.hasItems(Items.NETHERITE_HELMET, Items.NETHERITE_CHESTPLATE, Items.NETHERITE_LEGGINGS, Items.NETHERITE_BOOTS)
			)
			.save(output, "nether/netherite_armor");
		AdvancementHolder obtainCryingObsidian = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.CRYING_OBSIDIAN,
				Component.translatable("advancements.nether.obtain_crying_obsidian.title"),
				Component.translatable("advancements.nether.obtain_crying_obsidian.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion("crying_obsidian", InventoryChangeTrigger.TriggerInstance.hasItems(Items.CRYING_OBSIDIAN))
			.save(output, "nether/obtain_crying_obsidian");
		Advancement.Builder.advancement()
			.parent(obtainCryingObsidian)
			.display(
				Items.RESPAWN_ANCHOR,
				Component.translatable("advancements.nether.charge_respawn_anchor.title"),
				Component.translatable("advancements.nether.charge_respawn_anchor.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"charge_respawn_anchor",
				ItemUsedOnLocationTrigger.TriggerInstance.itemUsedOnBlock(
					LocationPredicate.Builder.location()
						.setBlock(
							BlockPredicate.Builder.block()
								.of(blocks, Blocks.RESPAWN_ANCHOR)
								.setProperties(StatePropertiesPredicate.Builder.properties().hasProperty(RespawnAnchorBlock.CHARGE, 4))
						),
					ItemPredicate.Builder.item().of(items, Blocks.GLOWSTONE)
				)
			)
			.save(output, "nether/charge_respawn_anchor");
		AdvancementHolder rideStrider = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.WARPED_FUNGUS_ON_A_STICK,
				Component.translatable("advancements.nether.ride_strider.title"),
				Component.translatable("advancements.nether.ride_strider.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"used_warped_fungus_on_a_stick",
				ItemDurabilityTrigger.TriggerInstance.changedDurability(
					Optional.of(EntityPredicate.wrap(EntityPredicate.Builder.entity().vehicle(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.STRIDER)))),
					Optional.of(ItemPredicate.Builder.item().of(items, Items.WARPED_FUNGUS_ON_A_STICK).build()),
					MinMaxBounds.Ints.ANY
				)
			)
			.save(output, "nether/ride_strider");
		Advancement.Builder.advancement()
			.parent(rideStrider)
			.display(
				Items.WARPED_FUNGUS_ON_A_STICK,
				Component.translatable("advancements.nether.ride_strider_in_overworld_lava.title"),
				Component.translatable("advancements.nether.ride_strider_in_overworld_lava.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"ride_entity_distance",
				DistanceTrigger.TriggerInstance.rideEntityInLava(
					EntityPredicate.Builder.entity()
						.located(LocationPredicate.Builder.inDimension(Level.OVERWORLD))
						.vehicle(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.STRIDER)),
					DistancePredicate.horizontal(MinMaxBounds.Doubles.atLeast(50.0))
				)
			)
			.save(output, "nether/ride_strider_in_overworld_lava");
		VanillaAdventureAdvancements.addBiomes(Advancement.Builder.advancement(), registries, MultiNoiseBiomeSourceParameterList.Preset.NETHER.usedBiomes().toList())
			.parent(rideStrider)
			.display(
				Items.NETHERITE_BOOTS,
				Component.translatable("advancements.nether.explore_nether.title"),
				Component.translatable("advancements.nether.explore_nether.description"),
				null,
				AdvancementType.CHALLENGE,
				true,
				true,
				false
			)
			.rewards(AdvancementRewards.Builder.experience(500))
			.save(output, "nether/explore_nether");
		AdvancementHolder findBastion = Advancement.Builder.advancement()
			.parent(root)
			.display(
				Items.POLISHED_BLACKSTONE_BRICKS,
				Component.translatable("advancements.nether.find_bastion.title"),
				Component.translatable("advancements.nether.find_bastion.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"bastion",
				PlayerTrigger.TriggerInstance.located(
					LocationPredicate.Builder.inStructure(registries.lookupOrThrow(Registries.STRUCTURE).getOrThrow(BuiltinStructures.BASTION_REMNANT))
				)
			)
			.save(output, "nether/find_bastion");
		Advancement.Builder.advancement()
			.parent(findBastion)
			.display(
				Blocks.CHEST,
				Component.translatable("advancements.nether.loot_bastion.title"),
				Component.translatable("advancements.nether.loot_bastion.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.requirements(AdvancementRequirements.Strategy.OR)
			.addCriterion("loot_bastion_other", LootTableTrigger.TriggerInstance.lootTableUsed(BuiltInLootTables.BASTION_OTHER))
			.addCriterion("loot_bastion_treasure", LootTableTrigger.TriggerInstance.lootTableUsed(BuiltInLootTables.BASTION_TREASURE))
			.addCriterion("loot_bastion_hoglin_stable", LootTableTrigger.TriggerInstance.lootTableUsed(BuiltInLootTables.BASTION_HOGLIN_STABLE))
			.addCriterion("loot_bastion_bridge", LootTableTrigger.TriggerInstance.lootTableUsed(BuiltInLootTables.BASTION_BRIDGE))
			.save(output, "nether/loot_bastion");
		Optional<ContextAwarePredicate> distractPiglinPlayerArmorPredicate = Optional.of(
			ContextAwarePredicate.create(
				LootItemEntityPropertyCondition.hasProperties(
						LootContext.EntityTarget.THIS,
						EntityPredicate.Builder.entity()
							.equipment(EntityEquipmentPredicate.Builder.equipment().head(ItemPredicate.Builder.item().of(items, ItemTags.PIGLIN_SAFE_ARMOR)))
					)
					.invert()
					.build(),
				LootItemEntityPropertyCondition.hasProperties(
						LootContext.EntityTarget.THIS,
						EntityPredicate.Builder.entity()
							.equipment(EntityEquipmentPredicate.Builder.equipment().chest(ItemPredicate.Builder.item().of(items, ItemTags.PIGLIN_SAFE_ARMOR)))
					)
					.invert()
					.build(),
				LootItemEntityPropertyCondition.hasProperties(
						LootContext.EntityTarget.THIS,
						EntityPredicate.Builder.entity()
							.equipment(EntityEquipmentPredicate.Builder.equipment().legs(ItemPredicate.Builder.item().of(items, ItemTags.PIGLIN_SAFE_ARMOR)))
					)
					.invert()
					.build(),
				LootItemEntityPropertyCondition.hasProperties(
						LootContext.EntityTarget.THIS,
						EntityPredicate.Builder.entity()
							.equipment(EntityEquipmentPredicate.Builder.equipment().feet(ItemPredicate.Builder.item().of(items, ItemTags.PIGLIN_SAFE_ARMOR)))
					)
					.invert()
					.build()
			)
		);
		Advancement.Builder.advancement()
			.parent(root)
			.requirements(AdvancementRequirements.Strategy.OR)
			.display(
				Items.GOLD_INGOT,
				Component.translatable("advancements.nether.distract_piglin.title"),
				Component.translatable("advancements.nether.distract_piglin.description"),
				null,
				AdvancementType.TASK,
				true,
				true,
				false
			)
			.addCriterion(
				"distract_piglin",
				PickedUpItemTrigger.TriggerInstance.thrownItemPickedUpByEntity(
					distractPiglinPlayerArmorPredicate,
					Optional.of(ItemPredicate.Builder.item().of(items, ItemTags.PIGLIN_LOVED).build()),
					Optional.of(
						EntityPredicate.wrap(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.PIGLIN).flags(EntityFlagsPredicate.Builder.flags().setIsBaby(false)))
					)
				)
			)
			.addCriterion(
				"distract_piglin_directly",
				PlayerInteractTrigger.TriggerInstance.itemUsedOnEntity(
					distractPiglinPlayerArmorPredicate,
					ItemPredicate.Builder.item().of(items, PiglinAi.BARTERING_ITEM),
					Optional.of(
						EntityPredicate.wrap(EntityPredicate.Builder.entity().of(entityTypes, EntityTypes.PIGLIN).flags(EntityFlagsPredicate.Builder.flags().setIsBaby(false)))
					)
				)
			)
			.save(output, "nether/distract_piglin");
	}
}