package net.minecraft.network.chat.contents.data;

import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.List;
import java.util.stream.Stream;
import net.minecraft.advancements.predicates.NbtPredicate;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.arguments.selector.EntitySelector;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.util.CompilableString;
import net.minecraft.world.entity.Entity;

public record EntityDataSource(CompilableString<EntitySelector> selector) implements DataSource {
	public static final MapCodec<EntityDataSource> MAP_CODEC = RecordCodecBuilder.mapCodec(
		i -> i.group(EntitySelector.COMPILABLE_CODEC.fieldOf("entity").forGetter(EntityDataSource::selector)).apply(i, EntityDataSource::new)
	);

	@Override
	public Stream<CompoundTag> getData(final CommandSourceStack sender) throws CommandSyntaxException {
		List<? extends Entity> entities = this.selector.compiled().findEntities(sender);
		return entities.stream().map(NbtPredicate::getEntityTagToCompare);
	}

	@Override
	public MapCodec<EntityDataSource> codec() {
		return MAP_CODEC;
	}
}