1 files changed, 74 insertions, 0 deletions
diff --git a/main/src/modules/Util.scala b/main/src/modules/Util.scala
new file mode 100644
index 00000000..029626fe
--- /dev/null
+++ b/main/src/modules/Util.scala
@@ -0,0 +1,74 @@
+package mill.modules
+
+
+import coursier.Repository
+import mill.api.{PathRef, IO}
+import mill.util.Ctx
+import mill.api.Loose
+
+
+object Util {
+  def cleanupScaladoc(v: String) = {
+    v.linesIterator.map(
+      _.dropWhile(_.isWhitespace)
+        .stripPrefix("/**")
+        .stripPrefix("*/")
+        .stripPrefix("*")
+        .dropWhile(_.isWhitespace)
+    ).toArray
+      .dropWhile(_.isEmpty)
+      .reverse
+      .dropWhile(_.isEmpty)
+      .reverse
+  }
+  def download(url: String, dest: os.RelPath = "download")(implicit ctx: Ctx.Dest) = {
+    val out = ctx.dest / dest
+
+    val website = new java.net.URI(url).toURL
+    val rbc = java.nio.channels.Channels.newChannel(website.openStream)
+    try{
+      val fos = new java.io.FileOutputStream(out.toIO)
+      try{
+        fos.getChannel.transferFrom(rbc, 0, java.lang.Long.MAX_VALUE)
+        PathRef(out)
+      } finally{
+        fos.close()
+      }
+    } finally{
+      rbc.close()
+    }
+  }
+
+  def downloadUnpackZip(url: String, dest: os.RelPath = "unpacked")
+                       (implicit ctx: Ctx.Dest) = {
+
+    val tmpName = if (dest == os.rel / "tmp.zip") "tmp2.zip" else "tmp.zip"
+    val downloaded = download(url, tmpName)
+    IO.unpackZip(downloaded.path, dest)
+  }
+
+
+  def millProjectModule(key: String,
+                        artifact: String,
+                        repositories: Seq[Repository],
+                        resolveFilter: os.Path => Boolean = _ => true,
+                        artifactSuffix: String = "_2.12") = {
+    val localPath = sys.props(key)
+    if (localPath != null) {
+      mill.api.Result.Success(
+        mill.api.Loose.Agg.from(localPath.split(',').map(p => PathRef(os.Path(p), quick = true)))
+      )
+    } else {
+      mill.modules.Jvm.resolveDependencies(
+        repositories,
+        Seq(
+          coursier.Dependency(
+            coursier.Module("com.lihaoyi", artifact + artifactSuffix),
+            sys.props("MILL_VERSION")
+          )
+        ),
+        Nil
+      ).map(_.filter(x => resolveFilter(x.path)))
+    }
+  }
+}