因此,在对此进行了深入研究之后,我想出了一个解决方案。首先,您需要将您的项目分成两个子项目。 gen
拥有包含您的生成器代码的所有源代码。 use
依赖gen
并使用生成器。
import sbt._
import Keys._
import java.io.{ File ⇒ JFile, FileOutputStream }
object OverallBuild extends Build {
lazy val root = Project(id = "overall", base = file(".")).aggregate(gen, use)
lazy val gen = Project(id = "generate", base = file("gen"))
val myCodeGenerator = TaskKey[Seq[File]]("mycode-generate", "Generate My Awesome Code")
lazy val use = Project(id = "use", base = file("use"),
settings = Defaults.defaultSettings ++ Seq(
sourceGenerators in Compile <+= (myCodeGenerator in Compile),
myCodeGenerator in Compile <<=
(javaSource in Compile, dependencyClasspath in Runtime in gen) map {
(javaSource, cp) ⇒ runMyCodeGenerator(javaSource, cp.files)
})).dependsOn(gen)
def runMyCodeGenerator(javaSource: File, cp: Seq[File]): Seq[File] = {
val mainClass = "com.yourcompany.myCodeGenerator"
val tmp = JFile.createTempFile("sources", ".txt")
val os = new FileOutputStream(tmp)
try {
val i = new Fork.ForkScala(mainClass).fork(None, Nil, cp,
Seq(javaSource.toString),
None,
false,
CustomOutput(os)).exitValue()
if (i != 0) {
error("Trouble with code generator")
}
} finally {
os.close()
}
scala.io.Source.fromFile(tmp).getLines.map(f ⇒ file(f)).toList
}
}
在这种情况下,我正在生成 .java 文件,因此我将其传递javaSource
给了生成器。
重要的是,当我们在这里使用 sourceGenerators 时,执行的任务必须返回Seq[File]
所有生成的文件中的一个,以便 sbt 可以管理它们。在这个实现中,我们的生成器将完整路径文件名输出到标准输出,然后我们将它们保存到一个临时文件中。
就像 Scala 和 SBT 一样,你可以做任何事情,只需要深入研究它。