1 files changed, 63 insertions, 0 deletions
diff --git a/integration/test/resources/play-json/headers.sc b/integration/test/resources/play-json/headers.sc
new file mode 100644
index 00000000..a66eaedd
--- /dev/null
+++ b/integration/test/resources/play-json/headers.sc
@@ -0,0 +1,63 @@
+import $ivy.`com.github.rockjam::license-headers:0.0.1`
+
+import ammonite.ops._
+import mill._, scalalib._
+import mill.eval.PathRef
+import de.heikoseeberger.sbtheader.{CommentStyle, FileType, HeaderCreator, License}
+
+trait Headers extends ScalaModule {
+
+  val headerMappings = Map(
+    FileType.scala.extension -> (FileType.scala, CommentStyle.cStyleBlockComment),
+    FileType.java.extension -> (FileType.java, CommentStyle.cStyleBlockComment)
+  )
+
+  val license = {
+    val currentYear = java.time.Year.now(java.time.Clock.systemUTC).getValue
+    License.Custom(
+      s"Copyright (C) 2009-$currentYear Lightbend Inc. <https://www.lightbend.com>"
+    )
+  }
+
+  def headerCreate() = T.command {
+    val withoutHeaders = filesWithoutHeaders(sources() ++ resources())
+
+    val updatedFiles = withoutHeaders.map {
+      case (file, updated) =>
+        write.over(file, updated)
+        file
+    }
+
+    if (updatedFiles.nonEmpty) {
+      T.ctx.log.info(
+        s"Headers created for ${updatedFiles.size} files:\n${updatedFiles.mkString("\n")}")
+    }
+    updatedFiles
+  }
+
+  def headerCheck() = T.command {
+    val withoutHeaders = filesWithoutHeaders(sources() ++ resources()).map(_._1)
+
+    if (withoutHeaders.nonEmpty) {
+      sys.error(
+        s"There are files without headers!\n${withoutHeaders.mkString("\n")}"
+      )
+    }
+  }
+
+  private def filesWithoutHeaders(input: Seq[PathRef]) = {
+    for {
+      ref <- input if exists(ref.path)
+      file <- ls.rec(ref.path) if file.isFile
+      (fileType, commentStyle) <- headerMappings.get(file.ext)
+      updatedContent <- HeaderCreator(
+        fileType,
+        commentStyle,
+        license,
+        log = _ => (),
+        read.getInputStream(file)
+      ).createText
+    } yield file -> updatedContent
+  }
+
+}