package net.minecraft.world.item.slot;

import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.minecraft.util.ExtraCodecs;

public class LimitSlotSource extends TransformedSlotSource {
	public static final MapCodec<LimitSlotSource> MAP_CODEC = RecordCodecBuilder.mapCodec(
		i -> commonFields(i).and(ExtraCodecs.POSITIVE_INT.fieldOf("limit").forGetter(t -> t.limit)).apply(i, LimitSlotSource::new)
	);
	private final int limit;

	private LimitSlotSource(final SlotSource slotSource, final int limit) {
		super(slotSource);
		this.limit = limit;
	}

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

	@Override
	protected SlotCollection transform(final SlotCollection slots) {
		return slots.limit(this.limit);
	}
}