package net.minecraft.core.particles;

import com.mojang.serialization.Codec;
import com.mojang.serialization.MapCodec;
import net.fabricmc.fabric.api.particle.v1.FabricBlockParticleOption;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;

public class BlockParticleOption implements ParticleOptions, FabricBlockParticleOption {
	private static final Codec<BlockState> BLOCK_STATE_CODEC = Codec.withAlternative(
		BlockState.CODEC, BuiltInRegistries.BLOCK.byNameCodec(), Block::defaultBlockState
	);
	private final ParticleType<BlockParticleOption> type;
	private final BlockState state;

	public static MapCodec<BlockParticleOption> codec(final ParticleType<BlockParticleOption> type) {
		return BLOCK_STATE_CODEC.<BlockParticleOption>xmap(state -> new BlockParticleOption(type, state), o -> o.state).fieldOf("block_state");
	}

	public static StreamCodec<? super RegistryFriendlyByteBuf, BlockParticleOption> streamCodec(final ParticleType<BlockParticleOption> type) {
		return ByteBufCodecs.idMapper(Block.BLOCK_STATE_REGISTRY).map(state -> new BlockParticleOption(type, state), o -> o.state);
	}

	public BlockParticleOption(final ParticleType<BlockParticleOption> type, final BlockState state) {
		this.type = type;
		this.state = state;
	}

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

	public BlockState getState() {
		return this.state;
	}
}