Scalafmt support

scalalib/src/mill/scalalib/scalafmt/ScalafmtModule.scala

@@ -0,0 +1,39 @@
+package mill.scalalib.scalafmt
+
+import ammonite.ops._
+import mill._
+import mill.define.{Command, Sources}
+import mill.scalalib._
+
+trait ScalafmtModule extends ScalaModule {
+
+  def reformat(): Command[Unit] = T.command {
+    ScalafmtWorkerModule
+      .worker()
+      .reformat(
+        filesToFormat(sources()),
+        scalafmtConfig().head,
+        scalafmtDeps().map(_.path)
+      )
+  }
+
+  def scalafmtVersion: T[String] = "1.5.1"
+
+  def scalafmtConfig: Sources = T.sources(pwd / ".scalafmt.conf")
+
+  def scalafmtDeps: T[Agg[PathRef]] = T {
+    Lib.resolveDependencies(
+      scalaWorker.repositories,
+      Lib.depToDependency(_, "2.12.4"),
+      Seq(ivy"com.geirsson::scalafmt-cli:${scalafmtVersion()}")
+    )
+  }
+
+  private def filesToFormat(sources: Seq[PathRef]) = {
+    for {
+      pathRef <- sources if exists(pathRef.path)
+      file <- ls.rec(pathRef.path) if file.isFile && file.ext == "scala"
+    } yield PathRef(file)
+  }
+
+}

scalalib/src/mill/scalalib/scalafmt/ScalafmtWorker.scala

@@ -0,0 +1,58 @@
+package mill.scalalib.scalafmt
+
+import ammonite.ops.{Path, exists}
+import mill._
+import mill.define.{Discover, ExternalModule, Worker}
+import mill.modules.Jvm
+import mill.util.Ctx
+
+import scala.collection.mutable
+
+object ScalafmtWorkerModule extends ExternalModule {
+  def worker: Worker[ScalafmtWorker] = T.worker { new ScalafmtWorker() }
+
+  lazy val millDiscover = Discover[this.type]
+}
+
+private[scalafmt] class ScalafmtWorker {
+  private val reformatted: mutable.Map[Path, Int] = mutable.Map.empty
+  private var configSig: Int = 0
+
+  def reformat(input: Seq[PathRef],
+               scalafmtConfig: PathRef,
+               scalafmtClasspath: Agg[Path])(implicit ctx: Ctx): Unit = {
+    val toFormat =
+      if (scalafmtConfig.sig != configSig) input
+      else
+        input.filterNot(ref => reformatted.get(ref.path).contains(ref.sig))
+
+    if (toFormat.nonEmpty) {
+      ctx.log.info(s"Formatting ${toFormat.size} Scala sources")
+      reformatAction(toFormat.map(_.path),
+                     scalafmtConfig.path,
+                     scalafmtClasspath)
+      reformatted ++= toFormat.map { ref =>
+        val updRef = PathRef(ref.path)
+        updRef.path -> updRef.sig
+      }
+      configSig = scalafmtConfig.sig
+    } else {
+      ctx.log.info(s"Everything is formatted already")
+    }
+  }
+
+  private val cliFlags = Seq("--non-interactive", "--quiet")
+
+  private def reformatAction(toFormat: Seq[Path],
+                             config: Path,
+                             classpath: Agg[Path])(implicit ctx: Ctx) = {
+    val configFlags =
+      if (exists(config)) Seq("--config", config.toString) else Seq.empty
+    Jvm.subprocess(
+      "org.scalafmt.cli.Cli",
+      classpath,
+      mainArgs = toFormat.map(_.toString) ++ configFlags ++ cliFlags
+    )
+  }
+
+}