From ccd03eeba20b8479565789fda546b33257794823 Mon Sep 17 00:00:00 2001
From: Eugene Yokota <eed3si9n@gmail.com>
Date: Fri, 27 Apr 2018 19:28:20 -0400
Subject: [PATCH] resolve Scala compiler using Ivy

When Scala compiler is not found in the boot directory, use Ivy to resolve it.
---
 .../test/scala/sbt/RunFromSourceMain.scala    | 48 +++++++++++++++++--
 1 file changed, 43 insertions(+), 5 deletions(-)

diff --git a/sbt/src/test/scala/sbt/RunFromSourceMain.scala b/sbt/src/test/scala/sbt/RunFromSourceMain.scala
index a625da9a8..7212d570e 100644
--- a/sbt/src/test/scala/sbt/RunFromSourceMain.scala
+++ b/sbt/src/test/scala/sbt/RunFromSourceMain.scala
@@ -67,9 +67,44 @@ object RunFromSourceMain {
     def apply[T](lockFile: File, run: java.util.concurrent.Callable[T]) = run.call()
   }
 
+  private lazy val bootDirectory: File = file(sys.props("user.home")) / ".sbt" / "boot"
+  private lazy val scalaHome: File = {
+    val scalaHome0 = bootDirectory / s"scala-$scalaVersion"
+    if (scalaHome0.exists) scalaHome0
+    else {
+      val target = new File("target").getAbsoluteFile
+      val fakeboot = target / "fakeboot"
+      val scalaHome1 = fakeboot / s"scala-$scalaVersion"
+      val scalaHome1Lib = scalaHome1 / "lib"
+      val scalaHome1Temp = scalaHome1 / "temp"
+      if (scalaHome1Lib.exists) ()
+      else {
+        IO.createDirectories(List(scalaHome1Lib, scalaHome1Temp))
+        val log = sbt.util.LogExchange.logger("run-from-source")
+        val lm = {
+          import sbt.librarymanagement.ivy.IvyDependencyResolution
+          val ivyConfig = InlineIvyConfiguration().withLog(log)
+          IvyDependencyResolution(ivyConfig)
+        }
+        val Name = """(.*)(\-[\d|\.]+)\.jar""".r
+        val module = "org.scala-lang" % "scala-compiler" % scalaVersion
+        lm.retrieve(module, scalaModuleInfo = None, scalaHome1Temp, log) match {
+          case Right(_) =>
+            (scalaHome1Temp ** "*.jar").get foreach { x =>
+              val Name(head, _) = x.getName
+              IO.copyFile(x, scalaHome1Lib / (head + ".jar"))
+            }
+          case Left(w) => sys.error(w.toString)
+        }
+      }
+      scalaHome1
+    }
+  }
+
   private def getConf(baseDir: File, args: Seq[String]): AppConfiguration = new AppConfiguration {
     def baseDirectory = baseDir
     def arguments = args.toArray
+
     def provider = new AppProvider { appProvider =>
       def scalaProvider = new ScalaProvider { scalaProvider =>
         def scalaOrg = "org.scala-lang"
@@ -80,7 +115,7 @@ object RunFromSourceMain {
           def app(id: xsbti.ApplicationID, version: String) = appProvider
           def topLoader = new java.net.URLClassLoader(Array(), null)
           def globalLock = noGlobalLock
-          def bootDirectory = file(sys.props("user.home")) / ".sbt" / "boot"
+          def bootDirectory = RunFromSourceMain.bootDirectory
           def ivyRepositories = Array()
           def appRepositories = Array()
           def isOverrideRepositories = false
@@ -88,11 +123,14 @@ object RunFromSourceMain {
           def checksums = Array("sha1", "md5")
         }
         def version = scalaVersion
-        def libDir: File = launcher.bootDirectory / s"scala-$version" / "lib"
+        lazy val libDir: File = RunFromSourceMain.scalaHome / "lib"
         def jar(name: String): File = libDir / s"$name.jar"
-        def libraryJar = jar("scala-library")
-        def compilerJar = jar("scala-compiler")
-        def jars = libDir.listFiles(f => !f.isDirectory && f.getName.endsWith(".jar"))
+        lazy val libraryJar = jar("scala-library")
+        lazy val compilerJar = jar("scala-compiler")
+        lazy val jars = {
+          assert(libDir.exists)
+          libDir.listFiles(f => !f.isDirectory && f.getName.endsWith(".jar"))
+        }
         def loader = new java.net.URLClassLoader(jars map (_.toURI.toURL), null)
         def app(id: xsbti.ApplicationID) = appProvider
       }