Merge pull request #102 from cvogt/chris

More classloading related fixes

8 files changed, 34 insertions, 35 deletions

diff --git a/stage2/AdminStage2.scala b/stage2/AdminStage2.scala
index 883b5ed..9d7dcb2 100644
--- a/stage2/AdminStage2.scala
+++ b/stage2/AdminStage2.scala
@@ -4,7 +4,7 @@ object AdminStage2 extends Stage2Base{
   def run( _args: Stage2Args ): Unit = {
     val args = _args.args.dropWhile(Seq("admin","direct") contains _)
     val lib = new Lib(_args.logger)
-    val adminTasks = new AdminTasks(lib, args, _args.cwd)
+    val adminTasks = new AdminTasks(lib, args, _args.cwd, _args.classLoaderCache)
     new lib.ReflectObject(adminTasks){
       def usage: String = "Available methods: " ++ lib.taskNames(adminTasks.getClass).mkString("  ")
     }.callNullary(args.lift(0))
diff --git a/stage2/AdminTasks.scala b/stage2/AdminTasks.scala
index 655b2b0..f189805 100644
--- a/stage2/AdminTasks.scala
+++ b/stage2/AdminTasks.scala
@@ -2,7 +2,7 @@ package cbt
 import scala.collection.immutable.Seq
 import java.io.{Console=>_,_}
 import java.nio.file._
-class AdminTasks(lib: Lib, args: Seq[String], cwd: File){
+class AdminTasks(lib: Lib, args: Seq[String], cwd: File, classLoaderCache: ClassLoaderCache){
   implicit val logger: Logger = lib.logger
   def resolve = {
     ClassPath.flatten(
@@ -31,14 +31,14 @@ class AdminTasks(lib: Lib, args: Seq[String], cwd: File){
     )
     // FIXME: this does not work quite yet, throws NoSuchFileException: /ammonite/repl/frontend/ReplBridge$.class
     lib.runMain(
-      "ammonite.repl.Main", Seq(), d.classLoader(new ClassLoaderCache(logger))
+      "ammonite.repl.Main", Seq(), d.classLoader(classLoaderCache)
     )
   }
   def scala = {
     val version = args.lift(1).getOrElse(constants.scalaVersion)
     val scalac = new ScalaCompilerDependency( version )
     lib.runMain(
-      "scala.tools.nsc.MainGenericRunner", Seq("-cp", scalac.classpath.string), scalac.classLoader(new ClassLoaderCache(logger))
+      "scala.tools.nsc.MainGenericRunner", Seq("-cp", scalac.classpath.string), scalac.classLoader(classLoaderCache)
     )
   }
   def scaffoldBasicBuild: Unit = lib.scaffoldBasicBuild( cwd )
diff --git a/stage2/BasicBuild.scala b/stage2/BasicBuild.scala
index 39cc9e3..2b49c93 100644
--- a/stage2/BasicBuild.scala
+++ b/stage2/BasicBuild.scala
@@ -23,7 +23,7 @@ class Build(val context: Context) extends Dependency with TriggerLoop with SbtDe
 
   override def canBeCached = false
   def enableConcurrency = false
-  final def projectDirectory: File = lib.realpath(context.cwd)
+  final def projectDirectory: File = lib.realpath(context.projectDirectory)
   assert( projectDirectory.exists, "projectDirectory does not exist: " ++ projectDirectory.string )
   final def usage: String = lib.usage(this.getClass, context)
 
@@ -89,7 +89,7 @@ class Build(val context: Context) extends Dependency with TriggerLoop with SbtDe
   ) = lib.ScalaDependency( groupId, artifactId, version, classifier, scalaVersion )
 
   final def BuildDependency(path: File) = cbt.BuildDependency(
-    context.copy( cwd = path, args = Seq() )
+    context.copy( projectDirectory = path, args = Seq() )
   )
 
   def triggerLoopFiles: Seq[File] = sources ++ transitiveDependencies.collect{ case b: TriggerLoop => b.triggerLoopFiles }.flatten
@@ -133,6 +133,7 @@ class Build(val context: Context) extends Dependency with TriggerLoop with SbtDe
 
   def test: ExitCode = lib.test(context)
 
+  /*
   context.logger.composition(">"*80)
   context.logger.composition("class   " ++ this.getClass.toString)
   context.logger.composition("dir     " ++ projectDirectory.string)
@@ -141,6 +142,7 @@ class Build(val context: Context) extends Dependency with TriggerLoop with SbtDe
   context.logger.composition("context " ++ context.toString)
   context.logger.composition("dependencyTree\n" ++ dependencyTree)
   context.logger.composition("<"*80)
+  */
 
   // ========== cbt internals ==========
   private[cbt] def finalBuild = this
diff --git a/stage2/BuildBuild.scala b/stage2/BuildBuild.scala
index 813b44b..c52c3c6 100644
--- a/stage2/BuildBuild.scala
+++ b/stage2/BuildBuild.scala
@@ -6,22 +6,20 @@ class BuildBuild(context: Context) extends Build(context){
   override def dependencies = Seq( CbtDependency()(context.logger) ) ++ super.dependencies
   def managedBuildDirectory: File = lib.realpath( projectDirectory.parent )
   val managedBuild = try{
-    val managedContext = context.copy( cwd = managedBuildDirectory )
-    val cl = new cbt.URLClassLoader(
-      exportedClasspath,
-      classOf[BuildBuild].getClassLoader // FIXME: this looks wrong. Should be ClassLoader.getSystemClassLoader but that crashes
-    )
-    cl
-      .loadClass(lib.buildClassName)
-      .getConstructor(classOf[Context])
-      .newInstance(managedContext)
-      .asInstanceOf[Build]
+    val managedContext = context.copy( projectDirectory = managedBuildDirectory )
+    val cl = classLoader(context.classLoaderCache)
+    logger.composition("Loading build at "+managedContext.projectDirectory)
+      cl
+        .loadClass(lib.buildClassName)
+        .getConstructor(classOf[Context])
+        .newInstance(managedContext)
+        .asInstanceOf[Build]
   } catch {
     case e: ClassNotFoundException if e.getMessage == lib.buildClassName => 
-      throw new Exception("You need to remove the directory or define a class Build in: "+context.cwd)
+      throw new Exception("You need to remove the directory or define a class Build in: "+context.projectDirectory)
     case e: Exception =>
-      throw new Exception("during build: "+context.cwd, e)
+      throw new Exception("during build: "+context.projectDirectory, e)
   }
   override def triggerLoopFiles = super.triggerLoopFiles ++ managedBuild.triggerLoopFiles
-  override def finalBuild = managedBuild.finalBuild
+  override def finalBuild = if( context.projectDirectory == context.cwd ) this else managedBuild.finalBuild
 }
diff --git a/stage2/BuildDependency.scala b/stage2/BuildDependency.scala
index 19357f9..8965cee 100644
--- a/stage2/BuildDependency.scala
+++ b/stage2/BuildDependency.scala
@@ -16,7 +16,7 @@ trait TriggerLoop extends Dependency{
 }
 /** You likely want to use the factory method in the BasicBuild class instead of this. */
 case class BuildDependency(context: Context) extends TriggerLoop{
-  override def show = this.getClass.getSimpleName ++ "(" ++ context.cwd.string ++ ")"
+  override def show = this.getClass.getSimpleName ++ "(" ++ context.projectDirectory.string ++ ")"
   final override lazy val logger = context.logger
   final override lazy val lib: Lib = new Lib(logger)
   private val root = lib.loadRoot( context.copy(args=Seq()) )
@@ -31,7 +31,7 @@ case class BuildDependency(context: Context) extends TriggerLoop{
 }
 /*
 case class DependencyOr(first: BuildDependency, second: JavaDependency) extends ProjectProxy with BuildDependencyBase{
-  val isFirst = new File(first.context.cwd).exists
+  val isFirst = new File(first.context.projectDirectory).exists
   def triggerLoopFiles = if(isFirst) first.triggerLoopFiles else Seq()
   protected val delegate = if(isFirst) first else second
 }
diff --git a/stage2/GitDependency.scala b/stage2/GitDependency.scala
index 27bf253..16423df 100644
--- a/stage2/GitDependency.scala
+++ b/stage2/GitDependency.scala
@@ -38,7 +38,7 @@ case class GitDependency(
 
     }
     val managedBuild = lib.loadDynamic(
-      context.copy( cwd = checkoutDirectory, args = Seq() )
+      context.copy( projectDirectory = checkoutDirectory, args = Seq() )
     )    
     Seq( managedBuild )
   }
diff --git a/stage2/Lib.scala b/stage2/Lib.scala
index 6b6263c..dbd6108 100644
--- a/stage2/Lib.scala
+++ b/stage2/Lib.scala
@@ -34,19 +34,19 @@ final class Lib(logger: Logger) extends Stage1Lib(logger) with Scaffold{
   This can either the Build itself, of if exists a BuildBuild or a BuildBuild for a BuildBuild and so on.
   */
   def loadRoot(context: Context, default: Context => Build = new Build(_)): Build = {
-    context.logger.composition( context.logger.showInvocation("Build.loadRoot",context) )
-    def findStartDir(cwd: File): File = {
-      val buildDir = realpath( cwd ++ "/build" )
-      if(buildDir.exists) findStartDir(buildDir) else cwd
+    context.logger.composition( context.logger.showInvocation("Build.loadRoot",context.projectDirectory) )
+    def findStartDir(projectDirectory: File): File = {
+      val buildDir = realpath( projectDirectory ++ "/build" )
+      if(buildDir.exists) findStartDir(buildDir) else projectDirectory
     }
 
-    val start = findStartDir(context.cwd)
+    val start = findStartDir(context.projectDirectory)
 
-    val useBasicBuildBuild = context.cwd == start
+    val useBasicBuildBuild = context.projectDirectory == start
 
     val rootBuildClassName = if( useBasicBuildBuild ) buildBuildClassName else buildClassName
     try{
-      if(useBasicBuildBuild) default( context ) else new cbt.BuildBuild( context.copy( cwd = start ) )
+      if(useBasicBuildBuild) default( context ) else new cbt.BuildBuild( context.copy( projectDirectory = start ) )
     } catch {
       case e:ClassNotFoundException if e.getMessage == rootBuildClassName =>
         throw new Exception(s"no class $rootBuildClassName found in " ++ start.string)
@@ -110,7 +110,7 @@ final class Lib(logger: Logger) extends Stage1Lib(logger) with Scaffold{
 
     logger.lib(s"invoke testDefault( $context )")
     val exitCode: ExitCode = loadDynamic(
-      context.copy( cwd = context.cwd ++ "/test", args = loggerArg.toVector ++ context.args ),
+      context.copy( projectDirectory = context.projectDirectory ++ "/test", args = loggerArg.toVector ++ context.args ),
       new Build(_) with mixins.Test
     ).run
     logger.lib(s"return testDefault( $context )")
@@ -145,7 +145,7 @@ final class Lib(logger: Logger) extends Stage1Lib(logger) with Scaffold{
     (
       (
         if( thisTasks.nonEmpty ){
-          s"""Methods provided by Build ${context.cwd}
+          s"""Methods provided by Build ${context.projectDirectory}
 
   ${thisTasks.mkString("  ")}
 
diff --git a/stage2/Stage2.scala b/stage2/Stage2.scala
index 4ae149c..ddadfb6 100644
--- a/stage2/Stage2.scala
+++ b/stage2/Stage2.scala
@@ -22,8 +22,8 @@ object Stage2 extends Stage2Base{
       0
     }
     val task = args.args.lift( taskIndex )
-
-    val context = Context( args.cwd, args.args.drop( taskIndex ), logger, args.cbtHasChanged, new ClassLoaderCache(logger) )
+    
+    val context = Context( args.cwd, args.cwd, args.args.drop( taskIndex ), logger, args.cbtHasChanged, args.classLoaderCache )
     val first = lib.loadRoot( context )
     val build = first.finalBuild
 
@@ -44,9 +44,8 @@ object Stage2 extends Stage2Base{
 
           case file if triggerFiles.exists(file.toString startsWith _.toString) =>
             val build = lib.loadDynamic(context)
-            val reflectBuild = new lib.ReflectBuild( build )
             logger.loop(s"Re-running $task for " ++ build.projectDirectory.toString)
-            reflectBuild.callNullary(task)
+            new lib.ReflectBuild(build).callNullary(task)
         }
       } else {
         new lib.ReflectBuild(build).callNullary(task)