package net.minecraft.core.particles;

import com.mojang.serialization.MapCodec;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStackTemplate;

public class ItemParticleOption implements ParticleOptions {
	private final ParticleType<ItemParticleOption> type;
	private final ItemStackTemplate itemStack;

	public static MapCodec<ItemParticleOption> codec(final ParticleType<ItemParticleOption> type) {
		return ItemStackTemplate.CODEC.<ItemParticleOption>xmap(stack -> new ItemParticleOption(type, stack), o -> o.itemStack).fieldOf("item");
	}

	public static StreamCodec<? super RegistryFriendlyByteBuf, ItemParticleOption> streamCodec(final ParticleType<ItemParticleOption> type) {
		return ItemStackTemplate.STREAM_CODEC.map(stack -> new ItemParticleOption(type, stack), o -> o.itemStack);
	}

	public ItemParticleOption(final ParticleType<ItemParticleOption> type, final Item item) {
		this(type, new ItemStackTemplate(item));
	}

	public ItemParticleOption(final ParticleType<ItemParticleOption> type, final ItemStackTemplate itemStack) {
		this.type = type;
		this.itemStack = itemStack;
	}

	@Override
	public ParticleType<ItemParticleOption> getType() {
		return this.type;
	}

	public ItemStackTemplate getItem() {
		return this.itemStack;
	}
}