package net.minecraft.advancements.predicates;

import com.mojang.logging.LogUtils;
import com.mojang.serialization.Codec;
import io.netty.buffer.ByteBuf;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.core.component.DataComponents;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.NbtUtils;
import net.minecraft.nbt.Tag;
import net.minecraft.nbt.TagParser;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.ProblemReporter;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.component.CustomData;
import net.minecraft.world.level.storage.TagValueOutput;
import org.jspecify.annotations.Nullable;
import org.slf4j.Logger;

public record NbtPredicate(CompoundTag tag) {
	private static final Logger LOGGER = LogUtils.getLogger();
	public static final Codec<NbtPredicate> CODEC = TagParser.LENIENT_CODEC.xmap(NbtPredicate::new, NbtPredicate::tag);
	public static final StreamCodec<ByteBuf, NbtPredicate> STREAM_CODEC = ByteBufCodecs.COMPOUND_TAG.map(NbtPredicate::new, NbtPredicate::tag);
	public static final String SELECTED_ITEM_TAG = "SelectedItem";

	public boolean matches(final DataComponentGetter components) {
		CustomData data = components.getOrDefault(DataComponents.CUSTOM_DATA, CustomData.EMPTY);
		return data.matchedBy(this.tag);
	}

	public boolean matches(final Entity entity) {
		return this.matches(getEntityTagToCompare(entity));
	}

	public boolean matches(@Nullable final Tag tag) {
		return tag != null && NbtUtils.compareNbt(this.tag, tag, true);
	}

	public static CompoundTag getEntityTagToCompare(final Entity entity) {
		CompoundTag var7;
		try (ProblemReporter.ScopedCollector reporter = new ProblemReporter.ScopedCollector(entity.problemPath(), LOGGER)) {
			TagValueOutput output = TagValueOutput.createWithContext(reporter, entity.registryAccess());
			entity.saveWithoutId(output);
			if (entity instanceof Player player) {
				ItemStack selected = player.getInventory().getSelectedItem();
				if (!selected.isEmpty()) {
					output.store("SelectedItem", ItemStack.CODEC, selected);
				}
			}

			var7 = output.buildResult();
		}

		return var7;
	}
}