3

我在多模块 SBT 项目中有模块 A 和模块 B。我想为模块 B 编写一个资源生成器任务，该任务从模块 A 调用代码。一种方法是从模块 A 中提取所有代码，project/但这是不可行的，因为模块 A 很大，我想把它放在哪里它是（见https://stackoverflow.com/a/47323703/471136）。我如何在 SBT 中做到这一点？

其次，是否有可能完全摆脱模块 B，即我希望模块 A 的资源生成器任务实际上从模块 A 调用代码，但模块 A 是根模块并且不存在于 SBT 中project？

注意：这个问题不是这个问题的重复：Defining sbt task that invokes method from project code？因为那个决定将代码移动到 SBT 的项目中，这是我在这里特别想避免的。

4

2 回答 2

2

我想我正在做类似你在以下项目中的第一部分的事情：我的模块gen相当于你的模块A，我的模块核心相当于你的模块B。未经测试，结构大致如下：

// taking inspiration from
// http://stackoverflow.com/questions/11509843/
lazy val ugenGenerator = TaskKey[Seq[File]]("ugen-generate", "Generate UGen class files")

lazy val gen = Project(id = "gen", base = file("gen")) ...

lazy val core = Project(id = "core", base = file("core"))
  .settings(
    sourceGenerators in Compile <+= ugenGenerator in Compile,
    ugenGenerator in Compile := {
      val src   = (sourceManaged       in Compile        ).value
      val cp    = (dependencyClasspath in Runtime in gen ).value
      val st    = streams.value
      runUGenGenerator(description.value, outputDir = src, 
        cp = cp.files, log = st.log)
    }
  )

def runUGenGenerator(name: String, outputDir: File, cp: Seq[File],
                    log: Logger): Seq[File] = {
  val mainClass   = "my.class.from.Gen"
  val tmp         = java.io.File.createTempFile("sources", ".txt")
  val os          = new java.io.FileOutputStream(tmp)

  log.info(s"Generating UGen source code in $outputDir for $name")

  try {
    val outs  = CustomOutput(os)
    val fOpt  = ForkOptions(javaHome = None, outputStrategy = Some(outs), bootJars = cp,
        workingDirectory = None, connectInput = false)
    val res: Int = Fork.scala(config = fOpt,
      arguments = mainClass :: "-d" :: outputDir.getAbsolutePath :: Nil)

    if (res != 0) {
      sys.error(s"UGen class file generator failed with exit code $res")
    }
  } finally {
    os.close()
  }
  val sources = scala.io.Source.fromFile(tmp).getLines().map(file).toList
  tmp.delete()
  sources
}

这在 sbt 0.13 中有效，但我没有时间弄清楚为什么它在 1.x 中不起作用。

sourceGenerators in Compile <+= ugenGenerator in Compile顺便说一句，如果没有不推荐使用的语法 ，我该如何编写？

于 2017-11-21T20:23:08.517 回答
1

基于上面的@0__ 答案，这是我的简化版本：

  /**
    * Util to run the main of a sub-module from within SBT
    *
    * @param cmd The cmd to run with the main class with args (if any)
    * @param module The sub-module
    */
  def runModuleMain(cmd: String, module: Reference) = Def.task {
    val log = streams.value.log
    log.info(s"Running $cmd ...")
    val classPath = (fullClasspath in Runtime in module).value.files
    val opt = ForkOptions(bootJars = classPath, outputStrategy = Some(LoggedOutput(log)))
    val res = Fork.scala(config = opt, arguments = cmd.split(' '))
    require(res == 0, s"$cmd exited with code $res")
  }

然后，您可以将其调用为：

resourceGenerators in Compile += Def.taskDyn {
  val dest = (resourceManaged in Compile).value
  IO.createDirectory(dest)
  runModuleMain(
    cmd = "com.mycompany.foo.ResourceGen $dest $arg2 $arg3 ...",
    module = $referenceToModule // submodule containing com.mycompany.foo.ResourceGen
  ).taskValue
  dest.listFiles()
}

还有另一种方法可以做到这一点：

resourceGenerators in Compile += Def.taskDyn {
  val dest = (resourceManaged in Compile).value
  IO.createDirectory(dest)      
  Def.task {
    val cmd = "com.mycompany.foo.ResourceGen $dest $arg2 $arg3 ..."        
    (runMain in Compile).toTask(" " + $cmd).value 
    dest.listFiles()
  }
}.taskValue
于 2017-11-22T18:27:53.037 回答