blob: d2fdcb53fdf0a262d68f4092e73269acf12474a8 (
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
74
75
76
77
78
|
package forge
import java.nio.{file => jnio}
import sourcecode.Enclosing
import scala.collection.mutable
class Evaluator(workspacePath: jnio.Path,
enclosingBase: DefCtx){
/**
* Takes the given targets, finds
*/
def topoSortedTransitiveTargets(sourceTargets: Seq[Target[_]]) = {
val transitiveTargetSet = mutable.Set.empty[Target[_]]
def rec(t: Target[_]): Unit = {
if (transitiveTargetSet.contains(t)) () // do nothing
else {
transitiveTargetSet.add(t)
t.inputs.foreach(rec)
}
}
sourceTargets.foreach(rec)
val transitiveTargets = transitiveTargetSet.toVector
val targetIndices = transitiveTargets.zipWithIndex.toMap
val numberedEdges =
for(i <- transitiveTargets.indices)
yield transitiveTargets(i).inputs.map(targetIndices)
val sortedClusters = Tarjans(numberedEdges)
val nonTrivialClusters = sortedClusters.filter(_.length > 1)
assert(nonTrivialClusters.isEmpty, nonTrivialClusters)
sortedClusters.flatten.map(transitiveTargets)
}
def apply[T](t: Target[T])
(implicit enclosing: Enclosing): T = {
jnio.Files.createDirectories(workspacePath)
val sortedTargets = topoSortedTransitiveTargets(Seq(t))
val results = mutable.Map.empty[Target[_], Any]
for (target <- sortedTargets){
val inputResults = target.inputs.map(results)
val targetDestPath = target.defCtx.staticEnclosing match{
case Some(enclosingStr) =>
val targetDestPath = workspacePath.resolve(
jnio.Paths.get(enclosingStr.stripSuffix(enclosingBase.staticEnclosing.getOrElse("")))
)
deleteRec(targetDestPath)
targetDestPath
case None => jnio.Files.createTempDirectory(null)
}
results(target) = target.evaluate(
new Args(inputResults.toIndexedSeq, targetDestPath)
)
}
results(t).asInstanceOf[T]
}
def deleteRec(path: jnio.Path) = {
if (jnio.Files.exists(path)){
import collection.JavaConverters._
jnio.Files.walk(path).iterator()
.asScala
.toArray
.reverseIterator
.map(jnio.Files.deleteIfExists)
}
}
}
|