Embryonic but functioning JSR223 support

Author: rjolly

compiler/src/dotty/tools/repl/Rendering.scala

@@ -26,7 +26,7 @@ private[repl] class Rendering(parentClassLoader: Option[ClassLoader] = None) {
   private[this] var myClassLoader: ClassLoader = _
 
   /** Class loader used to load compiled code */
-  private[this] def classLoader()(implicit ctx: Context) =
+  private[repl] def classLoader()(implicit ctx: Context) =
     if (myClassLoader != null) myClassLoader
     else {
       val parent = parentClassLoader.getOrElse {

compiler/src/dotty/tools/repl/ScriptEngine.scala

@@ -0,0 +1,77 @@
+package dotty.tools
+package repl
+
+import java.io.{Reader, StringWriter};
+import javax.script.{AbstractScriptEngine, Bindings, ScriptContext, ScriptEngine => JScriptEngine, ScriptEngineFactory, ScriptException, SimpleBindings}
+
+class ScriptEngine(factory: ScriptEngine.Factory) extends AbstractScriptEngine {
+  val driver = factory.driver
+  val rendering = new Rendering
+  var state: State = driver.initialState
+
+  def getFactory(): ScriptEngineFactory = factory
+
+  def createBindings: Bindings = new SimpleBindings
+
+  /* Evaluate with the given context. */
+  @throws[ScriptException]
+  def eval(script: String, context: ScriptContext): Object = {
+    val vid = state.valIndex
+    state = driver.run(script)(state)
+    val oid = state.objectIndex
+    Class.forName(s"rs$$line$$$oid", true, rendering.classLoader()(state.context))
+      .getDeclaredMethods.find(_.getName == s"res$vid")
+      .map(_.invoke(null))
+      .getOrElse(null)
+  }
+
+  @throws[ScriptException]
+  def eval(reader: Reader, context: ScriptContext): Object = eval(stringFromReader(reader), context)
+
+  def stringFromReader(reader: Reader) = {
+    val writer = new StringWriter()
+    var c = reader.read()
+    while(c != -1) {
+      writer.write(c)
+      c = reader.read()
+    }
+    reader.close()
+    writer.toString()
+  }
+}
+
+object ScriptEngine {
+  import java.util.Arrays.asList
+  import scala.util.Properties.versionString
+
+  class Factory extends ScriptEngineFactory {
+    val driver = new ReplDriver(Array("-usejavacp"), Console.out, None);
+
+    def getEngineName() = "Scala REPL"
+    def getEngineVersion() = "3.0"
+    def getExtensions() = asList("scala")
+    def getLanguageName() = "Scala"
+    def getLanguageVersion() = versionString
+    def getMimeTypes() = asList("application/x-scala")
+    def getNames() = asList("scala")
+
+    def getMethodCallSyntax(obj: String, m: String, args: String*): String = null
+
+    def getOutputStatement(toDisplay: String): String = null
+
+    def getParameter(key: String): Object = key match {
+      case JScriptEngine.ENGINE           => getEngineName
+      case JScriptEngine.ENGINE_VERSION   => getEngineVersion
+      case JScriptEngine.LANGUAGE         => getLanguageName
+      case JScriptEngine.LANGUAGE_VERSION => getLanguageVersion
+      case JScriptEngine.NAME             => getNames.get(0)
+      case _ => null
+    }
+
+    def getProgram(statements: String*): String = null
+
+    def getScriptEngine: JScriptEngine = {
+      new ScriptEngine(this)
+    }
+  }
+}

project/Build.scala

@@ -553,6 +553,13 @@ object Build {
         Seq(file)
       }.taskValue,
 
+      // Generate the ScriptEngineFactory service definition
+      resourceGenerators in Compile += Def.task {
+        val f = (resourceManaged in Compile).value / "META-INF" / "services" / "javax.script.ScriptEngineFactory"
+        IO.write(f, "dotty.tools.repl.ScriptEngine$Factory\n")
+        Seq(f)
+      }.taskValue,
+
       // get libraries onboard
       libraryDependencies ++= Seq(
         "org.scala-lang.modules" % "scala-asm" % "6.0.0-scala-1", // used by the backend

tests/run-with-compiler/scripting.check

@@ -0,0 +1,3 @@
+val res0: Int = 42
+
+42

tests/run-with-compiler/scripting.scala

@@ -0,0 +1,10 @@
+object Test {
+  def main(args: Array[String]): Unit = {
+//  val m = new javax.script.ScriptEngineManager()
+//  val e = m.getEngineByName("scala")
+    val f = new dotty.tools.repl.ScriptEngine.Factory()
+    val e = f.getScriptEngine()
+    println(e.eval("42"))
+  }
+}
+