li/cil/oc/server/component/Carriage.scala

package li.cil.oc.server.component

import li.cil.oc.api
import li.cil.oc.api.network._
import li.cil.oc.util.RedstoneInMotion
import net.minecraft.nbt.NBTTagCompound

class Carriage(controller: AnyRef) extends ManagedComponent {
  val node = api.Network.newNode(this, Visibility.Network).
    withComponent("carriage").
    create()

  private val names = Map(
    "negy" -> 0, "posy" -> 1, "negz" -> 2, "posz" -> 3, "negx" -> 4, "posx" -> 5,
    "down" -> 0, "up" -> 1, "north" -> 2, "south" -> 3, "west" -> 4, "east" -> 5)

  private var anchored = false
  private var direction = 0
  private var simulating = false

  private var shouldMove = false
  private var moving = false

  // ----------------------------------------------------------------------- //

  @LuaCallback("move")
  def move(context: Context, args: Arguments): Array[AnyRef] = {
    direction = checkDirection(args)
    simulating = if (args.count > 1) args.checkBoolean(1) else false
    shouldMove = true
    result(true)
  }

  @LuaCallback("simulate")
  def simulate(context: Context, args: Arguments): Array[AnyRef] = {
    direction = checkDirection(args)
    simulating = true
    shouldMove = true
    result(true)
  }

  @LuaCallback("getAnchored")
  def getAnchored(context: Context, args: Arguments): Array[AnyRef] =
    result(anchored)

  @LuaCallback("setAnchored")
  def setAnchored(context: Context, args: Arguments): Array[AnyRef] = {
    anchored = args.checkBoolean(0)
    result(anchored)
  }

  private def checkDirection(args: Arguments) = {
    if (!RedstoneInMotion.available)
      throw new Exception("Redstone in Motion not found")
    if (shouldMove || moving)
      throw new Exception("already moving")
    if (args.isString(0)) {
      val name = args.checkString(0).toLowerCase
      if (!names.contains(name))
        throw new IllegalArgumentException("invalid direction")
      names(name)
    }
    else {
      val index = args.checkInteger(0)
      if (index < 0 || index > 5)
        throw new ArrayIndexOutOfBoundsException("invalid direction")
      index
    }
  }

  // ----------------------------------------------------------------------- //

  override def update() {
    super.update()
    if (shouldMove) {
      shouldMove = false
      moving = true
      try {
        RedstoneInMotion.move(controller, direction, simulating, anchored)
        if (simulating || anchored) {
          // We won't get re-connected, so we won't send in onConnect. Do it here.
          node.sendToReachable("computer.signal", "carriage_moved", Boolean.box(true))
        }
      }
      catch {
        case e: Throwable =>
          node.sendToReachable("computer.signal", "carriage_moved", Unit, Option(e.getMessage).getOrElse(e.toString))
      }
      finally {
        moving = false
      }
    }
  }

  // ----------------------------------------------------------------------- //

  override def onConnect(node: Node) {
    super.onConnect(node)
    if (moving) {
      moving = false
      node.sendToReachable("computer.signal", "carriage_moved", Boolean.box(true))
    }
  }

  // ----------------------------------------------------------------------- //

  override def save(nbt: NBTTagCompound) = {
    super.save(nbt)
    nbt.setBoolean("moving", moving)
    nbt.setBoolean("anchored", anchored)
  }

  override def load(nbt: NBTTagCompound) {
    super.load(nbt)
    moving = nbt.getBoolean("moving")
    anchored = nbt.getBoolean("anchored")
  }
}