aboutsummaryrefslogtreecommitdiff
path: root/stage2/Stage2.scala
blob: 93f0a774ef8927cf2f777a19f5aff21c7b3f91ff (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package cbt
import java.io._
import java.util._

object Stage2 extends Stage2Base{
  def getBuild(context: Context) = {
    new Lib( context.logger ).loadRoot( context ).finalBuild
  }

  def run( args: Stage2Args ): ExitCode = {
    import args.logger
    val paths = CbtPaths(args.cbtHome,args.cache)
    import paths._
    val lib = new Lib(args.logger)
    logger.stage2(s"Stage2 start")
    val loop = args.args.lift(0) == Some("loop")

    val taskIndex = if (loop) {
      1
    } else {
      0
    }
    val task = args.args.lift( taskIndex )

    val context: Context = new ContextImplementation(
      args.cwd,
      args.cwd,
      args.args.drop( taskIndex +1 ).toArray,
      logger.enabledLoggers.toArray,
      logger.start,
      args.stage2LastModified,
      null,
      args.classLoaderCache.hashMap,
      args.transientCache,
      args.cache,
      args.cbtHome,
      args.cbtHome,
      args.compatibilityTarget,
      null
    )
    val first = lib.loadRoot( context )
    val build = first.finalBuild

    val res =
      if (loop) {
        // TODO: this should allow looping over task specific files, like test files as well
        val triggerFiles = first.triggerLoopFiles.map(lib.realpath)
        val triggerCbtFiles = Seq( nailgun, stage1, stage2 ).map(lib.realpath _)
        val allTriggerFiles = triggerFiles ++ triggerCbtFiles

        logger.loop("Looping change detection over:\n - "++allTriggerFiles.mkString("\n - "))

        lib.watch(allTriggerFiles){
          case file if triggerCbtFiles.exists(file.toString startsWith _.toString) =>
            logger.loop("Change is in CBT's own source code.")
            logger.loop("Restarting CBT.")
            scala.util.control.Breaks.break

          case file if triggerFiles.exists(file.toString startsWith _.toString) =>
            val build = lib.loadRoot(context).finalBuild
            logger.loop(s"Re-running $task for " ++ build.show)
            lib.callReflective(build, task)
        }
        ExitCode.Success
      } else {
        val code = lib.callReflective(build, task)
        logger.stage2(s"Stage2 end")
        code
      }

    res
  }
}