Link directly to a JAR

Author: nicolasstucki

compiler/src/dotty/tools/dotc/CompilationUnit.scala

@@ -3,8 +3,7 @@ package dotc
 
 import dotty.tools.dotc.core.Types.Type // Do not remove me #3383
 import util.SourceFile
-import ast.{tpd, untpd}
-import dotty.tools.dotc.ast.tpd.{ Tree, TreeTraverser }
+import dotty.tools.dotc.ast.{ tpd, untpd }
 import dotty.tools.dotc.core.Contexts.Context
 import dotty.tools.dotc.core.SymDenotations.ClassDenotation
 import dotty.tools.dotc.core.Symbols._
@@ -26,17 +25,17 @@ class CompilationUnit(val source: SourceFile) {
 object CompilationUnit {
 
   /** Make a compilation unit for top class `clsd` with the contends of the `unpickled` */
-  def mkCompilationUnit(clsd: ClassDenotation, unpickled: Tree, forceTrees: Boolean)(implicit ctx: Context): CompilationUnit = {
+  def mkCompilationUnit(clsd: ClassDenotation, unpickled: tpd.Tree, forceTrees: Boolean)(implicit ctx: Context): CompilationUnit = {
     assert(!unpickled.isEmpty, unpickled)
-    val unit1 = new CompilationUnit(new SourceFile(clsd.symbol.sourceFile, Seq()))
+    val unit1 = new CompilationUnit(new SourceFile(clsd.symbol.associatedFile, Seq()))
     unit1.tpdTree = unpickled
     if (forceTrees)
       force.traverse(unit1.tpdTree)
     unit1
   }
 
   /** Force the tree to be loaded */
-  private object force extends TreeTraverser {
-    def traverse(tree: Tree)(implicit ctx: Context): Unit = traverseChildren(tree)
+  private object force extends tpd.TreeTraverser {
+    def traverse(tree: tpd.Tree)(implicit ctx: Context): Unit = traverseChildren(tree)
   }
 }

compiler/src/dotty/tools/dotc/transform/LinkAll.scala

@@ -34,7 +34,7 @@ class LinkAll extends Phase {
       }
     }
 
-    if (ctx.settings.Xlink.value) allUnits(Set.empty, units.toSet, Set.empty)
+    if (doLink) allUnits(Set.empty, units.toSet, Set.empty)
     else units
   }
 
@@ -62,6 +62,16 @@ class LinkAll extends Phase {
       else acc + topClass
     }
   }
+
+  private def doLink(implicit ctx: Context): Boolean = {
+    val link = ctx.settings.Xlink
+    if (link.value) {
+      val out = ctx.settings.outputDir
+      if (out.value.endsWith(".jar"))
+        ctx.error("With " + link.name + " the output directory " + out.name + " should be a .jar\n" + out.value)
+      true
+    } else false
+  }
 }
 
 object LinkAll {

compiler/test/dotty/tools/dotc/LinkTests.scala

@@ -2,6 +2,8 @@ package dotty
 package tools
 package dotc
 
+import dotty.tools.io.Jar
+
 import java.io.{File => JFile}
 import java.nio.file.{Files, Path, Paths}
 
@@ -61,11 +63,11 @@ class LinkTests extends ParallelTesting {
       for (check <- new JFile(sourceDir).listFiles().filter(_.toString.endsWith(checkExt))) {
         val outDir = {
           def path(str: String) = str.substring(linkDir.length, str.length - checkExt.length)
-          defaultOutputDir + testName + path(check.toString) + "/"
+          defaultOutputDir + testName + path(check.toString) + "/linked.jar"
         }
         val expectedClasses = scala.io.Source.fromFile(check).getLines().toSet
-        val actualClasses = Files.walk(Paths.get(outDir)).iterator().asScala.collect {
-          case f if f.toString.endsWith(".class") => f.toString.substring(outDir.length, f.toString.length - ".class".length)
+        val actualClasses = new Jar(outDir).iterator.collect {
+          case f if f.toString.endsWith(".class") => f.toString.substring(0, f.toString.length - ".class".length)
         }.toSet
         assertEquals(check.toString, expectedClasses, actualClasses)
       }

compiler/test/dotty/tools/dotc/MissingCoreLibTests.scala

@@ -12,7 +12,7 @@ class MissingCoreLibTests {
   @Test def missingDottyLib: Unit = {
     val classPath = mkClassPath(Jars.dottyCompiler :: Jars.dottyInterfaces :: Jars.dottyExtras) // missing Jars.dottyLib
     val source = "../tests/neg/nolib/Foo.scala"
-    val options = Array("-classpath", classPath, source)
+    val options = Array("-classpath", classPath, "-d", "../out/", source)
     val reporter = Main.process(options)
     assertEquals(1, reporter.errorCount)
     val errorMessage = reporter.allErrors.head.message

compiler/test/dotty/tools/vulpix/ParallelTesting.scala

@@ -1027,18 +1027,22 @@ trait ParallelTesting extends RunnerOrchestration { self =>
   }
 
   /** Create out directory for directory `d` */
-  private def createOutputDirsForDir(d: JFile, sourceDir: JFile, outDir: String): JFile = {
+  private def createOutputDirsForDir(d: JFile, sourceDir: JFile, outDir: String, flags: TestFlags): JFile = {
     val targetDir = new JFile(outDir + s"${sourceDir.getName}/${d.getName}")
     targetDir.mkdirs()
-    targetDir
+    jaredIfNeeded(targetDir, flags)
   }
 
   /** Create out directory for `file` */
-  private def createOutputDirsForFile(file: JFile, sourceDir: JFile, outDir: String): JFile = {
+  private def createOutputDirsForFile(file: JFile, sourceDir: JFile, outDir: String, flags: TestFlags): JFile = {
     val uniqueSubdir = file.getName.substring(0, file.getName.lastIndexOf('.'))
     val targetDir = new JFile(outDir + s"${sourceDir.getName}/$uniqueSubdir")
     targetDir.mkdirs()
-    targetDir
+    jaredIfNeeded(targetDir, flags)
+  }
+
+  private def jaredIfNeeded(targetDir: JFile, flags: TestFlags): JFile = {
+    new JFile(targetDir + (if (flags.isLinkTest) "/linked.jar" else ""))
   }
 
   /** Make sure that directory string is as expected */
@@ -1073,7 +1077,7 @@ trait ParallelTesting extends RunnerOrchestration { self =>
       testGroup.name,
       Array(sourceFile),
       flags,
-      createOutputDirsForFile(sourceFile, parent, outDir)
+      createOutputDirsForFile(sourceFile, parent, outDir, flags)
     )
     new CompilationTest(target)
   }
@@ -1095,7 +1099,7 @@ trait ParallelTesting extends RunnerOrchestration { self =>
         (parent ne null) && parent.exists && parent.isDirectory,
       s"Source file: $f, didn't exist"
     )
-    val tastySource = createOutputDirsForFile(sourceFile, parent, outDir)
+    val tastySource = createOutputDirsForFile(sourceFile, parent, outDir, flags)
     val target = JointCompilationSource(
       testGroup.name,
       Array(sourceFile),
@@ -1180,8 +1184,8 @@ trait ParallelTesting extends RunnerOrchestration { self =>
     val (dirs, files) = compilationTargets(sourceDir)
 
     val targets =
-      files.map(f => JointCompilationSource(testGroup.name, Array(f), flags, createOutputDirsForFile(f, sourceDir, outDir))) ++
-      dirs.map(dir => SeparateCompilationSource(testGroup.name, dir, flags, createOutputDirsForDir(dir, sourceDir, outDir)))
+      files.map(f => JointCompilationSource(testGroup.name, Array(f), flags, createOutputDirsForFile(f, sourceDir, outDir, flags))) ++
+      dirs.map(dir => SeparateCompilationSource(testGroup.name, dir, flags, createOutputDirsForDir(dir, sourceDir, outDir, flags)))
 
     // Create a CompilationTest and let the user decide whether to execute a pos or a neg test
     new CompilationTest(targets)
@@ -1217,7 +1221,7 @@ trait ParallelTesting extends RunnerOrchestration { self =>
 
     val targets =
       files.map { f =>
-        val classpath = createOutputDirsForFile(f, sourceDir, outDir)
+        val classpath = createOutputDirsForFile(f, sourceDir, outDir, flags)
         JointCompilationSource(testGroup.name, Array(f), flags.withClasspath(classpath.getPath), classpath, fromTasty = true)
       }
     // TODO add SeparateCompilationSource from tasty?
@@ -1240,7 +1244,7 @@ trait ParallelTesting extends RunnerOrchestration { self =>
     val (_, files) = compilationTargets(sourceDir)
 
     val targets = files.map { file =>
-      JointCompilationSource(testGroup.name, Array(file), flags, createOutputDirsForFile(file, sourceDir, outDir))
+      JointCompilationSource(testGroup.name, Array(file), flags, createOutputDirsForFile(file, sourceDir, outDir, flags))
     }
 
     // Create a CompilationTest and let the user decide whether to execute a pos or a neg test

compiler/test/dotty/tools/vulpix/TestFlags.scala

@@ -15,6 +15,8 @@ final case class TestFlags(
     TestFlags(s"$defaultClassPath:$classPath", runClassPath, options)
 
   def all: Array[String] = Array("-classpath", defaultClassPath) ++ options
+
+  def isLinkTest: Boolean = options.contains("-Xlink")
 }
 
 object TestFlags {