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
|
package mill
package scalajslib
package worker
import java.io.File
import mill.eval.Result
import org.scalajs.core.tools.io._
import org.scalajs.core.tools.linker.{ModuleInitializer, Semantics, StandardLinker, ModuleKind => ScalaJSModuleKind}
import org.scalajs.core.tools.logging.ScalaConsoleLogger
import org.scalajs.jsenv.ConsoleJSConsole
import org.scalajs.jsenv.nodejs._
import org.scalajs.testadapter.TestAdapter
class ScalaJSWorkerImpl extends mill.scalajslib.ScalaJSWorkerApi {
def link(sources: Array[File],
libraries: Array[File],
dest: File,
main: String,
fullOpt: Boolean,
moduleKind: ModuleKind) = {
val semantics = fullOpt match {
case true => Semantics.Defaults.optimized
case false => Semantics.Defaults
}
val scalaJSModuleKind = moduleKind match {
case ModuleKind.NoModule => ScalaJSModuleKind.NoModule
case ModuleKind.CommonJSModule => ScalaJSModuleKind.CommonJSModule
}
val config = StandardLinker.Config()
.withOptimizer(fullOpt)
.withClosureCompilerIfAvailable(fullOpt)
.withSemantics(semantics)
.withModuleKind(scalaJSModuleKind)
val linker = StandardLinker(config)
val cache = new IRFileCache().newCache
val sourceIRs = sources.map(FileVirtualScalaJSIRFile)
val irContainers = FileScalaJSIRContainer.fromClasspath(libraries)
val libraryIRs = cache.cached(irContainers)
val destFile = AtomicWritableFileVirtualJSFile(dest)
val logger = new ScalaConsoleLogger
val initializer = Option(main).map { cls => ModuleInitializer.mainMethodWithArgs(cls, "main") }
try {
linker.link(sourceIRs ++ libraryIRs, initializer.toSeq, destFile, logger)
Result.Success(dest)
}catch {case e: org.scalajs.core.tools.linker.LinkingException =>
Result.Failure(e.getMessage)
}
}
def run(config: NodeJSConfig, linkedFile: File): Unit = {
nodeJSEnv(config)
.jsRunner(Seq(FileVirtualJSFile(linkedFile)))
.run(new ScalaConsoleLogger, ConsoleJSConsole)
}
def getFramework(config: NodeJSConfig,
frameworkName: String,
linkedFile: File): (() => Unit, sbt.testing.Framework) = {
val env = nodeJSEnv(config)
val tconfig = TestAdapter.Config().withLogger(new ScalaConsoleLogger)
val adapter =
new TestAdapter(env, Seq(FileVirtualJSFile(linkedFile)), tconfig)
(
() => adapter.close(),
adapter
.loadFrameworks(List(List(frameworkName)))
.flatten
.headOption
.getOrElse(throw new RuntimeException("Failed to get framework"))
)
}
def nodeJSEnv(config: NodeJSConfig): NodeJSEnv = {
new NodeJSEnv(
NodeJSEnv.Config()
.withExecutable(config.executable)
.withArgs(config.args)
.withEnv(config.env)
.withSourceMap(config.sourceMap))
}
}
|
