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
|
package mill.scalanativelib
import java.io.File
import java.net.URLClassLoader
import mill.define.{Discover, Worker}
import mill.{Agg, T}
import sbt.testing.Framework
class ScalaNativeWorker {
private var scalaInstanceCache = Option.empty[(Long, ScalaNativeWorkerApi)]
def impl(toolsClasspath: Agg[os.Path]): ScalaNativeWorkerApi = {
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 = new URLClassLoader(
toolsClasspath.map(_.toIO.toURI.toURL).toArray,
getClass.getClassLoader
)
try {
val bridge = cl
.loadClass("mill.scalanativelib.worker.ScalaNativeWorkerImpl")
.getDeclaredConstructor()
.newInstance()
.asInstanceOf[ScalaNativeWorkerApi]
scalaInstanceCache = Some((classloaderSig, bridge))
bridge
}
catch {
case e: Exception =>
e.printStackTrace()
throw e
}
}
}
}
// result wrapper to preserve some type safety
case class NativeConfig(config: Any)
trait ScalaNativeWorkerApi {
def discoverClang: os.Path
def discoverClangPP: os.Path
def discoverTarget(clang: os.Path, workDir: os.Path): String
def discoverCompileOptions: Seq[String]
def discoverLinkingOptions: Seq[String]
def config(nativeLibJar: os.Path,
mainClass: String,
classpath: Seq[os.Path],
nativeWorkdir: os.Path,
nativeClang: os.Path,
nativeClangPP: os.Path,
nativeTarget: String,
nativeCompileOptions: Seq[String],
nativeLinkingOptions: Seq[String],
nativeGC: String,
nativeLinkStubs: Boolean,
releaseMode: ReleaseMode,
logLevel: NativeLogLevel): NativeConfig
def defaultGarbageCollector: String
def nativeLink(nativeConfig: NativeConfig, outPath: os.Path): os.Path
def newScalaNativeFrameWork(framework: Framework, id: Int, testBinary: File,
logLevel: NativeLogLevel, envVars: Map[String, String]): Framework
}
object ScalaNativeWorkerApi extends mill.define.ExternalModule {
def scalaNativeWorker: Worker[ScalaNativeWorker] = T.worker { new ScalaNativeWorker() }
lazy val millDiscover = Discover[this.type]
}
|
