src/main/scala/li/cil/oc/common/nanomachines/Nanomachines.scala

package li.cil.oc.common.nanomachines

import li.cil.oc.Settings
import li.cil.oc.api
import li.cil.oc.api.nanomachines.BehaviorProvider
import li.cil.oc.api.nanomachines.Controller
import li.cil.oc.util.PlayerUtils
import net.minecraft.entity.player.EntityPlayer

import scala.collection.convert.WrapAsJava._
import scala.collection.mutable

object Nanomachines extends api.detail.NanomachinesAPI {
  val providers = mutable.Set.empty[BehaviorProvider]

  val serverControllers = mutable.WeakHashMap.empty[EntityPlayer, ControllerImpl]
  val clientControllers = mutable.WeakHashMap.empty[EntityPlayer, ControllerImpl]

  def controllers(player: EntityPlayer) = if (player.getEntityWorld.isRemote) clientControllers else serverControllers

  override def addProvider(provider: BehaviorProvider): Unit = providers += provider

  override def getProviders: java.lang.Iterable[BehaviorProvider] = providers

  def getController(player: EntityPlayer): Controller = {
    if (hasController(player)) controllers(player).getOrElseUpdate(player, new ControllerImpl(player))
    else null
  }

  def hasController(player: EntityPlayer) = {
    PlayerUtils.persistedData(player).getBoolean(Settings.namespace + "hasNanomachines")
  }

  def installController(player: EntityPlayer) = {
    if (!hasController(player)) {
      PlayerUtils.persistedData(player).setBoolean(Settings.namespace + "hasNanomachines", true)
    }
    getController(player) // Initialize controller instance.
  }

  override def uninstallController(player: EntityPlayer): Unit = {
    getController(player) match {
      case controller: ControllerImpl =>
        PlayerUtils.persistedData(player).removeTag(Settings.namespace + "hasNanomachines")
        controllers(player) -= player
        controller.dispose()
      case _ => // Doesn't have one anyway.
    }
  }
}