Allow custom output directory for class generated by run

Author: nicolasstucki

compiler/src/dotty/tools/dotc/quoted/ExprCompiler.scala

@@ -3,10 +3,10 @@ package quoted
 
 import dotty.tools.backend.jvm.GenBCode
 import dotty.tools.dotc.ast.tpd
-
 import dotty.tools.dotc.core.Contexts.Context
-import dotty.tools.dotc.core.Flags.{EmptyFlags, Method}
-import dotty.tools.dotc.core.{Mode, Phases}
+import dotty.tools.dotc.core.Decorators._
+import dotty.tools.dotc.core.Flags._
+import dotty.tools.dotc.core.Mode
 import dotty.tools.dotc.core.Phases.Phase
 import dotty.tools.dotc.core.Scopes.{EmptyScope, newScope}
 import dotty.tools.dotc.core.StdNames.nme
@@ -17,14 +17,14 @@ import dotty.tools.dotc.transform.ReifyQuotes
 import dotty.tools.dotc.typer.FrontEnd
 import dotty.tools.dotc.util.Positions.Position
 import dotty.tools.dotc.util.SourceFile
-import dotty.tools.io.{Path, PlainFile, VirtualDirectory}
+import dotty.tools.io.{AbstractFile, Path, PlainFile}
 
 import scala.quoted.Expr
 
 /** Compiler that takes the contents of a quoted expression `expr` and produces
  *  a class file with `class ' { def apply: Object = expr }`.
  */
-class ExprCompiler(directory: VirtualDirectory) extends Compiler {
+class ExprCompiler(directory: AbstractFile) extends Compiler {
   import tpd._
 
   /** A GenBCode phase that outputs to a virtual directory */
@@ -72,7 +72,7 @@ class ExprCompiler(directory: VirtualDirectory) extends Compiler {
       val pos = Position(0)
       val assocFile = new PlainFile(Path("<quote>"))
 
-      val cls = ctx.newCompleteClassSymbol(defn.RootClass, nme.QUOTE.toTypeName, EmptyFlags,
+      val cls = ctx.newCompleteClassSymbol(defn.RootClass, "Quoted".toTypeName, EmptyFlags,
         defn.ObjectType :: Nil, newScope, coord = pos, assocFile = assocFile).entered.asClass
       cls.enter(ctx.newDefaultConstructor(cls), EmptyScope)
       val meth = ctx.newSymbol(cls, nme.apply, Method, ExprType(defn.AnyType), coord = pos).entered

compiler/src/dotty/tools/dotc/quoted/QuoteDriver.scala

@@ -1,9 +1,9 @@
 package dotty.tools.dotc.quoted
 
 import dotty.tools.dotc.Driver
-import dotty.tools.dotc.core.Contexts.{Context, FreshContext}
+import dotty.tools.dotc.core.Contexts.Context
 import dotty.tools.dotc.core.StdNames._
-import dotty.tools.io.VirtualDirectory
+import dotty.tools.io.{AbstractFile, Directory, PlainDirectory, VirtualDirectory}
 import dotty.tools.repl.AbstractFileClassLoader
 
 import scala.quoted.Expr
@@ -17,13 +17,20 @@ class QuoteDriver extends Driver {
     val ctx: Context = initCtx.fresh
     ctx.settings.optimise.update(settings.optimise)(ctx)
 
-    val outDir = new VirtualDirectory("(memory)", None)
+    val outDir: AbstractFile = settings.outDir match {
+      case Some(out) =>
+        val dir = Directory(out)
+        dir.createDirectory()
+        new PlainDirectory(Directory(out))
+      case None =>
+        new VirtualDirectory("(memory)", None)
+    }
 
     new ExprCompiler(outDir).newRun(ctx).compileExpr(expr)
 
     val classLoader = new AbstractFileClassLoader(outDir, this.getClass.getClassLoader)
 
-    val clazz = classLoader.loadClass(nme.QUOTE.toString)
+    val clazz = classLoader.loadClass("Quoted")
     val method = clazz.getMethod("apply")
     val instance = clazz.newInstance()
 

compiler/src/dotty/tools/dotc/quoted/Runners.scala

@@ -13,7 +13,7 @@ object Runners {
 
   implicit def runner[T]: Runner[T] = new Runner[T] {
 
-    def run(expr: Expr[T]): T = Runners.run(expr, RunSettings())
+    def run(expr: Expr[T]): T = Runners.run(expr, RunSettings(outDir = Some("aa")))
 
     def show(expr: Expr[T]): String = expr match {
       case expr: ConstantExpr[T] =>
@@ -31,6 +31,7 @@ object Runners {
   }
 
   case class RunSettings(
-    optimise: Boolean = false
+    optimise: Boolean = false,
+    outDir: Option[String] = None
   )
 }

tests/run-with-compiler/quote-run-with-settings.check

@@ -5,5 +5,6 @@
 }
 foo
 5
+
 foo
 5

tests/run-with-compiler/quote-run-with-settings.scala

@@ -1,4 +1,6 @@
 
+import java.nio.file.{Files, Paths}
+
 import dotty.tools.dotc.quoted.Runners._
 
 import scala.quoted._
@@ -11,7 +13,17 @@ object Test {
       2 + a
     }
     println(expr.show)
-    println(run(expr.run, RunSettings(optimise = true)))
     println(expr.run)
+    println()
+
+    val outDir = Paths.get("../out/out-quoted-1")
+    val classFile = outDir.resolve("Quoted.class")
+
+    Files.deleteIfExists(classFile)
+
+    val settings = RunSettings(optimise = true, outDir = Some(outDir.toString))
+
+    println(run(expr, settings))
+    assert(Files.exists(classFile))
   }
 }