src/main/scala/li/cil/oc/common/container/DynamicComponentSlot.scala

package li.cil.oc.common.container

import li.cil.oc.api
import li.cil.oc.client.gui.Icons
import li.cil.oc.common.InventorySlots.{Tier, InventorySlot}
import net.minecraft.entity.player.EntityPlayer
import net.minecraft.inventory.{Slot, IInventory}
import net.minecraft.item.ItemStack

class DynamicComponentSlot(val container: Player, inventory: IInventory, index: Int, x: Int, y: Int, val info: Array[Array[InventorySlot]], val tierGetter: () => Int) extends Slot(inventory, index, x, y) with ComponentSlot {
  override def tier = {
    val mainTier = tierGetter()
    if (mainTier >= 0) info(mainTier)(slotNumber).tier
    else mainTier
  }

  def tierIcon = Icons.get(tier)

  def slot = {
    val mainTier = tierGetter()
    if (mainTier >= 0) info(tierGetter())(slotNumber).slot
    else api.driver.Slot.None
  }

  override def getBackgroundIconIndex = Icons.get(slot)

  override def getSlotStackLimit =
    slot match {
      case api.driver.Slot.Tool | api.driver.Slot.None => super.getSlotStackLimit
      case _ => 1
    }

  override def isItemValid(stack: ItemStack) = {
    inventory.isItemValidForSlot(index, stack)
  }

  override protected def clearIfInvalid(player: EntityPlayer) {
    if (getHasStack && !isItemValid(getStack)) {
      val stack = getStack
      putStack(null)
      if (!player.inventory.addItemStackToInventory(stack)) {
        player.dropPlayerItem(stack)
      }
    }
  }
}