blob: 5acdfd6dd0b84d846adaace085e3d18ec3b2d219 [file] [log] [blame] [raw]
package li.cil.oc.server.computer
import java.io.File
import java.io.FileOutputStream
import java.nio.channels.Channels
import scala.collection.mutable.Map
import com.naef.jnlua.LuaState
import com.naef.jnlua.NativeSupport
import com.naef.jnlua.NativeSupport.Loader
/**
* Factory singleton used to spawn new LuaState instances.
*
* This is realized as a singleton so that we only have to resolve shared
* library references once during initialization and can then re-use the
* already loaded ones.
*/
private[computer] object LuaStateFactory {
// ----------------------------------------------------------------------- //
// Initialization
// ----------------------------------------------------------------------- //
private val libraries = Map.empty[String, String]
private val basePath = "/assets/opencomputers/"
// Since we use native libraries we have to do some work. This includes
// figuring out what we're running on, so that we can load the proper shared
// libraries compiled for that system. It also means we have to unpack the
// shared libraries somewhere so that we can load them, because we cannot
// load them directly from a JAR.
{
val platform = System.getProperty("os.name").toLowerCase() match {
case name if (name.startsWith("linux")) => "linux"
case name if (name.startsWith("windows")) => "windows"
case name if (name.startsWith("mac")) => "mac"
}
val libPath = basePath + "lib/" + System.getProperty("os.arch") + "/" + platform + "/"
val libExt = platform match {
case "linux" => ".so"
case "windows" => ".dll"
case "mac" => ".dylib"
case _ => ""
}
val tmpPath = {
val path = System.getProperty("java.io.tmpdir")
if (path.endsWith("/") || path.endsWith("\\")) path
else path + "/"
}
for (library <- Array("lua52", "jnlua52")) {
val libraryUrl = classOf[Computer].getResource(libPath + library + libExt)
if (libraryUrl == null) {
throw new NotImplementedError("Unsupported platform.")
}
// Found file with proper extension. Create a temporary file.
val file = new File(tmpPath + library + libExt)
if (!file.exists()) {
file.deleteOnExit()
// Copy the file contents to the temporary file.
val in = Channels.newChannel(libraryUrl.openStream())
val out = new FileOutputStream(file).getChannel()
out.transferFrom(in, 0, Long.MaxValue)
in.close()
out.close()
}
// Remember the temporary file's location for later.
libraries += library -> file.getAbsolutePath()
}
// Register a custom library loader with JNLua to actually load the ones we
// just extracted.
NativeSupport.getInstance().setLoader(new Loader {
def load() {
System.load(libraries("lua52"))
System.load(libraries("jnlua52"))
}
})
}
// ----------------------------------------------------------------------- //
// Factory
// ----------------------------------------------------------------------- //
def createState(): LuaState = {
val state = new LuaState(Integer.MAX_VALUE)
try {
// Load all libraries.
state.openLib(LuaState.Library.BASE)
state.openLib(LuaState.Library.BIT32)
state.openLib(LuaState.Library.COROUTINE)
state.openLib(LuaState.Library.DEBUG)
state.openLib(LuaState.Library.ERIS)
state.openLib(LuaState.Library.MATH)
state.openLib(LuaState.Library.STRING)
state.openLib(LuaState.Library.TABLE)
state.pop(8)
// Prepare table for os stuff.
state.newTable()
state.setGlobal("os")
// Remove some other functions we don't need and are dangerous.
state.pushNil()
state.setGlobal("dofile")
state.pushNil()
state.setGlobal("loadfile")
state.pushNil()
state.setGlobal("module")
state.pushNil()
state.setGlobal("require")
}
catch {
case ex: Throwable => {
ex.printStackTrace()
state.close()
return null
}
}
return state
}
}