1 files changed, 99 insertions, 0 deletions
diff --git a/contrib/playlib/src/mill/playlib/Static.scala b/contrib/playlib/src/mill/playlib/Static.scala
new file mode 100644
index 00000000..e6cd7779
--- /dev/null
+++ b/contrib/playlib/src/mill/playlib/Static.scala
@@ -0,0 +1,99 @@
+package mill.playlib
+
+import java.net.URI
+import java.nio.file._
+import java.nio.file.attribute.BasicFileAttributes
+import java.util
+
+import mill.{PathRef, T}
+import mill.scalalib.{Lib, ScalaModule}
+
+trait Static extends ScalaModule {
+  /**
+    * project resources including configuration, webjars and static assets
+    */
+  override def resources = T.sources {
+    super.resources() :+ webJarResources() :+ staticAssets()
+  }
+
+  /**
+    * Resource base path of packaged assets (path they will appear in in the jar)
+    */
+  def assetsPath = T{ "public" }
+
+  /**
+    *  Directories to include assets from
+    */
+  def assetSources = T.sources{ millSourcePath / assetsPath() }
+
+  /*
+  Collected static assets for the project
+   */
+  def staticAssets = T {
+    val toPath = os.Path(assetsPath(), T.ctx().dest)
+    assetSources().foreach{ pathRef =>
+      val fromPath = pathRef.path
+      if (os.isDir(fromPath)) {
+        os.walk(fromPath).filter(os.isFile(_)).foreach{ p =>
+          os.copy(p, toPath / p.relativeTo(fromPath), createFolders = true)
+        }
+      }
+    }
+    PathRef(T.ctx().dest)
+  }
+
+  /**
+    * webjar dependencies - created from transitive ivy deps
+    */
+  def webJarDeps = T{
+    transitiveIvyDeps().filter(_.dep.module.organization == "org.webjars")
+  }
+
+  /**
+    * jar files of web jars
+    */
+  def webJars = T{
+    Lib.resolveDependencies(repositories, Lib.depToDependency(_, scalaVersion()), webJarDeps())
+  }
+
+  /**
+    * webjar resources extracted from their source jars with version from path removed
+    */
+  def webJarResources = T {
+    extractWebJars(webJars().toSeq, os.Path(assetsPath(), T.ctx().dest) / 'lib)
+    PathRef(T.ctx().dest)
+  }
+
+  private def extractWebJars(jars: Seq[PathRef], webJarBase: os.Path): Unit = {
+    import scala.collection.JavaConverters._
+    val prefix = "/META-INF/resources/webjars/"
+
+    jars.foreach{ jarRef =>
+      val uri = s"jar:file:${jarRef.path}"
+      val env = Map.empty[String,String].asJava
+
+      val zipFs = FileSystems.newFileSystem(URI.create(uri), env)
+      try {
+        for(root <- zipFs.getRootDirectories.asScala) {
+          Files.walkFileTree(root, util.EnumSet.noneOf(classOf[FileVisitOption]), Int.MaxValue,
+            new SimpleFileVisitor[Path] {
+              override def visitFile(file: Path, attrs: BasicFileAttributes) = {
+                if (file.startsWith(prefix)) {
+                  val rel = os.RelPath(file.toString.substring(prefix.length))
+                  val toFile = webJarBase / os.RelPath(rel.segments(0) +: rel.segments.drop(2), 0)
+                  //println(s"$file -> $toFile")
+                  os.makeDir.all(toFile / os.up)
+                  Files.copy(file, toFile.toNIO)
+                }
+                FileVisitResult.CONTINUE
+              }
+            }
+          )
+        }
+      }
+      finally {
+        zipFs.close()
+      }
+    }
+  }
+}