package net.minecraft.client.gui.screens.inventory;

import net.minecraft.resources.Identifier;
import net.minecraft.world.entity.EquipmentSlot;
import net.minecraft.world.entity.animal.nautilus.AbstractNautilus;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.inventory.NautilusInventoryMenu;
import org.jspecify.annotations.Nullable;

public class NautilusInventoryScreen extends AbstractMountInventoryScreen<NautilusInventoryMenu> {
	private static final Identifier SLOT_SPRITE = Identifier.withDefaultNamespace("container/slot");
	private static final Identifier NAUTILUS_INVENTORY_LOCATION = Identifier.withDefaultNamespace("textures/gui/container/nautilus.png");

	public NautilusInventoryScreen(final NautilusInventoryMenu menu, final Inventory inventory, final AbstractNautilus nautilus, final int inventoryColumns) {
		super(menu, inventory, nautilus.getDisplayName(), inventoryColumns, nautilus);
	}

	@Override
	protected Identifier getBackgroundTextureLocation() {
		return NAUTILUS_INVENTORY_LOCATION;
	}

	@Override
	protected Identifier getSlotSpriteLocation() {
		return SLOT_SPRITE;
	}

	@Nullable
	@Override
	protected Identifier getChestSlotsSpriteLocation() {
		return null;
	}

	@Override
	protected boolean shouldRenderSaddleSlot() {
		return this.mount.canUseSlot(EquipmentSlot.SADDLE);
	}

	@Override
	protected boolean shouldRenderArmorSlot() {
		return this.mount.canUseSlot(EquipmentSlot.BODY);
	}
}