summaryrefslogtreecommitdiff
path: root/scalajslib/src/ScalaJSWorkerApi.scala
blob: bcaeb2d306192f0239feeac0b718cd2d68eb6177 (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
package mill.scalajslib

import java.io.File
import java.net.URLClassLoader

import mill.define.Discover
import mill.api.Result
import mill.api.Ctx
import mill.{Agg, T}
import mill.scalajslib.api._
class ScalaJSWorker {
  private var scalaInstanceCache = Option.empty[(Long, ScalaJSWorkerApi)]

  private def bridge(toolsClasspath: Agg[os.Path])
                    (implicit ctx: Ctx.Home) = {
    val classloaderSig =
      toolsClasspath.map(p => p.toString().hashCode + os.mtime(p)).sum
    scalaInstanceCache match {
      case Some((sig, bridge)) if sig == classloaderSig => bridge
      case _ =>
        val cl = mill.api.ClassLoader.create(
          toolsClasspath.map(_.toIO.toURI.toURL).toVector,
          getClass.getClassLoader
        )
        val bridge = cl
          .loadClass("mill.scalajslib.worker.ScalaJSWorkerImpl")
          .getDeclaredConstructor()
          .newInstance()
          .asInstanceOf[ScalaJSWorkerApi]
        scalaInstanceCache = Some((classloaderSig, bridge))
        bridge
    }
  }

  def link(toolsClasspath: Agg[os.Path],
           sources: Agg[os.Path],
           libraries: Agg[os.Path],
           dest: File,
           main: Option[String],
           fullOpt: Boolean,
           moduleKind: ModuleKind)
          (implicit ctx: Ctx.Home): Result[os.Path] = {
    bridge(toolsClasspath).link(
      sources.items.map(_.toIO).toArray,
      libraries.items.map(_.toIO).toArray,
      dest,
      main.orNull,
      fullOpt,
      moduleKind
    ).map(os.Path(_))
  }

  def run(toolsClasspath: Agg[os.Path], config: NodeJSConfig, linkedFile: File)
         (implicit ctx: Ctx.Home): Unit = {
    bridge(toolsClasspath).run(config, linkedFile)
  }

  def getFramework(toolsClasspath: Agg[os.Path],
                   config: NodeJSConfig,
                   frameworkName: String,
                   linkedFile: File)
                  (implicit ctx: Ctx.Home): (() => Unit, sbt.testing.Framework) = {
    bridge(toolsClasspath).getFramework(config, frameworkName, linkedFile)
  }

}

object ScalaJSWorkerApi extends mill.define.ExternalModule {

  def scalaJSWorker = T.worker { new ScalaJSWorker() }
  lazy val millDiscover = Discover[this.type]
}