From c285ddaabe980b72e6cff240ed1c22664913a2e6 Mon Sep 17 00:00:00 2001
From: Nathan Fischer <nfischer921@gmail.com>
Date: Tue, 9 Apr 2019 14:04:57 -0700
Subject: Flyway module

---
 contrib/flyway/src/FlywayModule.scala | 59 +++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 contrib/flyway/src/FlywayModule.scala

(limited to 'contrib/flyway')

diff --git a/contrib/flyway/src/FlywayModule.scala b/contrib/flyway/src/FlywayModule.scala
new file mode 100644
index 00000000..86e7ea32
--- /dev/null
+++ b/contrib/flyway/src/FlywayModule.scala
@@ -0,0 +1,59 @@
+package mill
+package contrib.flyway
+
+import java.net.URLClassLoader
+
+import mill.scalalib.Lib.resolveDependencies
+import mill.scalalib._
+import org.flywaydb.core.Flyway
+import org.flywaydb.core.api.logging.LogFactory
+import org.flywaydb.core.internal.configuration.{ConfigUtils => flyway}
+import org.flywaydb.core.internal.logging.console.ConsoleLog.Level
+import org.flywaydb.core.internal.logging.console.ConsoleLogCreator
+
+import scala.collection.JavaConverters._
+
+
+trait FlywayModule extends JavaModule {
+
+  def flywayUrl: T[String]
+  def flywayUser: T[String] = T("")
+  def flywayPassword: T[String] = T("")
+  //def flywayClassLocationDeps: T[Agg[Dep]] = Agg.empty[Dep]
+  //def flywayClassLocations: T[Seq[String]] = T(Nil)
+  def flywayFileLocations: T[Seq[PathRef]] = T(resources().map(pr => PathRef(pr.path / "db" / "migration", pr.quick)))
+  def flywayDriverDeps: T[Agg[Dep]]
+  def jdbcClasspath = T ( resolveDependencies(
+    repositories,
+    Lib.depToDependencyJava(_),
+    flywayDriverDeps()
+  ))
+
+  private def tToOptPair[A](key: String, t: A) =
+    Option(t)
+      .filter {
+        case a: String => a.nonEmpty
+        case _ => true
+      }
+      .map(key -> _)
+
+  def flywayInstance = T {
+    val jdbcClassloader = new URLClassLoader(jdbcClasspath().map(_.path.toIO.toURI.toURL).toArray)
+
+    val configProps = Map(flyway.URL -> flywayUrl()) ++
+      tToOptPair(flyway.USER, flywayUser()) ++
+      tToOptPair(flyway.PASSWORD, flywayPassword())
+
+    LogFactory.setLogCreator(new ConsoleLogCreator(Level.INFO))
+
+    Flyway
+      .configure(jdbcClassloader)
+      .locations(flywayFileLocations().map("filesystem:" + _.path):_*)
+      .configuration(configProps.asJava)
+      .load
+  }
+
+  def flywayMigrate() = T.command (
+    flywayInstance().migrate()
+  )
+}
-- 
cgit v1.2.3