Add trait to configure twirl integration for playframework

2 files changed, 30 insertions, 1 deletions

diff --git a/build.sc b/build.sc
index 0478f97d..992210bd 100755
--- a/build.sc
+++ b/build.sc
@@ -248,7 +248,7 @@ object contrib extends MillModule {
   }
 
   object playlib extends MillModule {
-    def moduleDeps = Seq(scalalib, playlib.api)
+    def moduleDeps = Seq(scalalib, twirllib, playlib.api)
 
     def testArgs = T {
       val mapping = Map(
diff --git a/contrib/playlib/src/mill/playlib/Twirl.scala b/contrib/playlib/src/mill/playlib/Twirl.scala
new file mode 100644
index 00000000..23aba86b
--- /dev/null
+++ b/contrib/playlib/src/mill/playlib/Twirl.scala
@@ -0,0 +1,29 @@
+package mill
+package playlib
+
+import mill.twirllib._
+
+private[playlib] trait Twirl extends TwirlModule with Layout {
+
+  override def twirlSources=T.sources{ app() }
+
+  override def twirlAdditionalImports = Seq(
+    "_root_.play.twirl.api.TwirlFeatureImports._",
+    "_root_.play.twirl.api.TwirlHelperImports._",
+    "_root_.play.twirl.api.Html",
+    "_root_.play.twirl.api.JavaScript",
+    "_root_.play.twirl.api.Txt",
+    "_root_.play.twirl.api.Xml",
+    "models._",
+    "controllers._",
+    "play.api.i18n._",
+    "views.html._",
+    "play.api.templates.PlayMagic._",
+    "play.api.mvc._",
+    "play.api.data._"
+  )
+
+  def twirlOutput = T{Seq(compileTwirl().classes)}
+
+  override def generatedSources = T{ super.generatedSources() ++ twirlOutput() }
+}