假设我有一个图书馆 a.com。每次在每个文件中,我都需要导入很多包,例如
import a.com._
import a.com.b._
import a.com.c
import a.com.Implicits._
我不想每次都在另一个项目的每个文件中编写这些代码。
另外,如果我想更改a.com
为a.net
,我必须更改每个文件。
有没有办法防止这种情况发生?
您可以生成源
构建.sbt
lazy val commonSettings = Seq(
scalaVersion := "2.13.1",
)
lazy val in = project
.settings(
commonSettings,
)
lazy val out = project
.settings(
sourceGenerators in Compile += Def.task {
Generator.gen(
inputDir = sourceDirectory.in(in, Compile).value,
outputDir = sourceManaged.in(Compile).value
)
}.taskValue,
commonSettings,
)
项目/build.sbt
libraryDependencies += "org.scalameta" %% "scalameta" % "4.2.3"
项目/Generator.scala
import sbt._
object Generator {
def gen(inputDir: File, outputDir: File): Seq[File] = {
val finder: PathFinder = inputDir ** "*.scala"
for(inputFile <- finder.get) yield {
val inputStr = IO.read(inputFile)
val outputFile = outputDir / inputFile.toURI.toString.stripPrefix(inputDir.toURI.toString)
val outputStr = Transformer.transform(inputStr)
IO.write(outputFile, outputStr)
outputFile
}
}
}
项目/Transformer.scala
import scala.meta._
object Transformer {
def transform(input: String): String = transform(input.parse[Source].get).toString
def transform(input: Tree): Tree = input match {
case source"..${List(q"package $eref { ..$stats }")}" =>
q"""package $eref {
import a.com._
import a.com.b._
import a.com.c
import a.com.Implicits._
..$stats
}"""
}
}
在/src/main/scala/com/example/App.scala
package com.example
object App {
}
out/target/scala-2.13/src_managed/main/scala/com/example/App.scala(之后sbt "; project out; clean; compile"
)
package com.example
import a.com._
import a.com.b._
import a.com.c
import a.com.Implicits._
object App