package net.minecraft.world.level.levelgen.structure.templatesystem;

import com.mojang.serialization.MapCodec;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;

public class BlockMatchTest extends RuleTest {
	public static final MapCodec<BlockMatchTest> CODEC = BuiltInRegistries.BLOCK
		.byNameCodec()
		.fieldOf("block")
		.xmap(BlockMatchTest::new, blockMatchTest -> blockMatchTest.block);
	private final Block block;

	public BlockMatchTest(Block block) {
		this.block = block;
	}

	@Override
	public boolean test(BlockState state, RandomSource random) {
		return state.is(this.block);
	}

	@Override
	protected RuleTestType<?> getType() {
		return RuleTestType.BLOCK_TEST;
	}
}