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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
package mill.scalajslib
import java.io.File
import java.net.URLClassLoader
import ammonite.ops.Path
import mill.define.Discover
import mill.eval.Result
import mill.util.Ctx
import mill.{Agg, T}
sealed trait OptimizeMode
object FastOpt extends OptimizeMode
object FullOpt extends OptimizeMode
sealed trait ModuleKind
object ModuleKind{
object NoModule extends ModuleKind
object CommonJSModule extends ModuleKind
}
class ScalaJSWorker {
private var scalaInstanceCache = Option.empty[(Long, ScalaJSWorkerApi)]
private def bridge(toolsClasspath: Agg[Path])
(implicit ctx: Ctx.Home) = {
val classloaderSig =
toolsClasspath.map(p => p.toString().hashCode + p.mtime.toMillis).sum
scalaInstanceCache match {
case Some((sig, bridge)) if sig == classloaderSig => bridge
case _ =>
val cl = mill.util.ClassLoader.create(
toolsClasspath.map(_.toIO.toURI.toURL).toVector,
getClass.getClassLoader
)
val bridge = cl
.loadClass("mill.scalajslib.bridge.ScalaJSWorker")
.getDeclaredConstructor()
.newInstance()
.asInstanceOf[ScalaJSWorkerApi]
scalaInstanceCache = Some((classloaderSig, bridge))
bridge
}
}
def link(toolsClasspath: Agg[Path],
sources: Agg[Path],
libraries: Agg[Path],
dest: File,
main: Option[String],
fullOpt: Boolean,
moduleKind: ModuleKind)
(implicit ctx: Ctx.Home): Result[Path] = {
bridge(toolsClasspath).link(
sources.items.map(_.toIO).toArray,
libraries.items.map(_.toIO).toArray,
dest,
main.orNull,
fullOpt,
moduleKind
).map(Path(_))
}
def run(toolsClasspath: Agg[Path], config: NodeJSConfig, linkedFile: File)
(implicit ctx: Ctx.Home): Unit = {
bridge(toolsClasspath).run(config, linkedFile)
}
def getFramework(toolsClasspath: Agg[Path],
config: NodeJSConfig,
frameworkName: String,
linkedFile: File)
(implicit ctx: Ctx.Home): (() => Unit, sbt.testing.Framework) = {
bridge(toolsClasspath).getFramework(config, frameworkName, linkedFile)
}
}
trait ScalaJSWorkerApi {
def link(sources: Array[File],
libraries: Array[File],
dest: File,
main: String,
fullOpt: Boolean,
moduleKind: ModuleKind): Result[File]
def run(config: NodeJSConfig, linkedFile: File): Unit
def getFramework(config: NodeJSConfig,
frameworkName: String,
linkedFile: File): (() => Unit, sbt.testing.Framework)
}
object ScalaJSWorkerApi extends mill.define.ExternalModule {
def scalaJSWorker = T.worker { new ScalaJSWorker() }
lazy val millDiscover = Discover[this.type]
}
|