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
|
package mill
package contrib.scalapblib
import java.io.File
import java.lang.reflect.Method
import java.net.URLClassLoader
import mill.api.PathRef
class ScalaPBWorker {
private var scalaPBInstanceCache = Option.empty[(Long, ScalaPBWorkerApi)]
private def scalaPB(scalaPBClasspath: Agg[os.Path], protocPath: Option[String]) = {
val classloaderSig = scalaPBClasspath.map(p => p.toString().hashCode + os.mtime(p)).sum
scalaPBInstanceCache match {
case Some((sig, instance)) if sig == classloaderSig => instance
case _ =>
val cl = new URLClassLoader(scalaPBClasspath.map(_.toIO.toURI.toURL).toArray)
val scalaPBCompilerClass = cl.loadClass("scalapb.ScalaPBC")
val mainMethod = scalaPBCompilerClass.getMethod("main", classOf[Array[java.lang.String]])
val instance = new ScalaPBWorkerApi {
override def compileScalaPB(source: File, scalaPBOptions: String, generatedDirectory: File) {
val opts = if (scalaPBOptions.isEmpty) "" else scalaPBOptions + ":"
val args = protocPath.map(path => s"--protoc=$path").toList ++ List(
"--throw",
s"--scala_out=${opts}${generatedDirectory.getCanonicalPath}",
s"--proto_path=${source.getParentFile.getCanonicalPath}",
source.getCanonicalPath
)
mainMethod.invoke(null, args.toArray)
}
}
scalaPBInstanceCache = Some((classloaderSig, instance))
instance
}
}
def compile(scalaPBClasspath: Agg[os.Path], protocPath: Option[String], scalaPBSources: Seq[os.Path], scalaPBOptions: String, dest: os.Path)
(implicit ctx: mill.api.Ctx): mill.api.Result[PathRef] = {
val compiler = scalaPB(scalaPBClasspath, protocPath)
def compileScalaPBDir(inputDir: os.Path) {
// ls throws if the path doesn't exist
if (inputDir.toIO.exists) {
os.walk(inputDir).filter(_.last.matches(".*.proto"))
.foreach { proto =>
compiler.compileScalaPB(proto.toIO, scalaPBOptions, dest.toIO)
}
}
}
scalaPBSources.foreach(compileScalaPBDir)
mill.api.Result.Success(PathRef(dest))
}
}
trait ScalaPBWorkerApi {
def compileScalaPB(source: File, scalaPBOptions: String, generatedDirectory: File)
}
object ScalaPBWorkerApi {
def scalaPBWorker = new ScalaPBWorker()
}
|
